Update plugins/chatMemoryPlugin.py

plugins/chatMemoryPlugin.py

@@ -1,4 +1,3 @@
-#converterPlugin.py
 from datetime import datetime
 from typing import Annotated, Dict, List, Optional
 import uuid
@@ -22,11 +21,12 @@ class ChatMemoryPlugin:
             self.logger.error(f"Failed to log chat interaction: {str(e)}")
 
     @kernel_function(name="log_sql_query", description="Logs generated SQL queries")
-    async def log_sql_query(self, original_question: str, generated_sql: str):
+    async def log_sql_query(self, original_question: str, generated_sql: str, state:str="success"):
         
         try:
             await self.chat_memory_handler.log_sql_query(original_question=original_question,
-                                                         generated_sql=generated_sql)
+                                                         generated_sql=generated_sql,
+                                                         state=state)
         except Exception as e:
             self.logger.error(f"Failed to log SQL query: {str(e)}")