Update agent.py
Browse files
agent.py
CHANGED
|
@@ -55,7 +55,7 @@ class SmolagentToolWrapper(BaseTool):
|
|
| 55 |
|
| 56 |
|
| 57 |
|
| 58 |
-
class
|
| 59 |
def __init__(self, api_key: str, model_name: str = "gemini-2.0-flash"):
|
| 60 |
# Suppress warnings
|
| 61 |
import warnings
|
|
|
|
| 55 |
|
| 56 |
|
| 57 |
|
| 58 |
+
class Agent:
|
| 59 |
def __init__(self, api_key: str, model_name: str = "gemini-2.0-flash"):
|
| 60 |
# Suppress warnings
|
| 61 |
import warnings
|