diff --git a/services/rag/llamaindex/server.py b/services/rag/llamaindex/server.py index f65b1ab..eeb7ea4 100644 --- a/services/rag/llamaindex/server.py +++ b/services/rag/llamaindex/server.py @@ -8,6 +8,7 @@ from pathlib import Path from dotenv import load_dotenv from fastapi import FastAPI, HTTPException +from fastapi.middleware.cors import CORSMiddleware from loguru import logger from pydantic import BaseModel, Field @@ -41,6 +42,15 @@ setup_logging() app = FastAPI(title="LlamaIndex RAG API", version="1.0.0") +# Add CORS middleware +app.add_middleware( + CORSMiddleware, + allow_origins=["*"], # In production, configure this properly + allow_credentials=True, + allow_methods=["*"], + allow_headers=["*"], +) + class TestQueryRequest(BaseModel): query: str = Field(..., min_length=1, description="User query text")