osmosis-ai 0.1.7__tar.gz → 0.1.8__tar.gz

{osmosis_ai-0.1.7 → osmosis_ai-0.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: osmosis-ai
-Version: 0.1.7
+Version: 0.1.8
 Summary: Monkey patches LLM client libraries to print all prompts and responses
 Author-email: Gulp AI <jake@gulp.ai>
 License: MIT License

{osmosis_ai-0.1.7 → osmosis_ai-0.1.8}/osmosis_ai/__init__.py

@@ -15,7 +15,7 @@ Currently supported adapters:
 def _import_modules():
     global utils, logger, reconfigure_logger
     global set_log_destination, log_destination, LogDestination
-    global init, enabled, disable_osmosis, enable_osmosis
+    global init, enabled, disable_osmosis, enable_osmosis, osmosis_reward
 
     from . import utils
     from .logger import logger, reconfigure_logger, set_log_destination, log_destination
@@ -30,6 +30,9 @@ def _import_modules():
 
     # Re-export initialization function
     init = utils.init
+    
+    # Export the reward decorator
+    osmosis_reward = utils.osmosis_reward
 
     # Initialize wrappers as None
     global wrap_anthropic, wrap_openai, wrap_langchain

{osmosis_ai-0.1.7 → osmosis_ai-0.1.8}/osmosis_ai/consts.py

@@ -2,7 +2,7 @@ from enum import Enum
 
 # Extract package metadata
 package_name = "osmosis-ai"
-package_version = "0.1.7"
+package_version = "0.1.8"
 
 indent = 2  # Number of spaces to use for indentation in pretty print
 osmosis_api_url = "https://osmosis.gulp.dev"

{osmosis_ai-0.1.7 → osmosis_ai-0.1.8}/osmosis_ai/utils.py

@@ -4,8 +4,9 @@ Utility functions for osmosisadapters
 
 import json
 from datetime import datetime, timezone
-from typing import Any, Dict
+from typing import Any, Dict, Callable
 import xxhash
+import functools
 
 # Import constants
 from .consts import osmosis_api_url
@@ -95,3 +96,25 @@ def send_to_osmosis(
         )
     except Exception as e:
         logger.warning(f"Failed to send data to OSMOSIS API: {str(e)}")
+
+
+def osmosis_reward(func: Callable) -> Callable:
+    """
+    Decorator for reward functions.
+    
+    Args:
+        func: The reward function to be wrapped
+        
+    Returns:
+        The wrapped function
+        
+    Example:
+        @osmosis_reward
+        def calculate_reward(state, action):
+            return state.score + action.value
+    """
+    @functools.wraps(func)
+    def wrapper(*args, **kwargs):
+        return func(*args, **kwargs)
+    
+    return wrapper

{osmosis_ai-0.1.7 → osmosis_ai-0.1.8}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "osmosis-ai"
-version = "0.1.7"
+version = "0.1.8"
 description = "Monkey patches LLM client libraries to print all prompts and responses"
 readme = "README.md"
 authors = [