haystack-ml-stack 0.4.1__tar.gz → 0.4.3__tar.gz

{haystack_ml_stack-0.4.1 → haystack_ml_stack-0.4.3}/PKG-INFO

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

{haystack_ml_stack-0.4.1 → haystack_ml_stack-0.4.3}/pyproject.toml

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

{haystack_ml_stack-0.4.1 → haystack_ml_stack-0.4.3}/src/haystack_ml_stack/_kafka.py

@@ -7,6 +7,7 @@ import logging
 from .settings import Settings
 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
@@ -37,10 +39,12 @@ async def send_to_kafka(
         "meta": {
             "monitoring": monitoring_meta,
             "haystack_ml_stack_version": __version__,
+            "playlist_id": str(playlist.get("playlistId", "")),
             "playlist_category": playlist.get("category"),
             "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.3/src/haystack_ml_stack/_version.py

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

{haystack_ml_stack-0.4.1 → haystack_ml_stack-0.4.3}/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.1 → haystack_ml_stack-0.4.3}/src/haystack_ml_stack/dynamo.py

@@ -150,7 +150,6 @@ async def set_all_features(
     user_cache_misses = len(cache_miss) - stream_cache_misses
     valid_cache_delays = list(v for v in cache_delay_obj.values() if v > 0)
     cache_delay = min(valid_cache_delays) if valid_cache_delays else 0
-
     if not cache_miss:
         return FeatureRetrievalMeta(
             user_cache_misses=0,

{haystack_ml_stack-0.4.1 → haystack_ml_stack-0.4.3}/src/haystack_ml_stack.egg-info/PKG-INFO

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

haystack_ml_stack-0.4.1/src/haystack_ml_stack/_version.py

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