LangGraph Adapter

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