1.去掉无用的颜色

2.删掉API文档

airda/agent/assistants/assistant_manager.py

@@ -1,9 +1,7 @@
 from airda.agent.assistants.assistant_keys import AssistantKeys
 from airda.agent.assistants.chat_assistant.chat_assistant import ChatAssistant
 from airda.agent.assistants.sql_assistant.sql_assistant import SqlAssistant
-from airda.framework.agent.module.assistant_manager.assistant.assistant import (
-    Assistant,
-)
+from airda.framework.agent.module.assistant_manager.assistant.assistant import Assistant
 from airda.framework.agent.module.assistant_manager.assistant_manager import (
     AssistantManager,
 )

airda/agent/assistants/chat_assistant/chat_assistant.py

@@ -2,9 +2,7 @@
 
 from airda.agent.action.searcher import Searcher
 from airda.framework.action.action import Action
-from airda.framework.agent.module.assistant_manager.assistant.assistant import (
-    Assistant,
-)
+from airda.framework.agent.module.assistant_manager.assistant.assistant import Assistant
 
 
 class ChatAssistant(Assistant, ABC):

airda/agent/assistants/sql_assistant/sql_assistant.py

@@ -15,9 +15,7 @@
 from airda.agent.exception.action_type_error import ActionTypeError
 from airda.agent.llm.data_agent_llm_keys import DataAgentLLMKeys
 from airda.framework.action.action import Action
-from airda.framework.agent.module.assistant_manager.assistant.assistant import (
-    Assistant,
-)
+from airda.framework.agent.module.assistant_manager.assistant.assistant import Assistant
 from airda.framework.agent.module.assistant_manager.assistant.assistant_params import (
     AssistantParams,
 )

airda/agent/planner/data_agent_pipeline.py

@@ -7,9 +7,7 @@
 )
 from airda.agent.planner.data_agent_planner_params import DataAgentPlannerParams
 from airda.framework.agent.context import Context
-from airda.framework.agent.module.assistant_manager.assistant.assistant import (
-    Assistant,
-)
+from airda.framework.agent.module.assistant_manager.assistant.assistant import Assistant
 from airda.framework.agent.module.planner.pipeline.pipeline import Pipeline
 
 

airda/agent/planner/data_agent_planner_params.py

@@ -1,8 +1,6 @@
 from pydantic import BaseModel
 
-from airda.framework.agent.module.planner.pipeline.pipeline_params import (
-    PipelineParams,
-)
+from airda.framework.agent.module.planner.pipeline.pipeline_params import PipelineParams
 from airda.framework.agent.module.planner.planner_params import PlannerParams
 
 

airda/agent/rag/data_agent_rag.py

@@ -2,9 +2,7 @@
 
 from airda.agent import log
 from airda.agent.process_pool.DataAgentProcessPool import DataAgentProcessPool
-from airda.agent.rag.embedding.data_agent_embedding_model import (
-    DataAgentEmbeddingModel,
-)
+from airda.agent.rag.embedding.data_agent_embedding_model import DataAgentEmbeddingModel
 from airda.agent.rag.schema_linking.data_agent_schema_linking import (
     DataAgentSchemaLinking,
 )

airda/agent/rag/embedding/data_agent_embedding_model.py

@@ -2,9 +2,7 @@
 from sklearn.preprocessing import normalize
 
 from airda.agent.env import DataAgentEnv
-from airda.framework.agent.module.rag.embedding.embedding_model import (
-    EmbeddingModel,
-)
+from airda.framework.agent.module.rag.embedding.embedding_model import EmbeddingModel
 
 
 class DataAgentEmbeddingModel(EmbeddingModel):

airda/agent/rag/schema_linking/data_agent_schema_linking.py

@@ -1,12 +1,8 @@
 import numpy as np
 
 from airda.agent import log
-from airda.framework.agent.module.rag.embedding.embedding_model import (
-    EmbeddingModel,
-)
-from airda.framework.agent.module.rag.schema_linking.schema_linking import (
-    SchemaLinking,
-)
+from airda.framework.agent.module.rag.embedding.embedding_model import EmbeddingModel
+from airda.framework.agent.module.rag.schema_linking.schema_linking import SchemaLinking
 
 logger = log.getLogger()
 

airda/agent/storage/storage.py

@@ -7,9 +7,7 @@
 
 from airda.agent.env import DataAgentEnv
 from airda.agent.storage import StorageKey
-from airda.agent.storage.repositories.datasource_repository import (
-    DatasourceRepository,
-)
+from airda.agent.storage.repositories.datasource_repository import DatasourceRepository
 from airda.agent.storage.repositories.instruction_repository import (
     InstructionRepository,
 )

airda/framework/agent/module/assistant_manager/assistant_manager.py

@@ -1,9 +1,7 @@
 from abc import abstractmethod
 
 from airda.framework.agent.context import Context
-from airda.framework.agent.module.assistant_manager.assistant.assistant import (
-    Assistant,
-)
+from airda.framework.agent.module.assistant_manager.assistant.assistant import Assistant
 from airda.framework.agent.module.loader import Loader
 from airda.framework.module.keys import Keys
 

airda/framework/agent/module/planner/pipeline/pipeline.py

@@ -2,9 +2,7 @@
 from typing import AsyncGenerator, Generic, List, TypeVar
 
 from airda.framework.agent.context import Context
-from airda.framework.agent.module.assistant_manager.assistant.assistant import (
-    Assistant,
-)
+from airda.framework.agent.module.assistant_manager.assistant.assistant import Assistant
 
 from .pipeline_params import PipelineParams
 from .pipline_result import PipelineResult

airda/framework/agent/module/rag/rag.py

@@ -3,12 +3,8 @@
 from airda.framework.agent.context import Context
 from airda.framework.agent.module.loader import Loader
 from airda.framework.agent.module.rag import RagKey
-from airda.framework.agent.module.rag.embedding.embedding_model import (
-    EmbeddingModel,
-)
-from airda.framework.agent.module.rag.schema_linking.schema_linking import (
-    SchemaLinking,
-)
+from airda.framework.agent.module.rag.embedding.embedding_model import EmbeddingModel
+from airda.framework.agent.module.rag.schema_linking.schema_linking import SchemaLinking
 
 
 class Rag(Loader, ABC):

airda/framework/agent/module/rag/schema_linking/schema_linking.py

@@ -1,9 +1,7 @@
 from abc import abstractmethod
 
 from airda.framework import log
-from airda.framework.agent.module.rag.embedding.embedding_model import (
-    EmbeddingModel,
-)
+from airda.framework.agent.module.rag.embedding.embedding_model import EmbeddingModel
 from airda.framework.module.immediate import Immediate
 
 logger = log.getLogger()

airda/server/agent_server/__init__.py

@@ -23,9 +23,7 @@ def create_app(self):
     def run_server(self):
         import uvicorn
 
-        uvicorn.run(
-            self.app, host=self.host, port=self.port, log_level="info"
-        )
+        uvicorn.run(self.app, host=self.host, port=self.port, log_level="info")
 
     @overrides
     def add_routes(self):

pyproject.toml

@@ -40,4 +40,3 @@ airda = "airda.cli.startup:main"
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
-