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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: osmosis-ai
-Version: 0.1.5
+Version: 0.1.7
 Summary: Monkey patches LLM client libraries to print all prompts and responses
 Author-email: Gulp AI <jake@gulp.ai>
 License: MIT License
@@ -24,8 +24,8 @@ License: MIT License
         LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
         OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
         SOFTWARE. 
-Project-URL: Homepage, https://github.com/Gulp-AI/osmosis-ai
-Project-URL: Issues, https://github.com/Gulp-AI/osmosis-ai/issues
+Project-URL: Homepage, https://github.com/Gulp-AI/osmosis-sdk-python
+Project-URL: Issues, https://github.com/Gulp-AI/osmosis-sdk-python/issues
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
@@ -57,11 +57,11 @@ Requires-Dist: langchain-openai>=0.0.200; extra == "all"
 Requires-Dist: langchain-anthropic>=0.0.200; extra == "all"
 Dynamic: license-file
 
-[![Run Tests](https://github.com/Gulp-AI/osmosis-ai/actions/workflows/test.yml/badge.svg)](https://github.com/Gulp-AI/osmosis-ai/actions/workflows/test.yml)
+[![Run Tests](https://github.com/Gulp-AI/osmosis-sdk-python/actions/workflows/test.yml/badge.svg)](https://github.com/Gulp-AI/osmosis-sdk-python/actions/workflows/test.yml)
 
-# Osmosis Wrap
+# Osmosis
 
-A Python library that monkey patches LLM client libraries to send all prompts and responses to the OSMOSIS API for logging and monitoring.
+A Python library that monkey patches LLM client libraries to send all prompts and responses to the Osmosis API for logging and monitoring.
 
 ## Supported Libraries
 
@@ -93,7 +93,7 @@ pip install "osmosis-ai[all]"
 Or install from source:
 
 ```bash
-git clone https://github.com/your-username/osmosis-ai.git
+git clone https://github.com/your-username/osmosis-sdk-python.git
 cd osmosis-ai
 pip install -e .
 ```
@@ -106,7 +106,7 @@ pip install -r requirements.txt
 
 ## Environment Setup
 
-Osmosis Wrap requires a OSMOSIS API key to log LLM usage. Create a `.env` file in your project directory:
+osmosisrequires a OSMOSIS API key to log LLM usage. Create a `.env` file in your project directory:
 
 ```bash
 # Copy the sample .env file
@@ -128,7 +128,7 @@ OPENAI_API_KEY=your_openai_key_here
 
 ## Usage
 
-First, import and initialize Osmosis Wrap with your OSMOSIS API key:
+First, import and initialize osmosiswith your OSMOSIS API key:
 
 ```python
 import os

{osmosis_ai-0.1.5 → osmosis_ai-0.1.7}/README.md

@@ -1,8 +1,8 @@
-[![Run Tests](https://github.com/Gulp-AI/osmosis-ai/actions/workflows/test.yml/badge.svg)](https://github.com/Gulp-AI/osmosis-ai/actions/workflows/test.yml)
+[![Run Tests](https://github.com/Gulp-AI/osmosis-sdk-python/actions/workflows/test.yml/badge.svg)](https://github.com/Gulp-AI/osmosis-sdk-python/actions/workflows/test.yml)
 
-# Osmosis Wrap
+# Osmosis
 
-A Python library that monkey patches LLM client libraries to send all prompts and responses to the OSMOSIS API for logging and monitoring.
+A Python library that monkey patches LLM client libraries to send all prompts and responses to the Osmosis API for logging and monitoring.
 
 ## Supported Libraries
 
@@ -34,7 +34,7 @@ pip install "osmosis-ai[all]"
 Or install from source:
 
 ```bash
-git clone https://github.com/your-username/osmosis-ai.git
+git clone https://github.com/your-username/osmosis-sdk-python.git
 cd osmosis-ai
 pip install -e .
 ```
@@ -47,7 +47,7 @@ pip install -r requirements.txt
 
 ## Environment Setup
 
-Osmosis Wrap requires a OSMOSIS API key to log LLM usage. Create a `.env` file in your project directory:
+osmosisrequires a OSMOSIS API key to log LLM usage. Create a `.env` file in your project directory:
 
 ```bash
 # Copy the sample .env file
@@ -69,7 +69,7 @@ OPENAI_API_KEY=your_openai_key_here
 
 ## Usage
 
-First, import and initialize Osmosis Wrap with your OSMOSIS API key:
+First, import and initialize osmosiswith your OSMOSIS API key:
 
 ```python
 import os

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

@@ -10,38 +10,41 @@ 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
-    
+
     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
-    
+
     # 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 +52,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 +69,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 +86,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 +103,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 +120,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-0.1.5 → osmosis_ai-0.1.7}/osmosis_ai/adapters/__init__.py

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