req updates

Author: JakeTrock

README.md

@@ -15,7 +15,20 @@ A Python library that monkey patches LLM client libraries to send all prompts an
 [pypi](https://pypi.org/project/osmosis-wrap/)
 
 ```bash
+# Basic installation with minimal dependencies
 pip install osmosis-wrap
+
+# Install with specific provider support
+pip install "osmosis-wrap[openai]"     # Only OpenAI support
+pip install "osmosis-wrap[anthropic]"  # Only Anthropic support
+
+# Install with LangChain support
+pip install "osmosis-wrap[langchain]"         # Base LangChain support
+pip install "osmosis-wrap[langchain-openai]"  # LangChain + OpenAI support
+pip install "osmosis-wrap[langchain-anthropic]" # LangChain + Anthropic support
+
+# Install with all dependencies
+pip install "osmosis-wrap[all]"
 ```
 
 Or install from source:

osmosis_wrap/__init__.py

@@ -12,13 +12,12 @@
 
 # Use lazy imports to avoid importing modules during setup
 def _import_modules():
-    global utils, logger, reconfigure_logger, anthropic, openai, langchain
+    global utils, logger, reconfigure_logger
     global enabled, use_stderr, pretty_print, print_messages
-    global init, wrap_anthropic, wrap_openai, disable_hoover, enable_hoover, wrap_langchain
+    global init, disable_hoover, enable_hoover
 
     from . import utils
     from .logger import logger, reconfigure_logger
-    from .adapters import anthropic, openai
 
     # Re-export configuration flags for easy access
     enabled = utils.enabled
@@ -29,19 +28,47 @@ def _import_modules():
     # Re-export initialization function
     init = utils.init
 
-    # Export adapter functions
-    wrap_anthropic = anthropic.wrap_anthropic
-    wrap_openai = openai.wrap_openai
-    
     # Export disable and enable functions
     disable_hoover = utils.disable_hoover
     enable_hoover = utils.enable_hoover
 
-    # Automatically apply patches when the module is imported
-    wrap_anthropic()
-    wrap_openai()
+    # 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.")
 
-    # Conditionally load and apply LangChain wrapper
+    # Lazily set up LangChain wrapper if available
     try:
         from .adapters import langchain
         wrap_langchain = langchain.wrap_langchain
@@ -51,9 +78,34 @@ def _import_modules():
         except Exception as e:
             logger.warning(f"Failed to wrap LangChain: {str(e)}")
     except ImportError:
-        # LangChain not installed or not compatible
         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

osmosis_wrap/adapters/__init__.py

@@ -2,6 +2,8 @@
 Adapters package for Osmosis Wrap
 
 This package contains various adapters for different LLM providers.
+Each adapter is conditionally imported based on installed dependencies.
 """
 
-__all__ = ['anthropic', 'openai', 'langchain', 'langchain_anthropic', 'langchain_openai'] 
+# Don't include adapters in __all__ as they're imported conditionally
+__all__ = [] 

osmosis_wrap/consts.py

@@ -1,6 +1,6 @@
 # Extract package metadata
 package_name = "osmosis-wrap"
-package_version = "0.1.0"
+package_version = "0.1.1"
 
 indent = 2  # Number of spaces to use for indentation in pretty print
 hoover_api_url = "https://ftgrv77m9f.execute-api.us-west-2.amazonaws.com/prod"

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "osmosis-wrap"
-version = "0.1.0"
+version = "0.1.1"
 description = "Monkey patches LLM client libraries to print all prompts and responses"
 readme = "README.md"
 authors = [
@@ -30,4 +30,12 @@ Issues = "https://github.com/Gulp-AI/osmosis-wrap/issues"
 anthropic = ["anthropic>=0.5.0"]
 openai = ["openai>=0.27.0"]
 langchain = ["langchain>=0.0.200"]
-all = ["anthropic>=0.5.0", "openai>=0.27.0", "langchain>=0.0.200"] 
+langchain-openai = ["langchain>=0.0.200", "langchain-openai>=0.0.200", "openai>=0.27.0"]
+langchain-anthropic = ["langchain>=0.0.200", "langchain-anthropic>=0.0.200", "anthropic>=0.5.0"]
+all = [
+    "anthropic>=0.5.0", 
+    "openai>=0.27.0", 
+    "langchain>=0.0.200",
+    "langchain-openai>=0.0.200",
+    "langchain-anthropic>=0.0.200"
+] 

requirements.txt

@@ -3,8 +3,8 @@ python-dotenv>=0.19.0
 requests>=2.25.0
 
 # LLM provider clients - install the ones you need
-anthropic>=0.5.0  # For Claude API
-openai>=0.27.0    # For OpenAI API
+anthropic>=0.49.0  # For Claude API
+openai>=1.27.0   # For OpenAI API
 langchain>=0.0.200  # For LangChain integration
 langchain-openai>=0.0.200  # For LangChain OpenAI integration
 langchain-anthropic>=0.0.200  # For LangChain Anthropic integration