diff --git a/edgee/__init__.py b/edgee/__init__.py index 57f3387..fd930a9 100644 --- a/edgee/__init__.py +++ b/edgee/__init__.py @@ -60,10 +60,7 @@ class InputObject: tool_choice: str | dict | None = None tags: list[str] | None = None compression_model: str | None = ( - None # Compression model: agentic, claude, opencode, cursor, customer (gateway-internal) - ) - compression_configuration: dict | None = ( - None # Configuration for compression model: {"rate": 0.7, "semantic_preservation_threshold": 60} + None # Compression model: claude, opencode, cursor, customer (gateway-internal) ) @@ -219,21 +216,18 @@ def send( tool_choice = None tags = None compression_model = None - compression_configuration = None elif isinstance(input, InputObject): messages = input.messages tools = input.tools tool_choice = input.tool_choice tags = input.tags compression_model = input.compression_model - compression_configuration = input.compression_configuration else: messages = input.get("messages", []) tools = input.get("tools") tool_choice = input.get("tool_choice") tags = input.get("tags") compression_model = input.get("compression_model") - compression_configuration = input.get("compression_configuration") body: dict = {"model": model, "messages": messages} if stream: @@ -246,8 +240,6 @@ def send( body["tags"] = tags if compression_model is not None: body["compression_model"] = compression_model - if compression_configuration is not None: - body["compression_configuration"] = compression_configuration request = Request( f"{self.base_url}{API_ENDPOINT}", diff --git a/example/compression.py b/example/compression.py index 2b76548..b9a5788 100644 --- a/example/compression.py +++ b/example/compression.py @@ -92,10 +92,7 @@ "messages": [ {"role": "user", "content": user_message}, ], - "compression_model": "agentic", - "compression_configuration": { - "rate": 0.5, - }, + "compression_model": "claude", }, ) diff --git a/pyproject.toml b/pyproject.toml index 733ced4..bf151e6 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "edgee" -version = "1.0.3" +version = "1.0.4" description = "Lightweight Python SDK for Edgee AI Gateway" readme = "README.md" license = "Apache-2.0" diff --git a/uv.lock b/uv.lock index 6b24728..7963285 100644 --- a/uv.lock +++ b/uv.lock @@ -22,7 +22,7 @@ wheels = [ [[package]] name = "edgee" -version = "1.0.3" +version = "1.0.4" source = { editable = "." } [package.optional-dependencies]