Update app/schemas/categories.py

app/schemas/categories.py

@@ -1,10 +1,26 @@
+from typing import Optional
+
 from pydantic import BaseModel, Field
 
 
 class CategorizeRequest(BaseModel):
     notes: str = Field(..., min_length=1, description="Full transaction note.")
+    user_id: Optional[str] = Field(
+        None, description="Optional user identifier associated with the request."
+    )
 
 
 class CategoryPrediction(BaseModel):
     title: str = Field(..., description="High-level category title.")
     subcategory: str = Field(..., description="Specific subcategory chosen by the model.")
+
+# from pydantic import BaseModel, Field
+
+
+# class CategorizeRequest(BaseModel):
+#     notes: str = Field(..., min_length=1, description="Full transaction note.")
+
+
+# class CategoryPrediction(BaseModel):
+#     title: str = Field(..., description="High-level category title.")
+#     subcategory: str = Field(..., description="Specific subcategory chosen by the model.")