binaychandra commited on
Commit
10d2525
·
1 Parent(s): 7f85b45

Trying fix chromadb issue2

Browse files
Files changed (1) hide show
  1. langchain_helper.py +2 -2
langchain_helper.py CHANGED
@@ -70,8 +70,8 @@ def get_few_shot_db_chain(user_message):
70
 
71
  print(f"This is the intermediate query : {intermediate_sql_query}")
72
 
73
- if intermediate_sql_query.startswith('sql '):
74
- intermediate_sql_query = intermediate_sql_query.replace("sql ", "")
75
 
76
  result_df = pd.read_sql_query(intermediate_sql_query, engine)
77
  print("Printing results")
 
70
 
71
  print(f"This is the intermediate query : {intermediate_sql_query}")
72
 
73
+ if intermediate_sql_query.startswith('SQLQuery: '):
74
+ intermediate_sql_query = intermediate_sql_query.replace("SQLQuery: ", "")
75
 
76
  result_df = pd.read_sql_query(intermediate_sql_query, engine)
77
  print("Printing results")