OpenRouter

export OPENROUTER_API_KEY="sk-or-..."

from upsonic import infer_model

# Access any supported model
model = infer_model("openrouter/anthropic/claude-3-5-sonnet")

from upsonic.models.openai import OpenAIChatModel, OpenAIChatModelSettings

settings = OpenAIChatModelSettings(
    max_tokens=2048,
    temperature=0.7
)

model = OpenAIChatModel(
    model_name="anthropic/claude-3-5-sonnet",
    provider="openrouter",
    settings=settings
)

from upsonic import Agent, Task, infer_model

model = infer_model("openrouter/anthropic/claude-3-5-sonnet")
agent = Agent(model=model)

task = Task("Explain neural networks")
result = agent.do(task)

from upsonic import infer_model

# Anthropic
model_claude = infer_model("openrouter/anthropic/claude-3-5-sonnet")

# OpenAI
model_gpt = infer_model("openrouter/openai/gpt-4o")

# Google
model_gemini = infer_model("openrouter/google/gemini-2.5-flash")

# Meta
model_llama = infer_model("openrouter/meta-llama/llama-3.1-70b-instruct")

from upsonic import Agent, Task, infer_model

model = infer_model("openrouter/anthropic/claude-3-5-sonnet")
agent = Agent(model=model)

task = Task("Write a detailed article about AI")

async for chunk in agent.do_stream(task):
    print(chunk, end="", flush=True)

from upsonic import Agent, Task, infer_model

def get_weather(location: str) -> str:
    """Get weather for a location."""
    return f"Weather in {location}: Sunny"

model = infer_model("openrouter/openai/gpt-4o")
agent = Agent(model=model, tools=[get_weather])

task = Task("What's the weather in Tokyo?")
result = agent.do(task)

from upsonic import infer_model

# Some models are free on OpenRouter
model = infer_model("openrouter/google/gemini-2.5-flash-lite-free")
agent = Agent(model=model)

task = Task("Tell me about machine learning")
result = agent.do(task)

from upsonic import Agent, Task, infer_model
from upsonic.storage.memory import Memory
from upsonic.storage.providers.in_memory import InMemoryStorage

storage = InMemoryStorage()
memory = Memory(storage=storage, session_id="session-123")

model = infer_model("openrouter/anthropic/claude-3-5-sonnet")
agent = Agent(model=model, memory=memory)

from upsonic.models.openai import OpenAIChatModel, OpenAIChatModelSettings

settings = OpenAIChatModelSettings(
    max_tokens=2048,
    temperature=0.7,
    top_p=0.9,
    presence_penalty=0.1,
    frequency_penalty=0.1
)

model = OpenAIChatModel(
    model_name="anthropic/claude-3-5-sonnet",
    provider="openrouter",
    settings=settings
)

# Use cost-effective models for simple tasks
simple_model = infer_model("openrouter/google/gemini-2.5-flash")

# Reserve expensive models for complex tasks
complex_model = infer_model("openrouter/anthropic/claude-opus-4")

# Use free models for testing
test_model = infer_model("openrouter/google/gemini-2.5-flash-lite-free")

from upsonic import Agent, Task, infer_model
from upsonic.utils.package.exception import ModelHTTPError
import asyncio

async def request_with_retry(agent, task, max_retries=3):
    for attempt in range(max_retries):
        try:
            return agent.do(task)
        except ModelHTTPError as e:
            if e.status_code == 429:  # Rate limit
                wait_time = 2 ** attempt
                print(f"Rate limited, waiting {wait_time}s...")
                await asyncio.sleep(wait_time)
            elif e.status_code >= 500:  # Server error
                wait_time = 2 ** attempt
                print(f"Server error, retrying in {wait_time}s...")
                await asyncio.sleep(wait_time)
            else:
                raise
    raise Exception("Max retries exceeded")