diff --git a/agents/customer_explorer_agent.py b/agents/customer_explorer_agent.py index a1d8adc..f23bbda 100644 --- a/agents/customer_explorer_agent.py +++ b/agents/customer_explorer_agent.py @@ -6,6 +6,7 @@ import dotenv from agno.agent import Agent from agno.models.openai import OpenAIChat +from agno.models.nebius import Nebius from pydantic import BaseModel from datetime import datetime, timedelta import json @@ -116,9 +117,13 @@ def create_customer_explorer_agent() -> Agent: """Create and return the customer explorer agent""" return Agent( name="CustomerExplorerAgent", - model=OpenAIChat( - id="gpt-4o-mini", - api_key=os.getenv("OPENAI_API_KEY"), + # model=OpenAIChat( + # id="gpt-4o-mini", + # api_key=os.getenv("OPENAI_API_KEY"), + # ), + model=Nebius( + id="openai/gpt-oss-120b", + api_key=os.getenv("NEBIUS_API_KEY"), ), description=CUSTOMER_EXPLORER_SYSTEM_PROMPT, add_history_to_context=False, diff --git a/agents/next_message_agent.py b/agents/next_message_agent.py index 90e956d..1fbf194 100644 --- a/agents/next_message_agent.py +++ b/agents/next_message_agent.py @@ -6,6 +6,7 @@ import dotenv from agno.agent import Agent from agno.models.openai import OpenAIChat +from agno.models.nebius import Nebius from pydantic import BaseModel dotenv.load_dotenv() @@ -131,9 +132,13 @@ def create_next_message_agent() -> Agent: """Create and return the next message agent""" return Agent( name="NextMessageAgent", - model=OpenAIChat( - id="gpt-4o-mini", - api_key=os.getenv("OPENAI_API_KEY"), + # model=OpenAIChat( + # id="gpt-4o-mini", + # api_key=os.getenv("OPENAI_API_KEY"), + # ), + model=Nebius( + id="openai/gpt-oss-120b", + api_key=os.getenv("NEBIUS_API_KEY"), ), description=NEXT_MESSAGE_SYSTEM_PROMPT, add_history_to_context=False, diff --git a/agents/summary_agent.py b/agents/summary_agent.py index 272871b..4002b96 100644 --- a/agents/summary_agent.py +++ b/agents/summary_agent.py @@ -6,6 +6,7 @@ import dotenv from agno.agent import Agent from agno.models.openai import OpenAIChat +from agno.models.nebius import Nebius from pydantic import BaseModel dotenv.load_dotenv() @@ -46,9 +47,13 @@ def create_summary_agent() -> Agent: """Create and return the summary agent""" return Agent( name="SummaryAgent", - model=OpenAIChat( - id="gpt-4o-mini", - api_key=os.getenv("OPENAI_API_KEY"), + # model=OpenAIChat( + # id="gpt-4o-mini", + # api_key=os.getenv("OPENAI_API_KEY"), + # ), + model=Nebius( + id="openai/gpt-oss-120b", + api_key=os.getenv("NEBIUS_API_KEY"), ), description=SUMMARY_SYSTEM_PROMPT, add_history_to_context=False, # Each call is independent diff --git a/main_support_agent.py b/main_support_agent.py index faa8a26..c21ddd4 100644 --- a/main_support_agent.py +++ b/main_support_agent.py @@ -10,6 +10,7 @@ import dotenv from agno.agent import Agent from agno.models.openai import OpenAIChat +from agno.models.nebius import Nebius # Import our specialized agents as tools from agents.summary_agent import summarize_conversation @@ -179,9 +180,13 @@ def escalate_to_human(reason: str, urgency: str = "medium") -> str: # Create the main support agent support_agent = Agent( name="BankCustomerSupportAgent", - model=OpenAIChat( - id="gpt-4o-mini", - api_key=os.getenv("OPENAI_API_KEY"), + # model=OpenAIChat( + # id="gpt-4o-mini", + # api_key=os.getenv("OPENAI_API_KEY"), + # ), + model=Nebius( + id="openai/gpt-oss-120b", + api_key=os.getenv("NEBIUS_API_KEY"), ), tools=[ get_conversation_summary, diff --git a/tests/test_main_support_agent.py b/tests/test_main_support_agent.py index 1376f81..d28f4ff 100644 --- a/tests/test_main_support_agent.py +++ b/tests/test_main_support_agent.py @@ -17,7 +17,7 @@ from main_support_agent import support_agent dotenv.load_dotenv() -scenario.configure(default_model="openai/gpt-4o-mini") +scenario.configure(default_model="nebius/openai/gpt-oss-120b") class BankSupportAgentAdapter(scenario.AgentAdapter):