[langchain공부] SimpleJsonOutputParser (json으로 결과도출!!)

chatgpt의 결과물을  json 으로 paring  할수 있어요!!

엄청 좋아요!!!

 

## 주요 패키지 및 키 설정

from langchain.chat_models import ChatOpenAI
from langchain.document_loaders import YoutubeLoader, UnstructuredURLLoader
from langchain.chains.summarize import load_summarize_chain
from langchain.prompts import PromptTemplate
from langchain.output_parsers.json import SimpleJsonOutputParser

openai_api_key = '{내 key}'
model =  "gpt-3.5-turbo"

 

 

## 코드 작동!!!!

# Create a JSON prompt
json_prompt = PromptTemplate.from_template(
    "Return a JSON object with `대통령` and `취임일` key that answers the following question: {question}"
)

# Initialize the JSON parser
json_parser = SimpleJsonOutputParser()

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


# Create a chain with the prompt, model, and parser
json_chain = json_prompt | llm | json_parser

# Stream through the results
result_list = list(json_chain.stream({"question": "한국 미국 러시아의 대통령 이름과 취임일은??"}))

result_list[-1]

 

결과물 !! 완전 멋져!!