Code&Data Insights

[LangChain] LLMChain 만들기 ( Chatgpt-4o & prompt Template 객체 생성 ) 본문

Artificial Intelligence/Prompt Engineering

[LangChain] LLMChain 만들기 ( Chatgpt-4o & prompt Template 객체 생성 )

paka_corn 2024. 5. 20. 16:42


1. Google Colab 초기 세팅 

1

 

 

2. OpenAI API 키 받아오기 & 입력 

import os

# OpenAI API 키 설정
os.environ['OPENAI_API_KEY'] = "your_key" 

 

 

 

3.  모델 설정 

**  " model = " 객체 생성 시 옵션 값 지정 할 수 있음 

=> temperature | max_tokens

 

 

 

4. Chain 실행 &  PromptTemplate 객체 생성 ( input : 유저의 질문 ) 

# chain 실행
llm.invoke('한국에서 가장 좋은 신용카드는?')
 
# ChatPromptTemplate.from_template() - 문자열 형태의 템플릿을 인자로 받아 해당 형식에 맞는 프롬프트 객체를 생성
from langchain_core.prompts import ChatPromptTemplate

prompt = ChatPromptTemplate.from_template("You are a senior Credit Card Specialist. Ansser the question. <Question>: {input}")
prompt

 

 

 

5. Chain 연결 (LCEL : LangChain Expression Language ) 

=> 한 구성 요소의 출력을 다음 구성 요소의 입력으로 전달

# chain 연결하기 (LCEL)
chain = prompt | llm

 

 

 

6. Chain 호출 

 

 

 

 

 

 

 

 

Reference

https://wikidocs.net/231363

 

Comments