pip install multirun

# With LangGraph support
pip install multirun[langgraph]

import asyncio
import multirun
from multirun import run, step
from multirun.models import Budget

multirun.init("http://localhost:8989", api_key="your-api-key")

@step(retry=3, checkpoint=True)
async def call_llm(prompt: str) -> str:
    return f"Response to: {prompt}"

@run(budget=Budget(max_tokens=10000, max_cost_usd=0.10))
async def my_agent(query: str) -> dict:
    response = await call_llm(query)
    return {"result": response}

async def main():
    async with multirun.get_client():
        result = await my_agent("What is the weather?")
        print(result)

asyncio.run(main())