修改服务名称为AirdaServer, 取消datasource的enable和disable选项，添加每次进入问答datasource的指定

airda/cli/startup.py

@@ -18,7 +18,7 @@
 from airda.agent.storage.entity.datasource import Datasource, Kind
 from airda.agent.storage.repositories.datasource_repository import DatasourceRepository
 from airda.connector.mysql import MysqlConnector
-from airda.server.agent_server import DataAgentServer
+from airda.server.agent_server.airda_server import AirdaServer
 
 style = Style.from_dict(
     {
@@ -108,8 +108,8 @@ async def execute():
     help="服务端口号",
 )
 def server(port: int):
-    data_agent_server = DataAgentServer(port=port)
-    data_agent_server.run_server()
+    airda_server = AirdaServer(port=port)
+    airda_server.run_server()
     pass
 
 

airda/connector/mysql.py

@@ -68,3 +68,4 @@ def query_schema(self):
                 table_comment=table[1],
             )
             self.context.sync_instruction(instruction)
+        cursor.close()

pyproject.toml

@@ -30,7 +30,6 @@ torch = "2.0.1"
 pymongo = "^4.6.2"
 prompt-toolkit = "^3.0.43"
 pyyaml = "^6.0.1"
-mysql-connector-python = "^8.3.0"
 fastapi = "0.99.0"
 
 [tool.poetry.scripts]