Spaces:

real-jiakai
/

Agent_Course_Final_Assignment_Demo

Paused

real-jiakai commited on Apr 26

Commit

0527a8f

verified ·

1 Parent(s): 71701ce

Update agent.py

Files changed (1) hide show

agent.py CHANGED Viewed

@@ -24,7 +24,6 @@ from smolagents import (
     CodeAgent,
     DuckDuckGoSearchTool,
     PythonInterpreterTool,
-    BaseModel,
     tool,
 )
@@ -48,7 +47,7 @@ def _download_file(file_id: str) -> bytes:
 # --------------------------------------------------------------------------- #
 # Model wrapper
 # --------------------------------------------------------------------------- #
-class GeminiModel(BaseModel):
     """
     Thin adapter around google-genai.Client so it can be used by smolagents.
     """

     CodeAgent,
     DuckDuckGoSearchTool,
     PythonInterpreterTool,
     tool,
 )
 # --------------------------------------------------------------------------- #
 # Model wrapper
 # --------------------------------------------------------------------------- #
+class GeminiModel:
     """
     Thin adapter around google-genai.Client so it can be used by smolagents.
     """