add connector file

pilot/connections/base.py

@@ -0,0 +1,8 @@
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+"""We need to design a base class.  That other connector can Write with this"""
+
+class BaseConnection:
+    pass
+

pilot/connections/clickhouse.py

@@ -0,0 +1,7 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+
+class ClickHouseConnector:
+    """ClickHouseConnector"""
+    pass

pilot/connections/es.py

@@ -0,0 +1,7 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+
+class ElasticSearchConnector:
+    """ElasticSearchConnector"""
+    pass

pilot/connections/mongo.py

@@ -0,0 +1,6 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+class MongoConnector:
+    """MongoConnector is a class which connect to mongo and chat with LLM"""
+    pass

pilot/connections/mysql.py

@@ -4,7 +4,11 @@
 import pymysql
 
 class MySQLOperator:
-    """Connect MySQL Database fetch MetaData For LLM Prompt """
+    """Connect MySQL Database fetch MetaData For LLM Prompt 
+        Args:
+
+        Usage:
+    """
 
     default_db = ["information_schema", "performance_schema", "sys", "mysql"]
     def __init__(self, user, password, host="localhost", port=3306) -> None:
@@ -26,6 +30,9 @@ class MySQLOperator:
             cursor.execute(_sql)
             results = cursor.fetchall()
             return results
+    
+    def get_index(self, schema_name):
+        pass
 
     def get_db_list(self):
         with self.conn.cursor() as cursor:
@@ -38,5 +45,7 @@ class MySQLOperator:
             dbs = [d["Database"] for d in results if d["Database"] not in self.default_db]
             return dbs
 
+    def get_meta(self, schema_name):
+        pass
 
 

pilot/connections/oracle.py

@@ -0,0 +1,6 @@
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+class OracleConnector:
+    """OracleConnector"""
+    pass

pilot/connections/pg_conn.py

@@ -1,2 +0,0 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-

pilot/connections/postgres.py

@@ -0,0 +1,8 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+
+
+class PostgresConnector:
+    """PostgresConnector is a class which Connector to chat with LLM"""
+    pass

pilot/connections/redis.py

@@ -0,0 +1,7 @@
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+
+class RedisConnector:
+    """RedisConnector"""
+    pass