[PYTHON/LANGCHAIN] ChatPromptTemplate 클래스 : from_messages 정적 메소드를 사용해 퓨샷 프롬프트 만들기

■ ChatPromptTemplate 클래스의 from_messages 정적 메소드를 사용해 퓨샷 프롬프트를 만드는 방법을 보여준다.

▶ main.py


import os

from typing import Optional
from typing import Union

from langchain_core.pydantic_v1 import BaseModel, Field
from langchain_openai           import ChatOpenAI
from langchain_core.prompts     import ChatPromptTemplate

os.environ["OPENAI_API_KEY"] = "<OPENAI_API_KEY>"

class Joke(BaseModel):
    """Joke to tell user."""

    setup     : str           = Field(description = "The setup of the joke"              )
    punchline : str           = Field(description = "The punchline to the joke"          )
    rating    : Optional[int] = Field(description = "How funny the joke is, from 1 to 10")

class ConversationalResponse(BaseModel):
    """Respond in a conversational manner. Be kind and helpful."""

    response : str = Field(description = "A conversational response to the user's query")

class Response(BaseModel):
    output : Union[Joke, ConversationalResponse]

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

structuredOutputRunnableSequence = chatOpenAI.with_structured_output(Response)

systemMessage = """You are a hilarious comedian. Your specialty is knock-knock jokes. \
Return a joke which has the setup (the response to "Who's there?") and the final punchline (the response to "<setup> who?").

Here are some examples of jokes:

example_user : Tell me a joke about planes
example_assistant : {{"setup" : "Why don't planes ever get tired?", "punchline" : "Because they have rest wings!", "rating" : 2}}

example_user : Tell me another joke about planes
example_assistant : {{"setup" : "Cargo", "punchline" : "Cargo 'vroom vroom', but planes go 'zoom zoom'!", "rating" : 10}}

example_user : Now about caterpillars
example_assistant : {{"setup" : "Caterpillar", "punchline" : "Caterpillar really slow, but watch me turn into a butterfly and steal the show!", "rating" : 5}}"""

fewShotChatPromptTemplate = ChatPromptTemplate.from_messages([("system", systemMessage), ("human", "{input}")])

fewShotRunnableSequence = fewShotChatPromptTemplate | structuredOutputRunnableSequence

response = fewShotRunnableSequence.invoke("what's something funny about woodpeckers")

print(response)

"""
output=Joke(setup='Wooden', punchline='Wooden you like to know why woodpeckers are such great comedians? They always knock knock on wood!', rating=8)
"""

import os

from typing import Optional

from typing import Union

from langchain_core.pydantic_v1 import BaseModel, Field

from langchain_openai import ChatOpenAI

from langchain_core.prompts import ChatPromptTemplate

os.environ["OPENAI_API_KEY"] = "<OPENAI_API_KEY>"

class Joke(BaseModel):

"""Joke to tell user."""

setup : str = Field(description = "The setup of the joke" )

punchline : str = Field(description = "The punchline to the joke" )

rating : Optional[int] = Field(description = "How funny the joke is, from 1 to 10")

class ConversationalResponse(BaseModel):

"""Respond in a conversational manner. Be kind and helpful."""

response : str = Field(description = "A conversational response to the user's query")

class Response(BaseModel):

output : Union[Joke, ConversationalResponse]

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

structuredOutputRunnableSequence = chatOpenAI.with_structured_output(Response)

systemMessage = """You are a hilarious comedian. Your specialty is knock-knock jokes. \

Return a joke which has the setup (the response to "Who's there?") and the final punchline (the response to "<setup> who?").

Here are some examples of jokes:

example_user : Tell me a joke about planes

example_assistant : {{"setup" : "Why don't planes ever get tired?", "punchline" : "Because they have rest wings!", "rating" : 2}}

example_user : Tell me another joke about planes

example_assistant : {{"setup" : "Cargo", "punchline" : "Cargo 'vroom vroom', but planes go 'zoom zoom'!", "rating" : 10}}

example_user : Now about caterpillars

example_assistant : {{"setup" : "Caterpillar", "punchline" : "Caterpillar really slow, but watch me turn into a butterfly and steal the show!", "rating" : 5}}"""

fewShotChatPromptTemplate = ChatPromptTemplate.from_messages([("system", systemMessage), ("human", "{input}")])

fewShotRunnableSequence = fewShotChatPromptTemplate | structuredOutputRunnableSequence

response = fewShotRunnableSequence.invoke("what's something funny about woodpeckers")

print(response)

"""

output=Joke(setup='Wooden', punchline='Wooden you like to know why woodpeckers are such great comedians? They always knock knock on wood!', rating=8)

"""

▶ requirements.txt


annotated-types==0.7.0
anyio==4.4.0
certifi==2024.6.2
charset-normalizer==3.3.2
distro==1.9.0
exceptiongroup==1.2.1
h11==0.14.0
httpcore==1.0.5
httpx==0.27.0
idna==3.7
jsonpatch==1.33
jsonpointer==3.0.0
langchain-core==0.2.7
langchain-openai==0.1.8
langsmith==0.1.77
openai==1.34.0
orjson==3.10.5
packaging==24.1
pydantic==2.7.4
pydantic_core==2.18.4
PyYAML==6.0.1
regex==2024.5.15
requests==2.32.3
sniffio==1.3.1
tenacity==8.3.0
tiktoken==0.7.0
tqdm==4.66.4
typing_extensions==4.12.2
urllib3==2.2.1

annotated-types==0.7.0

anyio==4.4.0

certifi==2024.6.2

charset-normalizer==3.3.2

distro==1.9.0

exceptiongroup==1.2.1

h11==0.14.0

httpcore==1.0.5

httpx==0.27.0

idna==3.7

jsonpatch==1.33

jsonpointer==3.0.0

langchain-core==0.2.7

langchain-openai==0.1.8

langsmith==0.1.77

openai==1.34.0

orjson==3.10.5

packaging==24.1

pydantic==2.7.4

pydantic_core==2.18.4

PyYAML==6.0.1

regex==2024.5.15

requests==2.32.3

sniffio==1.3.1

tenacity==8.3.0

tiktoken==0.7.0

tqdm==4.66.4

typing_extensions==4.12.2

urllib3==2.2.1

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