#!/usr/bin/env python3
"""Sample client for the daimon sidecar.

Usage:
    pip install daimon-client          # from PyPI
    # pip install -e ../../sdk/python  # local dev
    python chat.py                  # runs both components
    python chat.py llama            # run one component
    python chat.py gpt4o claude     # run specific components
"""

import sys

import daimon_client as daimon


def main() -> None:
    client = daimon.Client()
    components = sys.argv[1:] or ["gpt4o", "claude"]
    messages = [
        daimon.Message(role="user", content="What is the capital of France? Answer in one sentence."),
    ]

    for component in components:
        print(f"=== {component} ===")
        for text in client.stream(
            component,
            messages=messages,
            on_tool_call=lambda tc: print(f"\n[calling: {tc.name}]", flush=True),
        ):
            print(text, end="", flush=True)
        print()


if __name__ == "__main__":
    main()