Cook AI

OpenAI LLM 사용해보기 - 기본

앤써니킴 2024. 8. 3. 01:09
728x90

주요내용

from dotenv import load_dotenv

# .env로부터 API KEY 정보 가져오기
load_dotenv()
True
from langchain_openai import ChatOpenAI

# OpenAI Chat 전용 LLM 객체 생성
# https://api.python.langchain.com/en/latest/chat_models/langchain_openai.chat_models.base.ChatOpenAI.html
llm = ChatOpenAI(
    temperature=0.1,  # 창의성 (0.0 ~ 2.0)
    max_tokens=2048,  # 채팅 결과로 생성되는 토큰의 최대개수
    model_name="gpt-3.5-turbo",  # 모델명
)

# 질의 내용
question = "대한민국의 꽃은 무엇인가요?"

# LLM에게 질의 요청
print(f"[답변]: {llm.invoke(question)}")
[답변]: content='대한민국의 꽃은 무궁화(무궁화꽃)입니다. 무궁화는 대한민국의 국화로 지정되어 있으며, 우리나라의 아름다움과 숭고함을 상징하는 꽃으로 사랑받고 있습니다.' response_metadata={'token_usage': {'completion_tokens': 92, 'prompt_tokens': 26, 'total_tokens': 118}, 'model_name': 'gpt-3.5-turbo', 'system_fingerprint': None, 'finish_reason': 'stop', 'logprobs': None} id='run-fcb5e3c0-0a22-438d-ad6c-1de8637e18f0-0'
728x90