Spaces:
Running
Running
Commit
·
6df28e0
1
Parent(s):
3649991
Update app.py
Browse files
app.py
CHANGED
@@ -133,11 +133,17 @@ class ChatDocumentQA:
|
|
133 |
vectorstore = self._create_vector_store_from_text_chunks(text_chunks)
|
134 |
return "file uploaded", {"knowledge_base": vectorstore}
|
135 |
elif file_extension == '.csv':
|
136 |
-
df = pd.read_csv(file_path)
|
137 |
-
pd_agent = create_pandas_dataframe_agent(OpenAI(temperature=0), df, verbose=True)
|
138 |
-
tools = self.get_agent_tools(pd_agent)
|
139 |
-
memory,tools,prompt = self.create_memory_for_csv_qa(tools)
|
140 |
-
agent_chain = self.create_agent_chain_for_csv_qa(memory,tools,prompt)
|
|
|
|
|
|
|
|
|
|
|
|
|
141 |
return "file uploaded", {"knowledge_base": agent_chain}
|
142 |
|
143 |
else:
|
|
|
133 |
vectorstore = self._create_vector_store_from_text_chunks(text_chunks)
|
134 |
return "file uploaded", {"knowledge_base": vectorstore}
|
135 |
elif file_extension == '.csv':
|
136 |
+
# df = pd.read_csv(file_path)
|
137 |
+
# pd_agent = create_pandas_dataframe_agent(OpenAI(temperature=0), df, verbose=True)
|
138 |
+
# tools = self.get_agent_tools(pd_agent)
|
139 |
+
# memory,tools,prompt = self.create_memory_for_csv_qa(tools)
|
140 |
+
# agent_chain = self.create_agent_chain_for_csv_qa(memory,tools,prompt)
|
141 |
+
agent_chain = create_csv_agent(
|
142 |
+
OpenAI(temperature=0),
|
143 |
+
file_path,
|
144 |
+
verbose=True,
|
145 |
+
agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
|
146 |
+
)
|
147 |
return "file uploaded", {"knowledge_base": agent_chain}
|
148 |
|
149 |
else:
|