[PYTHON/LANGCHAIN] RunnableWithMessageHistory 클래스 : 템플리트 사용시 생성자의 input_messages_key/history_messages

■ RunnableWithMessageHistory 클래스에서 템플리트 사용시 생성자의 input_messages_key/history_messages_key 인자를 설정하는 방법을 보여준다.

※ OPENAI_API_KEY 환경 변수 값은 .env 파일에 정의한다.

▶ main.py


from dotenv                                     import load_dotenv
from langchain_community.chat_message_histories import SQLChatMessageHistory
from langchain_core.prompts                     import ChatPromptTemplate
from langchain_core.prompts                     import MessagesPlaceholder
from langchain_openai                           import ChatOpenAI
from langchain_core.output_parsers              import StrOutputParser
from langchain_core.runnables.history           import RunnableWithMessageHistory

load_dotenv()

def getSessionHistory(sessionID):
    return SQLChatMessageHistory(sessionID, "sqlite:///memory.db")

chatPromptTemplate = ChatPromptTemplate.from_messages(
    [
        (
            "system",
            "You're an assistant who can speaks in {language}. Respond in 20 words or fewer",
        ),
        MessagesPlaceholder(variable_name = "history"),
        (
            "human",
            "{input}"
        )
    ]
)

chatOpenAI = ChatOpenAI(model = "gpt-3.5-turbo-0125")

runnableSequence = chatPromptTemplate | chatOpenAI | StrOutputParser()

runnableWithMessageHistory = RunnableWithMessageHistory(
    runnableSequence,
    getSessionHistory,
    input_messages_key   = "input",
    history_messages_key = "history",
)

responseString1 = runnableWithMessageHistory.invoke(
    {"language" : "korean", "input" : "hi im bob!"},
    config = {"configurable" : {"session_id" : "1"}}
)

print(responseString1)
print()

responseString2 = runnableWithMessageHistory.invoke(
    {"language" : "korean", "input" : "whats my name?"},
    config = {"configurable" : {"session_id" : "1"}}
)

print(responseString2)
print()

"""
안녕하세요, 밥님! 도와드릴게요. 무엇을 도와드릴까요?

당신의 이름은 밥입니다! 어떻게 도와드릴까요?
"""

from dotenv import load_dotenv

from langchain_community.chat_message_histories import SQLChatMessageHistory

from langchain_core.prompts import ChatPromptTemplate

from langchain_core.prompts import MessagesPlaceholder

from langchain_openai import ChatOpenAI

from langchain_core.output_parsers import StrOutputParser

from langchain_core.runnables.history import RunnableWithMessageHistory

load_dotenv()

def getSessionHistory(sessionID):

return SQLChatMessageHistory(sessionID, "sqlite:///memory.db")

chatPromptTemplate = ChatPromptTemplate.from_messages(

[

(

"system",

"You're an assistant who can speaks in {language}. Respond in 20 words or fewer",

MessagesPlaceholder(variable_name = "history"),

(

"human",

"{input}"

)

]

)

chatOpenAI = ChatOpenAI(model = "gpt-3.5-turbo-0125")

runnableSequence = chatPromptTemplate | chatOpenAI | StrOutputParser()

runnableWithMessageHistory = RunnableWithMessageHistory(

runnableSequence,

getSessionHistory,

input_messages_key = "input",

history_messages_key = "history",

)

responseString1 = runnableWithMessageHistory.invoke(

{"language" : "korean", "input" : "hi im bob!"},

config = {"configurable" : {"session_id" : "1"}}

)

print(responseString1)

print()

responseString2 = runnableWithMessageHistory.invoke(

{"language" : "korean", "input" : "whats my name?"},

config = {"configurable" : {"session_id" : "1"}}

)

print(responseString2)

print()

"""

안녕하세요, 밥님! 도와드릴게요. 무엇을 도와드릴까요?

당신의 이름은 밥입니다! 어떻게 도와드릴까요?

"""

▶ requirements.txt


aiohttp==3.9.5
aiosignal==1.3.1
annotated-types==0.7.0
anyio==4.4.0
async-timeout==4.0.3
attrs==23.2.0
certifi==2024.6.2
charset-normalizer==3.3.2
dataclasses-json==0.6.7
distro==1.9.0
exceptiongroup==1.2.1
frozenlist==1.4.1
greenlet==3.0.3
h11==0.14.0
httpcore==1.0.5
httpx==0.27.0
idna==3.7
jsonpatch==1.33
jsonpointer==3.0.0
langchain==0.2.5
langchain-community==0.2.5
langchain-core==0.2.9
langchain-openai==0.1.9
langchain-text-splitters==0.2.1
langsmith==0.1.81
marshmallow==3.21.3
multidict==6.0.5
mypy-extensions==1.0.0
numpy==1.26.4
openai==1.35.3
orjson==3.10.5
packaging==24.1
pydantic==2.7.4
pydantic_core==2.18.4
python-dotenv==1.0.1
PyYAML==6.0.1
regex==2024.5.15
requests==2.32.3
sniffio==1.3.1
SQLAlchemy==2.0.31
tenacity==8.4.1
tiktoken==0.7.0
tqdm==4.66.4
typing-inspect==0.9.0
typing_extensions==4.12.2
urllib3==2.2.2
yarl==1.9.4

aiohttp==3.9.5

aiosignal==1.3.1

annotated-types==0.7.0

anyio==4.4.0

async-timeout==4.0.3

attrs==23.2.0

certifi==2024.6.2

charset-normalizer==3.3.2

dataclasses-json==0.6.7

distro==1.9.0

exceptiongroup==1.2.1

frozenlist==1.4.1

greenlet==3.0.3

h11==0.14.0

httpcore==1.0.5

httpx==0.27.0

idna==3.7

jsonpatch==1.33

jsonpointer==3.0.0

langchain==0.2.5

langchain-community==0.2.5

langchain-core==0.2.9

langchain-openai==0.1.9

langchain-text-splitters==0.2.1

langsmith==0.1.81

marshmallow==3.21.3

multidict==6.0.5

mypy-extensions==1.0.0

numpy==1.26.4

openai==1.35.3

orjson==3.10.5

packaging==24.1

pydantic==2.7.4

pydantic_core==2.18.4

python-dotenv==1.0.1

PyYAML==6.0.1

regex==2024.5.15

requests==2.32.3

sniffio==1.3.1

SQLAlchemy==2.0.31

tenacity==8.4.1

tiktoken==0.7.0

tqdm==4.66.4

typing-inspect==0.9.0

typing_extensions==4.12.2

urllib3==2.2.2

yarl==1.9.4

※ pip install python-dotenv langchain langchain-community langchain-openai 명령을 실행했다.