Update app.py
Browse files
app.py
CHANGED
|
@@ -29,9 +29,9 @@ class LLaMaAgent:
|
|
| 29 |
model=model,
|
| 30 |
)
|
| 31 |
|
| 32 |
-
|
| 33 |
-
|
| 34 |
-
|
| 35 |
|
| 36 |
class CompoundAgent:
|
| 37 |
def __init__(self):
|
|
|
|
| 29 |
model=model,
|
| 30 |
)
|
| 31 |
|
| 32 |
+
def __call__(self, question: str) -> str:
|
| 33 |
+
response = self.agent.run(question)
|
| 34 |
+
return response
|
| 35 |
|
| 36 |
class CompoundAgent:
|
| 37 |
def __init__(self):
|