env key
Browse files
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 |
-
|
| 295 |
-
|
| 296 |
-
|
| 297 |
-
|
|
|
|
| 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}")
|