[PYTHON/LANGCHAIN] trimMessageList 함수 : strategy 인자에서 "first" 값 사용하기

■ trimMessageList 함수의 strategy 인자에서 "first" 값을 사용하는 방법을 보여준다.

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

▶ main.py


from dotenv                  import load_dotenv
from langchain_core.messages import SystemMessage
from langchain_core.messages import HumanMessage
from langchain_core.messages import AIMessage
from langchain_openai        import ChatOpenAI
from langchain_core.messages import trim_messages

load_dotenv()

messageList = [
    SystemMessage("you're a good assistant, you always respond with a joke."),
    HumanMessage("i wonder why it's called langchain"),
    AIMessage('Well, I guess they thought "WordRope" and "SentenceString" just didn\'t have the same ring to it!'),
    HumanMessage("and who is harrison chasing anyways"),
    AIMessage("Hmmm let me think.\n\nWhy, he's probably chasing after the last cup of coffee in the office!"),
    HumanMessage("what do you call a speechless parrot")
]

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

trimMessageList = trim_messages(
    messageList,
    max_tokens     = 45,
    strategy       = "first",
    token_counter  = chatOpenAI
)

print(trimMessageList)

"""
[
    SystemMessage(
        content           = "you're a good assistant, you always respond with a joke.",
        additional_kwargs = {},
        response_metadata = {}
    ),
    HumanMessage(
        content           = "i wonder why it's called langchain",
        additional_kwargs = {},
        response_metadata = {}
    )
]
"""

from dotenv import load_dotenv

from langchain_core.messages import SystemMessage

from langchain_core.messages import HumanMessage

from langchain_core.messages import AIMessage

from langchain_openai import ChatOpenAI

from langchain_core.messages import trim_messages

load_dotenv()

messageList = [

SystemMessage("you're a good assistant, you always respond with a joke."),

HumanMessage("i wonder why it's called langchain"),

AIMessage('Well, I guess they thought "WordRope" and "SentenceString" just didn\'t have the same ring to it!'),

HumanMessage("and who is harrison chasing anyways"),

AIMessage("Hmmm let me think.\n\nWhy, he's probably chasing after the last cup of coffee in the office!"),

HumanMessage("what do you call a speechless parrot")

]

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

trimMessageList = trim_messages(

messageList,

max_tokens = 45,

strategy = "first",

token_counter = chatOpenAI

)

print(trimMessageList)

"""

[

SystemMessage(

content = "you're a good assistant, you always respond with a joke.",

additional_kwargs = {},

response_metadata = {}

HumanMessage(

content = "i wonder why it's called langchain",

additional_kwargs = {},

response_metadata = {}

)

]

"""

▶ requirements.txt


annotated-types==0.7.0
anyio==4.7.0
certifi==2024.12.14
charset-normalizer==3.4.0
colorama==0.4.6
distro==1.9.0
h11==0.14.0
httpcore==1.0.7
httpx==0.28.1
idna==3.10
jiter==0.8.2
jsonpatch==1.33
jsonpointer==3.0.0
langchain-core==0.3.25
langchain-openai==0.2.12
langsmith==0.2.3
openai==1.57.4
orjson==3.10.12
packaging==24.2
pydantic==2.10.3
pydantic_core==2.27.1
python-dotenv==1.0.1
PyYAML==6.0.2
regex==2024.11.6
requests==2.32.3
requests-toolbelt==1.0.0
sniffio==1.3.1
tenacity==9.0.0
tiktoken==0.8.0
tqdm==4.67.1
typing_extensions==4.12.2
urllib3==2.2.3

annotated-types==0.7.0

anyio==4.7.0

certifi==2024.12.14

charset-normalizer==3.4.0

colorama==0.4.6

distro==1.9.0

h11==0.14.0

httpcore==1.0.7

httpx==0.28.1

idna==3.10

jiter==0.8.2

jsonpatch==1.33

jsonpointer==3.0.0

langchain-core==0.3.25

langchain-openai==0.2.12

langsmith==0.2.3

openai==1.57.4

orjson==3.10.12

packaging==24.2

pydantic==2.10.3

pydantic_core==2.27.1

python-dotenv==1.0.1

PyYAML==6.0.2

regex==2024.11.6

requests==2.32.3

requests-toolbelt==1.0.0

sniffio==1.3.1

tenacity==9.0.0

tiktoken==0.8.0

tqdm==4.67.1

typing_extensions==4.12.2

urllib3==2.2.3

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