Kshitijk20 commited on
Commit
25ef9d2
·
1 Parent(s): 9eb858f

fixing check query

Browse files
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}")],