Abraham E. Tavarez commited on
Commit ·
233fe2e
1
Parent(s): 93b922c
deps added to requirements
Browse files- agent.py +13 -6
- requirements.txt +3 -1
agent.py
CHANGED
|
@@ -22,7 +22,12 @@ load_dotenv()
|
|
| 22 |
login(os.environ["HF_API_KEY"])
|
| 23 |
|
| 24 |
# Tools
|
| 25 |
-
tools = [
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
|
| 27 |
question = QUESTIONS[0]
|
| 28 |
|
|
@@ -46,9 +51,11 @@ codeAgent = CodeAgent(
|
|
| 46 |
|
| 47 |
codeAgent.logger.console.width = 66
|
| 48 |
|
| 49 |
-
#
|
| 50 |
-
#
|
| 51 |
-
#
|
|
|
|
|
|
|
| 52 |
|
| 53 |
|
| 54 |
# Tool Calling Agent
|
|
@@ -62,5 +69,5 @@ toolCallingAgent = ToolCallingAgent(
|
|
| 62 |
|
| 63 |
toolCallingAgent.logger.console.width = 66
|
| 64 |
|
| 65 |
-
#
|
| 66 |
-
# print(
|
|
|
|
| 22 |
login(os.environ["HF_API_KEY"])
|
| 23 |
|
| 24 |
# Tools
|
| 25 |
+
tools = [
|
| 26 |
+
DuckDuckGoSearchTool(),
|
| 27 |
+
VisitWebpageTool(),
|
| 28 |
+
PythonInterpreterTool(),
|
| 29 |
+
FinalAnswerTool(),
|
| 30 |
+
]
|
| 31 |
|
| 32 |
question = QUESTIONS[0]
|
| 33 |
|
|
|
|
| 51 |
|
| 52 |
codeAgent.logger.console.width = 66
|
| 53 |
|
| 54 |
+
# answer = codeAgent.run(question)
|
| 55 |
+
# answer = codeAgent.run(
|
| 56 |
+
# "Who are the pitchers with the number before and after Taishō Tamai's number as of July 2023? Give them to me in the form Pitcher Before, Pitcher After, use their last names only, in Roman characters."
|
| 57 |
+
# )
|
| 58 |
+
# print(answer)
|
| 59 |
|
| 60 |
|
| 61 |
# Tool Calling Agent
|
|
|
|
| 69 |
|
| 70 |
toolCallingAgent.logger.console.width = 66
|
| 71 |
|
| 72 |
+
# answer = toolCallingAgent.run(question)
|
| 73 |
+
# print(answer)
|
requirements.txt
CHANGED
|
@@ -3,4 +3,6 @@ requests
|
|
| 3 |
smolagents
|
| 4 |
dotenv
|
| 5 |
duckduckgo-search
|
| 6 |
-
markdownify
|
|
|
|
|
|
|
|
|
| 3 |
smolagents
|
| 4 |
dotenv
|
| 5 |
duckduckgo-search
|
| 6 |
+
markdownify
|
| 7 |
+
pandas
|
| 8 |
+
numpy
|