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

osmosis_ai/logger.py

@@ -1,77 +0,0 @@
-"""
-Logging utilities for Osmosis
-"""
-
-import sys
-import logging
-import os
-from .consts import LogDestination, DEFAULT_LOG_DESTINATION
-
-# Create logger
-logger = logging.getLogger("osmosis_ai")
-logger.setLevel(logging.INFO)
-
-# Default configuration - no logging
-logger.propagate = False
-
-# Default log destination
-log_destination = DEFAULT_LOG_DESTINATION
-logger.log_destination = log_destination
-
-
-def set_log_destination(destination: LogDestination) -> None:
-    """
-    Set the log destination for the logger
-    """
-    global log_destination
-    log_destination = destination
-    logger.log_destination = destination
-
-
-def configure_logger() -> None:
-    """
-    Configure the logger based on the log_destination setting in consts.py
-    """
-    # Remove any existing handlers
-    for handler in logger.handlers[:]:
-        logger.removeHandler(handler)
-
-    # Configure based on log_destination
-    if log_destination == "none":
-        # No logging
-        return
-    elif log_destination == "stdout":
-        handler = logging.StreamHandler(sys.stdout)
-        logger.addHandler(handler)
-    elif log_destination == "stderr":
-        handler = logging.StreamHandler(sys.stderr)
-        logger.addHandler(handler)
-    elif log_destination == "file":
-        # Log to a file in the current directory
-        log_file = os.environ.get("OSMOSIS_LOG_FILE", "osmosis_ai.log")
-        handler = logging.FileHandler(log_file)
-        logger.addHandler(handler)
-    else:
-        # Invalid setting, log to stderr as fallback
-        handler = logging.StreamHandler(sys.stderr)
-        logger.addHandler(handler)
-        logger.warning(f"Invalid log_destination: {log_destination}, using stderr")
-
-    # Set formatter
-    formatter = logging.Formatter(
-        "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
-    )
-    for handler in logger.handlers:
-        handler.setFormatter(formatter)
-
-
-# Initialize the logger on module import
-configure_logger()
-
-
-# Function to force reconfiguration if log_destination is changed
-def reconfigure_logger() -> None:
-    """
-    Reconfigure the logger, should be called if log_destination changes
-    """
-    configure_logger()

osmosis_ai-0.1.8.dist-info/METADATA

@@ -1,281 +0,0 @@
-Metadata-Version: 2.4
-Name: osmosis-ai
-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
-        
-        Copyright (c) 2024 Gulp AI
-        
-        Permission is hereby granted, free of charge, to any person obtaining a copy
-        of this software and associated documentation files (the "Software"), to deal
-        in the Software without restriction, including without limitation the rights
-        to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-        copies of the Software, and to permit persons to whom the Software is
-        furnished to do so, subject to the following conditions:
-        
-        The above copyright notice and this permission notice shall be included in all
-        copies or substantial portions of the Software.
-        
-        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-        AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-        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-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
-Requires-Python: >=3.6
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: python-dotenv>=0.19.0
-Requires-Dist: requests>=2.25.0
-Requires-Dist: xxhash>=3.4.1
-Provides-Extra: anthropic
-Requires-Dist: anthropic>=0.5.0; extra == "anthropic"
-Provides-Extra: openai
-Requires-Dist: openai>=0.27.0; extra == "openai"
-Provides-Extra: langchain
-Requires-Dist: langchain>=0.0.200; extra == "langchain"
-Provides-Extra: langchain-openai
-Requires-Dist: langchain>=0.0.200; extra == "langchain-openai"
-Requires-Dist: langchain-openai>=0.0.200; extra == "langchain-openai"
-Requires-Dist: openai>=0.27.0; extra == "langchain-openai"
-Provides-Extra: langchain-anthropic
-Requires-Dist: langchain>=0.0.200; extra == "langchain-anthropic"
-Requires-Dist: langchain-anthropic>=0.0.200; extra == "langchain-anthropic"
-Requires-Dist: anthropic>=0.5.0; extra == "langchain-anthropic"
-Provides-Extra: all
-Requires-Dist: anthropic>=0.5.0; extra == "all"
-Requires-Dist: openai>=0.27.0; extra == "all"
-Requires-Dist: langchain>=0.0.200; extra == "all"
-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-sdk-python/actions/workflows/test.yml/badge.svg)](https://github.com/Gulp-AI/osmosis-sdk-python/actions/workflows/test.yml)
-
-# Osmosis
-
-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
-
-- **Anthropic**: Logs all Claude API requests and responses (both sync and async clients)
-- **OpenAI**: Logs all OpenAI API requests and responses (supports v1 and v2 API versions, both sync and async clients)
-- **LangChain**: Currently supports prompt template logging (LLM and ChatModel support varies by LangChain version)
-
-## Installation
-
-[pypi](https://pypi.org/project/osmosis-ai/)
-
-```bash
-# Basic installation with minimal dependencies
-pip install osmosis-ai
-
-# Install with specific provider support
-pip install "osmosis-ai[openai]"     # Only OpenAI support
-pip install "osmosis-ai[anthropic]"  # Only Anthropic support
-
-# Install with LangChain support
-pip install "osmosis-ai[langchain]"         # Base LangChain support
-pip install "osmosis-ai[langchain-openai]"  # LangChain + OpenAI support
-pip install "osmosis-ai[langchain-anthropic]" # LangChain + Anthropic support
-
-# Install with all dependencies
-pip install "osmosis-ai[all]"
-```
-
-Or install from source:
-
-```bash
-git clone https://github.com/your-username/osmosis-sdk-python.git
-cd osmosis-ai
-pip install -e .
-```
-
-For development, you can install all dependencies using:
-
-```bash
-pip install -r requirements.txt
-```
-
-## Environment Setup
-
-osmosisrequires a OSMOSIS API key to log LLM usage. Create a `.env` file in your project directory:
-
-```bash
-# Copy the sample .env file
-cp .env.sample .env
-
-# Edit the .env file with your API keys
-```
-
-Edit the `.env` file to add your API keys:
-
-```
-# Required for logging
-OSMOSIS_API_KEY=your_osmosis_api_key_here
-
-# Optional: Only needed if you're using these services
-ANTHROPIC_API_KEY=your_anthropic_key_here
-OPENAI_API_KEY=your_openai_key_here
-```
-
-## Usage
-
-First, import and initialize osmosiswith your OSMOSIS API key:
-
-```python
-import os
-import osmosis_ai
-
-# Initialize with your OSMOSIS API key
-osmosis_ai.init("your-osmosis-api-key")
-
-# Or load from environment variable
-osmosis_api_key = os.environ.get("OSMOSIS_API_KEY")
-osmosis_ai.init(osmosis_api_key)
-```
-
-Once you import `osmosis_ai` and initialize it, the library automatically patches the supported LLM clients. You can then use your LLM clients normally, and all API calls will be logged to OSMOSIS:
-
-### Anthropic Example
-
-```python
-# Import osmosis_ai first and initialize it
-import osmosis_ai
-osmosis_ai.init(os.environ.get("OSMOSIS_API_KEY"))
-
-# Then import and use Anthropic as normal
-from anthropic import Anthropic
-
-# Create and use the Anthropic client as usual - it's already patched
-client = Anthropic(api_key=os.environ.get("ANTHROPIC_API_KEY"))
-
-# All API calls will now be logged to OSMOSIS automatically
-response = client.messages.create(
-    model="claude-3-haiku-20240307",
-    max_tokens=1000,
-    messages=[
-        {"role": "user", "content": "Hello, Claude!"}
-    ]
-)
-
-# Async client is also supported and automatically patched
-from anthropic import AsyncAnthropic
-import asyncio
-
-async def call_claude_async():
-    async_client = AsyncAnthropic(api_key=os.environ.get("ANTHROPIC_API_KEY"))
-    response = await async_client.messages.create(
-        model="claude-3-haiku-20240307",
-        max_tokens=1000,
-        messages=[
-            {"role": "user", "content": "Hello, async Claude!"}
-        ]
-    )
-    return response
-
-# All async API calls will be logged to OSMOSIS as well
-asyncio.run(call_claude_async())
-```
-
-### OpenAI Example
-
-```python
-# Import osmosis_ai first and initialize it
-import osmosis_ai
-osmosis_ai.init(os.environ.get("OSMOSIS_API_KEY"))
-
-# Then import and use OpenAI as normal
-from openai import OpenAI
-
-# Create and use the OpenAI client as usual - it's already patched
-client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
-
-# All API calls will now be logged to OSMOSIS automatically
-response = client.chat.completions.create(
-    model="gpt-4o-mini",
-    max_tokens=150,
-    messages=[
-        {"role": "user", "content": "Hello, GPT!"}
-    ]
-)
-
-# Async client is also supported and automatically patched
-from openai import AsyncOpenAI
-import asyncio
-
-async def call_openai_async():
-    async_client = AsyncOpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
-    response = await async_client.chat.completions.create(
-        model="gpt-4o-mini",
-        max_tokens=150,
-        messages=[
-            {"role": "user", "content": "Hello, async GPT!"}
-        ]
-    )
-    return response
-
-# All async API calls will be logged to OSMOSIS as well
-asyncio.run(call_openai_async())
-```
-
-### LangChain Example
-
-```python
-# Import osmosis_ai first and initialize it
-import osmosis_ai
-osmosis_ai.init(os.environ.get("OSMOSIS_API_KEY"))
-
-# Then use LangChain as normal
-from langchain_core.prompts import PromptTemplate
-
-# Use LangChain prompt templates as usual
-template = PromptTemplate(
-    input_variables=["topic"],
-    template="Write a short paragraph about {topic}."
-)
-
-# Formatting the prompt will be logged to OSMOSIS automatically
-formatted_prompt = template.format(topic="artificial intelligence")
-print(f"Formatted prompt: {formatted_prompt}")
-
-# Multiple prompt templates are also captured
-template2 = PromptTemplate(
-    input_variables=["name", "profession"],
-    template="My name is {name} and I work as a {profession}."
-)
-formatted_prompt2 = template2.format(name="Alice", profession="data scientist")
-print(f"Formatted prompt 2: {formatted_prompt2}")
-```
-
-## Configuration
-
-You can configure the behavior of the library by modifying the following variables:
-
-```python
-import osmosis_ai
-
-# Disable logging to OSMOSIS (default: True)
-osmosis_ai.enabled = False
-```
-
-## How it Works
-
-This library uses monkey patching to override the LLM clients' methods that make API calls. When you import the `osmosis_ai` module, it automatically patches the supported LLM client libraries. When methods are called on these clients, the library intercepts the calls and sends the request parameters and response data to the OSMOSIS API for logging and monitoring.
-
-The data sent to OSMOSIS includes:
-- Timestamp (UTC)
-- Request parameters
-- Response data
-- HTTP status code
-
-## License
-
-MIT 

osmosis_ai-0.1.8.dist-info/RECORD

@@ -1,15 +0,0 @@
-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=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.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,,