sample_agent.py

"""Simple Q&A agent that answers questions about Python using the Anthropic SDK."""

import anthropic


def create_python_qa_agent():
    """Create a multi-turn Q&A agent that answers Python questions."""
    client = anthropic.Anthropic()

    system_prompt = (
        "You are a Python expert. Answer questions about Python clearly and concisely. "
        "Include short code examples when helpful. If a question is not about Python, "
        "politely redirect the user to ask a Python-related question."
    )

    messages = []

    def ask(question: str) -> str:
        """Send a question and return the answer."""
        messages.append({"role": "user", "content": question})

        with client.messages.stream(
            model="claude-opus-4-6",
            max_tokens=16000,
            system=system_prompt,
            thinking={"type": "adaptive"},
            messages=messages,
        ) as stream:
            response = stream.get_final_message()

        assistant_text = next(
            (b.text for b in response.content if b.type == "text"), ""
        )
        messages.append({"role": "assistant", "content": response.content})

        return assistant_text

    return ask


def main():
    """Interactive Q&A loop for Python questions."""
    ask = create_python_qa_agent()

    print("Python Q&A Agent (powered by Claude)")
    print("Type 'quit' to exit.\n")

    while True:
        question = input("Ask about Python: ").strip()
        if question.lower() in ("quit", "exit", "q"):
            break
        if not question:
            continue

        try:
            answer = ask(question)
            print(f"\n{answer}\n")
        except anthropic.APIError as e:
            print(f"\nAPI error: {e}\n")


if __name__ == "__main__":
    main()