Make 뉴스레터 자동화를 LangChain으로: claude와 json parser

from langchain.prompts import ChatPromptTemplate

prompt = ChatPromptTemplate.from_messages([
    ("human", "다음 정보로 뉴스레터를 작성해주세요: {topic}")
])

from langchain_anthropic import ChatAnthropic

model = ChatAnthropic(
    model="claude-3-sonnet-20240229",
    temperature=0.7,
    max_tokens=4000,
    anthropic_api_key=ANTHROPIC_API_KEY
)

from langchain_core.output_parsers import JsonOutputParser
from pydantic import BaseModel, Field

class NewsletterContent(BaseModel):
    title: str = Field(description="뉴스레터 제목")
    content: str = Field(description="뉴스레터 내용")

parser = JsonOutputParser(pydantic_object=NewsletterContent)

from langchain.schema.runnable import RunnableSequence

newsletter_chain = RunnableSequence(
    prompt,
    model,
    parser
)

result = newsletter_chain.invoke({"topic": "건강한 식습관"})

import os
import yaml
import json
from pydantic import BaseModel, Field
from dotenv import load_dotenv
from langchain_anthropic import ChatAnthropic
from langchain_core.output_parsers import JsonOutputParser
from langchain.prompts import ChatPromptTemplate
from langchain.schema.runnable import RunnableSequence

load_dotenv()

def load_yaml_and_create_prompt(file_path: str, user_name: str, topic: str, title: str, habit: str) -> ChatPromptTemplate:
    with open(file_path, 'r', encoding='utf-8') as file:
        yaml_data = yaml.safe_load(file)

    # YAML 데이터를 사용하여 프롬프트 템플릿 생성
    prompt_content = f"""주간 건강습관 뉴스레터를 작성해주세요. 다음 정보를 사용하세요:

사용자 이름: {user_name}
주제: {topic}
제목: {title}
이번 주 건강습관: {habit}

다음 가이드라인을 따라주세요:
"""
    # YAML 데이터를 기반으로 프롬프트 내용 구성
    ...

    return ChatPromptTemplate.from_messages([
        ("human", prompt_content)
    ])

name_placeholder: "ㅎ이름자리ㅎ"
guide:
  sections:
    - section: "사용자 이름 언급"
      details:
        - "중간중간 사용자의 이름을 언급하세요. 사용자의 이름은 'ㅎ이름자리ㅎ'으로 표기"
        - "예: 이번주는 ㅎ이름자리ㅎ님의 날입니다."

    - section: "제목"
      details:
        - "제공된 제목을 사용하세요."

    - section: "도입부 (약 150단어)"
      details:
        - "독자의 관심을 끄는 질문이나 통계로 시작하세요."
        - "이번 주 건강습관과 관련된 건강 주제를 소개하세요."

    - section: "이번 주 건강습관 소개 (약 200단어)"
      details:
        - "실천할 구체적인 건강습관을 명확히 설명하세요."
        - "언제, 어디서, 어떻게 실천할 수 있는지 구체적으로 안내하세요."

    - section: "과학적 근거 및 기전 설명 (약 600단어)"
      details:
        - "해당 습관의 효과를 뒷받침하는 2-3개의 핵심 연구나 관련 기사를 인용하세요."
        - "각 연구나 기사에 대해 출처(저자, 발행년도, 저널명 또는 웹사이트 등)를 구체적으로 제시하세요."
        - "이 습관이 건강에 도움이 되는 자세한 기전과 원리를 쉽게 설명하세요."
        - "복잡한 생물학적 과정을 일상생활의 비유를 사용해 설명하세요."

    - section: "기대효과 (약 300단어)"
      details:
        - "이 습관을 꾸준히 실천했을 때 기대할 수 있는 단기 및 장기 효과를 설명하세요."
        - "가능하다면 수치나 구체적인 예시를 포함하세요."
        - "앞서 설명한 기전과 연결 지어 왜 이런 효과가 나타나는지 설명하세요."

    - section: "실천 팁 (약 300단어)"
      details:
        - "습관을 쉽게 실천할 수 있는 4-5가지 팁을 제공하세요."
        - "각 팁에 대해 상세한 설명과 예시를 덧붙이세요."
        - "가능하다면 각 팁이 어떻게 앞서 설명한 기전과 연관되는지 설명하세요."

    - section: "주의사항 (약 150단어)"
      details:
        - "습관 실천 시 주의해야 할 점이나 피해야 할 사항을 안내하세요."
        - "필요하다면 전문가의 조언을 구해야 하는 상황도 명시하세요."

    - section: "추가 정보 및 관련 연구 (약 400단어)"
      details:
        - "주제와 관련된 추가적인 연구 결과나 최신 트렌드를 소개하세요."
        - "이 정보들이 어떻게 주간 건강습관과 연관되는지 설명하세요."
        - "가능하다면 이 분야의 향후 연구 방향이나 전망에 대해 언급하세요."

    - section: "독자 참여 유도 (약 100단어)"
      details:
        - "독자들에게 이번 주 습관 실천 경험을 공유해달라고 요청하세요."
        - "다음 주 건강습관에 대한 아이디어나 의견을 물어보세요."

    - section: "마무리 (약 100단어)"
      details:
        - "이번 주 건강습관의 핵심 메시지를 다시 한 번 강조하세요."
        - "실천을 독려하는 긍정적인 메시지로 마무리하세요."

additional_guidelines:
  - "전체 길이는 2000-2200단어로 유지하세요."
  - "'-습니다', '-입니다'와 '-요', '-죠'를 2:3 비율로 적절히 섞어 사용하세요."
  - "전문 용어는 가능한 쉽게 풀어서 설명하세요."
  - "문장은 간결하게, 단락은 3-4문장으로 구성하세요."
  - "중간중간 흥미로운 부제목을 사용하세요."
  - "실천 가능성과 효과를 강조하되, 과장된 표현은 피하세요."
  - "복잡한 개념을 설명할 때는 일상생활의 예시나 비유를 활용하세요."
  - "중간중간 사용자의 이름을 불러주세요. 사용자의 이름은 'ㅎ이름자리ㅎ'으로 표기하세요."

output_format:
  - title: "주간 건강습관 뉴스레터: [주제]"
  - introduction: "도입부 내용"
  - habit_intro: "이번 주 건강습관 소개 내용"
  - scientific_evidence: "과학적 근거 및 기전 설명 내용"
  - expected_effects: "기대효과 내용"
  - practical_tips: "실천 팁 내용"
  - precautions: "주의사항 내용"
  - additional_info: "추가 정보 및 관련 연구 내용"
  - reader_participation: "독자 참여 유도 내용"
  - conclusion: "마무리 내용"

parser = JsonOutputParser(pydantic_object=NewsletterContent)
prompt = load_yaml_and_create_prompt(prompt_file, user_name, topic, title, habit)
model = ChatAnthropic(
    model="claude-3-sonnet-20240229",
    temperature=0.7,
    max_tokens=4000,
    anthropic_api_key=ANTHROPIC_API_KEY
)

newsletter_chain = RunnableSequence(
    prompt,
    model,
    parser
)

def create_image_prompts(newsletter_content: NewsletterContent) -> RunnableSequence:
    prompt_template = """다음 뉴스레터 내용을 바탕으로 3개의 이미지 생성 프롬프트를 만들어주세요:

1. 도입부: {introduction}
2. 이번 주 건강습관: {habit_intro}
3. 기대효과: {expected_effects}

각 이미지 프롬프트는 다음 형식을 따라야 합니다:
[구체적인 동작 또는 상황 설명]. The overall color scheme is pastel, with brand colors (#004a30, #ff4a1c) used sparingly for emphasis on key elements. The background is a soft pastel blue. The illustration style is flat and simple with white space around the main elements. The design is a vector graphic in a minimalist flat style with no shadows, suitable for various media applications. Do not include any logos or text. --ar 16:9 --stylize 1000 --v 5.2

응답은 반드시 다음 형식의 유효한 JSON 객체로만 제공해주세요:

{format_instructions}
"""

    prompt = ChatPromptTemplate.from_template(prompt_template)

    model = ChatAnthropic(
        model="claude-3-sonnet-20240229",
        temperature=0.7,
        max_tokens=1000,
        anthropic_api_key=ANTHROPIC_API_KEY
    )

    parser = JsonOutputParser(pydantic_object=ImagePrompts)

    return RunnableSequence(
        prompt,
        model,
        parser
    )