Kshitijk20 commited on
Commit
b7755d4
·
1 Parent(s): 129b6d6

fixing check query issue

Browse files
Files changed (1) hide show
  1. app/services/sql_agent.py +2 -2
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 {