Skip to content

Commit 6b8e441

Browse files
EvanthxVegetarianOrcclaudejmaeagle99jssmith
authored
Sample code for Nexus messaging (#290)
* Working on adding Nexus messaging sample code * Fix to the tests * Updates from a code review * Working on adding Nexus messaging sample code * Fix to the tests * Updates from a code review * Bump pandas to 2.3.3 for Python 3.14 wheel support * removed some unneeded code * Removing workflow.unsafe.imports_passed_through * Bringing in changes from main * Linting * Nexus samples: use business IDs for workflow IDs (#297) * Nexus samples: use business IDs for workflow IDs. Stop using the Nexus request ID (nexus_cancel) or bare uuid4 (hello_nexus, nexus_multiple_args) as the backing workflow ID. Build a meaningful business ID from the operation input and append a uuid4 suffix for uniqueness. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com> * remove UUID from the nexus sample workflow ids to better communicate intent --------- Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com> * Changes from code reviews * Update SDK to 1.27.0 (#302) * Add workflow_streams samples (#300) * Add workflow_streams samples: order_workflow scenario Initial samples directory for temporalio.contrib.workflow_streams, the workflow-hosted durable event stream contrib (experimental, contrib/pubsub branch of sdk-python). The order_workflow scenario covers the basic publisher path: a workflow binds a typed topic in @workflow.init, an activity publishes events via the topic handle, and a starter subscribes with WorkflowStreamClient and prints events as they arrive. Also enables the uv supply-chain cooldown options in the lockfile. * samples: workflow_stream: add reconnecting-subscriber scenario Adds a second scenario demonstrating the central Workflow Streams use case: a consumer disconnects mid-stream and resumes later via subscribe(from_offset=...), with no events lost or duplicated. The existing OrderWorkflow finishes too quickly to make the pattern visible, so this introduces a multi-stage PipelineWorkflow paced with workflow.sleep between stages. The runner reads a couple of events, persists item.offset + 1 to a temp file, sleeps "disconnected" while the workflow keeps publishing, then opens a fresh Client + WorkflowStreamClient and resumes from the persisted offset — the same shape that works across actual process restarts. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com> * samples: workflow_stream: add external-publisher scenario Adds a third scenario covering the third publisher shape: a backend service or scheduled job pushing events into a workflow it didn't itself start. The earlier scenarios publish either from inside the workflow or from one of its activities; this one uses WorkflowStreamClient.create() externally. HubWorkflow is a passive stream host — it does no work of its own and just waits to be told to close, fitting the event-bus pattern. The runner publishes a series of news headlines, runs a subscriber task alongside, signals close, and exits when both tasks complete. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com> * samples: workflow_stream: add truncating-ticker scenario Adds a fourth scenario for long-running workflows that need to bound their event log: the workflow publishes events at a fixed cadence and calls self.stream.truncate(...) periodically to keep only the most recent entries. The runner subscribes twice — fast and slow — to make the trade visible: the fast subscriber sees every offset in order; the slow one falls behind a truncation, has its iterator transparently jump forward to the new base offset, and shows the offset gap that intermediate events fell into. This is the model for high-volume long-running streams: bounded log size, slow consumers may miss intermediate events but always see the most recent state. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com> * samples: rename workflow_stream → workflow_streams; migrate to topic handles - Directory and module path renamed to plural to match sdk-python `temporalio.contrib.workflow_streams` rename. - Workflow-side: bind a typed topic handle in `@workflow.init` and call `topic.publish(value)` — the removed `WorkflowStream.publish` form is gone. Same change applied to the activity and external-publisher. - Activity: `WorkflowStreamClient.from_activity()` → `from_within_activity()`. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com> * samples: workflow_streams review polish - README: fix scenario count (two -> four), document subscriber start position and continue-as-new semantics for stream_state - hub_workflow: drop stale comment referencing a README race note that does not exist in this sample - payment_activity: trim long publisher_id/dedup caveat — moved out of the first sample's docstring to keep it approachable * workflow_streams: deliver terminal events + fix run_publisher subscribe shape End-to-end runs of the four workflow_streams scenarios surfaced two sample-side issues, both fixed here. run_publisher's consumer asserted ``isinstance(item.data, Payload)`` and called ``payload_converter.from_payload(item.data, T)``. The contrib's ``subscribe()`` defaults to converter-decoded data, not raw payloads, so this assertion fired on the first run. Switch to ``result_type=RawValue`` (the documented escape hatch for heterogeneous topics) and read ``item.data.payload``. Items published in the same workflow task that returns from ``@workflow.run`` were not delivered to subscribers — the in-memory log dies with the workflow and the next subscriber poll lands on a completed workflow. Fix: each scenario now uses an in-band terminator that subscribers break on, and each workflow holds the run open with ``await workflow.sleep(timedelta(milliseconds=500))`` so that final publish is fetched before the workflow exits: - OrderWorkflow / PipelineWorkflow: the workflow's own ``StatusEvent(kind="complete")`` / ``StageEvent(stage="complete")`` is the terminator (consumers already broke on it). - HubWorkflow: the *publisher* in run_external_publisher emits a sentinel ``NewsEvent(headline="__done__")`` immediately before signaling close; the consumer breaks on the sentinel. - TickerWorkflow: the final tick (n == count - 1) is the terminator; ``keep_last`` guarantees that offset survives the last truncation, so even slow consumers reach it. Because subscribers stop polling on the terminator, by the time ``workflow.run`` returns there are no in-flight poll handlers — no ``UnfinishedUpdateHandlersWarning`` from the SDK and no need for ``detach_pollers()`` / ``wait_condition(all_handlers_finished)`` in the workflow exit path. Two consecutive end-to-end runs of all four scenarios pass cleanly against ``temporal server start-dev --headless``. * workflow_streams README: document the stream-end pattern Subscribers don't exit on their own when the host workflow completes — they need an in-band terminator, and the workflow needs to hold open briefly so the final publish is fetched before run() returns. Both pieces show up in every scenario here, so document them in one place and update scenario 3's description to mention the sentinel headline the publisher emits. * samples: workflow_streams: README and wheel packages cleanup Now that temporalio 1.27.0 has shipped (and main has bumped to it in #302), drop the README's "install sdk-python from a branch" callout and point at >=1.27.0 instead. Also add workflow_streams to the wheel packages list alongside the other samples. * samples: workflow_streams: drop force_flush=True from charge_card The activity's final publish was using force_flush=True, which sets the flush_event so the background flusher fires immediately. Triggering a flush right before __aexit__ runs the activity into the WorkflowStreamClient's cancel-mid-flush path: __aexit__ cancels the flusher task while it's awaiting the publish signal RPC, the cancel propagates into the in-flight signal, and the activity hangs until the StartToClose timeout fires. Empirically the workflow then retries the activity indefinitely. Without force_flush=True the buffered "card charged" event flushes via the regular 200ms batch interval and the flusher is sleeping in wait_for(...) when __aexit__ cancels it — a clean cancellation path. The user-visible publish ordering is unchanged. The underlying SDK bug should be fixed separately by switching __aexit__ from cancel() to a cooperative-stop flag so the in-flight signal completes before the flusher exits. * samples: workflow_streams: drop temp-file resume offset; add stats column The reconnecting-subscriber demo previously persisted its resume offset to a temp file between phases. Inside one process that's theatrical: the disconnect/reconnect shape comes from creating a fresh Client + WorkflowStreamClient with from_offset=N, not from where N happens to be stored. Replace the file with a local int and a comment about durable storage in production (a DB row keyed by user_id/run_id, etc.). Restructure output around a stats column so the demo conveys what's happening to the stream at all times, not just between phases. A background poller calls WorkflowStreamClient.get_offset() throughout and emits a heartbeat line once a second; every emit prints current proc/avail/pend in a left column followed by the phase or event message. Watching pend grow during the disconnect window and shrink again as phase 2 catches up is the demo's core point. * samples: workflow_streams: surface multiple truncation jumps in ticker The truncating-ticker demo is meant to make the bounded-log trade visible: fast subscriber sees every event, slow subscriber loses intermediate ones to truncation. The previous parameters (truncate_every=5, keep_last=3, interval_ms=400, slow_delay=1.5s) produced at most one tiny jump near the end of the run — easy to miss. Tighter parameters (truncate_every=2, keep_last=1, interval_ms=200, count=30) keep the workflow log at one or two entries between truncations. That shrinks the slow subscriber's per-poll batch, so it re-polls more often, and most polls land after a truncation that has passed its position. The result is several visible jumps over the demo, not a single batched one at the end. Switch the output to two lanes (fast on the left, slow on the right with explicit "↪ jumped offset=N → M (K dropped)" markers) so the divergence reads at a glance instead of being lost in interleaved single-stream output. Also extend the docstring to call out the opposite trade — never truncating means slow consumers eventually catch up at the cost of unbounded workflow history — so readers know when this pattern is the wrong fit. * samples: workflow_streams: add LLM-streaming scenario Adds a fifth scenario to workflow_streams/ that streams an OpenAI chat completion to the terminal through a Workflow Stream. Activity is the publisher (it owns the non-deterministic API call), workflow hosts the stream and runs the activity, runner subscribes and renders to stdout as deltas arrive. Layout: * `chat_shared.py` — types and topics for this scenario, kept out of the cross-scenario `shared.py` because no other scenario uses them * `workflows/chat_workflow.py` — `ChatWorkflow` runs `stream_completion` with `RetryPolicy(maximum_attempts=3)` and the same 500ms hold-open pattern the other four samples use * `activities/chat_activity.py` — `stream_completion` calls `AsyncOpenAI(...).chat.completions.create(stream=True)` with `gpt-5-mini`, publishes each token chunk on the `delta` topic, the full text on `complete`, and a `RetryEvent` on `retry` when running on attempt > 1. `force_flush=True` is intentionally omitted to avoid the `__aexit__` cancel-mid-flight hang in `temporalio.contrib.workflow_streams` 1.27.0; the 200ms `batch_interval` is fast enough for an interactive feel. * `run_chat.py` — subscribes to all three topics, prints deltas to stdout as they stream, and on a retry event uses plain ANSI escapes (`\033[<n>A`, `\033[J`) to rewind the rendered output before the retried attempt re-publishes * `run_chat_worker.py` — runs on its own task queue (`workflow-stream-chat-task-queue`), registering only `ChatWorkflow` and `stream_completion`; the openai dependency and the `OPENAI_API_KEY` requirement stay isolated to this one scenario The split worker also makes the retry-handling demo trivial to run: the user kills the chat worker mid-stream, brings it back up, and the activity retries — no synthetic failure injection needed. Adds `chat-stream = ["openai>=1.0,<2"]` as a new optional dependency group; `uv sync --group chat-stream` and an `OPENAI_API_KEY` are documented in the README. * samples: workflow_streams: drop chat-stream openai upper cap openai-agents (the existing langsmith-tracing / openai-agents extras) already pulls openai>=2.26.0. Capping chat-stream at openai<2 made the two extras unsatisfiable together. Drop the cap; the chat activity uses APIs that are stable across openai 1.x and 2.x. * samples: workflow_streams: chat consumer header + cursor save/restore Two display fixes for run_chat.py: 1. Print a header line right after start_workflow so the user sees immediate feedback ("[chat <id>] streaming response from gpt-5-mini, awaiting first token...") instead of a blank screen until the first delta arrives. 2. Replace the newline-counting ANSI clear with cursor save/restore (\033[s / \033[u\033[J). The previous version counted text newlines to decide how far up to move the cursor on retry, which undercounts when the terminal has wrapped long lines — the failed attempt's first wrapped lines stayed on screen above the retry marker. save/restore rewinds to a fixed position regardless of wrapping. Bumps the prompt to a 500-word distributed-systems comparison (Paxos vs Raft vs Viewstamped Replication) so there is enough output to comfortably kill the worker mid-stream and watch the retried attempt re-render from scratch. * samples: workflow_streams: rename chat -> llm in scenario 5 "Chat" implies multi-turn conversation. The new scenario is a one-shot LLM completion stream, not a chat. Rename to make the scope clear: - chat_shared.py -> llm_shared.py - workflows/chat_workflow.py -> workflows/llm_workflow.py - activities/chat_activity.py -> activities/llm_activity.py - run_chat.py -> run_llm.py - run_chat_worker.py -> run_llm_worker.py - ChatInput / ChatWorkflow -> LLMInput / LLMWorkflow - CHAT_TASK_QUEUE -> LLM_TASK_QUEUE ("workflow-stream-chat-task-queue" -> "workflow-stream-llm-task-queue") - chat-stream extra -> llm-stream - workflow id prefix workflow-stream-chat-... -> workflow-stream-llm-... The activity's `stream_completion` defn name and the topic constants (`delta`, `complete`, `retry`) stay the same — those already describe what they do without the "chat" framing. README, docstrings, and run instructions updated to match. * samples: workflow_streams: race the LLM consumer with workflow result If the LLM activity exhausts its retries (bad OPENAI_API_KEY, provider outage, etc.), the workflow fails before the activity publishes the `complete` terminator. The consumer's previous async-for loop only exited on `complete`, so the script blocked indefinitely on a terminator that would never arrive instead of surfacing the workflow failure. Wrap the subscriber in a `consume()` coroutine and run it through the existing `race_with_workflow` helper (the same pattern `run_publisher.py` uses): if the workflow finishes first the subscriber gets cancelled and the workflow's exception propagates; if the subscriber sees `complete` first, the helper waits for the workflow result and returns it. Found in a Codex code review of today's workflow_streams changes. * samples: workflow_streams: drop race_with_workflow helper The helper wrapped the consumer in an asyncio.gather that cancelled the subscriber when the workflow result settled — defensive logic for a case the SDK already handles. WorkflowStreamClient.subscribe() exits cleanly on every workflow terminal state (return, continue-as-new, failure) via its AcceptedUpdateCompletedWorkflow, WorkflowUpdateRPCTimeoutOrCancelledError, and NOT_FOUND branches in sdk-python. The async-for loop ends naturally when the workflow terminates without a publish, so we don't need a separate task to race against handle.result(). Replace the helper with the obvious shape in both runners: async for item in stream.subscribe(...): ... if item.is_terminator: break result = await handle.result() # raises on workflow failure Either path reaches handle.result(): an explicit break on the in-band terminator (workflow still running, hold-open lets the poll deliver the event), or the iterator naturally exhausting when the workflow has already terminated. handle.result() then either returns or raises the workflow's failure — covering the LLM "activity exhausted retries" case that prompted the helper to be added in the first place. Smoke tested: uv run workflow_streams/run_publisher.py uv run workflow_streams/run_llm.py * samples: workflow_streams: reorganize README; drop closing section Two fixes: 1. Reorganize so the README doesn't jump back and forth between scenarios. The previous shape introduced 1-4, then put scenario 5's full description plus its setup and run instructions inline, then jumped back to a "Run it" section that only covered 1-4. New shape: all five scenarios up front (parallel structure), one unified "Run it" section that covers worker setup for both groups and all five runner scripts in one block, then expected output, then notes. 2. Drop the inline "Ending the stream" section. The same material is in documentation/docs/develop/python/libraries/workflow-streams.mdx under the "Closing the stream" anchor, so the README links there from the Notes block instead of duplicating the explanation. The scenario 5 "split-out worker" rationale (extra dependency, secret, retry-via-Ctrl-C) collapses to a single sentence at the end of its bullet block. * samples: workflow_streams: drop README Notes section The Notes block (subscriber start position, continue-as-new, closing the stream) was a small docs summary tacked onto the end of the README. The samples themselves cover these points: docstrings in each runner / workflow / activity explain the from_offset behavior, the stream_state field, and the in-band terminator + hold-open pattern. Readers who want the full conceptual treatment go to the docs page; the README sticks to "what the scenarios are and how to run them". * samples: workflow_streams: lock llm-stream dependency group The llm-stream dependency group was introduced in pyproject.toml without a corresponding uv.lock update, so `uv sync --frozen --group llm-stream` would fail or force a relock before scenario 5 could run. Add the two missing entries (the package-optional-dependencies list and the package-metadata requires-dev list) so frozen installs work against the committed lock. Found in a Codex review of the day's workflow_streams changes. * samples: workflow_streams: fix lint failures (ruff isort + format) CI's `poe lint` step was failing on three small things across four files: * `run_external_publisher.py`, `ticker_workflow.py`: ruff isort (`I001`) wanted the `workflow_streams.shared` imports re-sorted and a stray blank line removed. Apply the auto-fix. * `run_external_publisher.py`, `run_reconnecting_subscriber.py`, `run_truncating_ticker.py`: ruff format wanted three line-wrapped function calls collapsed back to single lines. Apply the formatter. * `run_truncating_ticker.py`: the formatter joined an adjacent pair of f-strings into an awkward `f"..." f"..."` one-liner. Consolidate them into a single f-string for readability — the resulting line is comfortably under the 88-char limit. `poe lint` (ruff isort + ruff format --check + mypy --all-groups --check-untyped-defs) now passes locally. * samples: workflow_streams: drop BFF jargon and Expected output block Two README/comment cleanups: * "BFF" (backend-for-frontend) is not a widely-known term outside certain front-end-architecture circles. Replace with the more obvious "web backends" in the README intro and "production web backend" in the run_reconnecting_subscriber.py comment about where the resume offset would live durably. * Drop the "Expected output" section. It only covered scenarios 1 and 2; with five scenarios it is no longer pulling its weight. Anyone running the script can see the output for themselves. * Apply suggestion from @brianstrauch * Apply suggestion from @brianstrauch --------- Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com> Co-authored-by: Brian Strauch <brian@brianstrauch.com> * Working on adding Nexus messaging sample code * Fix to the tests * Updates from a code review * Working on adding Nexus messaging sample code * Bump pandas to 2.3.3 for Python 3.14 wheel support * removed some unneeded code * Removing workflow.unsafe.imports_passed_through * Bringing in changes from main * Linting * Changes from code reviews --------- Co-authored-by: Alex Mazzeo <alex.mazzeo@temporal.io> Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com> Co-authored-by: Justin Anderson <44687433+jmaeagle99@users.noreply.github.com> Co-authored-by: Johann Schleier-Smith <johann.schleiersmith@temporal.io> Co-authored-by: Brian Strauch <brian@brianstrauch.com>
1 parent 118083b commit 6b8e441

35 files changed

Lines changed: 1241 additions & 305 deletions

README.md

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -75,6 +75,9 @@ Some examples require extra dependencies. See each sample's directory for specif
7575
* [message_passing/introduction](message_passing/introduction/) - Introduction to queries, signals, and updates.
7676
* [message_passing/safe_message_handlers](message_passing/safe_message_handlers/) - Safely handling updates and signals.
7777
* [message_passing/update_with_start/lazy_initialization](message_passing/update_with_start/lazy_initialization/) - Use update-with-start to update a Shopping Cart, starting it if it does not exist.
78+
* [Nexus Messaging](nexus_messaging): Demonstrates how send signal, update and query messages through Nexus.
79+
This contains two samples, one sending messages to an existing workflow and a second that creates a workflow through Nexus
80+
and sends messages to it.
7881
* [open_telemetry](open_telemetry) - Trace workflows with OpenTelemetry.
7982
* [patching](patching) - Alter workflows safely with `patch` and `deprecate_patch`.
8083
* [polling](polling) - Recommended implementation of an activity that needs to periodically poll an external resource waiting its successful completion.

nexus_messaging/README.md

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
This sample shows how to expose a long-running Workflow's queries, updates, and signals as Nexus
2+
operations. There are two self-contained examples, each in its own directory:
3+
4+
| | `callerpattern/` | `ondemandpattern/` |
5+
|--------------------------------|--------------------------------------|--------------------------------------------------------------|
6+
| **Pattern** | Signal an existing Workflow | Create and run Workflows on demand, and send signals to them |
7+
| **Who creates the Workflow?** | The handler worker starts it on boot | The caller starts it via a Nexus operation |
8+
| **Who knows the Workflow ID?** | Only the handler | The caller chooses and passes it in every operation |
9+
| **Nexus service** | `NexusGreetingService` | `NexusRemoteGreetingService` |
10+
11+
Each directory is fully self-contained for clarity. The `GreetingWorkflow`, activity, and
12+
`Language` enum are **identical** between the two -- only the Nexus service definition and its
13+
handler implementation differ. This highlights that the same Workflow can be exposed through
14+
Nexus in different ways depending on whether the caller needs lifecycle control.
15+
16+
See each directory's README for running instructions.
Lines changed: 56 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,56 @@
1+
## Caller pattern
2+
3+
The handler worker starts a `GreetingWorkflow` for a User ID.
4+
`NexusGreetingServiceHandler` holds that ID and routes every Nexus operation to it.
5+
The caller's input does not have that Workflow ID as the caller doesn't know it -- but the caller
6+
sends in the User ID, and `NexusGreetingServiceHandler` knows how to get the desired Workflow ID
7+
from that User ID (see the `get_workflow_id` call).
8+
9+
The handler worker uses the same `get_workflow_id` call to generate a Workflow ID from a Wser ID
10+
when it launches the Workflow.
11+
12+
The caller Workflow:
13+
1. Queries for supported languages (`get_languages` -- backed by a `@workflow.query`)
14+
2. Changes the language to Arabic (`set_language` -- backed by a `@workflow.update` that calls an activity)
15+
3. Confirms the change via a second query (`get_language`)
16+
4. Approves the Workflow (`approve` -- backed by a `@workflow.signal`)
17+
18+
### Running
19+
20+
Start a Temporal server:
21+
22+
```bash
23+
temporal server start-dev
24+
```
25+
26+
Create the namespaces and Nexus endpoint:
27+
28+
```bash
29+
temporal operator namespace create --namespace nexus-messaging-handler-namespace
30+
temporal operator namespace create --namespace nexus-messaging-caller-namespace
31+
32+
temporal operator nexus endpoint create \
33+
--name nexus-messaging-nexus-endpoint \
34+
--target-namespace nexus-messaging-handler-namespace \
35+
--target-task-queue nexus-messaging-handler-task-queue
36+
```
37+
38+
In one terminal, start the handler worker:
39+
40+
```bash
41+
uv run python -m nexus_messaging.callerpattern.handler.worker
42+
```
43+
44+
In another terminal, run the following command to start the example:
45+
46+
```bash
47+
uv run python -m nexus_messaging.callerpattern.caller.app
48+
```
49+
50+
Expected output:
51+
52+
```
53+
Supported languages: [<Language.CHINESE: 2>, <Language.ENGLISH: 3>]
54+
Language changed: ENGLISH -> ARABIC
55+
Workflow approved
56+
```
File renamed without changes.

nexus_sync_operations/caller/app.py renamed to nexus_messaging/callerpattern/caller/app.py

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -6,15 +6,13 @@
66
from temporalio.envconfig import ClientConfig
77
from temporalio.worker import Worker
88

9-
from nexus_sync_operations.caller.workflows import CallerWorkflow
9+
from nexus_messaging.callerpattern.caller.workflows import CallerWorkflow
1010

11-
NAMESPACE = "nexus-sync-operations-caller-namespace"
12-
TASK_QUEUE = "nexus-sync-operations-caller-task-queue"
11+
NAMESPACE = "nexus-messaging-caller-namespace"
12+
TASK_QUEUE = "nexus-messaging-caller-task-queue"
1313

1414

15-
async def execute_caller_workflow(
16-
client: Optional[Client] = None,
17-
) -> None:
15+
async def execute_caller_workflow(client: Optional[Client] = None) -> None:
1816
if client is None:
1917
config = ClientConfig.load_client_connect_config()
2018
config.setdefault("target_host", "localhost:7233")
@@ -28,7 +26,8 @@ async def execute_caller_workflow(
2826
):
2927
log = await client.execute_workflow(
3028
CallerWorkflow.run,
31-
id=str(uuid.uuid4()),
29+
arg="user-1",
30+
id=f"nexus-messaging-caller-{uuid.uuid4()}",
3231
task_queue=TASK_QUEUE,
3332
)
3433
for line in log:
Lines changed: 73 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,73 @@
1+
"""
2+
A caller workflow that executes Nexus operations. The caller does not have information
3+
about how these operations are implemented by the Nexus service.
4+
"""
5+
6+
from temporalio import workflow
7+
from temporalio.exceptions import ApplicationError
8+
9+
from nexus_messaging.callerpattern.service import (
10+
ApproveInput,
11+
GetLanguageInput,
12+
GetLanguagesInput,
13+
Language,
14+
NexusGreetingService,
15+
SetLanguageInput,
16+
)
17+
18+
NEXUS_ENDPOINT = "nexus-messaging-nexus-endpoint"
19+
20+
21+
@workflow.defn
22+
class CallerWorkflow:
23+
@workflow.run
24+
async def run(self, user_id: str) -> list[str]:
25+
log: list[str] = []
26+
nexus_client = workflow.create_nexus_client(
27+
service=NexusGreetingService,
28+
endpoint=NEXUS_ENDPOINT,
29+
)
30+
31+
# Call a Nexus operation backed by a query against the entity workflow.
32+
# The workflow must already be running on the handler, otherwise you will
33+
# get an error saying the workflow has already terminated.
34+
languages_output = await nexus_client.execute_operation(
35+
NexusGreetingService.get_languages,
36+
GetLanguagesInput(include_unsupported=False, user_id=user_id),
37+
)
38+
log.append(f"Supported languages: {languages_output.languages}")
39+
workflow.logger.info("Supported languages: %s", languages_output.languages)
40+
41+
# Following are examples for each of the three messaging types -
42+
# update, query, then signal.
43+
44+
# Call a Nexus operation backed by an update against the entity workflow.
45+
previous_language = await nexus_client.execute_operation(
46+
NexusGreetingService.set_language,
47+
SetLanguageInput(language=Language.ARABIC, user_id=user_id),
48+
)
49+
50+
# Call a Nexus operation backed by a query to confirm the language change.
51+
current_language = await nexus_client.execute_operation(
52+
NexusGreetingService.get_language,
53+
GetLanguageInput(user_id=user_id),
54+
)
55+
if current_language != Language.ARABIC:
56+
raise ApplicationError(f"Expected language ARABIC, got {current_language}")
57+
58+
log.append(
59+
f"Language changed: {previous_language.name} -> {Language.ARABIC.name}"
60+
)
61+
workflow.logger.info(
62+
"Language changed from %s to %s", previous_language, Language.ARABIC
63+
)
64+
65+
# Call a Nexus operation backed by a signal against the entity workflow.
66+
await nexus_client.execute_operation(
67+
NexusGreetingService.approve,
68+
ApproveInput(name="caller", user_id=user_id),
69+
)
70+
log.append("Workflow approved")
71+
workflow.logger.info("Workflow approved")
72+
73+
return log

nexus_messaging/callerpattern/handler/__init__.py

Whitespace-only changes.
Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
1+
import asyncio
2+
from typing import Optional
3+
4+
from temporalio import activity
5+
6+
from nexus_messaging.callerpattern.service import Language
7+
8+
9+
@activity.defn
10+
async def call_greeting_service(language: Language) -> Optional[str]:
11+
"""Simulates a call to a remote greeting service. Returns None if unsupported."""
12+
greetings = {
13+
Language.ARABIC: "\u0645\u0631\u062d\u0628\u0627 \u0628\u0627\u0644\u0639\u0627\u0644\u0645",
14+
Language.CHINESE: "\u4f60\u597d\uff0c\u4e16\u754c",
15+
Language.ENGLISH: "Hello, world",
16+
Language.FRENCH: "Bonjour, monde",
17+
Language.HINDI: "\u0928\u092e\u0938\u094d\u0924\u0947 \u0926\u0941\u0928\u093f\u092f\u093e",
18+
Language.PORTUGUESE: "Ol\u00e1 mundo",
19+
Language.SPANISH: "Hola mundo",
20+
}
21+
await asyncio.sleep(0.2)
22+
return greetings.get(language)

0 commit comments

Comments
 (0)