Added Rag Featured

ai_service/api.py

@@ -1,6 +1,11 @@
 """
 FastAPI application for the AI service.
 This service acts as a backend for OpenWebUI, providing OpenWebUI-compatible API endpoints.
+
+The service supports document-based question answering using OpenWebUI's knowledge database:
+- Set use_rag=True in API requests to enable Retrieval Augmented Generation
+- When enabled, the service will use OpenWebUI's knowledge database to find relevant information
+- Documents uploaded to OpenWebUI will be used to augment the model's responses
 """
 
 from fastapi import FastAPI, HTTPException
@@ -203,6 +208,43 @@ async def test_chat_completion():
             "ollama_url": config.OLLAMA_API_URL
         }
 
+@app.post("/test-rag")
+async def test_rag_completion(query: str = "What information do you have in your knowledge database?"):
+    """
+    Test the RAG (Retrieval Augmented Generation) functionality with a query.
+
+    This endpoint tests the integration with OpenWebUI's knowledge database.
+
+    Args:
+        query: The question to ask about documents in the knowledge database.
+
+    Returns:
+        Model response using RAG.
+    """
+    try:
+        # Use the model service directly with RAG enabled
+        response = model_service.generate_response(
+            model_id=config.DEFAULT_MODEL,
+            prompt=query,
+            context=[],
+            use_rag=True  # Enable RAG
+        )
+
+        return {
+            "status": "success",
+            "model": config.DEFAULT_MODEL,
+            "query": query,
+            "use_rag": True,
+            "response": response,
+            "openwebui_url": config.OPENWEBUI_URL
+        }
+    except Exception as e:
+        return {
+            "status": "error",
+            "message": f"Failed to get RAG completion: {str(e)}",
+            "openwebui_url": config.OPENWEBUI_URL
+        }
+
 @app.post("/test-ollama-direct")
 async def test_ollama_direct():
     """