binaychandra commited on
Commit
7f85b45
·
1 Parent(s): 501f9c8

Trying fix chromadb issue6

Browse files
Files changed (1) hide show
  1. langchain_helper.py +3 -2
langchain_helper.py CHANGED
@@ -62,11 +62,12 @@ def get_few_shot_db_chain(user_message):
62
  chain = SQLDatabaseChain.from_llm(llm, db, verbose=True, prompt=few_shot_prompt, return_intermediate_steps = True)
63
  print(chain)
64
  response_llm = chain.invoke(user_message)
65
-
 
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 '):
 
62
  chain = SQLDatabaseChain.from_llm(llm, db, verbose=True, prompt=few_shot_prompt, return_intermediate_steps = True)
63
  print(chain)
64
  response_llm = chain.invoke(user_message)
65
+ print("========PRINTING response LLM==========")
66
+ print(response_llm)
67
  print(f"sql query : {response_llm['intermediate_steps'][1]}")
68
  if 'sql_cmd' in response_llm['intermediate_steps'][2].keys():
69
  intermediate_sql_query = response_llm['intermediate_steps'][2]['sql_cmd']
70
+
71
  print(f"This is the intermediate query : {intermediate_sql_query}")
72
 
73
  if intermediate_sql_query.startswith('sql '):