import asyncio

from temporalio.client import Client
from workflows.tool_workflow import ToolWorkflow


async def main():
    # Create client connected to server at the given address
    client = await Client.connect("localhost:7233")

    workflow_id = "ollama-agent"

    handle = client.get_workflow_handle_for(ToolWorkflow.run, workflow_id)

    # Sends a signal to the workflow
    await handle.signal(ToolWorkflow.end_chat)


if __name__ == "__main__":
    print("Sending signal to end chat.")
    asyncio.run(main())