Spaces:

Murali-rade
/

final-gaia-agent

Sleeping

Murali-rade commited on Oct 5

Commit

be45591

verified ·

1 Parent(s): f6ff3ab

Update agent.py

Files changed (1) hide show

agent.py CHANGED Viewed

@@ -20,7 +20,7 @@ load_dotenv()
 #     model_id=os.getenv("GEMINI_MODEL"), api_key=os.getenv("KEY")
 # )
-client = OpenAIServerModel(
     base_url="https://router.huggingface.co/v1",
     api_key=os.environ["HF_TOKEN"],
     model_id="openai/gpt-oss-120b:novita"
@@ -101,7 +101,7 @@ class GaiaAgent:
         ]
         self.agent = CodeAgent(
-            llm_client=client,
             tools=tools,
             add_base_tools=True,
             additional_authorized_imports=["pandas", "numpy", "csv", "subprocess"],

 #     model_id=os.getenv("GEMINI_MODEL"), api_key=os.getenv("KEY")
 # )
+modell = OpenAIServerModel(
     base_url="https://router.huggingface.co/v1",
     api_key=os.environ["HF_TOKEN"],
     model_id="openai/gpt-oss-120b:novita"
         ]
         self.agent = CodeAgent(
+            model=modell,
             tools=tools,
             add_base_tools=True,
             additional_authorized_imports=["pandas", "numpy", "csv", "subprocess"],