[improvement][chat] Move python code out of chat-core module

chat/python/services/sql/output_parser.py

@@ -0,0 +1,57 @@
+# -*- coding:utf-8 -*-
+import re
+
+import os
+import sys
+
+sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
+sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+sys.path.append(os.path.dirname(os.path.abspath(__file__)))
+
+from instances.logging_instance import logger
+
+
+def schema_link_parse(schema_link_output):
+    try:
+        schema_link_output = schema_link_output.strip()
+        pattern = r"Schema_links:(.*)"
+        schema_link_output = re.findall(pattern, schema_link_output, re.DOTALL)[0].strip()
+    except Exception as e:
+        logger.exception(e)
+        schema_link_output = None
+
+    return schema_link_output
+
+
+def combo_schema_link_parse(schema_linking_sql_combo_output: str):
+    try:
+        schema_linking_sql_combo_output = schema_linking_sql_combo_output.strip()
+        pattern = r"Schema_links:(\[.*?\])"
+        schema_links_match = re.search(pattern, schema_linking_sql_combo_output)
+
+        if schema_links_match:
+            schema_links = schema_links_match.group(1)
+        else:
+            schema_links = None
+    except Exception as e:
+        logger.info(e)
+        schema_links = None
+
+    return schema_links
+
+
+def combo_sql_parse(schema_linking_sql_combo_output: str):
+    try:
+        schema_linking_sql_combo_output = schema_linking_sql_combo_output.strip()
+        pattern = r"SQL:(.*)"
+        sql_match = re.search(pattern, schema_linking_sql_combo_output)
+
+        if sql_match:
+            sql = sql_match.group(1)
+        else:
+            sql = None
+    except Exception as e:
+        logger.exception(e)
+        sql = None
+
+    return sql