Spaces:
Sleeping
Sleeping
Commit ·
25ef9d2
1
Parent(s): 9eb858f
fixing check query
Browse files- app/requirements.txt +2 -1
- app/services/sql_agent.py +4 -0
app/requirements.txt
CHANGED
|
@@ -15,4 +15,5 @@ ipykernel
|
|
| 15 |
passlib
|
| 16 |
python-multipart
|
| 17 |
bcrypt==4.3.0
|
| 18 |
-
psycopg2-binary
|
|
|
|
|
|
| 15 |
passlib
|
| 16 |
python-multipart
|
| 17 |
bcrypt==4.3.0
|
| 18 |
+
psycopg2-binary
|
| 19 |
+
langchain-google-genai
|
app/services/sql_agent.py
CHANGED
|
@@ -662,6 +662,10 @@ class SQLAgent:
|
|
| 662 |
|
| 663 |
try:
|
| 664 |
checked_query = self.query_checker_tool.invoke(query)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 665 |
print(f"Query checked: {checked_query}")
|
| 666 |
return {
|
| 667 |
"messages": [AIMessage(content=f"Query checked: {checked_query}")],
|
|
|
|
| 662 |
|
| 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 {
|
| 671 |
"messages": [AIMessage(content=f"Query checked: {checked_query}")],
|