send_message.py

import asyncio
import sys

from temporalio.client import Client
from temporalio.envconfig import ClientConfig
from workflows import BedrockParams, EntityBedrockWorkflow


async def main(prompt):
    # Create client connected to server at the given address
    config = ClientConfig.load_client_connect_config()
    config.setdefault("target_host", "localhost:7233")
    client = await Client.connect(**config)

    workflow_id = "entity-bedrock-workflow"

    # Sends a signal to the workflow (and starts it if needed)
    await client.start_workflow(
        EntityBedrockWorkflow.run,
        BedrockParams(None, None),
        id=workflow_id,
        task_queue="bedrock-task-queue",
        start_signal="user_prompt",
        start_signal_args=[prompt],
    )


if __name__ == "__main__":
    if len(sys.argv) != 2:
        print("Usage: python send_message.py '<prompt>'")
        print("Example: python send_message.py 'What animals are marsupials?'")
    else:
        asyncio.run(main(sys.argv[1]))