mdicio commited on
Commit
4274fe2
·
1 Parent(s): 1aa70af
Files changed (1) hide show
  1. agent.py +6 -9
agent.py CHANGED
@@ -1,9 +1,6 @@
1
  import os
2
-
3
  from dotenv import load_dotenv
4
 
5
- load_dotenv()
6
-
7
  # Import models from SmolaAgents
8
  from smolagents import CodeAgent, LiteLLMModel, OpenAIServerModel
9
 
@@ -289,9 +286,9 @@ class BoomBot:
289
  return final_answer
290
 
291
 
292
- # Example of how to use this code (commented out)
293
- if __name__ == "__main__":
294
- agent = BoomBot(provider="gemma")
295
- question = "In the year 2020, where were koi fish found in the watershed with the id 02040203? Give only the name of the pond, lake, or stream where the fish were found, and not the name of the city or county."
296
- response = agent.run(question=question, task_id="1", to_download=False)
297
- print(f"Response: {response}")
 
1
  import os
 
2
  from dotenv import load_dotenv
3
 
 
 
4
  # Import models from SmolaAgents
5
  from smolagents import CodeAgent, LiteLLMModel, OpenAIServerModel
6
 
 
286
  return final_answer
287
 
288
 
289
+ # # Example of how to use this code (commented out)
290
+ # if __name__ == "__main__":
291
+ # agent = BoomBot(provider="gemma")
292
+ # question = "In the year 2020, where were koi fish found in the watershed with the id 02040203? Give only the name of the pond, lake, or stream where the fish were found, and not the name of the city or county."
293
+ # response = agent.run(question=question, task_id="1", to_download=False)
294
+ # print(f"Response: {response}")