Added asyncio
Browse files
app.py
CHANGED
|
@@ -3,6 +3,7 @@ import gradio as gr
|
|
| 3 |
import requests
|
| 4 |
import inspect
|
| 5 |
import pandas as pd
|
|
|
|
| 6 |
from llama_index.tools.duckduckgo import DuckDuckGoSearchToolSpec
|
| 7 |
from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
|
| 8 |
from llama_index.core.agent.workflow import AgentWorkflow
|
|
@@ -21,7 +22,7 @@ class BasicAgent:
|
|
| 21 |
self.agent = AgentWorkflow.from_tools_or_functions([self.search_tool], llm=self.llm)
|
| 22 |
print("BasicAgent initialized.")
|
| 23 |
|
| 24 |
-
def __call__(self, question: str) -> str:
|
| 25 |
response = await self.agent.run(question)
|
| 26 |
return response
|
| 27 |
|
|
|
|
| 3 |
import requests
|
| 4 |
import inspect
|
| 5 |
import pandas as pd
|
| 6 |
+
import asyncio
|
| 7 |
from llama_index.tools.duckduckgo import DuckDuckGoSearchToolSpec
|
| 8 |
from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
|
| 9 |
from llama_index.core.agent.workflow import AgentWorkflow
|
|
|
|
| 22 |
self.agent = AgentWorkflow.from_tools_or_functions([self.search_tool], llm=self.llm)
|
| 23 |
print("BasicAgent initialized.")
|
| 24 |
|
| 25 |
+
async def __call__(self, question: str) -> str:
|
| 26 |
response = await self.agent.run(question)
|
| 27 |
return response
|
| 28 |
|