Merge branch 'feat/queue-based-graph-engine' into chore/merge-graph-engine

api/core/plugin/impl/agent.py

@@ -17,7 +17,7 @@ class PluginAgentClient(BasePluginClient):
         Fetch agent providers for the given tenant.
         """
 
-        def transformer(json_response: dict[str, Any]) -> dict:
+        def transformer(json_response: dict[str, Any]):
             for provider in json_response.get("data", []):
                 declaration = provider.get("declaration", {}) or {}
                 provider_name = declaration.get("identity", {}).get("name")
@@ -49,7 +49,7 @@ class PluginAgentClient(BasePluginClient):
         """
         agent_provider_id = GenericProviderID(provider)
 
-        def transformer(json_response: dict[str, Any]) -> dict:
+        def transformer(json_response: dict[str, Any]):
             # skip if error occurs
             if json_response.get("data") is None or json_response.get("data", {}).get("declaration") is None:
                 return json_response

api/core/plugin/impl/exc.py

@@ -8,7 +8,7 @@ from extensions.ext_logging import get_request_id
 class PluginDaemonError(Exception):
     """Base class for all plugin daemon errors."""
 
-    def __init__(self, description: str) -> None:
+    def __init__(self, description: str):
         self.description = description
 
     def __str__(self) -> str:

api/core/plugin/impl/model.py

@@ -415,7 +415,7 @@ class PluginModelClient(BasePluginClient):
         model: str,
         credentials: dict,
         language: Optional[str] = None,
-    ) -> list[dict]:
+    ):
         """
         Get tts model voices
         """

api/core/plugin/impl/tool.py

@@ -20,7 +20,7 @@ class PluginToolManager(BasePluginClient):
         Fetch tool providers for the given tenant.
         """
 
-        def transformer(json_response: dict[str, Any]) -> dict:
+        def transformer(json_response: dict[str, Any]):
             for provider in json_response.get("data", []):
                 declaration = provider.get("declaration", {}) or {}
                 provider_name = declaration.get("identity", {}).get("name")
@@ -55,7 +55,7 @@ class PluginToolManager(BasePluginClient):
         """
         tool_provider_id = ToolProviderID(provider)
 
-        def transformer(json_response: dict[str, Any]) -> dict:
+        def transformer(json_response: dict[str, Any]):
             data = json_response.get("data")
             if data:
                 for tool in data.get("declaration", {}).get("tools", []):