diff --git a/pretor/core/global_state_machine/model_provider/__init__.py b/pretor/core/global_state_machine/model_provider/__init__.py index 5cb81a3..e0d1fcd 100644 --- a/pretor/core/global_state_machine/model_provider/__init__.py +++ b/pretor/core/global_state_machine/model_provider/__init__.py @@ -15,5 +15,4 @@ 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 -from pretor.core.global_state_machine.model_provider.deepseek_provider import DeepseekProvider -__all__ = ["Provider", "ProviderArgs", "OpenAIProvider", "ClaudeProvider", "DeepseekProvider"] +__all__ = ["Provider", "ProviderArgs", "OpenAIProvider", "ClaudeProvider"] diff --git a/pretor/core/global_state_machine/provider_manager.py b/pretor/core/global_state_machine/provider_manager.py index b714700..a126a9b 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, DeepseekProvider +from pretor.core.global_state_machine.model_provider import Provider, OpenAIProvider, ClaudeProvider 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": DeepseekProvider} + "deepseek": OpenAIProvider} 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 c4d4645..ec42f29 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 = ray_actor_hook("workflow_running_engine").workflow_running_engine await workflow_running_engine.put_event.remote(payload) except Exception as e: self.logger.error(f"SupervisoryNode: 无法将事件放入 WorkflowRunningEngine: {e}")