mirror of
https://github.com/furyhawk/ai_agent.git
synced 2026-07-20 09:45:44 +00:00
fix: update model import and adjust model instantiation in _build_model function
This commit is contained in:
@@ -19,7 +19,7 @@ from pydantic_ai.messages import (
|
||||
TextPart,
|
||||
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.settings import ModelSettings
|
||||
|
||||
@@ -37,7 +37,7 @@ def _build_model(model_name: str):
|
||||
provider_kwargs: dict[str, Any] = {"api_key": settings.OPENAI_API_KEY}
|
||||
if settings.OPENAI_BASE_URL:
|
||||
provider_kwargs["base_url"] = settings.OPENAI_BASE_URL
|
||||
return OpenAIResponsesModel(
|
||||
return OpenAIModel(
|
||||
model_name or settings.AI_MODEL,
|
||||
provider=OpenAIProvider(**provider_kwargs),
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user