Sborole commited on
Commit
7131ab4
·
verified ·
1 Parent(s): a15700b

Update src/agent.py

Browse files
Files changed (1) hide show
  1. src/agent.py +1 -1
src/agent.py CHANGED
@@ -7,7 +7,7 @@ import random
7
  from litellm import completion
8
 
9
  settings = Settings()
10
- class GaiaAgent():
11
  def __init__(self):
12
  self.model = LiteLLMModel(
13
  model_id=settings.llm_model_id,
 
7
  from litellm import completion
8
 
9
  settings = Settings()
10
+ class BasicAgent():
11
  def __init__(self):
12
  self.model = LiteLLMModel(
13
  model_id=settings.llm_model_id,