From fa22999d8c510eee5396a0fa239aaf998cae8e9c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9C=9D=E5=A4=95?= Date: Tue, 28 Apr 2026 14:21:00 +0800 Subject: [PATCH] Revert "Feat/deepseek adapter dropin 14224636701039833263 (#51)" (#52) This reverts commit 9d7b9807699cecf729fc175150a2ce9c1f1cf97e. --- pretor/core/global_state_machine/model_provider/__init__.py | 3 ++- pretor/core/global_state_machine/provider_manager.py | 4 ++-- pretor/core/individual/supervisory_node/supervisory_node.py | 2 +- 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/pretor/core/global_state_machine/model_provider/__init__.py b/pretor/core/global_state_machine/model_provider/__init__.py index e0d1fcd..5cb81a3 100644 --- a/pretor/core/global_state_machine/model_provider/__init__.py +++ b/pretor/core/global_state_machine/model_provider/__init__.py @@ -15,4 +15,5 @@ from pretor.core.global_state_machine.model_provider.base_provider import Provider, ProviderArgs from pretor.core.global_state_machine.model_provider.openai_provider import OpenAIProvider from pretor.core.global_state_machine.model_provider.claude_provider import ClaudeProvider -__all__ = ["Provider", "ProviderArgs", "OpenAIProvider", "ClaudeProvider"] +from pretor.core.global_state_machine.model_provider.deepseek_provider import DeepseekProvider +__all__ = ["Provider", "ProviderArgs", "OpenAIProvider", "ClaudeProvider", "DeepseekProvider"] diff --git a/pretor/core/global_state_machine/provider_manager.py b/pretor/core/global_state_machine/provider_manager.py index a126a9b..b714700 100644 --- a/pretor/core/global_state_machine/provider_manager.py +++ b/pretor/core/global_state_machine/provider_manager.py @@ -12,7 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. -from pretor.core.global_state_machine.model_provider import Provider, OpenAIProvider, ClaudeProvider +from pretor.core.global_state_machine.model_provider import Provider, OpenAIProvider, ClaudeProvider, DeepseekProvider from typing import Dict, Type class ProviderManager: @@ -29,7 +29,7 @@ class ProviderManager: def __init__(self, postgres): self.provider_mapper = {"openai": OpenAIProvider, "claude": ClaudeProvider, - "deepseek": OpenAIProvider} + "deepseek": DeepseekProvider} self.provider_register = {} async def init_provider_register(self, postgres) -> None: diff --git a/pretor/core/individual/supervisory_node/supervisory_node.py b/pretor/core/individual/supervisory_node/supervisory_node.py index ec42f29..c4d4645 100644 --- a/pretor/core/individual/supervisory_node/supervisory_node.py +++ b/pretor/core/individual/supervisory_node/supervisory_node.py @@ -111,7 +111,7 @@ class SupervisoryNode: if isinstance(payload, PretorEvent): payload.context["workflow_template"] = result.workflow_template try: - workflow_running_engine = ray_actor_hook("workflow_running_engine").workflow_running_engine + workflow_running_engine = ray_actor_hook("workflow_running_engine") await workflow_running_engine.put_event.remote(payload) except Exception as e: self.logger.error(f"SupervisoryNode: 无法将事件放入 WorkflowRunningEngine: {e}")