orbulat commited on
Commit
3f12852
·
verified ·
1 Parent(s): ff7ca8b

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
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.0-flash"
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="deepseek"):
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()