How to create a chat app using OpenAI API

Code

Python

import os
import morph
from morph import MorphGlobalContext
from morph_lib.stream import create_chunk
from openai import OpenAI

@morph.func
def llm_chat_app(context: MorphGlobalContext):
    client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
    prompt = context.vars["prompt"]
    # thread_id can be used to identify the chat thread
    thread_id = context.vars["thread_id"]

    # chat
    messages = [{"role": "user", "content": prompt}]
    response = client.chat.completions.create(
        model="gpt-4o",
        messages=messages,
        stream=True,
    )

    for chunk in response:
        yield create_chunk(chunk.choices[0].delta.content)

    return response

MDX

export const title = "LLM Chat App"

# LLM Chat App

<LLM postData="llm_chat_app" />

Result