osmosis-ai 0.1.7__py3-none-any.whl → 0.1.8__py3-none-any.whl

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/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/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.dist-info → osmosis_ai-0.1.8.dist-info}/METADATA

@@ -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.dist-info → osmosis_ai-0.1.8.dist-info}/RECORD

@@ -1,15 +1,15 @@
-osmosis_ai/__init__.py,sha256=esOnyFqlu7-AKuxTV5fkLVb34-kvCReMEDFkFpxBH7M,4374
-osmosis_ai/consts.py,sha256=1HavvPFM9nJCDqy1RK7VH_g9MbAgXNyBqN8Rt3FAeDE,408
+osmosis_ai/__init__.py,sha256=p6SXDKMc-AUoFSJSOCgM6zvIEYJ6IrqtymqBNAB8lW4,4471
+osmosis_ai/consts.py,sha256=tZ3duER7kZ5bxMdMfMiwFlsSRmOaIpTReu19NPBuZPc,408
 osmosis_ai/logger.py,sha256=J2Yw_QMGqXySzJlj_QQqHLs52S1oKYXoELnU66YCAuI,2170
-osmosis_ai/utils.py,sha256=pJDJTO1ov0Nyyj6UMbPGZIAPyoF-wY9QM3hYoxtIuo4,2490
+osmosis_ai/utils.py,sha256=l46vQYLZEUCFSCCUlM9tCgNgDV-WzIwzwTwR8K3vZ4I,2992
 osmosis_ai/adapters/__init__.py,sha256=MlS3aGG8AQqQUQlV196sPdChTvfenK6ZdSaP0Tsr8eQ,262
 osmosis_ai/adapters/anthropic.py,sha256=Q9pb_AxgaSTyjNYkGGZL5mslyVnfTbGI04WbxLDhvLA,22959
 osmosis_ai/adapters/langchain.py,sha256=oVIIRmpdadgozoysJBxoJsTVwJcjrtoA63SjRchXr9E,27673
 osmosis_ai/adapters/langchain_anthropic.py,sha256=qWtcNAQ3tQpJfTttV9y5mMCkPujuRodoFkaoYx9pOW8,14557
 osmosis_ai/adapters/langchain_openai.py,sha256=8sq4vGSqm96-UjeeiW0ZGWMOUXiv85A1BLGwaeboBF8,24815
 osmosis_ai/adapters/openai.py,sha256=B-dNQvA8XcmX4kB5ZZGf4EoS_VAVZPo3aTfem_xe-8I,39437
-osmosis_ai-0.1.7.dist-info/licenses/LICENSE,sha256=WLvxAm2-fPuT8qECE6ejD8ezG9btIanJl7zysXvtQgM,1064
-osmosis_ai-0.1.7.dist-info/METADATA,sha256=wVZ0LQAJIfGKinOoQrgCUKwCOf667InqQxOYIWBBv_Q,9255
-osmosis_ai-0.1.7.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-osmosis_ai-0.1.7.dist-info/top_level.txt,sha256=UPNRTKIBSrxsJVNxwXnLCqSoBS4bAiL_3jMtjvf5zEY,11
-osmosis_ai-0.1.7.dist-info/RECORD,,
+osmosis_ai-0.1.8.dist-info/licenses/LICENSE,sha256=WLvxAm2-fPuT8qECE6ejD8ezG9btIanJl7zysXvtQgM,1064
+osmosis_ai-0.1.8.dist-info/METADATA,sha256=Zeuhm80tBeJXhG6OMvVplbH52VjQnOFsd5z2PAEBfEU,9255
+osmosis_ai-0.1.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+osmosis_ai-0.1.8.dist-info/top_level.txt,sha256=UPNRTKIBSrxsJVNxwXnLCqSoBS4bAiL_3jMtjvf5zEY,11
+osmosis_ai-0.1.8.dist-info/RECORD,,

{osmosis_ai-0.1.7.dist-info → osmosis_ai-0.1.8.dist-info}/WHEEL

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