import { query } from '@anthropic-ai/claude-agent-sdk';

// Set environment variables before importing
// ANTHROPIC_BASE_URL=https://router.requesty.ai
// ANTHROPIC_AUTH_TOKEN=your_requesty_api_key

for await (const message of query({
  prompt: 'Your prompt here',
  options: {
    systemPrompt: 'You are a helpful assistant.',
  },
})) {
  // Handle messages
  if (message.type === 'assistant') {
    console.log(message.message.content);
  }
}

"""Minimal agent using Claude Agent SDK."""
import os
import asyncio

from claude_agent_sdk import (
    ClaudeSDKClient,
    ClaudeAgentOptions,
    tool,
    create_sdk_mcp_server,
    AssistantMessage,
    TextBlock,
)

os.environ["ANTHROPIC_BASE_URL"] = "https://router.requesty.ai"
# ANTHROPIC_AUTH_TOKEN should be set as an environment variable


async def main():
    @tool("greet", "Greet a user", {"name": str})
    async def greet_user(args):
        return {
            "content": [
                {"type": "text", "text": f"Hello, {args['name']}!"}
            ]
        }

    server = create_sdk_mcp_server(
        name="my-tools",
        version="1.0.0",
        tools=[greet_user]
    )

    options = ClaudeAgentOptions(
        mcp_servers={"tools": server},
        allowed_tools=["mcp__tools__greet"]
    )

    async with ClaudeSDKClient(options=options) as client:
        await client.query("Greet World")
        
        last_assistant_message = None
        async for msg in client.receive_response():
            if isinstance(msg, AssistantMessage):
                last_assistant_message = msg

        if last_assistant_message:
            print(last_assistant_message.content[0].text)


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