终端字体黑白背景改为自动

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/cli/startup.py

@@ -16,9 +16,7 @@
 from airda.agent.planner.data_agent_planner_params import DataAgentPlannerParams
 from airda.agent.storage import StorageKey
 from airda.agent.storage.entity.datasource import Datasource, Kind
-from airda.agent.storage.repositories.datasource_repository import (
-    DatasourceRepository,
-)
+from airda.agent.storage.repositories.datasource_repository import DatasourceRepository
 from airda.connector.mysql import MysqlConnector
 from airda.server.agent_server import DataAgentServer
 
@@ -28,7 +26,6 @@
         "output": "#3993d4",
         "enabled": "#5c962c",
         "success": "#4fc414",
-        "disabled": "#ffffff",
         "error": "#f0524f bold",
     }
 )
@@ -311,14 +308,18 @@ def env():
 )
 def load(path: str):
     import shutil
+
     if os.path.exists(path):
         shutil.copy(path, env_path)
 
 
 @env.command()
 def ls():
     import json
-    print_formatted_text(FormattedText([('class:json', json.dumps(DataAgentEnv().__dict__, indent=4))]))
+
+    print_formatted_text(
+        FormattedText([("class:json", json.dumps(DataAgentEnv().__dict__, indent=4))])
+    )
 
 
 @main.group()
@@ -336,14 +337,16 @@ def log():
 )
 def load(path: str):
     import shutil
+
     if os.path.exists(path):
         shutil.copy(path, log_path)
 
 
 @log.command()
 def ls():
     import json
-    print_formatted_text(FormattedText([('class:json', json.dumps(log_config, indent=4))]))
+
+    print_formatted_text(FormattedText([("class:json", json.dumps(log_config, indent=4))]))
 
 
 if __name__ == "__main__":

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):