edgee 1.0.0__py3-none-any.whl → 1.0.1__py3-none-any.whl

edgee/__init__.py

@@ -45,6 +45,13 @@ class InputObject:
     messages: list[dict]
     tools: list[dict] | None = None
     tool_choice: str | dict | None = None
+    tags: list[str] | None = None
+    enable_compression: bool | None = (
+        None  # Enable token compression (gateway-internal, not sent to providers)
+    )
+    compression_rate: float | None = (
+        None  # Compression rate 0.0-1.0 (gateway-internal, not sent to providers)
+    )
 
 
 @dataclass
@@ -61,10 +68,18 @@ class Usage:
     total_tokens: int
 
 
+@dataclass
+class Compression:
+    input_tokens: int
+    saved_tokens: int
+    rate: float
+
+
 @dataclass
 class SendResponse:
     choices: list[Choice]
     usage: Usage | None = None
+    compression: Compression | None = None
 
     @property
     def text(self) -> str | None:
@@ -188,14 +203,23 @@ class Edgee:
             messages = [{"role": "user", "content": input}]
             tools = None
             tool_choice = None
+            tags = None
+            enable_compression = None
+            compression_rate = None
         elif isinstance(input, InputObject):
             messages = input.messages
             tools = input.tools
             tool_choice = input.tool_choice
+            tags = input.tags
+            enable_compression = input.enable_compression
+            compression_rate = input.compression_rate
         else:
             messages = input.get("messages", [])
             tools = input.get("tools")
             tool_choice = input.get("tool_choice")
+            tags = input.get("tags")
+            enable_compression = input.get("enable_compression")
+            compression_rate = input.get("compression_rate")
 
         body: dict = {"model": model, "messages": messages}
         if stream:
@@ -204,6 +228,12 @@ class Edgee:
             body["tools"] = tools
         if tool_choice:
             body["tool_choice"] = tool_choice
+        if tags:
+            body["tags"] = tags
+        if enable_compression is not None:
+            body["enable_compression"] = enable_compression
+        if compression_rate is not None:
+            body["compression_rate"] = compression_rate
 
         request = Request(
             f"{self.base_url}{API_ENDPOINT}",
@@ -246,7 +276,15 @@ class Edgee:
                 total_tokens=data["usage"]["total_tokens"],
             )
 
-        return SendResponse(choices=choices, usage=usage)
+        compression = None
+        if "compression" in data:
+            compression = Compression(
+                input_tokens=data["compression"]["input_tokens"],
+                saved_tokens=data["compression"]["saved_tokens"],
+                rate=data["compression"]["rate"],
+            )
+
+        return SendResponse(choices=choices, usage=usage, compression=compression)
 
     def _handle_streaming_response(self, request: Request):
         """Handle streaming response, yielding StreamChunk objects."""

{edgee-1.0.0.dist-info → edgee-1.0.1.dist-info}/METADATA

@@ -1,10 +1,10 @@
 Metadata-Version: 2.4
 Name: edgee
-Version: 1.0.0
+Version: 1.0.1
 Summary: Lightweight Python SDK for Edgee AI Gateway
-Project-URL: Homepage, https://github.com/edgee-cloud/python-sdk
-Project-URL: Repository, https://github.com/edgee-cloud/python-sdk
-Author-email: Edgee <support@edgee.cloud>
+Project-URL: Homepage, https://github.com/edgee-ai/python-sdk
+Project-URL: Repository, https://github.com/edgee-ai/python-sdk
+Author-email: Edgee <support@edgee.ai>
 License-Expression: Apache-2.0
 License-File: LICENSE
 Keywords: ai,anthropic,gateway,llm,openai
@@ -24,7 +24,7 @@ Description-Content-Type: text/markdown
 
 # Edgee Python SDK
 
-Lightweight, type-safe Python SDK for the [Edgee AI Gateway](https://www.edgee.cloud).
+Lightweight, type-safe Python SDK for the [Edgee AI Gateway](https://www.edgee.ai).
 
 [![PyPI version](https://img.shields.io/pypi/v/edgee.svg)]( )
 [![License](https://img.shields.io/badge/license-Apache%202.0-blue.svg)](LICENSE)
@@ -66,6 +66,15 @@ response = edgee.send(
 print(response.text)           # Text content
 print(response.finish_reason)  # Finish reason
 print(response.tool_calls)     # Tool calls (if any)
+
+# Access usage and compression info
+if response.usage:
+    print(f"Tokens used: {response.usage.total_tokens}")
+
+if response.compression:
+    print(f"Input tokens: {response.compression.input_tokens}")
+    print(f"Saved tokens: {response.compression.saved_tokens}")
+    print(f"Compression rate: {response.compression.rate}")
 ```
 
 ## Stream Method
@@ -88,19 +97,20 @@ for chunk in edgee.stream("gpt-4o", "Tell me a story"):
 - ✅ **Streaming** - Real-time response streaming with generators
 - ✅ **Tool calling** - Full support for function calling
 - ✅ **Flexible input** - Accept strings, dicts, or InputObject
+- ✅ **Compression info** - Access token compression metrics in responses
 - ✅ **Zero dependencies** - Uses only Python standard library
 
 ## Documentation
 
 For complete documentation, examples, and API reference, visit:
 
-**👉 [Official Python SDK Documentation](https://www.edgee.cloud/docs/sdk/python)**
+**👉 [Official Python SDK Documentation](https://www.edgee.ai/docs/sdk/python)**
 
 The documentation includes:
-- [Configuration guide](https://www.edgee.cloud/docs/sdk/python/configuration) - Multiple ways to configure the SDK
-- [Send method](https://www.edgee.cloud/docs/sdk/python/send) - Complete guide to non-streaming requests
-- [Stream method](https://www.edgee.cloud/docs/sdk/python/stream) - Streaming responses guide
-- [Tools](https://www.edgee.cloud/docs/sdk/python/tools) - Function calling guide
+- [Configuration guide](https://www.edgee.ai/docs/sdk/python/configuration) - Multiple ways to configure the SDK
+- [Send method](https://www.edgee.ai/docs/sdk/python/send) - Complete guide to non-streaming requests
+- [Stream method](https://www.edgee.ai/docs/sdk/python/stream) - Streaming responses guide
+- [Tools](https://www.edgee.ai/docs/sdk/python/tools) - Function calling guide
 
 ## License
 

edgee-1.0.1.dist-info/RECORD

@@ -0,0 +1,6 @@
+edgee/__init__.py,sha256=kokFp3wCXmib5e_HvE2KkE8gmRYyrqubyJGGGNWUz54,10700
+edgee/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+edgee-1.0.1.dist-info/METADATA,sha256=b9A9YRZrdQlHbmQNeHR_c-0afAwYrmhCrnoXsRIkXZA,3622
+edgee-1.0.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+edgee-1.0.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+edgee-1.0.1.dist-info/RECORD,,

edgee-1.0.0.dist-info/RECORD

@@ -1,6 +0,0 @@
-edgee/__init__.py,sha256=RZLLPbFiWAObCugQy8TZ7FXOO2uMjSEyhxuw61btXbs,9300
-edgee/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-edgee-1.0.0.dist-info/METADATA,sha256=ezHz8hxvm_o4i87jCKobLr6HBH7FN8LjS6_h_7Te2l8,3247
-edgee-1.0.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-edgee-1.0.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-edgee-1.0.0.dist-info/RECORD,,