Skip to content

Commit e237680

Browse files
committed
update
1 parent a12879a commit e237680

File tree

1 file changed

+6
-7
lines changed

1 file changed

+6
-7
lines changed

libs/agno/agno/models/openai/responses.py

+6-7
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22
from typing import Any, AsyncIterator, Dict, Iterator, List, Optional, Tuple, Union
33

44
import httpx
5-
65
from pydantic import BaseModel
76

87
from agno.exceptions import ModelProviderError
@@ -201,15 +200,15 @@ def request_kwargs(self) -> Dict[str, Any]:
201200
base_params["text"] = {"format": {"type": "json_object"}}
202201

203202
# Filter out None values
204-
request_params = {k: v for k, v in base_params.items() if v is not None}
203+
request_params: Dict[str, Any] = {k: v for k, v in base_params.items() if v is not None}
205204

206205
if self.web_search:
207-
request_params.setdefault("tools", [])
206+
request_params.setdefault("tools", []) # type: ignore
208207
request_params["tools"].append({"type": "web_search_preview"})
209208

210209
# Add tools
211210
if self._functions is not None and len(self._functions) > 0:
212-
request_params.setdefault("tools", [])
211+
request_params.setdefault("tools", []) # type: ignore
213212
for function in self._functions.values():
214213
function_dict = function.to_dict()
215214
for prop in function_dict["parameters"]["properties"].values():
@@ -452,7 +451,7 @@ async def ainvoke_stream(self, messages: List[Message]) -> AsyncIterator[Respons
452451
stream=True,
453452
**self.request_kwargs,
454453
)
455-
async for chunk in async_stream:
454+
async for chunk in async_stream: # type: ignore
456455
yield chunk
457456
except RateLimitError as e:
458457
logger.error(f"Rate limit error from OpenAI API: {e}")
@@ -519,7 +518,7 @@ def parse_provider_response(self, response: Response) -> ModelResponse:
519518

520519
if response.error:
521520
raise ModelProviderError(
522-
message=response.error.get("message", "Unknown model error"),
521+
message=response.error.message,
523522
model_name=self.name,
524523
model_id=self.id,
525524
)
@@ -611,7 +610,7 @@ def _process_stream_response(
611610

612611
elif stream_event.type == "response.output_item.done" and tool_use:
613612
model_response = ModelResponse()
614-
model_response.tool_calls = tool_use
613+
model_response.tool_calls = [tool_use]
615614
if assistant_message.tool_calls is None:
616615
assistant_message.tool_calls = []
617616
assistant_message.tool_calls.append(tool_use)

0 commit comments

Comments
 (0)