From 596d9aaf657b84c15d76c7badcf0883c8d8e73c5 Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Wed, 8 Jul 2026 04:23:49 +0000 Subject: [PATCH] test(responses): use httpx.Response and assert callback fired in retrieve-path tests Co-Authored-By: Ishaan Jaffer <155045088+ishaan-berri@users.noreply.github.com> --- .../test_responses_background_cost.py | 18 +++++------------- 1 file changed, 5 insertions(+), 13 deletions(-) diff --git a/tests/test_litellm/integrations/test_responses_background_cost.py b/tests/test_litellm/integrations/test_responses_background_cost.py index c3c13e61d75..a5648c50a65 100644 --- a/tests/test_litellm/integrations/test_responses_background_cost.py +++ b/tests/test_litellm/integrations/test_responses_background_cost.py @@ -3,12 +3,12 @@ Integration tests for responses API background cost tracking """ import asyncio -import json import os from datetime import datetime from types import SimpleNamespace from unittest.mock import AsyncMock, MagicMock, Mock, patch +import httpx import pytest import litellm @@ -670,18 +670,8 @@ def _fake_get_response(status, with_usage): ), } - class _Resp: - status_code = 200 - headers = {} - text = json.dumps(payload) - - def json(self): - return payload - - def raise_for_status(self): - return None - - return _Resp() + request = httpx.Request("GET", "https://api.openai.com/v1/responses/resp_finalize_test") + return httpx.Response(status_code=200, json=payload, request=request) async def _capture_aget_responses(status, with_usage): @@ -704,6 +694,8 @@ async def _capture_aget_responses(status, with_usage): await asyncio.sleep(0.1) finally: litellm.callbacks = original_callbacks + + assert capture.events, "success logging callback never fired for the retrieved response" return capture.events