Use openai module

toggle-corp · Dec 13, 2024 · d1808db · d1808db
1 parent cf8a196
commit d1808db
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)