LangGraph Adapter

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 import Agent
7
from thenvoi.adapters import LangGraphAdapter
8
from thenvoi.config import load_agent_config
9
10
async def main():
11
    load_dotenv()
12
13
    # Load agent credentials
14
    agent_id, api_key = load_agent_config("my_agent")
15
16
    # Create adapter with LLM and checkpointer
17
    adapter = LangGraphAdapter(
18
        llm=ChatOpenAI(model="gpt-4o"),
19
        checkpointer=InMemorySaver(),
20
    )
21
22
    # Create and run the agent
23
    agent = Agent.create(
24
        adapter=adapter,
25
        agent_id=agent_id,
26
        api_key=api_key,
27
        ws_url=os.getenv("THENVOI_WS_URL"),
28
        rest_url=os.getenv("THENVOI_REST_URL"),
29
    )
30
31
    print("Agent is running! Press Ctrl+C to stop.")
32
    await agent.run()
33
34
if __name__ == "__main__":
35
    asyncio.run(main())

1
adapter = LangGraphAdapter(
2
    llm=ChatOpenAI(model="gpt-4o"),
3
    checkpointer=InMemorySaver(),
4
    custom_section="""
5
    You are a helpful assistant that specializes in answering
6
    questions about Python programming. Be concise and include
7
    code examples when helpful.
8
    """,
9
)

1
from langchain_core.tools import tool
2
3
@tool
4
def calculate(operation: str, a: float, b: float) -> str:
5
    """Perform a mathematical calculation.
6
7
    Args:
8
        operation: The operation (add, subtract, multiply, divide)
9
        a: First number
10
        b: Second number
11
    """
12
    operations = {
13
        "add": lambda x, y: x + y,
14
        "subtract": lambda x, y: x - y,
15
        "multiply": lambda x, y: x * y,
16
        "divide": lambda x, y: x / y if y != 0 else "Cannot divide by zero",
17
    }
18
    if operation not in operations:
19
        return f"Unknown operation: {operation}"
20
    return str(operations[operation](a, b))

1
adapter = LangGraphAdapter(
2
    llm=ChatOpenAI(model="gpt-4o"),
3
    checkpointer=InMemorySaver(),
4
    additional_tools=[calculate],
5
    custom_section="Use the calculator for math questions.",
6
)

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

1
import asyncio
2
import os
3
import logging
4
from dotenv import load_dotenv
5
from langchain_openai import ChatOpenAI
6
from langgraph.checkpoint.memory import InMemorySaver
7
from thenvoi import Agent
8
from thenvoi.adapters import LangGraphAdapter
9
from thenvoi.config import load_agent_config
10
11
# Enable debug logging for the SDK
12
logging.basicConfig(
13
    level=logging.WARNING,
14
    format="%(asctime)s [%(levelname)s] %(name)s: %(message)s",
15
    datefmt="%Y-%m-%d %H:%M:%S",
16
)
17
logging.getLogger("thenvoi").setLevel(logging.DEBUG)
18
19
async def main():
20
    load_dotenv()
21
    agent_id, api_key = load_agent_config("my_agent")
22
23
    adapter = LangGraphAdapter(
24
        llm=ChatOpenAI(model="gpt-4o"),
25
        checkpointer=InMemorySaver(),
26
    )
27
28
    agent = Agent.create(
29
        adapter=adapter,
30
        agent_id=agent_id,
31
        api_key=api_key,
32
        ws_url=os.getenv("THENVOI_WS_URL"),
33
        rest_url=os.getenv("THENVOI_REST_URL"),
34
    )
35
36
    print("Agent running with DEBUG logging. Press Ctrl+C to stop.")
37
    await agent.run()
38
39
if __name__ == "__main__":
40
    asyncio.run(main())