From 5945ba02d6ed9937f7cf1b6b4f24c9325dd3415c Mon Sep 17 00:00:00 2001
From: "Julianne H." <julianne@edgee.ai>
Date: Wed, 30 Sep 2026 17:13:51 +0200
Subject: [PATCH] feat!: remove compression_model input field

---
 edgee/__init__.py   | 13 -------------
 tests/test_edgee.py | 15 ---------------
 2 files changed, 28 deletions(-)

diff --git a/edgee/__init__.py b/edgee/__init__.py
index c9323ec..ea04f3e 100644
--- a/edgee/__init__.py
+++ b/edgee/__init__.py
@@ -3,7 +3,6 @@
 import json
 import os
 import ssl
-import warnings
 from dataclasses import dataclass
 from urllib.error import HTTPError
 from urllib.request import Request, urlopen
@@ -76,8 +75,6 @@ class InputObject:
     tools: list[dict] | None = None
     tool_choice: str | dict | None = None
     tags: list[str] | None = None
-    # Deprecated: any value turns tool-result trimming on. Use tool_result_trimming instead.
-    compression_model: str | None = None
     # Per-request overrides of the API key settings. None keeps the key setting.
     tool_result_trimming: bool | None = None
     tool_surface_reduction: bool | None = None
@@ -235,21 +232,18 @@ def send(
             tools = None
             tool_choice = None
             tags = None
-            compression_model = None
             toggles = {}
         elif isinstance(input, InputObject):
             messages = input.messages
             tools = input.tools
             tool_choice = input.tool_choice
             tags = input.tags
-            compression_model = input.compression_model
             toggles = {field: getattr(input, field) for field in COMPRESSION_HEADERS}
         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")
             toggles = {field: input.get(field) for field in COMPRESSION_HEADERS}
 
         body: dict = {"model": model, "messages": messages}
@@ -261,13 +255,6 @@ def send(
             body["tool_choice"] = tool_choice
         if tags:
             body["tags"] = tags
-        if compression_model is not None:
-            warnings.warn(
-                "compression_model is deprecated; use tool_result_trimming=True instead",
-                DeprecationWarning,
-                stacklevel=2,
-            )
-            body["compression_model"] = compression_model
 
         request = Request(
             f"{self.base_url}{API_ENDPOINT}",
diff --git a/tests/test_edgee.py b/tests/test_edgee.py
index 0a9fef5..f1907e3 100644
--- a/tests/test_edgee.py
+++ b/tests/test_edgee.py
@@ -460,18 +460,3 @@ def test_streaming_request_sends_toggles(self, mock_urlopen):
         request, _ = self._sent(mock_urlopen)
         assert request.get_header(self.SURFACE.capitalize()) == "true"
         assert not request.has_header(self.TRIM.capitalize())
-
-    @patch("edgee.urlopen")
-    def test_compression_model_is_deprecated_but_still_sent(self, mock_urlopen):
-        mock_urlopen.return_value = self._ok()
-        with pytest.warns(DeprecationWarning, match="tool_result_trimming"):
-            Edgee("test-api-key").send(
-                model="gpt-4",
-                input={
-                    "messages": [{"role": "user", "content": "Hello"}],
-                    "compression_model": "claude",
-                },
-            )
-
-        _, body = self._sent(mock_urlopen)
-        assert body["compression_model"] == "claude"