Fixed sql_retriever.py bug (run-llama#16445)

271285808 · Oct 9, 2024 · 4101976 · 4101976
1 parent bba274d
commit 4101976
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llama-index-core/llama_index/core/indices/struct_store/sql_retriever.py b/llama-index-core/llama_index/core/indices/struct_store/sql_retriever.py
@@ -143,7 +143,7 @@ def parse_response_to_sql(self, response: str, query_bundle: QueryBundle) -> str
         sql_result_start = response.find("SQLResult:")
         if sql_result_start != -1:
             response = response[:sql_result_start]
-        return response.strip().strip("```sql").strip("```").strip()
+        return response.replace("```sql", "").replace("```", "").strip()
 
 
 class PGVectorSQLParser(BaseSQLParser):