From ec93b5d4157d707629324a11f1df0235661b8466 Mon Sep 17 00:00:00 2001 From: Andrei Fajardo Date: Sat, 3 Aug 2024 15:45:25 -0400 Subject: [PATCH] cr --- poetry.lock | 2 +- snowflake_cybersyn_demo/core_services/control_plane.py | 5 ++--- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/poetry.lock b/poetry.lock index c6f293b..e09014b 100644 --- a/poetry.lock +++ b/poetry.lock @@ -1399,7 +1399,7 @@ redis = ["redis (>=5.0.7,<6.0.0)"] type = "git" url = "git@github.com:run-llama/llama-agents.git" reference = "nerdai-fix-pipeline-orchestrator" -resolved_reference = "5d87da4c9d386391d09be6ba81d3a321b35e124f" +resolved_reference = "d66cd537569c8d40c16b69e75a804e8dcb987fc6" [[package]] name = "llama-index-agent-openai" diff --git a/snowflake_cybersyn_demo/core_services/control_plane.py b/snowflake_cybersyn_demo/core_services/control_plane.py index 4ddcc9a..587c7a6 100644 --- a/snowflake_cybersyn_demo/core_services/control_plane.py +++ b/snowflake_cybersyn_demo/core_services/control_plane.py @@ -1,8 +1,7 @@ import asyncio import uvicorn -from llama_agents import ControlPlaneServer, PipelineOrchestrator -from llama_agents.orchestrators.router import RouterOrchestrator +from llama_agents import ControlPlaneServer, PipelineOrchestrator, OrchestratorRouter from llama_agents.message_queues.rabbitmq import RabbitMQMessageQueue from llama_index.core.query_pipeline import QueryPipeline, RouterComponent from llama_index.core.selectors import PydanticSingleSelector @@ -49,7 +48,7 @@ general_pipeline = QueryPipeline(chain=[funny_agent_component]) general_pipeline_orchestrator = PipelineOrchestrator(general_pipeline) -pipeline_orchestrator = RouterOrchestrator( +pipeline_orchestrator = OrchestratorRouter( selector=PydanticSingleSelector.from_defaults(llm=OpenAI()), orchestrators=[timeseries_pipeline_orchestrator, general_pipeline_orchestrator], choices=[timeseries_task_pipeline_desc, funny_agent_server.description],