Ishanpardeshi commited on
Commit
dd2be67
·
verified ·
1 Parent(s): fdb6c1f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -11
app.py CHANGED
@@ -8,7 +8,6 @@ from langchain_google_genai import ChatGoogleGenerativeAI
8
  from langchain.agents.agent_types import AgentType
9
  from langchain_experimental.agents.agent_toolkits import create_csv_agent
10
  from langchain_groq import ChatGroq
11
- import asyncio
12
 
13
  # API keys-----------------move them to ENV
14
  os.environ["OPENAI_API_KEY"] = "NA"
@@ -26,9 +25,9 @@ class tools:
26
  PDFtool = PDFSearchTool(
27
  config=dict(
28
  llm=dict(
29
- provider="google",
30
  config=dict(
31
- model="gemini-1.5-flash-latest",
32
  ),
33
  ),
34
  embedder=dict(
@@ -47,9 +46,9 @@ class tools:
47
  Filetool = FileReadTool(
48
  config=dict(
49
  llm=dict(
50
- provider="google",
51
  config=dict(
52
- model="gemini-1.5-flash-latest",
53
  ),
54
  ),
55
  embedder=dict(
@@ -68,9 +67,9 @@ class tools:
68
  Docstool = DOCXSearchTool(
69
  config=dict(
70
  llm=dict(
71
- provider="google",
72
  config=dict(
73
- model="gemini-1.5-flash-latest",
74
  ),
75
  ),
76
  embedder=dict(
@@ -92,7 +91,7 @@ class AgentLoader:
92
 
93
  def csvReaderAgent(path):
94
  agent = create_csv_agent(
95
- ChatGoogleGenerativeAI(temperature=0.6, model="gemini-1.5-flash-latest"),
96
  path,
97
  verbose=True,
98
  agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION
@@ -173,7 +172,7 @@ def getTasks(query, agent, exp):
173
  return [task_read, task_write]
174
 
175
  # Gradio interface function
176
- async def process_file(file, query, expected_output):
177
  path = file.name
178
 
179
  if path.endswith(".pdf"):
@@ -184,7 +183,7 @@ async def process_file(file, query, expected_output):
184
  agent = AgentLoader.fileReaderAgent(path)
185
  elif path.endswith(".csv"):
186
  agent = AgentLoader.csvReaderAgent(path)
187
- results = await anyio.to_thread.run_sync(agent.run, query)
188
  else:
189
  return 'File NOT supported'
190
 
@@ -195,7 +194,7 @@ async def process_file(file, query, expected_output):
195
  tasks=task1,
196
  verbose=True
197
  )
198
- results = await anyio.to_thread.run_sync(mycrew.kickoff)
199
 
200
  return results
201
 
 
8
  from langchain.agents.agent_types import AgentType
9
  from langchain_experimental.agents.agent_toolkits import create_csv_agent
10
  from langchain_groq import ChatGroq
 
11
 
12
  # API keys-----------------move them to ENV
13
  os.environ["OPENAI_API_KEY"] = "NA"
 
25
  PDFtool = PDFSearchTool(
26
  config=dict(
27
  llm=dict(
28
+ provider="ollama",
29
  config=dict(
30
+ model="llama3-70b-8192",
31
  ),
32
  ),
33
  embedder=dict(
 
46
  Filetool = FileReadTool(
47
  config=dict(
48
  llm=dict(
49
+ provider="ollama",
50
  config=dict(
51
+ model="llama3-70b-8192",
52
  ),
53
  ),
54
  embedder=dict(
 
67
  Docstool = DOCXSearchTool(
68
  config=dict(
69
  llm=dict(
70
+ provider="ollama",
71
  config=dict(
72
+ model="llama3-70b-8192",
73
  ),
74
  ),
75
  embedder=dict(
 
91
 
92
  def csvReaderAgent(path):
93
  agent = create_csv_agent(
94
+ ChatGroq(temperature=0.6, model="llama3-70b-8192"),
95
  path,
96
  verbose=True,
97
  agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION
 
172
  return [task_read, task_write]
173
 
174
  # Gradio interface function
175
+ def process_file(file, query, expected_output):
176
  path = file.name
177
 
178
  if path.endswith(".pdf"):
 
183
  agent = AgentLoader.fileReaderAgent(path)
184
  elif path.endswith(".csv"):
185
  agent = AgentLoader.csvReaderAgent(path)
186
+ results = agent.run(query)
187
  else:
188
  return 'File NOT supported'
189
 
 
194
  tasks=task1,
195
  verbose=True
196
  )
197
+ results = mycrew.kickoff()
198
 
199
  return results
200