karthikeyan-r commited on
Commit
9d2baf0
·
verified ·
1 Parent(s): 5fecd33

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -141,7 +141,7 @@ class ChatDocumentQA:
141
  # memory,tools,prompt = self.create_memory_for_csv_qa(tools)
142
  # agent_chain = self.create_agent_chain_for_csv_qa(memory,tools,prompt)
143
  agent_chain = create_csv_agent(
144
- AzureChatOpenAI(azure_deployment = "GPT-3"),
145
  file_path,
146
  verbose=True,
147
  agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
@@ -175,7 +175,7 @@ class ChatDocumentQA:
175
  #************************
176
  def create_agent(self,file_path):
177
  agent_chain = create_csv_agent(
178
- AzureChatOpenAI(azure_deployment = "GPT-3"),
179
  file_path,
180
  verbose=True,
181
  agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
@@ -212,7 +212,7 @@ class ChatDocumentQA:
212
 
213
  def create_agent_chain_for_csv_qa(self,memory,tools,prompt):
214
 
215
- llm_chain = LLMChain(llm=AzureChatOpenAI(azure_deployment = "GPT-3"), prompt=prompt)
216
  agent = ZeroShotAgent(llm_chain=llm_chain, tools=tools, verbose=True)
217
  agent_chain = AgentExecutor.from_agent_and_tools(
218
  agent=agent, tools=tools, verbose=True, memory=memory
 
141
  # memory,tools,prompt = self.create_memory_for_csv_qa(tools)
142
  # agent_chain = self.create_agent_chain_for_csv_qa(memory,tools,prompt)
143
  agent_chain = create_csv_agent(
144
+ AzureChatOpenAI(azure_deployment = "gpt-4o"),
145
  file_path,
146
  verbose=True,
147
  agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
 
175
  #************************
176
  def create_agent(self,file_path):
177
  agent_chain = create_csv_agent(
178
+ AzureChatOpenAI(azure_deployment = "gpt-4o"),
179
  file_path,
180
  verbose=True,
181
  agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
 
212
 
213
  def create_agent_chain_for_csv_qa(self,memory,tools,prompt):
214
 
215
+ llm_chain = LLMChain(llm=AzureChatOpenAI(azure_deployment = "gpt-4o"), prompt=prompt)
216
  agent = ZeroShotAgent(llm_chain=llm_chain, tools=tools, verbose=True)
217
  agent_chain = AgentExecutor.from_agent_and_tools(
218
  agent=agent, tools=tools, verbose=True, memory=memory