From 82da594384a448e136ea9a01af2a740c22afa0d2 Mon Sep 17 00:00:00 2001 From: Massimiliano Pippi Date: Thu, 17 Oct 2024 15:26:23 +0200 Subject: [PATCH] remove timeout from the underlying network service --- llama_deploy/deploy/network_workflow.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/llama_deploy/deploy/network_workflow.py b/llama_deploy/deploy/network_workflow.py index 1aa914a9..e5bddaf3 100644 --- a/llama_deploy/deploy/network_workflow.py +++ b/llama_deploy/deploy/network_workflow.py @@ -1,6 +1,6 @@ from typing import Any, Dict, Optional -from llama_index.core.workflow import Workflow, StopEvent, StartEvent, step +from llama_index.core.workflow import StartEvent, StopEvent, Workflow, step from llama_index.core.workflow.service import ServiceManager, ServiceNotFoundError from llama_deploy.client.async_client import AsyncLlamaDeployClient @@ -60,7 +60,7 @@ def get(self, name: str, default: Optional["Workflow"] = None) -> "Workflow": # If the remove service exists, swap it in if remote_service is not None: - return NetworkWorkflow(self.control_plane_config, name) + return NetworkWorkflow(self.control_plane_config, name, timeout=None) # else default to the local workflow -- if it exists if local_workflow is None: