Skip to content

Commit

Permalink
remove timeout from the underlying network service
Browse files Browse the repository at this point in the history
  • Loading branch information
masci committed Oct 17, 2024
1 parent 0cfab66 commit 82da594
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions llama_deploy/deploy/network_workflow.py
Original file line number Diff line number Diff line change
@@ -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
Expand Down Expand Up @@ -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:
Expand Down

0 comments on commit 82da594

Please sign in to comment.