achuthc1298 commited on
Commit
0a6edbc
1 Parent(s): 5c1c23c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,7 +15,7 @@ from llama_index.core.objects import ObjectIndex
15
  from llama_index.core.agent import ReActAgent
16
 
17
  # Function to process files and create document tools
18
- async def create_doc_tools(document_fp: str, doc_name: str, verbose: bool = True) -> Tuple[QueryEngineTool,]:
19
  documents = SimpleDirectoryReader(input_files=[document_fp]).load_data()
20
 
21
  Settings.llm = Groq(model="mixtral-8x7b-32768")
@@ -59,7 +59,7 @@ def main():
59
  paper_to_tools_dict = {}
60
  for paper in tex_files:
61
  path = Path(paper)
62
- vector_tool = await create_doc_tools(doc_name=path.stem, document_fp=path)
63
  paper_to_tools_dict[path.stem] = [vector_tool]
64
 
65
  initial_tools = [t for paper in tex_files for t in paper_to_tools_dict[Path(paper).stem]]
 
15
  from llama_index.core.agent import ReActAgent
16
 
17
  # Function to process files and create document tools
18
+ def create_doc_tools(document_fp: str, doc_name: str, verbose: bool = True) -> Tuple[QueryEngineTool,]:
19
  documents = SimpleDirectoryReader(input_files=[document_fp]).load_data()
20
 
21
  Settings.llm = Groq(model="mixtral-8x7b-32768")
 
59
  paper_to_tools_dict = {}
60
  for paper in tex_files:
61
  path = Path(paper)
62
+ vector_tool = create_doc_tools(doc_name=path.stem, document_fp=path)
63
  paper_to_tools_dict[path.stem] = [vector_tool]
64
 
65
  initial_tools = [t for paper in tex_files for t in paper_to_tools_dict[Path(paper).stem]]