folder-classifier 0.1.0__tar.gz

folder_classifier-0.1.0/PKG-INFO

@@ -0,0 +1,13 @@
+Metadata-Version: 2.4
+Name: folder-classifier
+Version: 0.1.0
+Summary: Deploy folder classifier API to a Ray cluster
+Author: Crispin Almodovar
+Author-email: 
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.12
+Description-Content-Type: text/markdown
+
+# Folder classifier API

folder_classifier-0.1.0/README.md

@@ -0,0 +1 @@
+# Folder classifier API

folder_classifier-0.1.0/folder_classifier/__init__.py

@@ -0,0 +1 @@
+from folder_classifier.deploy import build_app

folder_classifier-0.1.0/folder_classifier/app.py

@@ -0,0 +1,27 @@
+import logging
+
+from fastapi import FastAPI
+from ray import serve
+
+from folder_classifier.dto import ModelConfig, FolderClassificationResponse, FolderClassificationRequest
+
+web_api = FastAPI(title=f"Folder Classifier API")
+
+
+@serve.deployment
+@serve.ingress(web_api)
+class FolderClassifierAPI:
+    def __init__(self, model_config: ModelConfig):
+        assert model_config, "model_config is required"
+        assert model_config.app_name and model_config.deployment, "Invalid ModelConfig values"
+        logging.basicConfig(level=logging.INFO)
+        self.logger = logging.getLogger(__name__)
+        self.logger.info(f"Initializing model: {model_config}")
+        #self.model_handle = serve.get_deployment_handle(app_name=model_config.app_name, deployment_name=model_config.deployment)
+        self.logger.info(f"Successfully initialized Folder Classifier API")
+
+    @web_api.post("/predict")
+    async def predict(self, request: FolderClassificationRequest) -> FolderClassificationResponse:
+        result = ("matter", 0.9)  #await self.model_handle.remote(request)
+        self.logger.info(f"Received request: {request}")
+        return FolderClassificationResponse(category=result[0], confidence=result[1])

folder_classifier-0.1.0/folder_classifier/deploy.py

@@ -0,0 +1,12 @@
+from ray.serve import Application
+
+from folder_classifier.app import FolderClassifierAPI
+from folder_classifier.dto import AppConfig
+
+
+def build_app(args: AppConfig) -> Application:
+    assert args and args.model, "AppConfig model is required"
+    assert args.model.app_name and args.model.deployment, "Model's app_name and deployment are required"
+
+    app = FolderClassifierAPI.bind(args.model)
+    return app

folder_classifier-0.1.0/folder_classifier/dto.py

@@ -0,0 +1,50 @@
+from __future__ import annotations
+
+from textwrap import dedent
+from typing import List, Union, Literal
+
+from pydantic import BaseModel, Field, confloat
+
+
+class ModelConfig(BaseModel):
+    app_name: str
+    deployment: str
+
+
+class AppConfig(BaseModel):
+    model: ModelConfig
+
+
+class File(BaseModel):
+    name: str
+    type: Literal["file"]
+
+
+class Folder(BaseModel):
+    name: str
+    type: Literal["folder"]
+    # Discriminated union: 'type' field is used to select between File and Folder
+    items: List[Union[File, Folder]] = Field(default_factory=list)
+    model_config = {
+        "json_schema_extra": {
+            # Override the OpenAPI example to avoid the default 'string' entry
+            "example": dedent("""{
+                "name": "string",
+                "type": "folder",
+                "items": [
+                    {
+                      "name": "string",
+                      "type": "file"
+                    }
+                  ]
+                }""")
+        }
+    }
+
+
+Folder.model_rebuild()
+FolderClassificationRequest = Folder
+
+class FolderClassificationResponse(BaseModel):
+    category: Literal["matter", "other"]
+    confidence: confloat(ge=0.0, le=1.0)

folder_classifier-0.1.0/folder_classifier.egg-info/PKG-INFO

@@ -0,0 +1,13 @@
+Metadata-Version: 2.4
+Name: folder-classifier
+Version: 0.1.0
+Summary: Deploy folder classifier API to a Ray cluster
+Author: Crispin Almodovar
+Author-email: 
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.12
+Description-Content-Type: text/markdown
+
+# Folder classifier API

folder_classifier-0.1.0/folder_classifier.egg-info/SOURCES.txt

@@ -0,0 +1,11 @@
+README.md
+pyproject.toml
+setup.cfg
+folder_classifier/__init__.py
+folder_classifier/app.py
+folder_classifier/deploy.py
+folder_classifier/dto.py
+folder_classifier.egg-info/PKG-INFO
+folder_classifier.egg-info/SOURCES.txt
+folder_classifier.egg-info/dependency_links.txt
+folder_classifier.egg-info/top_level.txt

folder_classifier-0.1.0/folder_classifier.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

folder_classifier-0.1.0/folder_classifier.egg-info/top_level.txt

@@ -0,0 +1 @@
+folder_classifier

folder_classifier-0.1.0/pyproject.toml

@@ -0,0 +1,3 @@
+[build-system]
+requires = ["setuptools>=61.0"]
+build-backend = "setuptools.build_meta"

folder_classifier-0.1.0/setup.cfg

@@ -0,0 +1,21 @@
+[metadata]
+name = folder-classifier
+version = 0.1.0
+author = Crispin Almodovar
+author_email = 
+description = Deploy folder classifier API to a Ray cluster
+long_description = file: README.md
+long_description_content_type = text/markdown
+classifiers = 
+	Programming Language :: Python :: 3
+	License :: OSI Approved :: MIT License
+	Operating System :: OS Independent
+
+[options]
+packages = find:
+python_requires = >=3.12
+
+[egg_info]
+tag_build = 
+tag_date = 0
+