카테고리 없음

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})