Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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="
|
30 |
config=dict(
|
31 |
-
model="
|
32 |
),
|
33 |
),
|
34 |
embedder=dict(
|
@@ -47,9 +46,9 @@ class tools:
|
|
47 |
Filetool = FileReadTool(
|
48 |
config=dict(
|
49 |
llm=dict(
|
50 |
-
provider="
|
51 |
config=dict(
|
52 |
-
model="
|
53 |
),
|
54 |
),
|
55 |
embedder=dict(
|
@@ -68,9 +67,9 @@ class tools:
|
|
68 |
Docstool = DOCXSearchTool(
|
69 |
config=dict(
|
70 |
llm=dict(
|
71 |
-
provider="
|
72 |
config=dict(
|
73 |
-
model="
|
74 |
),
|
75 |
),
|
76 |
embedder=dict(
|
@@ -92,7 +91,7 @@ class AgentLoader:
|
|
92 |
|
93 |
def csvReaderAgent(path):
|
94 |
agent = create_csv_agent(
|
95 |
-
|
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 |
-
|
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 =
|
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 =
|
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 |
|