diff --git a/backend/src/bot/modules/ai/agent.py b/backend/src/bot/modules/ai/agent.py index ffb7386..2419e8b 100644 --- a/backend/src/bot/modules/ai/agent.py +++ b/backend/src/bot/modules/ai/agent.py @@ -41,7 +41,7 @@ def create_text_agent( model = GoogleModel(model_name, provider=provider) base_prompt = system_prompt or DEFAULT_SYSTEM_PROMPT full_prompt = f"{base_prompt} {LATEX_INSTRUCTION}" - return Agent(model, system_prompt=full_prompt) + return Agent(model, instructions=full_prompt) def create_follow_up_agent( @@ -52,7 +52,7 @@ def create_follow_up_agent( provider = GoogleProvider(api_key=api_key) model = GoogleModel(model_name, provider=provider) prompt = system_prompt or DEFAULT_FOLLOW_UP - return Agent(model, output_type=FollowUpOptions, system_prompt=prompt) + return Agent(model, output_type=FollowUpOptions, instructions=prompt) def build_message_history(history: list[dict[str, str]]) -> list[ModelMessage]: