This reverts commit 9d7b980769.
This commit is contained in:
parent
9d7b980769
commit
fa22999d8c
|
|
@ -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.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.openai_provider import OpenAIProvider
|
||||||
from pretor.core.global_state_machine.model_provider.claude_provider import ClaudeProvider
|
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"]
|
||||||
|
|
|
||||||
|
|
@ -12,7 +12,7 @@
|
||||||
# See the License for the specific language governing permissions and
|
# See the License for the specific language governing permissions and
|
||||||
# limitations under the License.
|
# 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
|
from typing import Dict, Type
|
||||||
|
|
||||||
class ProviderManager:
|
class ProviderManager:
|
||||||
|
|
@ -29,7 +29,7 @@ class ProviderManager:
|
||||||
def __init__(self, postgres):
|
def __init__(self, postgres):
|
||||||
self.provider_mapper = {"openai": OpenAIProvider,
|
self.provider_mapper = {"openai": OpenAIProvider,
|
||||||
"claude": ClaudeProvider,
|
"claude": ClaudeProvider,
|
||||||
"deepseek": OpenAIProvider}
|
"deepseek": DeepseekProvider}
|
||||||
self.provider_register = {}
|
self.provider_register = {}
|
||||||
|
|
||||||
async def init_provider_register(self, postgres) -> None:
|
async def init_provider_register(self, postgres) -> None:
|
||||||
|
|
|
||||||
|
|
@ -111,7 +111,7 @@ class SupervisoryNode:
|
||||||
if isinstance(payload, PretorEvent):
|
if isinstance(payload, PretorEvent):
|
||||||
payload.context["workflow_template"] = result.workflow_template
|
payload.context["workflow_template"] = result.workflow_template
|
||||||
try:
|
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)
|
await workflow_running_engine.put_event.remote(payload)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
self.logger.error(f"SupervisoryNode: 无法将事件放入 WorkflowRunningEngine: {e}")
|
self.logger.error(f"SupervisoryNode: 无法将事件放入 WorkflowRunningEngine: {e}")
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue