removed unused code

Author: priyankinfinnov

src/openlayer/lib/integrations/portkey_tracer.py

@@ -532,7 +532,6 @@ def extract_usage(obj: Any) -> Dict[str, Optional[int]]:
     This function attempts to extract token usage information from various
     locations where it might be stored, including:
     - Direct `usage` attribute
-    - `_hidden_params` (for streaming chunks)
     - `model_dump()` dictionary (for streaming chunks)
     
     Parameters
@@ -556,27 +555,6 @@ def extract_usage(obj: Any) -> Dict[str, Optional[int]]:
                 "completion_tokens": getattr(usage, "completion_tokens", None),
             }
 
-        # Check for usage in _hidden_params (primarily for streaming chunks)
-        if hasattr(obj, "_hidden_params"):
-            hidden_params = obj._hidden_params
-            # Check if usage is a direct attribute
-            if hasattr(hidden_params, "usage") and hidden_params.usage is not None:
-                usage = hidden_params.usage
-                return {
-                    "total_tokens": getattr(usage, "total_tokens", None),
-                    "prompt_tokens": getattr(usage, "prompt_tokens", None),
-                    "completion_tokens": getattr(usage, "completion_tokens", None),
-                }
-            # Check if usage is a dictionary key
-            elif _supports_membership_check(hidden_params) and "usage" in hidden_params:
-                usage = hidden_params["usage"]
-                if usage:
-                    return {
-                        "total_tokens": usage.get("total_tokens", None),
-                        "prompt_tokens": usage.get("prompt_tokens", None),
-                        "completion_tokens": usage.get("completion_tokens", None),
-                    }
-        
         # Check if object model dump has usage (primarily for streaming chunks)
         if hasattr(obj, "model_dump"):
             obj_dict = obj.model_dump()

tests/test_portkey_integration.py

@@ -230,15 +230,6 @@ def test_extract_usage_from_chunk(self) -> None:
             "completion_tokens": 80,
         }
 
-        chunk_hidden = SimpleNamespace(
-            _hidden_params={"usage": {"total_tokens": 30, "prompt_tokens": 10, "completion_tokens": 20}}
-        )
-        assert extract_usage(chunk_hidden) == {
-            "total_tokens": 30,
-            "prompt_tokens": 10,
-            "completion_tokens": 20,
-        }
-
         class ChunkWithModelDump:  # pylint: disable=too-few-public-methods
             def model_dump(self) -> Dict[str, Any]:
                 return {"usage": {"total_tokens": 12, "prompt_tokens": 5, "completion_tokens": 7}}