Skip to content

Commit 31f8164

Browse files
committed
fix: Toolcall data structure
1 parent f0e76ca commit 31f8164

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

openagent/conf/llm_provider.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ def get_available_providers() -> Dict[str, BaseChatModel]:
6464
providers = {}
6565

6666
provider_configs = [
67-
(["gpt-4o-mini", "gpt-4o"], get_openai_provider),
67+
(["gpt-4o-mini", "gpt-4o", "gpt-3.5-turbo"], get_openai_provider),
6868
(["claude-3-5-sonnet"], get_anthropic_provider),
6969
(["gemini-1.5-pro", "gemini-1.5-flash"], get_gemini_provider),
7070
]
@@ -84,8 +84,10 @@ def get_available_providers() -> Dict[str, BaseChatModel]:
8484
def get_openai_provider(model: str) -> BaseChatModel | None:
8585
return ChatOpenAI(model=model) if settings.OPENAI_API_KEY else None
8686

87+
8788
def get_anthropic_provider(model: str) -> BaseChatModel | None:
88-
return ChatAnthropic(model="claude-3-5-sonnet-20240620",) if settings.ANTHROPIC_API_KEY else None
89+
return ChatAnthropic(model="claude-3-5-sonnet-20240620", ) if settings.ANTHROPIC_API_KEY else None
90+
8991

9092
def get_gemini_provider(model: str) -> BaseChatModel | None:
9193
if settings.VERTEX_PROJECT_ID:

openagent/router/openai.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,7 @@ class ChatCompletionResponse(BaseModel):
8080
class DeltaMessage(BaseModel):
8181
role: Optional[str] = None
8282
content: Optional[str] = None
83+
tool_calls: Optional[List[ToolCall]] = None
8384

8485

8586
class StreamChoice(BaseModel):

0 commit comments

Comments
 (0)