model2service 0.0.1__py3-none-any.whl

model2service/__init__.py

@@ -0,0 +1,9 @@
+from .src import *
+
+def service2model_mlflowredis(cls) -> Model2Service:
+    return Model2Service(
+        service_registry=RedisServices(),
+        model_repository=MlflowRepository()
+        )
+
+DefaultModel2Service: Model2Service = service2model_mlflowredis()

model2service/src/__init__.py

@@ -0,0 +1,5 @@
+from .model_repositories import *
+from .service_registries import *
+from .model2service import *
+
+

model2service/src/model2service.py

@@ -0,0 +1,33 @@
+from __future__ import annotations
+
+import abc
+from .service_registries import ServiceRegistry, ModelServiceRegistry
+from .model_repositories import ModelRepository, Model
+
+from pydantic import BaseModel
+
+# class Model2ServiceMapping(BaseModel):
+#     service_registry: ServiceRegistry
+#     model_repository: ModelRepository
+
+
+class Model2Service():
+    
+    def __init__(self, 
+                 service_registry: ServiceRegistry,
+                 model_repository: ModelRepository
+                 ):
+        self.mode
+
+    def model_save(self, model: Model): 
+        self.model_repository.save(model)
+
+    def model_load(self, model_name: str) -> Model:
+        return self.model_repository.load(model_name)
+
+    def service_register(self, model_service_registry: ModelServiceRegistry):
+        self.service_registry.register(model_service_registry)
+
+    def service_discover(self, model_name: str) -> ModelServiceRegistry:
+        return self.service_registry.discovery(model_name)
+    

model2service/src/model_repositories/__init__.py

@@ -0,0 +1,2 @@
+from .model_repository import *
+from .mlflow_repository import *

model2service/src/model_repositories/mlflow_repository.py

@@ -0,0 +1,12 @@
+from mlflow.pyfunc.model import PythonModel, PythonModelContext
+from .model_repository import Model
+
+from .model_repository import ModelRepository
+
+class MlflowRepository(ModelRepository):
+    def save(self, model: Model): 
+        pass
+
+    def load(self) -> Model: 
+        pass
+ 

model2service/src/model_repositories/model_repository.py

@@ -0,0 +1,12 @@
+import abc
+
+class Model:
+    name: str
+
+class ModelRepository(abc.ABC):
+
+    @abc.abstractmethod
+    def save(self, model: Model): ...
+
+    @abc.abstractmethod
+    def load(self, model_name: str) -> Model: ...

model2service/src/service_registries/__init__.py

@@ -0,0 +1,2 @@
+from .service_registry import *
+from .redis_registry import *

model2service/src/service_registries/redis_registry.py

@@ -0,0 +1,39 @@
+from .service_registry import ServiceRegistry, ModelServiceRegistry
+import redis
+import os
+from typing import Optional
+
+ENVIRONMENT_VARIABLE_HOST = "REDIS_HOST"
+ENVIRONMENT_VARIABLE_PORT = "REDIS_PORT"
+
+class RedisServices(ServiceRegistry):
+    _instace = None
+
+    def __new__(cls):
+        if cls._instance is None:
+            instance = super().__new__(cls)
+            instance.client = redis.Redis(
+                host=instance.host,
+                port=int(instance.port),
+                decode_responses=True
+            )
+            cls._instance = instance
+        return cls._instance
+
+    def register(self, model_service: ModelServiceRegistry): 
+        # registra o nome do modelo e a fila no serviço
+        key = f"model_queue:{model_service.service_name}"
+        self.client.set(key, model_service.queue_name)
+
+    def discovery(self, service_name: str) -> Optional[str]: 
+        # retorna o nome da fila associada ao modelo
+        key = f"model_queue:{service_name}"
+        return self.client.get(key)
+    
+    @property
+    def host(self):
+        return os.getenv(ENVIRONMENT_VARIABLE_HOST)
+
+    @property
+    def port(self):
+        return os.getenv(ENVIRONMENT_VARIABLE_PORT)

model2service/src/service_registries/service_registry.py

@@ -0,0 +1,27 @@
+import abc
+from pydantic import BaseModel
+from collections.abc import Callable
+from typing import Optional
+from pika.spec import BasicProperties
+
+
+#TODO this class does not belong in this file, maybe a MessageConsumer interface
+class CallBackArguments(BaseModel):
+    body : bytes
+    #TODO make less coupling, maybe use dict[str, Any] for now
+    properties: BasicProperties
+
+class ModelServiceRegistry(BaseModel):
+    service_name: str
+    queue_name: str
+    #TODO 'process' does not belong in this class, this class is only for registration
+    process: Callable[[CallBackArguments], Optional[BaseModel]]
+
+@abc.abstractmethod
+class ServiceRegistry(abc.ABC):
+
+    @abc.abstractmethod
+    def register(self, model_service: ModelServiceRegistry): ...
+
+    @abc.abstractmethod
+    def discovery(self, service_name: str) -> Optional[str]: ...

model2service-0.0.1.dist-info/METADATA

@@ -0,0 +1,17 @@
+Metadata-Version: 2.4
+Name: model2service
+Version: 0.0.1
+Author-email: Diego Pinheiro <diego.silva@unicap.br>
+Project-URL: Homepage, https://github.com/projeto-carcara/model2service
+Keywords: service
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: pydantic==2.13.2
+Requires-Dist: debugpy==1.8.20
+Requires-Dist: pika==1.3.2
+Requires-Dist: redis==7.4.0
+
+# model2service

model2service-0.0.1.dist-info/RECORD

@@ -0,0 +1,13 @@
+model2service/__init__.py,sha256=5toIgenSzxa_HnWEoKNnmbiQhhA23u_UshDna0LxFyo,261
+model2service/src/__init__.py,sha256=VL8NWeATFuncXuBV7M7U3WMWuJ0WpWNVUgcyt8eldYE,99
+model2service/src/model2service.py,sha256=YV_uKYR1lu6CBxpeI9CbVF92F1kHSkSrmuFDbTDuchE,992
+model2service/src/model_repositories/__init__.py,sha256=9oor5e8p4hoyHnb3WEdyXfRoJ8mJCpkXeH_Xdym7q7s,64
+model2service/src/model_repositories/mlflow_repository.py,sha256=QnZW-tfyJ1KP2pG7uAuNN7RHJHDYoAcsc8KaxYX-hcI,282
+model2service/src/model_repositories/model_repository.py,sha256=PbXsVnAgpvyUcUdiaxdRQb9uhzHME2P2Bm3DhPm7kxc,210
+model2service/src/service_registries/__init__.py,sha256=Qxy8TPYAOUCSQVLCKQwO8I0MmaCUYsR9buD4vkDxXzc,61
+model2service/src/service_registries/redis_registry.py,sha256=zfFMf2jXohv1Cgvrzn_oEGpUIxHqzFBD3bDI7BU2FVA,1206
+model2service/src/service_registries/service_registry.py,sha256=_KXQ_IsV-3CUdBjZ-rhSI8V6VW3CBwN84ulB2xUKiiQ,841
+model2service-0.0.1.dist-info/METADATA,sha256=ohujwzvDFZYG58626Ny72abFcIlbXBUHnBybqNMhB3g,544
+model2service-0.0.1.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+model2service-0.0.1.dist-info/top_level.txt,sha256=U0j5cvdpPM8qMQ298RD6JqavG0TzGd_ijNhAw-hhPBQ,14
+model2service-0.0.1.dist-info/RECORD,,

model2service-0.0.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

model2service-0.0.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+model2service