osmosis-ai 0.1.9__tar.gz → 0.2.1__tar.gz

{osmosis_ai-0.1.9 → osmosis_ai-0.2.1}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: osmosis-ai
-Version: 0.1.9
-Summary: Adds reward functionality to LLM client libraries
+Version: 0.2.1
+Summary: A Python library for reward function validation with strict type enforcement.
 Author-email: Osmosis AI <jake@osmosis.ai>
 License: MIT License
         

osmosis_ai-0.2.1/osmosis_ai/__init__.py

@@ -0,0 +1,15 @@
+"""
+osmosis-ai: A Python library for reward function validation with strict type enforcement.
+
+This library provides the @osmosis_reward decorator that enforces standardized
+function signatures for reward functions used in LLM applications.
+
+Features:
+- Type-safe reward function decoration
+- Parameter name and type validation
+- Support for optional configuration parameters
+"""
+
+from .utils import osmosis_reward
+
+__all__ = ["osmosis_reward"]

{osmosis_ai-0.1.9 → osmosis_ai-0.2.1}/osmosis_ai/consts.py

@@ -1,3 +1,3 @@
 # package metadata
 package_name = "osmosis-ai"
-package_version = "0.1.9"
+package_version = "0.2.1"

{osmosis_ai-0.1.9 → osmosis_ai-0.2.1}/osmosis_ai/utils.py

@@ -54,6 +54,7 @@ def osmosis_reward(func: Callable) -> Callable:
 
     @functools.wraps(func)
     def wrapper(*args, **kwargs):
+        kwargs.pop("data_source", None)
         result = func(*args, **kwargs)
         if not isinstance(result, float):
             raise TypeError(f"Function {func.__name__} must return a float, got {type(result).__name__}")

{osmosis_ai-0.1.9 → osmosis_ai-0.2.1}/pyproject.toml

@@ -4,8 +4,8 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "osmosis-ai"
-version = "0.1.9"
-description = "Adds reward functionality to LLM client libraries"
+version = "0.2.1"
+description = "A Python library for reward function validation with strict type enforcement."
 readme = "README.md"
 authors = [
     {name = "Osmosis AI", email = "jake@osmosis.ai"}

osmosis_ai-0.1.9/osmosis_ai/__init__.py

@@ -1,137 +0,0 @@
-"""
-osmosis-ai: A library for monkey patching LLM APIs to print all prompts and responses.
-
-This module patches various LLM client libraries to send all prompts and responses
-to the OSMOSIS API for logging and monitoring.
-
-Currently supported adapters:
-- Anthropic (both sync and async clients)
-- OpenAI (both sync and async clients, v1 and v2 API versions)
-- LangChain (LLMs, Chat Models, and Prompt Templates)
-"""
-
-
-# Use lazy imports to avoid importing modules during setup
-def _import_modules():
-    global utils, logger, reconfigure_logger
-    global set_log_destination, log_destination, LogDestination
-    global init, enabled, disable_osmosis, enable_osmosis, osmosis_reward
-
-    from . import utils
-    from .logger import logger, reconfigure_logger, set_log_destination, log_destination
-    from .consts import LogDestination
-
-    # Re-export configuration flags for easy access
-    enabled = utils.enabled
-
-    # Export disable and enable functions
-    disable_osmosis = utils.disable_osmosis
-    enable_osmosis = utils.enable_osmosis
-
-    # 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
-    global wrap_langchain_openai, wrap_langchain_anthropic
-
-    wrap_anthropic = None
-    wrap_openai = None
-    wrap_langchain = None
-    wrap_langchain_openai = None
-    wrap_langchain_anthropic = None
-
-    # Lazily set up anthropic wrapper if available
-    try:
-        from .adapters import anthropic
-
-        wrap_anthropic = anthropic.wrap_anthropic
-        # Apply the wrapper, but don't fail if it doesn't work
-        try:
-            wrap_anthropic()
-        except Exception as e:
-            logger.warning(f"Failed to wrap Anthropic: {str(e)}")
-    except ImportError:
-
-        def wrap_anthropic():
-            logger.warning(
-                "Anthropic support not available. Install Anthropic to use this feature."
-            )
-
-    # Lazily set up OpenAI wrapper if available
-    try:
-        from .adapters import openai
-
-        wrap_openai = openai.wrap_openai
-        # Apply the wrapper, but don't fail if it doesn't work
-        try:
-            wrap_openai()
-        except Exception as e:
-            logger.warning(f"Failed to wrap OpenAI: {str(e)}")
-    except ImportError:
-
-        def wrap_openai():
-            logger.warning(
-                "OpenAI support not available. Install OpenAI to use this feature."
-            )
-
-    # Lazily set up LangChain wrapper if available
-    try:
-        from .adapters import langchain
-
-        wrap_langchain = langchain.wrap_langchain
-        # Apply the wrapper, but don't fail if it doesn't work
-        try:
-            wrap_langchain()
-        except Exception as e:
-            logger.warning(f"Failed to wrap LangChain: {str(e)}")
-    except ImportError:
-
-        def wrap_langchain():
-            logger.warning(
-                "LangChain support not available. Install LangChain to use this feature."
-            )
-
-    # Lazily set up LangChain-OpenAI wrapper if available
-    try:
-        from .adapters import langchain_openai
-
-        wrap_langchain_openai = langchain_openai.wrap_langchain_openai
-        # Apply the wrapper, but don't fail if it doesn't work
-        try:
-            wrap_langchain_openai()
-        except Exception as e:
-            logger.warning(f"Failed to wrap LangChain-OpenAI: {str(e)}")
-    except ImportError:
-
-        def wrap_langchain_openai():
-            logger.warning(
-                "LangChain-OpenAI support not available. Install LangChain and OpenAI to use this feature."
-            )
-
-    # Lazily set up LangChain-Anthropic wrapper if available
-    try:
-        from .adapters import langchain_anthropic
-
-        wrap_langchain_anthropic = langchain_anthropic.wrap_langchain_anthropic
-        # Apply the wrapper, but don't fail if it doesn't work
-        try:
-            wrap_langchain_anthropic()
-        except Exception as e:
-            logger.warning(f"Failed to wrap LangChain-Anthropic: {str(e)}")
-    except ImportError:
-
-        def wrap_langchain_anthropic():
-            logger.warning(
-                "LangChain-Anthropic support not available. Install LangChain and Anthropic to use this feature."
-            )
-
-
-# Initialize the module on first import, but not during installation
-import sys
-
-if "pip" not in sys.modules:
-    _import_modules()