basic_usage.py

"""Basic usage example for the Lumo SDK."""

from lumo_sdk import LumoClient, Message, MessageRole

# Initialize the client
client = LumoClient(api_key="your-api-key-here")

# Example 1: Simple task
print("Example 1: Simple task")
response = client.run_task(
    task="What is the weather in Berlin?",
    model="gpt-4.1-nano",
    base_url="https://api.openai.com/v1/chat/completions",
    tools=["ExaSearchTool", "VisitWebsite"],
)

print(f"Final Answer: {response.final_answer}")
print(f"Number of steps: {len(response.steps)}")
for i, step in enumerate(response.steps, 1):
    print(f"  Step {i}:")
    if step.llm_output:
        print(f"    LLM Output: {step.llm_output[:100]}...")
    if step.tool_calls:
        print(f"    Tools used: {len(step.tool_calls)}")
        for tool_call in step.tool_calls:
            tool_name = tool_call.function.name
            arguments = tool_call.function.arguments
            print(f"      - {tool_name} (ID: {tool_call.id}, Type: {tool_call.type}, Arguments: {arguments})")
    if step.token_usage:
        print(f"    Token Usage: {step.token_usage.total_tokens} tokens")
    else:
        print("    Tools used: None")
print(f"Total Token Usage: {response.token_usage.total_tokens} tokens")
print()

# Example 2: Task with conversation history
print("Example 2: Task with conversation history")
history = [
    Message(role=MessageRole.USER, content="Hello!"),
    Message(role=MessageRole.ASSISTANT, content="Hi! How can I help you?"),
]

response = client.run_task(
    task="What did I just say?",
    model="gpt-4.1-nano",
    base_url="https://api.openai.com/v1/chat/completions",
    history=history,
)

print(f"Final Answer: {response.final_answer}")
print(f"Number of steps: {len(response.steps)}")
for i, step in enumerate(response.steps, 1):
    print(f"  Step {i}:")
    if step.llm_output:
        print(f"    LLM Output: {step.llm_output[:100]}...")
    if step.tool_calls:
        print(f"    Tools used: {len(step.tool_calls)}")
        for tool_call in step.tool_calls:
            tool_name = tool_call.function.name
            arguments = tool_call.function.arguments
            print(f"      - {tool_name} (ID: {tool_call.id}, Type: {tool_call.type}, Arguments: {arguments})")
    else:
        print("    Tools used: None")
print()