[langchain공부] CommaSeparatedListOutputParser(, 결과도출!!)

 

 

from langchain.output_parsers import CommaSeparatedListOutputParser
from langchain.prompts import PromptTemplate
from langchain_openai import ChatOpenAI
openai_api_key = '{나의 키}'
model =  "gpt-3.5-turbo"

output_parser = CommaSeparatedListOutputParser()

format_instructions = output_parser.get_format_instructions()
prompt = PromptTemplate(
    template="{subject}.\n{format_instructions}",
    input_variables=["subject"],
    partial_variables={"format_instructions": format_instructions},
)

llm = ChatOpenAI(temperature=0, model=model, openai_api_key=openai_api_key)

chain = prompt | llm | output_parser

 

chain.invoke({"subject": "축구 잘하는 국가들 이름"})

결과!!

리스트, 콤마로 구분되어 잘 나옵니다!

['브라질', '아르헨티나', '독일', '이탈리아', '스페인', '프랑스', '네덜란드', '벨기에', '포르투갈', '잉글랜드']