razvanfischer commited on
Commit
e5a4011
·
verified ·
1 Parent(s): 2ee9938

Added asyncio

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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