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 |
|