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

osmosis_ai/__init__.py

@@ -10,38 +10,44 @@ Currently supported adapters:
 - 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
-    
+    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:
@@ -49,12 +55,16 @@ def _import_modules():
         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.")
-    
+            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:
@@ -62,12 +72,16 @@ def _import_modules():
         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.")
-    
+            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:
@@ -75,12 +89,16 @@ def _import_modules():
         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.")
-            
+            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:
@@ -88,12 +106,16 @@ def _import_modules():
         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.")
-            
+            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:
@@ -101,10 +123,15 @@ def _import_modules():
         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.")
+            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() 
+
+if "pip" not in sys.modules:
+    _import_modules()

osmosis_ai/adapters/__init__.py

@@ -6,4 +6,4 @@ Each adapter is conditionally imported based on installed dependencies.
 """
 
 # Don't include adapters in __all__ as they're imported conditionally
-__all__ = [] 
+__all__ = []