sajjadpsavoji commited on
Commit ·
e2f4a03
1
Parent(s): 94f6a00
add a simple code agent
Browse files- agent.py +5 -5
- requirements.txt +0 -2
agent.py
CHANGED
|
@@ -1,8 +1,8 @@
|
|
|
|
|
|
|
|
| 1 |
class BasicAgent:
|
| 2 |
def __init__(self):
|
| 3 |
-
|
| 4 |
def __call__(self, question: str) -> str:
|
| 5 |
-
|
| 6 |
-
|
| 7 |
-
print(f"Agent returning fixed answer: {fixed_answer}")
|
| 8 |
-
return fixed_answer
|
|
|
|
| 1 |
+
from smolagents import CodeAgent, InferenceClientModel
|
| 2 |
+
|
| 3 |
class BasicAgent:
|
| 4 |
def __init__(self):
|
| 5 |
+
self.agent = CodeAgent(model=InferenceClientModel("Qwen/Qwen2.5-Coder-32B-Instruct"))
|
| 6 |
def __call__(self, question: str) -> str:
|
| 7 |
+
answer = self.agent.run(question)
|
| 8 |
+
return answer
|
|
|
|
|
|
requirements.txt
CHANGED
|
@@ -1,5 +1,3 @@
|
|
| 1 |
gradio
|
| 2 |
requests
|
| 3 |
-
duckduckgo_search
|
| 4 |
-
markdownify
|
| 5 |
smolagents
|
|
|
|
| 1 |
gradio
|
| 2 |
requests
|
|
|
|
|
|
|
| 3 |
smolagents
|