niro commited on
Commit
e5b1596
·
1 Parent(s): 577064b
Files changed (1) hide show
  1. chatbot_csv.py +5 -1
chatbot_csv.py CHANGED
@@ -45,6 +45,8 @@ def main():
45
  layout.show_header()
46
  user_api_key = utils.load_api_key()
47
 
 
 
48
  if not user_api_key:
49
  layout.show_api_key_missing()
50
  else:
@@ -87,7 +89,9 @@ def main():
87
  if query != "":
88
  old_stdout = sys.stdout
89
  sys.stdout = captured_output = StringIO()
90
- agent = create_csv_agent(ChatOpenAI(temperature=0), uploaded_file_content, verbose=True,
 
 
91
  max_iterations=4)
92
 
93
  result = agent.run(query)
 
45
  layout.show_header()
46
  user_api_key = utils.load_api_key()
47
 
48
+
49
+
50
  if not user_api_key:
51
  layout.show_api_key_missing()
52
  else:
 
89
  if query != "":
90
  old_stdout = sys.stdout
91
  sys.stdout = captured_output = StringIO()
92
+ agent = create_csv_agent(ChatOpenAI(temperature=0),
93
+ uploaded_file_content,
94
+ verbose=True,
95
  max_iterations=4)
96
 
97
  result = agent.run(query)