Merge pull request #9 from FJS-Services-Inc/dev_deploy

fix: migrate OpenAIModel to provider API (pydantic-ai 0.4.x)
This commit is contained in:
2026-02-02 21:39:26 -05:00
committed by GitHub

View File

@ -6,6 +6,7 @@ import openai
import praw
from pydantic_ai import Agent
from pydantic_ai.models.openai import OpenAIModel
from pydantic_ai.providers.openai import OpenAIProvider
from backend.core.consts import AI_MODEL, default_system_prompt
from backend.db.base import Base
@ -70,7 +71,10 @@ class SwotAgentDeps:
swot_agent = Agent(
OpenAIModel(model_name=AI_MODEL, api_key=get_val("OPENAI_API_KEY")),
OpenAIModel(
model_name=AI_MODEL,
provider=OpenAIProvider(api_key=get_val("OPENAI_API_KEY")),
),
deps_type=SwotAgentDeps,
result_type=SwotAnalysis,
system_prompt=default_system_prompt,