Skip to content

Commit 6575d1b

Browse files
committed
Fix ID incorrect on model logging
1 parent ecf537e commit 6575d1b

File tree

7 files changed

+11
-6
lines changed

7 files changed

+11
-6
lines changed

libs/agno/agno/models/base.py

+5
Original file line numberDiff line numberDiff line change
@@ -89,6 +89,7 @@ def __post_init__(self):
8989
if self.provider is None and self.name is not None:
9090
self.provider = f"{self.name} ({self.id})"
9191

92+
9293
def to_dict(self) -> Dict[str, Any]:
9394
fields = {"name", "id", "provider"}
9495
_dict = {field: getattr(self, field) for field in fields if getattr(self, field) is not None}
@@ -161,6 +162,7 @@ def response(self, messages: List[Message]) -> ModelResponse:
161162
ModelResponse: The model's response
162163
"""
163164
logger.debug(f"---------- {self.get_provider()} Response Start ----------")
165+
logger.debug(f"---------- Model: {self.id} ----------")
164166
self._log_messages(messages)
165167
model_response = ModelResponse()
166168

@@ -223,6 +225,7 @@ async def aresponse(self, messages: List[Message]) -> ModelResponse:
223225
ModelResponse: The model's response
224226
"""
225227
logger.debug(f"---------- {self.get_provider()} Async Response Start ----------")
228+
logger.debug(f"---------- Model: {self.id} ----------")
226229
self._log_messages(messages)
227230
model_response = ModelResponse()
228231

@@ -460,6 +463,7 @@ def response_stream(self, messages: List[Message]) -> Iterator[ModelResponse]:
460463
Iterator[ModelResponse]: Iterator of model responses
461464
"""
462465
logger.debug(f"---------- {self.get_provider()} Response Stream Start ----------")
466+
logger.debug(f"---------- Model: {self.id} ----------")
463467
self._log_messages(messages)
464468

465469
while True:
@@ -555,6 +559,7 @@ async def aresponse_stream(self, messages: List[Message]) -> AsyncIterator[Model
555559
AsyncIterator[ModelResponse]: Async iterator of model responses
556560
"""
557561
logger.debug(f"---------- {self.get_provider()} Async Response Stream Start ----------")
562+
logger.debug(f"---------- Model: {self.id} ----------")
558563
self._log_messages(messages)
559564

560565
while True:

libs/agno/agno/models/deepinfra/deepinfra.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ class DeepInfra(OpenAILike):
2222

2323
id: str = "meta-llama/Llama-2-70b-chat-hf"
2424
name: str = "DeepInfra"
25-
provider: str = "DeepInfra " + id
25+
provider: str = "DeepInfra"
2626

2727
api_key: Optional[str] = getenv("DEEPINFRA_API_KEY", None)
2828
base_url: str = "https://api.deepinfra.com/v1/openai"

libs/agno/agno/models/lmstudio/lmstudio.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,6 @@ class LMStudio(OpenAILike):
1717

1818
id: str = "qwen2.5-7b-instruct-1m"
1919
name: str = "LMStudio"
20-
provider: str = "LMStudio " + id
20+
provider: str = "LMStudio"
2121

2222
base_url: str = "http://127.0.0.1:1234/v1"

libs/agno/agno/models/nvidia/nvidia.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ class Nvidia(OpenAILike):
2020

2121
id: str = "meta/llama-3.3-70b-instruct"
2222
name: str = "Nvidia"
23-
provider: str = "Nvidia " + id
23+
provider: str = "Nvidia"
2424

2525
api_key: Optional[str] = getenv("NVIDIA_API_KEY", None)
2626
base_url: str = "https://integrate.api.nvidia.com/v1"

libs/agno/agno/models/openrouter/openrouter.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ class OpenRouter(OpenAILike):
2121

2222
id: str = "gpt-4o"
2323
name: str = "OpenRouter"
24-
provider: str = "OpenRouter: " + id
24+
provider: str = "OpenRouter"
2525

2626
api_key: Optional[str] = getenv("OPENROUTER_API_KEY")
2727
base_url: str = "https://openrouter.ai/api/v1"

libs/agno/agno/models/perplexity/perplexity.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ class Perplexity(OpenAILike):
2323

2424
id: str = "sonar"
2525
name: str = "Perplexity"
26-
provider: str = "Perplexity: " + id
26+
provider: str = "Perplexity"
2727

2828
api_key: Optional[str] = getenv("PERPLEXITY_API_KEY")
2929
base_url: str = "https://api.perplexity.ai/"

libs/agno/agno/models/together/together.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,6 @@ class Together(OpenAILike):
2020

2121
id: str = "mistralai/Mixtral-8x7B-Instruct-v0.1"
2222
name: str = "Together"
23-
provider: str = "Together " + id
23+
provider: str = "Together"
2424
api_key: Optional[str] = getenv("TOGETHER_API_KEY")
2525
base_url: str = "https://api.together.xyz/v1"

0 commit comments

Comments
 (0)