카테고리 없음
llm
rlarudals
2024. 9. 20. 21:12
langchain 을 이용한 번역 기능
class TranslateAPIView(APIView):
permission_classes = [AllowAny]
def post(self, request):
# LLM 모델 인스턴스 생성
llm = ChatOpenAI(model="gpt-4o-mini", api_key=settings.API_KEY)
# 요청에서 번역할 텍스트 가져오기
text_to_translate = request.data.get('text', '')
# LLM에 들어갈 메시지 설정
messages = [
SystemMessage(content="Translate the following text from English to Korean."),
HumanMessage(content=text_to_translate),
]
# LLM 호출하여 결과 받기
result = llm.invoke(messages)
# 결과 반환
return Response({'내용': result})