osmosis-ai 0.1.6__py3-none-any.whl → 0.1.8__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.6"
+package_version = "0.1.8"
 
 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

@@ -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

@@ -4,10 +4,13 @@ Utility functions for osmosisadapters
 
 import json
 from datetime import datetime, timezone
-from typing import Any, Dict
+from typing import Any, Dict, Callable
 import xxhash
+import functools
+
 # Import constants
 from .consts import osmosis_api_url
+
 # Import logger
 from .logger import logger
 
@@ -16,10 +19,11 @@ enabled = True
 osmosis_api_key = None  # Will be set by init()
 _initialized = False
 
+
 def init(api_key: str) -> None:
     """
     Initialize osmosiswith the OSMOSIS API key.
-    
+
     Args:
         api_key: The OSMOSIS API key for logging LLM usage
     """
@@ -27,18 +31,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
@@ -54,35 +63,58 @@ def send_to_osmosis(query: Dict[str, Any], response: Dict[str, Any], status: int
     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)}")
+
+
+def osmosis_reward(func: Callable) -> Callable:
+    """
+    Decorator for reward functions.
+    
+    Args:
+        func: The reward function to be wrapped
+        
+    Returns:
+        The wrapped function
+        
+    Example:
+        @osmosis_reward
+        def calculate_reward(state, action):
+            return state.score + action.value
+    """
+    @functools.wraps(func)
+    def wrapper(*args, **kwargs):
+        return func(*args, **kwargs)
+    
+    return wrapper

{osmosis_ai-0.1.6.dist-info → osmosis_ai-0.1.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: osmosis-ai
-Version: 0.1.6
+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

osmosis_ai-0.1.8.dist-info/RECORD

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

{osmosis_ai-0.1.6.dist-info → osmosis_ai-0.1.8.dist-info}/WHEEL

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

osmosis_ai-0.1.6.dist-info/RECORD

@@ -1,15 +0,0 @@
-osmosis_ai/__init__.py,sha256=nQJwkT7efmsSQ0ydygcH1OUW8y2rKxHCnwKD-Br51Co,4266
-osmosis_ai/consts.py,sha256=EqrQmFe6kXMHgK5J2xdtzUFnuRVaV3KDDsbQATQphwo,441
-osmosis_ai/logger.py,sha256=eG-GJoUo7bcrFkLOS0HGzCSbshaZRoowjSI0ZsUZH1U,2160
-osmosis_ai/utils.py,sha256=_CLtNAxli8bTzIzDY86MWefpJpFoAdta9RHC2DnEpP0,2470
-osmosis_ai/adapters/__init__.py,sha256=IOaShk3HYVdKYRd6aPLu1Shvt_LeUjYaCrtOE3BIrJw,262
-osmosis_ai/adapters/anthropic.py,sha256=GzTw03L9zth7GjCKEbNgEgDQQApwohmm6MpWbsa52CE,21796
-osmosis_ai/adapters/langchain.py,sha256=s_T7LM-o8VR3UMGeudj-XSjC_jDQvAHHAWdgTUJcOKA,26397
-osmosis_ai/adapters/langchain_anthropic.py,sha256=VME8PJMjjtikL00JiFaH7VWtySAWJ9eYez2gFyhGN8U,13350
-osmosis_ai/adapters/langchain_openai.py,sha256=1Jo3X5Yv14SztvidLZM4usdnF8lcjxhVhECO85hgWl4,22908
-osmosis_ai/adapters/openai.py,sha256=8pVtVkD5rUGI5tpNmGIYVBTOyPmheQTknmGHNDjrYd8,38620
-osmosis_ai-0.1.6.dist-info/licenses/LICENSE,sha256=WLvxAm2-fPuT8qECE6ejD8ezG9btIanJl7zysXvtQgM,1064
-osmosis_ai-0.1.6.dist-info/METADATA,sha256=Yl84ihmD7UbC7lS0atAC7lIriauKyXXKq0DNZ0oxVK8,9255
-osmosis_ai-0.1.6.dist-info/WHEEL,sha256=DK49LOLCYiurdXXOXwGJm6U4DkHkg4lcxjhqwRa0CP4,91
-osmosis_ai-0.1.6.dist-info/top_level.txt,sha256=UPNRTKIBSrxsJVNxwXnLCqSoBS4bAiL_3jMtjvf5zEY,11
-osmosis_ai-0.1.6.dist-info/RECORD,,