Spaces:
Sleeping
Sleeping
Commit ·
b7755d4
1
Parent(s): 129b6d6
fixing check query issue
Browse files
app/services/sql_agent.py
CHANGED
|
@@ -663,8 +663,8 @@ class SQLAgent:
|
|
| 663 |
try:
|
| 664 |
checked_query = self.query_checker_tool.invoke(query)
|
| 665 |
## if checked query contains ``` anywhere remove it
|
| 666 |
-
if "```" in checked_query:
|
| 667 |
-
checked_query = checked_query.replace("```", "")
|
| 668 |
|
| 669 |
print(f"Query checked: {checked_query}")
|
| 670 |
return {
|
|
|
|
| 663 |
try:
|
| 664 |
checked_query = self.query_checker_tool.invoke(query)
|
| 665 |
## if checked query contains ``` anywhere remove it
|
| 666 |
+
if "```" in checked_query or "sql" in checked_query:
|
| 667 |
+
checked_query = checked_query.replace("```", "").replace("sql", "").strip()
|
| 668 |
|
| 669 |
print(f"Query checked: {checked_query}")
|
| 670 |
return {
|