chore: add more corebase

apis/__init__.py

@@ -0,0 +1,6 @@
+from fastapi import APIRouter
+from .v1.routes.documents import router as documents_router
+
+
+api_v1_router = APIRouter(prefix="/v1")
+api_v1_router.include_router(documents_router)

apis/create_app.py

@@ -0,0 +1,21 @@
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+from .v1.configs.swagger_config import swagger_config
+
+
+# Define create_app function.
+# Avoid circular import by using this function
+# to create FastAPI app instance.
+def create_app():
+    app = FastAPI(**swagger_config)
+
+    # CORs handling
+    app.add_middleware(
+        CORSMiddleware,
+        allow_origins=["*"],
+        allow_credentials=True,
+        allow_methods=["*"],
+        allow_headers=["*"],
+    )
+
+    return app

apis/v1/configs/swagger_config.py

@@ -0,0 +1,16 @@
+# Swagger configuration
+swagger_config = {
+    "title": "RAG System API",
+    "description": "API for the Retrieval-Augmented Generation (RAG) System",
+    "docs_url": "/",
+    "openapi_tags": [
+        {
+            "name": "Document",
+            "description": "Operations with documents"
+        },
+        {
+            "name": "RAG",
+            "description": "Retrieve and Generate operations"
+        }
+    ]
+}

apis/v1/routes/documents.py

@@ -0,0 +1,13 @@
+from typing import Annotated
+from io import BytesIO
+from fastapi import APIRouter, Depends, BackgroundTasks
+from ..interfaces.document_interface import DocumentUploadResponseInterface
+router = APIRouter(prefix="/documents", tags=["Documents"])
+
+@router.post("/upload", response_model=DocumentResponse)
+async def upload_document(file: Annotated[BytesIO, Field(..., description="File to upload")], background_tasks: BackgroundTasks):
+    """
+    Upload a document
+    """
+    
+    return DocumentUploadResponseInterface(file=file)

apis/v1/schemas/document_schema.py

@@ -0,0 +1,7 @@
+from typing import AnyStr, Dict
+from pydantic import BaseModel, Field
+from ..providers import jd_db
+
+
+class DocumentResponse(BaseModel):
+