osmosis-ai 0.1.5__py3-none-any.whl → 0.1.7__py3-none-any.whl

osmosis_ai/consts.py

@@ -2,15 +2,16 @@ from enum import Enum
 
 # Extract package metadata
 package_name = "osmosis-ai"
-package_version = "0.1.5"
+package_version = "0.1.7"
 
 indent = 2  # Number of spaces to use for indentation in pretty print
-osmosis_api_url = "https://ftgrv77m9f.execute-api.us-west-2.amazonaws.com/prod"
+osmosis_api_url = "https://osmosis.gulp.dev"
 
 DEFAULT_LOG_DESTINATION = "none"  # "none" or "stdout" or "stderr" or "file"
 
+
 class LogDestination(Enum):
     NONE = "none"
     STDOUT = "stdout"
     STDERR = "stderr"
-    FILE = "file"
+    FILE = "file"

osmosis_ai/logger.py

@@ -1,5 +1,5 @@
 """
-Logging utilities for Osmosis Wrap
+Logging utilities for Osmosis
 """
 
 import sys
@@ -8,7 +8,7 @@ import os
 from .consts import LogDestination, DEFAULT_LOG_DESTINATION
 
 # Create logger
-logger = logging.getLogger('osmosis_ai')
+logger = logging.getLogger("osmosis_ai")
 logger.setLevel(logging.INFO)
 
 # Default configuration - no logging
@@ -18,6 +18,7 @@ logger.propagate = False
 log_destination = DEFAULT_LOG_DESTINATION
 logger.log_destination = log_destination
 
+
 def set_log_destination(destination: LogDestination) -> None:
     """
     Set the log destination for the logger
@@ -26,6 +27,7 @@ def set_log_destination(destination: LogDestination) -> None:
     log_destination = destination
     logger.log_destination = destination
 
+
 def configure_logger() -> None:
     """
     Configure the logger based on the log_destination setting in consts.py
@@ -33,7 +35,7 @@ def configure_logger() -> None:
     # Remove any existing handlers
     for handler in logger.handlers[:]:
         logger.removeHandler(handler)
-    
+
     # Configure based on log_destination
     if log_destination == "none":
         # No logging
@@ -54,18 +56,22 @@ def configure_logger() -> None:
         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')
+    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() 
+    configure_logger()

osmosis_ai/utils.py

@@ -1,13 +1,15 @@
 """
-Utility functions for Osmosis Wrap adapters
+Utility functions for osmosisadapters
 """
 
 import json
 from datetime import datetime, timezone
 from typing import Any, Dict
 import xxhash
+
 # Import constants
 from .consts import osmosis_api_url
+
 # Import logger
 from .logger import logger
 
@@ -16,10 +18,11 @@ enabled = True
 osmosis_api_key = None  # Will be set by init()
 _initialized = False
 
+
 def init(api_key: str) -> None:
     """
-    Initialize Osmosis Wrap with the OSMOSIS API key.
-    
+    Initialize osmosiswith the OSMOSIS API key.
+
     Args:
         api_key: The OSMOSIS API key for logging LLM usage
     """
@@ -27,18 +30,23 @@ def init(api_key: str) -> None:
     osmosis_api_key = api_key
     _initialized = True
 
+
 def disable_osmosis() -> None:
     global enabled
     enabled = False
 
+
 def enable_osmosis() -> None:
     global enabled
     enabled = True
 
-def send_to_osmosis(query: Dict[str, Any], response: Dict[str, Any], status: int = 200) -> None:
+
+def send_to_osmosis(
+    query: Dict[str, Any], response: Dict[str, Any], status: int = 200
+) -> None:
     """
     Send query and response data to the OSMOSIS API using AWS Firehose.
-    
+
     Args:
         query: The query/request data
         response: The response data
@@ -48,41 +56,42 @@ def send_to_osmosis(query: Dict[str, Any], response: Dict[str, Any], status: int
         return
 
     if not _initialized:
-        logger.warning("Osmosis Wrap not initialized. Call osmosis_ai.init(api_key) first.")
+        logger.warning("osmosisnot initialized. Call osmosis_ai.init(api_key) first.")
         return
 
     try:
         # Import requests only when needed
         import requests
-        
+
         # Create headers
-        headers = {
-            "Content-Type": "application/json",
-            "x-api-key": osmosis_api_key
-        }
-            
+        headers = {"Content-Type": "application/json", "x-api-key": osmosis_api_key}
+
         # Prepare main data payload
         data = {
-            "owner": xxhash.xxh32(osmosis_api_key.encode('utf-8')).hexdigest(),
+            "owner": xxhash.xxh32(osmosis_api_key.encode("utf-8")).hexdigest(),
             "date": int(datetime.now(timezone.utc).timestamp()),
             "query": query,
             "response": response,
-            "status": status
+            "status": status,
         }
 
         logger.info(f"Sending data to OSMOSIS API: {data}")
-        
+
         # Send main data payload
         response_data = requests.post(
-            f"{osmosis_api_url}/data",
+            f"{osmosis_api_url}/ingest",
             headers=headers,
-            data=json.dumps(data).replace('\n', '') + '\n'
+            data=json.dumps(data).replace("\n", "") + "\n",
         )
-        
+
         if response_data.status_code != 200:
-            logger.warning(f"OSMOSIS API returned status {response_data.status_code} for data")
-    
+            logger.warning(
+                f"OSMOSIS API returned status {response_data.status_code} for data with error: {response_data.text}"
+            )
+
     except ImportError:
-        logger.warning("Requests library not installed. Please install it with 'pip install requests'.")
+        logger.warning(
+            "Requests library not installed. Please install it with 'pip install requests'."
+        )
     except Exception as e:
-        logger.warning(f"Failed to send data to OSMOSIS API: {str(e)}")
+        logger.warning(f"Failed to send data to OSMOSIS API: {str(e)}")

{osmosis_ai-0.1.5.dist-info → osmosis_ai-0.1.7.dist-info}/METADATA

@@ -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.7.dist-info/RECORD

@@ -0,0 +1,15 @@
+osmosis_ai/__init__.py,sha256=esOnyFqlu7-AKuxTV5fkLVb34-kvCReMEDFkFpxBH7M,4374
+osmosis_ai/consts.py,sha256=1HavvPFM9nJCDqy1RK7VH_g9MbAgXNyBqN8Rt3FAeDE,408
+osmosis_ai/logger.py,sha256=J2Yw_QMGqXySzJlj_QQqHLs52S1oKYXoELnU66YCAuI,2170
+osmosis_ai/utils.py,sha256=pJDJTO1ov0Nyyj6UMbPGZIAPyoF-wY9QM3hYoxtIuo4,2490
+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.7.dist-info/licenses/LICENSE,sha256=WLvxAm2-fPuT8qECE6ejD8ezG9btIanJl7zysXvtQgM,1064
+osmosis_ai-0.1.7.dist-info/METADATA,sha256=wVZ0LQAJIfGKinOoQrgCUKwCOf667InqQxOYIWBBv_Q,9255
+osmosis_ai-0.1.7.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+osmosis_ai-0.1.7.dist-info/top_level.txt,sha256=UPNRTKIBSrxsJVNxwXnLCqSoBS4bAiL_3jMtjvf5zEY,11
+osmosis_ai-0.1.7.dist-info/RECORD,,

{osmosis_ai-0.1.5.dist-info → osmosis_ai-0.1.7.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (77.0.3)
+Generator: setuptools (78.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

osmosis_ai-0.1.5.dist-info/RECORD

@@ -1,15 +0,0 @@
-osmosis_ai/__init__.py,sha256=nQJwkT7efmsSQ0ydygcH1OUW8y2rKxHCnwKD-Br51Co,4266
-osmosis_ai/consts.py,sha256=X47sKF5fDss23UIJJwwlIsP-Py78afwiqOcxDJPKiYQ,441
-osmosis_ai/logger.py,sha256=bUok07WG2vaY0lDdEnn_JM_rI_UaRu3PcidVKaTZpPA,2165
-osmosis_ai/utils.py,sha256=A0A2kcPc2os7QRLMHFyXjdhOzszRsAoHhwqWhtpETFY,2488
-osmosis_ai/adapters/__init__.py,sha256=Lq9r0h-jwKoZy2iabwUxpbwlf6ybobRRpbGi6NfeX7U,267
-osmosis_ai/adapters/anthropic.py,sha256=ans7rYdtlgXtjcbGNyuLIST18gQKfU1l4kghD9E3ugM,21801
-osmosis_ai/adapters/langchain.py,sha256=0AF6S7he-XBfjaYq2J8e-Zn_X2wT0P2YsoEi8pmyqKU,26402
-osmosis_ai/adapters/langchain_anthropic.py,sha256=IQ3uhn-3mgvK3ln4kqR_3OyPaI_10X6WFRgoiebuN_4,13355
-osmosis_ai/adapters/langchain_openai.py,sha256=8VctxATCHi6M2jy3JtLOWez9jww1bMXg4t8A1c78oTk,22913
-osmosis_ai/adapters/openai.py,sha256=E036slghKoe80futj8wsLY2exDjbwgtTxpnR3UCgfRY,38625
-osmosis_ai-0.1.5.dist-info/licenses/LICENSE,sha256=WLvxAm2-fPuT8qECE6ejD8ezG9btIanJl7zysXvtQgM,1064
-osmosis_ai-0.1.5.dist-info/METADATA,sha256=rd9ObQ2l-RgCU_hnQhiFdRN_Ec2l36BF19aF706Roh4,9232
-osmosis_ai-0.1.5.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
-osmosis_ai-0.1.5.dist-info/top_level.txt,sha256=UPNRTKIBSrxsJVNxwXnLCqSoBS4bAiL_3jMtjvf5zEY,11
-osmosis_ai-0.1.5.dist-info/RECORD,,