From d1808dbcf31e691228a2ff62e97d71464f375f34 Mon Sep 17 00:00:00 2001 From: Ranjan Shrestha Date: Fri, 13 Dec 2024 09:43:38 +0545 Subject: [PATCH] Use openai module --- content/views.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/content/views.py b/content/views.py index 41ccc18..be39896 100644 --- a/content/views.py +++ b/content/views.py @@ -4,12 +4,12 @@ from rest_framework.generics import GenericAPIView from rest_framework.response import Response -from chatbotcore.llm import OllamaHandler +from chatbotcore.llm import OpenAIHandler from content.serializers import UserQuerySerializer class UserQuery(GenericAPIView): - llm = OllamaHandler() + llm = OpenAIHandler() def post(self, request, *arg, **kwargs): serializer = UserQuerySerializer(data=request.data)