Spaces:
Running
Running
Commit
·
7f85b45
1
Parent(s):
501f9c8
Trying fix chromadb issue6
Browse files- 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 '):
|