furyhawk/ai_agent
1
0
Fork 0
mirror of https://github.com/furyhawk/ai_agent.git synced 2026-07-21 10:15:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: update model import and adjust model instantiation in _build_model function

Browse Source
This commit is contained in:
furyhawk
2026-06-12 15:10:47 +08:00
parent 21cee0c2bd
commit 5abd1d7175
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/app/agents/assistant.py
+2 -2
View File
@@ -19,7 +19,7 @@ from pydantic_ai.messages import (
TextPart, TextPart,
UserPromptPart, UserPromptPart,
) )
from pydantic_ai.models.openai import OpenAIResponsesModel from pydantic_ai.models.openai import OpenAIModel
from pydantic_ai.providers.openai import OpenAIProvider from pydantic_ai.providers.openai import OpenAIProvider
from pydantic_ai.settings import ModelSettings from pydantic_ai.settings import ModelSettings
@@ -37,7 +37,7 @@ def _build_model(model_name: str):
provider_kwargs: dict[str, Any] = {"api_key": settings.OPENAI_API_KEY} provider_kwargs: dict[str, Any] = {"api_key": settings.OPENAI_API_KEY}
if settings.OPENAI_BASE_URL: if settings.OPENAI_BASE_URL:
provider_kwargs["base_url"] = settings.OPENAI_BASE_URL provider_kwargs["base_url"] = settings.OPENAI_BASE_URL
return OpenAIResponsesModel( return OpenAIModel(
model_name or settings.AI_MODEL, model_name or settings.AI_MODEL,
provider=OpenAIProvider(**provider_kwargs), provider=OpenAIProvider(**provider_kwargs),
) )