!python --version

Python 3.12.4

# https://python.langchain.com/docs/how_to/streaming/

from constants import OPENAI_API_KEY

!pip show langchain-openai | grep "Version:"

Version: 0.2.9

import os

os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY

from langchain_openai import ChatOpenAI

model = ChatOpenAI(model="gpt-4o-mini")

from langchain_core.runnables import chain

tech_dict = {
    "py" : "python",
    "ja" : "java"
}

@chain
async def expand_word(word: str):
    if word in tech_dict.keys():
        return tech_dict[word]
    return "unknown"

@chain
async def fill_info(word: str):
    expanded_word = await expand_word.ainvoke(word)
    return expanded_word.title()

await fill_info.ainvoke("py")

'Python'