Skip to content

Commit 572a1f2

Browse files
shah-sidddgustavocidornelas
authored andcommitted
fix: fixed model names and OpenLayer to Openlayer.
1 parent d666470 commit 572a1f2

File tree

2 files changed

+8
-8
lines changed

2 files changed

+8
-8
lines changed

src/openlayer/lib/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,7 @@ def trace_litellm():
177177
>>> response = litellm.completion(
178178
... model="gpt-3.5-turbo",
179179
... messages=[{"role": "user", "content": "Hello!"}],
180-
... inference_id="custom-id-123" # Optional OpenLayer parameter
180+
... inference_id="custom-id-123" # Optional Openlayer parameter
181181
... )
182182
"""
183183
# pylint: disable=import-outside-toplevel

src/openlayer/lib/integrations/litellm_tracer.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ def trace_litellm() -> None:
5454
>>> response = litellm.completion(
5555
... model="gpt-3.5-turbo",
5656
... messages=[{"role": "user", "content": "Hello!"}],
57-
... inference_id="custom-id-123" # Optional OpenLayer parameter
57+
... inference_id="custom-id-123" # Optional Openlayer parameter
5858
... )
5959
"""
6060
if not HAVE_LITELLM:
@@ -480,17 +480,17 @@ def detect_provider_from_model_name(model_name: str) -> str:
480480
model_lower = model_name.lower()
481481

482482
if model_lower.startswith(('gpt-', 'o1-', 'text-davinci', 'text-curie', 'text-babbage', 'text-ada')):
483-
return "openai"
483+
return "OpenAI"
484484
elif model_lower.startswith(('claude-', 'claude')):
485-
return "anthropic"
485+
return "Anthropic"
486486
elif 'gemini' in model_lower or 'palm' in model_lower:
487-
return "google"
487+
return "Google"
488488
elif 'llama' in model_lower:
489-
return "meta"
489+
return "Meta"
490490
elif model_lower.startswith('mistral'):
491-
return "mistral"
491+
return "Mistral"
492492
elif model_lower.startswith('command'):
493-
return "cohere"
493+
return "Cohere"
494494
else:
495495
return "unknown"
496496

0 commit comments

Comments
 (0)