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