Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -23,7 +23,7 @@ configure(api_key=os.getenv("GOOGLE_API_KEY"))
|
|
| 23 |
#logger = logging.getLogger(__name__)
|
| 24 |
|
| 25 |
# --- Model Configuration ---
|
| 26 |
-
GEMINI_MODEL_NAME = "gemini/gemini-2.
|
| 27 |
OPENAI_MODEL_NAME = "openai/gpt-4o"
|
| 28 |
GROQ_MODEL_NAME = "groq/llama3-70b-8192"
|
| 29 |
DEEPSEEK_MODEL_NAME = "deepseek/deepseek-chat"
|
|
@@ -165,7 +165,7 @@ class FileAttachmentQueryTool(Tool):
|
|
| 165 |
|
| 166 |
# --- Basic Agent Definition ---
|
| 167 |
class BasicAgent:
|
| 168 |
-
def __init__(self, provider="
|
| 169 |
print("BasicAgent initialized.")
|
| 170 |
model = self.select_model(provider)
|
| 171 |
client = InferenceClientModel()
|
|
|
|
| 23 |
#logger = logging.getLogger(__name__)
|
| 24 |
|
| 25 |
# --- Model Configuration ---
|
| 26 |
+
GEMINI_MODEL_NAME = "gemini/gemini-2.5-pro"
|
| 27 |
OPENAI_MODEL_NAME = "openai/gpt-4o"
|
| 28 |
GROQ_MODEL_NAME = "groq/llama3-70b-8192"
|
| 29 |
DEEPSEEK_MODEL_NAME = "deepseek/deepseek-chat"
|
|
|
|
| 165 |
|
| 166 |
# --- Basic Agent Definition ---
|
| 167 |
class BasicAgent:
|
| 168 |
+
def __init__(self, provider="gemini"):
|
| 169 |
print("BasicAgent initialized.")
|
| 170 |
model = self.select_model(provider)
|
| 171 |
client = InferenceClientModel()
|