From b516cfc61a707336c893fe5f61e8e0f35b7432cb Mon Sep 17 00:00:00 2001
From: Parker Erickson <parker.erickson@tigergraph.com>
Date: Sun, 28 Jan 2024 19:41:26 -0600
Subject: [PATCH] fix(logs): fix warn deprecation

---
 app/main.py                         | 6 +++---
 app/tools/generate_function.py      | 6 +++---
 app/tools/map_schema_to_question.py | 2 +-
 3 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/app/main.py b/app/main.py
index e4856d82..429644d9 100644
--- a/app/main.py
+++ b/app/main.py
@@ -150,15 +150,15 @@ def retrieve_answer(graphname, query: NaturalLanguageQuery, credentials: Annotat
             resp.natural_language_response = steps["output"]
             resp.query_sources = {"agent_history": str(steps)}
             resp.answered_question = False
-            logger.warn(f"/{graphname}/query request_id={req_id_cv.get()} agent execution failed due to unknown exception")
+            logger.warning(f"/{graphname}/query request_id={req_id_cv.get()} agent execution failed due to unknown exception")
     except MapQuestionToSchemaException as e:
         resp.natural_language_response = ""
         resp.query_sources = {}
         resp.answered_question = False
-        logger.warn(f"/{graphname}/query request_id={req_id_cv.get()} agent execution failed due to MapQuestionToSchemaException")
+        logger.warning(f"/{graphname}/query request_id={req_id_cv.get()} agent execution failed due to MapQuestionToSchemaException")
     except Exception as e:
         resp.natural_language_response = ""
         resp.query_sources = {}
         resp.answered_question = False
-        logger.warn(f"/{graphname}/query request_id={req_id_cv.get()} agent execution failed due to unknown exception")
+        logger.warning(f"/{graphname}/query request_id={req_id_cv.get()} agent execution failed due to unknown exception")
     return resp
diff --git a/app/tools/generate_function.py b/app/tools/generate_function.py
index 2927e146..edb2fd76 100644
--- a/app/tools/generate_function.py
+++ b/app/tools/generate_function.py
@@ -90,7 +90,7 @@ def _run(self, question: str,
                             target_vertex_attributes, 
                             target_edge_attributes)
         except MapQuestionToSchemaException as e:
-            logger.warn(f"request_id={req_id_cv.get()} WARN input schema not valid")
+            logger.warning(f"request_id={req_id_cv.get()} WARN input schema not valid")
             return e
 
         lookup_question = question + " "
@@ -122,7 +122,7 @@ def _run(self, question: str,
         try:
             generated = validate_function_call(self.conn, generated, docs)
         except InvalidFunctionCallException as e:
-            logger.warn(f"request_id={req_id_cv.get()} EXIT GenerateFunction._run() with exception={e}")
+            logger.warning(f"request_id={req_id_cv.get()} EXIT GenerateFunction._run() with exception={e}")
             return e
 
         try:
@@ -131,7 +131,7 @@ def _run(self, question: str,
             logger.info(f"request_id={req_id_cv.get()} EXIT GenerateFunction._run()")
             return "Function {} produced the result {}".format(generated, json.dumps(loc["res"]))
         except Exception as e:
-            logger.warn(f"request_id={req_id_cv.get()} EXIT GenerateFunction._run() with exception={e}")
+            logger.warning(f"request_id={req_id_cv.get()} EXIT GenerateFunction._run() with exception={e}")
             raise ToolException("The function {} did not execute correctly. Please rephrase your question and try again".format(generated))
 
 
diff --git a/app/tools/map_schema_to_question.py b/app/tools/map_schema_to_question.py
index 831ca481..104df970 100644
--- a/app/tools/map_schema_to_question.py
+++ b/app/tools/map_schema_to_question.py
@@ -103,7 +103,7 @@ def _run(self, query: str) -> str:
                             parsed_q.target_vertex_attributes, 
                             parsed_q.target_edge_attributes)
         except MapQuestionToSchemaException as e:
-            logger.warn(f"request_id={req_id_cv.get()} WARN MapQuestionToSchema to validate schema")
+            logger.warning(f"request_id={req_id_cv.get()} WARN MapQuestionToSchema to validate schema")
             raise e
         logger.info(f"request_id={req_id_cv.get()} EXIT MapQuestionToSchema._run()")
         return parsed_q