Spaces:
Running
Running
Commit
·
10d2525
1
Parent(s):
7f85b45
Trying fix chromadb issue2
Browse files- 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('
|
| 74 |
-
intermediate_sql_query = intermediate_sql_query.replace("
|
| 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")
|