add bucket in CRD

app/main.py

@@ -2,7 +2,6 @@ from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.openapi.utils import get_openapi
 from app.routers import documents
-from app.config import settings
 from app.logger import setup_logging
 from app.middleware.auth import AuthMiddleware
 

app/middleware/auth.py

@@ -1,6 +1,5 @@
 from fastapi import Request
 from starlette.middleware.base import BaseHTTPMiddleware
-from starlette.responses import JSONResponse
 from app.logger import get_logger
 
 logger = get_logger(__name__)

app/models.py

@@ -1,6 +1,5 @@
 from pydantic import BaseModel, Field
 from datetime import datetime
-from typing import Optional
 from app.enums import DocumentType
 
 class DocumentMetadata(BaseModel):

app/pdf.py

@@ -1,6 +1,4 @@
-import os
 from pypdf import PdfReader
-from typing import Any
 
 def discover_fields(pdf_path: str) -> list[dict]:
     """

app/routers/__init__.py

@@ -1 +1,3 @@
-from app.routers import documents
+from app.routers import documents as documents
+
+__all__ = ["documents"]

app/routers/documents.py

@@ -1,6 +1,5 @@
 import os
-from fastapi import APIRouter, HTTPException, UploadFile, File, Form, Request
+from fastapi import APIRouter, HTTPException, UploadFile, File, Request
-from typing import Optional
 from datetime import datetime
 
 from app import s3, pdf, utils

app/s3.py

@@ -50,15 +50,24 @@ def upload_file(file: UploadFile, s3_key: str, content_type: str, metadata: dict
     """Upload file to S3 with metadata"""
     client = get_client()
     
+    # Read file content
+    file.file.seek(0, os.SEEK_END)
+    file_size = file.file.tell()
+    file.file.seek(0)
+    file_content = file.file.read()
+    file.file.seek(0)
+    
     extra_args = {"ContentType": content_type}
     if metadata:
         extra_args["Metadata"] = metadata
     
-    client.upload_fileobj(
+    client.put_object(
-        file.file,
+        Bucket=settings.s3_bucket,
-        settings.s3_bucket,
+        Key=s3_key,
-        s3_key,
+        Body=file_content,
-        ExtraArgs=extra_args
+        ContentLength=file_size,
+        ContentType=content_type,
+        Metadata=metadata
     )
     return s3_key
 

ops/chart/values.yaml

@@ -56,3 +56,15 @@ service:
         port: 8082
         protocol: HTTP
 
+rawResources:
+  document-service-bucket:
+    enabled: true
+    apiVersion: objectstorage.k8s.io/v1alpha1
+    kind: BucketClaim
+    suffix: bucket
+    spec:
+      spec:
+        bucketClassName: seaweedfs
+        protocols:
+          - s3
+