Langchainで作成したAIエージェントをチャットアプリ化する

Prerequisites

1. langchainのパッケージをインストールする

pip install langchain

2. OPENAI_KEYを.envにセットする

OPENAI_KEY=xxxxxxxx

Code

Python

import os
import morph
from morph import MorphGlobalContext
from langchain_openai import ChatOpenAI
from langchain_core.messages import HumanMessage

@morph.func
def langchain_chat_app(context: MorphGlobalContext):
    prompt = context.vars["prompt"]
    # thread_id can be used to identify the chat thread
    thread_id = context.vars["thread_id"]
    model = ChatOpenAI(model="gpt-4o-mini")

    # chat
    messages = [HumanMessage(content=prompt)]
    response = model.invoke(messages)

    for token in model.stream(messages):
        yield token.content

MDX

export const title = "LangChain Chat App"

# LangChain Chat App

<LLM postData="langchain_chat_app" />

Result