Build AI Agents.
Operate with Confidence.

import os

from maivn import Agent, depends_on_tool
from maivn.messages import HumanMessage

agent = Agent(
    name="my-assistant",
    api_key=os.environ["MAIVN_API_KEY"],
    system_prompt="You are a helpful assistant.",
)

@agent.toolify(description="Get current weather for a city")
def get_weather(city: str) -> dict:
    return {"city": city, "forecast": "Sunny, 72F"}

@agent.toolify(description="Summarize the weather forecast", final_tool=True)
@depends_on_tool(get_weather, arg_name="forecast")
def summarize_weather(forecast: dict) -> str:
    return f"Weather in {forecast['city']}: {forecast['forecast']}"

result = agent.invoke(
    [HumanMessage(content="What's the weather in Austin?")],
    force_final_tool=True,
)
print(result.result)