import asyncio

from temporalio.client import Client
from temporalio.worker import Worker

from activities import build_prompt, call_llm, evaluate_answer
from workflow import AgentOptimizationWorkflow, BenchmarkWorkflow


async def main():
    client = await Client.connect("localhost:7233")

    worker = Worker(
        client,
        task_queue="agent-optimization",
        workflows=[AgentOptimizationWorkflow, BenchmarkWorkflow],
        activities=[build_prompt, call_llm, evaluate_answer],
    )

    print("Temporal worker started.")
    print("Task queue: agent-optimization")
    await worker.run()


if __name__ == "__main__":
    asyncio.run(main())