haystack-ml-stack 0.4.0__tar.gz → 0.4.2__tar.gz

{haystack_ml_stack-0.4.0 → haystack_ml_stack-0.4.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: haystack-ml-stack
-Version: 0.4.0
+Version: 0.4.2
 Summary: Functions related to Haystack ML
 Author-email: Oscar Vega <oscar@haystack.tv>
 License: MIT

{haystack_ml_stack-0.4.0 → haystack_ml_stack-0.4.2}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "haystack-ml-stack"
-version = "0.4.0"
+version = "0.4.2"
 description = "Functions related to Haystack ML"
 readme = "README.md"
 authors = [{ name = "Oscar Vega", email = "oscar@haystack.tv" }]

{haystack_ml_stack-0.4.0 → haystack_ml_stack-0.4.2}/src/haystack_ml_stack/__init__.py

@@ -4,11 +4,11 @@ try:
     from .app import create_app
 
     __all__ = ["create_app"]
-except ImportError:
+except ImportError as e:
     pass
 
+
 from ._serializers import SerializerRegistry, FeatureRegistryId
+from ._version import __version__
 
 __all__ = [*__all__, "SerializerRegistry", "FeatureRegistryId"]
-
-__version__ = "0.4.0"

{haystack_ml_stack-0.4.0 → haystack_ml_stack-0.4.2}/src/haystack_ml_stack/_kafka.py

@@ -5,8 +5,9 @@ import base64
 import os
 import logging
 from .settings import Settings
-from . import __version__
+from ._version import __version__
 import hashlib
+import datetime
 
 logger = logging.getLogger(__name__)
 SECURITY_PROTOCOL = "SASL_SSL"
@@ -22,6 +23,7 @@ async def send_to_kafka(
     state: dict,
     model_output: dict,
     monitoring_meta: dict,
+    processed_at: datetime.datetime,
 ) -> None:
     if topic is None or producer is None:
         return
@@ -41,6 +43,7 @@ async def send_to_kafka(
             "user_features": state.get("user_features", []),
             "stream_features": state.get("stream_features", []),
         },
+        "processed_at": processed_at.isoformat(),
     }
     delivery_future = await producer.produce(
         topic, orjson.dumps(message, default=default_serialization)

haystack_ml_stack-0.4.2/src/haystack_ml_stack/_version.py

@@ -0,0 +1 @@
+__version__ = "0.4.2"

{haystack_ml_stack-0.4.0 → haystack_ml_stack-0.4.2}/src/haystack_ml_stack/app.py

@@ -9,6 +9,7 @@ from contextlib import asynccontextmanager, AsyncExitStack
 import traceback
 import json
 import asyncio
+import datetime
 
 import aiobotocore.session
 from aiobotocore.config import AioConfig
@@ -320,6 +321,7 @@ def create_app(
                     model_output,
                 )
             if should_log_user(userid=userid, kafka_fraction=cfg.kafka_fraction):
+                processed_at = datetime.datetime.now(tz=datetime.UTC)
                 background_tasks.add_task(
                     send_to_kafka,
                     producer=state["kafka_producer"],
@@ -330,6 +332,7 @@ def create_app(
                     state=state,
                     model_output=model_output,
                     monitoring_meta=monitoring_meta,
+                    processed_at=processed_at,
                 )
             return jsonable_encoder(model_output)
 

{haystack_ml_stack-0.4.0 → haystack_ml_stack-0.4.2}/src/haystack_ml_stack.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: haystack-ml-stack
-Version: 0.4.0
+Version: 0.4.2
 Summary: Functions related to Haystack ML
 Author-email: Oscar Vega <oscar@haystack.tv>
 License: MIT

{haystack_ml_stack-0.4.0 → haystack_ml_stack-0.4.2}/src/haystack_ml_stack.egg-info/SOURCES.txt

@@ -3,6 +3,7 @@ pyproject.toml
 src/haystack_ml_stack/__init__.py
 src/haystack_ml_stack/_kafka.py
 src/haystack_ml_stack/_serializers.py
+src/haystack_ml_stack/_version.py
 src/haystack_ml_stack/app.py
 src/haystack_ml_stack/cache.py
 src/haystack_ml_stack/dynamo.py