Merge pull request #1671 from swirlai/DS-4207_disable_rag_features_no_ai_provider

[DS-4207] Disable search RAG features when no AI provider configured

Author: erikspears

swirl/urls.py

@@ -43,6 +43,7 @@
 
     path('search/search', views.SearchViewSet.as_view({'get': 'list'}), name='search'),
     path('sapi/detail-search-rag/', views.DetailSearchRagView.as_view(), name='detail-search-rag'),
+    path('sapi/is_chat_ai_provider_exists', views.IsChatAIProviderExists.as_view({'get': 'list'}), name='is-chat-ai-provider-exists'),
 
     path('', views.index, name='index'),
     path('index.html', views.index, name='index'),

swirl/views.py

@@ -995,3 +995,13 @@ def list(self, request):
             return JsonResponse({}, status=status.HTTP_200_OK)
         else:
             return Response('Logo Object Not Found', status=status.HTTP_404_NOT_FOUND)
+
+
+class IsChatAIProviderExists(viewsets.ModelViewSet):
+
+    def list(self, request):
+        # TODO: The Community Edition does not include an AI provider model. 
+        # For now, we are setting the status to True as a temporary workaround. 
+        # This should be updated once the AI provider model is implemented.
+        # This validation is used to enable or disable the second row of Search RAG.
+        return Response({'status': 'True'}, status=status.HTTP_200_OK)

swirl/views_helpers/search_rag.py

@@ -61,14 +61,13 @@ def get_rag_result(self) -> tuple[str, dict[str, str]]:
         instances[self.search_id] = rag_processor
         if rag_processor.validate():
             result = rag_processor.process(should_return=True)
-            try:
-                if self.search_id in instances:
-                    del instances[self.search_id]
-                return result.json_results[0]["body"][0]
-            except:
-                if self.search_id in instances:
-                    del instances[self.search_id]
-                return False
+            if result == 0:
+                return "Please check the OpenAI or Azure/OpenAI credentials in your environment."
+
+            if self.search_id in instances:
+                del instances[self.search_id]
+
+            return result.json_results[0]["body"][0]
 
     def process_rag(self) -> dict[str, str]:
         result = ""