binaychandra commited on
Commit
501f9c8
·
1 Parent(s): 0cf0063

Trying fix chromadb issue4

Browse files
Files changed (1) hide show
  1. langchain_helper.py +5 -0
langchain_helper.py CHANGED
@@ -66,7 +66,12 @@ def get_few_shot_db_chain(user_message):
66
  print(f"sql query : {response_llm['intermediate_steps'][1]}")
67
  if 'sql_cmd' in response_llm['intermediate_steps'][2].keys():
68
  intermediate_sql_query = response_llm['intermediate_steps'][2]['sql_cmd']
 
 
69
 
 
 
 
70
  result_df = pd.read_sql_query(intermediate_sql_query, engine)
71
  print("Printing results")
72
  print(result_df)
 
66
  print(f"sql query : {response_llm['intermediate_steps'][1]}")
67
  if 'sql_cmd' in response_llm['intermediate_steps'][2].keys():
68
  intermediate_sql_query = response_llm['intermediate_steps'][2]['sql_cmd']
69
+
70
+ print(f"This is the intermediate query : {intermediate_sql_query}")
71
 
72
+ if intermediate_sql_query.startswith('sql '):
73
+ intermediate_sql_query = intermediate_sql_query.replace("sql ", "")
74
+
75
  result_df = pd.read_sql_query(intermediate_sql_query, engine)
76
  print("Printing results")
77
  print(result_df)