Built-in Agent

1
import asyncio
2
import os
3
from dotenv import load_dotenv
4
from langchain_openai import ChatOpenAI
5
from langgraph.checkpoint.memory import InMemorySaver
6
from thenvoi.agent.langgraph import create_langgraph_agent
7
from thenvoi.config import load_agent_config
8
9
async def main():
10
    load_dotenv()
11
12
    # Load agent credentials
13
    agent_id, api_key = load_agent_config("my_agent")
14
15
    # Create and start the agent
16
    await create_langgraph_agent(
17
        agent_id=agent_id,
18
        api_key=api_key,
19
        llm=ChatOpenAI(model="gpt-4o"),
20
        checkpointer=InMemorySaver(),
21
        ws_url=os.getenv("THENVOI_WS_URL"),
22
        thenvoi_restapi_url=os.getenv("THENVOI_REST_API_URL"),
23
    )
24
25
    print("Agent is running! Press Ctrl+C to stop.")
26
27
if __name__ == "__main__":
28
    asyncio.run(main())

1
await create_langgraph_agent(
2
    agent_id=agent_id,
3
    api_key=api_key,
4
    llm=ChatOpenAI(model="gpt-4o"),
5
    checkpointer=InMemorySaver(),
6
    ws_url=os.getenv("THENVOI_WS_URL"),
7
    thenvoi_restapi_url=os.getenv("THENVOI_REST_API_URL"),
8
    custom_instructions="""
9
    You are a helpful assistant that specializes in answering
10
    questions about Python programming. Be concise and include
11
    code examples when helpful.
12
    """,
13
)

1
# tools.py
2
from langchain_core.tools import tool
3
4
@tool
5
def calculate(operation: str, a: float, b: float) -> str:
6
    """Perform a mathematical calculation.
7
8
    Args:
9
        operation: The operation (add, subtract, multiply, divide)
10
        a: First number
11
        b: Second number
12
    """
13
    operations = {
14
        "add": lambda x, y: x + y,
15
        "subtract": lambda x, y: x - y,
16
        "multiply": lambda x, y: x * y,
17
        "divide": lambda x, y: x / y if y != 0 else "Cannot divide by zero",
18
    }
19
    if operation not in operations:
20
        return f"Unknown operation: {operation}"
21
    return str(operations[operation](a, b))
22
23
@tool
24
async def get_weather(city: str) -> str:
25
    """Get the current weather for a city."""
26
    # TODO: Implement with a weather API (e.g., Open-Meteo)
27
    return "Weather tool not implemented."

1
from tools import calculate, get_weather
2
3
await create_langgraph_agent(
4
    agent_id=agent_id,
5
    api_key=api_key,
6
    llm=ChatOpenAI(model="gpt-4o"),
7
    checkpointer=InMemorySaver(),
8
    ws_url=os.getenv("THENVOI_WS_URL"),
9
    thenvoi_restapi_url=os.getenv("THENVOI_REST_API_URL"),
10
    additional_tools=[calculate, get_weather],
11
    custom_instructions="Use the calculator for math and weather tool for forecasts.",
12
)

1
import asyncio
2
import os
3
from dotenv import load_dotenv
4
from langchain_openai import ChatOpenAI
5
from langchain_core.tools import tool
6
from langgraph.checkpoint.memory import InMemorySaver
7
from thenvoi.agent.langgraph import create_langgraph_agent
8
from thenvoi.config import load_agent_config
9
10
@tool
11
def calculate(operation: str, a: float, b: float) -> str:
12
    """Perform a mathematical calculation.
13
14
    Args:
15
        operation: The operation (add, subtract, multiply, divide)
16
        a: First number
17
        b: Second number
18
    """
19
    operations = {
20
        "add": lambda x, y: x + y,
21
        "subtract": lambda x, y: x - y,
22
        "multiply": lambda x, y: x * y,
23
        "divide": lambda x, y: x / y if y != 0 else "Cannot divide by zero",
24
    }
25
    if operation not in operations:
26
        return f"Unknown operation: {operation}"
27
    return str(operations[operation](a, b))
28
29
async def main():
30
    load_dotenv()
31
    agent_id, api_key = load_agent_config("my_agent")
32
33
    await create_langgraph_agent(
34
        agent_id=agent_id,
35
        api_key=api_key,
36
        llm=ChatOpenAI(model="gpt-4o"),
37
        checkpointer=InMemorySaver(),
38
        ws_url=os.getenv("THENVOI_WS_URL"),
39
        thenvoi_restapi_url=os.getenv("THENVOI_REST_API_URL"),
40
        additional_tools=[calculate],
41
        custom_instructions="""
42
        You are a helpful math tutor. When users ask math questions:
43
        1. Use the calculator tool for computations
44
        2. Explain the steps clearly
45
        3. Offer to help with follow-up questions
46
        """,
47
    )
48
49
    print("Math tutor agent is running! Press Ctrl+C to stop.")
50
51
if __name__ == "__main__":
52
    asyncio.run(main())