langgraph_openai_serve 0.0.1

Openai Compatible Langgraph Server

LangGraph OpenAI Serve

A package that provides an OpenAI-compatible API for LangGraph instances.

Features

• Expose your LangGraph instances through an OpenAI-compatible API
• Register multiple graphs and map them to different model names
• Use with any FastAPI application
• Support for both streaming and non-streaming completions

Installation

pip install langgraph-openai-serve

Quick Start

Here's a simple example of how to use LangGraph OpenAI Serve:

from fastapi import FastAPI
from langgraph_openai_serve import LangchainOpenaiApiServe

# Import your LangGraph instances
from your_graphs import simple_graph_1, advanced_graph

# Create a FastAPI app
app = FastAPI(
    title="LangGraph OpenAI API",
    version="1.0",
    description="OpenAI API exposing LangGraph agents",
)

graph_serve = LangchainOpenaiApiServe(
    app=app,
    graphs={
        "simple_graph_1": simple_graph_1,
        "advanced_graph": advanced_graph
    },
    default_graph="simple_graph_1"  # Optional: specify which graph to use by default
)

# Bind the OpenAI-compatible endpoints
graph_serve.bind_openai_chat_completion(prefix="/v1")

# Run the app with uvicorn
if __name__ == "__main__":
    import uvicorn
    uvicorn.run(graph_serve.app, host="0.0.0.0", port=8000)

Using with the OpenAI Client

Once your API is running, you can use any OpenAI-compatible client to interact with it:

from openai import OpenAI

# Create a client pointing to your API
client = OpenAI(
    base_url="http://localhost:8000/v1",
    api_key="any-value"  # API key is not verified
)

# Use a specific graph by specifying its name as the model
response = client.chat.completions.create(
    model="simple_graph_1",  # This maps to the graph name in your registry
    messages=[
        {"role": "user", "content": "Hello, how can you help me today?"}
    ]
)

print(response.choices[0].message.content)

# You can also use streaming
stream = client.chat.completions.create(
    model="advanced_graph",
    messages=[
        {"role": "user", "content": "Write a short poem about AI."}
    ],
    stream=True
)

for chunk in stream:
    if chunk.choices[0].delta.content:
        print(chunk.choices[0].delta.content, end="")

Docker Usage

To run with Docker:

# Start the server
docker compose up -d langgraph-openai-serve-dev

# For a complete example with open-webui
docker compose up -d open-webui