Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

wip v2 experimentation #182

Open
wants to merge 12 commits into
base: main
Choose a base branch
from
Open

wip v2 experimentation #182

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
1fde9d6
wip v2 experimentation
steinwaywhw Sep 10, 2024
f8ca09f
hooking it up to the runner, pening more investigations of the trigge…
steinwaywhw Sep 11, 2024
5491960
more draft work on the runtime
steinwaywhw Sep 12, 2024
b5968ee
hooking up the runner to actually run the functions
steinwaywhw Sep 13, 2024
6d81998
a bit more work on propagating the events, and on runtime
steinwaywhw Sep 14, 2024
4eb7290
continue working on context propagation
steinwaywhw Sep 15, 2024
2c6d6b4
context propagation works now, working on result listening
steinwaywhw Sep 15, 2024
d0ea7c6
iterator adapters, more WIP for workflow run listener
steinwaywhw Sep 18, 2024
9cbd0e3
wip
steinwaywhw Sep 18, 2024
f8360a1
more changes to generator adaptors and workers. working on the main t…
steinwaywhw Sep 18, 2024
4307666
more changes
steinwaywhw Sep 22, 2024
f854d1d
now it is a fully working @function decorator
steinwaywhw Sep 23, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
24 changes: 18 additions & 6 deletions hatchet_sdk/clients/admin.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@

import grpc
from google.protobuf import timestamp_pb2
from loguru import logger

from hatchet_sdk.clients.rest.tenacity_utils import tenacity_retry
from hatchet_sdk.clients.run_event_listener import new_listener
Expand Down Expand Up @@ -339,13 +340,9 @@ def run_workflow(
if self.namespace != "" and not workflow_name.startswith(self.namespace):
workflow_name = f"{self.namespace}{workflow_name}"

request = self._prepare_workflow_request(workflow_name, input, options)
resp: TriggerWorkflowResponse = self.client.TriggerWorkflow(
request,
metadata=get_metadata(self.token),
)
id = self.trigger_workflow(workflow_name, input, options)
return WorkflowRunRef(
workflow_run_id=resp.workflow_run_id,
workflow_run_id=id,
workflow_listener=self.pooled_workflow_listener,
workflow_run_event_listener=self.listener_client,
)
Expand All @@ -355,6 +352,21 @@ def run_workflow(

raise ValueError(f"gRPC error: {e}")

def trigger_workflow(
self,
workflow_name: str,
input,
options: TriggerWorkflowOptions = None,
) -> str:
request = self._prepare_workflow_request(workflow_name, input, options)

logger.trace("trigger proto: {}", request)
resp: TriggerWorkflowResponse = self.client.TriggerWorkflow(
request,
metadata=get_metadata(self.token),
)
return resp.workflow_run_id

def run(
self,
function: Union[str, Callable[[Any], T]],
Expand Down
Loading