From dff8308e922d4f61471a0b696fe6ae1a95495488 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Tue, 11 Mar 2025 16:53:18 -0700
Subject: [PATCH] add transform_response_api_response

---
 .../llms/base_llm/responses/transformation.py | 19 +++++++++++++------
 .../llms/openai/responses/transformation.py   |  2 +-
 2 files changed, 14 insertions(+), 7 deletions(-)

diff --git a/litellm/llms/base_llm/responses/transformation.py b/litellm/llms/base_llm/responses/transformation.py
index 76b0a5b329d..cc0a85b6f54 100644
--- a/litellm/llms/base_llm/responses/transformation.py
+++ b/litellm/llms/base_llm/responses/transformation.py
@@ -2,10 +2,13 @@ import types
 from abc import ABC, abstractmethod
 from typing import TYPE_CHECKING, Any, Optional, Union
 
+import httpx
+
 from litellm.types.llms.openai import (
     ResponseInputParam,
     ResponsesAPIOptionalRequestParams,
     ResponsesAPIRequestParams,
+    ResponsesAPIResponse,
 )
 from litellm.types.router import GenericLiteLLMParams
 from litellm.types.utils import ModelInfo
@@ -85,7 +88,7 @@ class BaseResponsesAPIConfig(ABC):
         return api_base
 
     @abstractmethod
-    def transform_request(
+    def transform_responses_api_request(
         self,
         model: str,
         input: Union[str, ResponseInputParam],
@@ -95,8 +98,12 @@ class BaseResponsesAPIConfig(ABC):
     ) -> ResponsesAPIRequestParams:
         pass
 
-    # @abstractmethod
-    # def transform_response(
-    #     self,
-    # ):
-    #     pass
+    @abstractmethod
+    def transform_response_api_response(
+        self,
+        model: str,
+        raw_response: httpx.Response,
+        model_response: ResponsesAPIResponse,
+        logging_obj: LiteLLMLoggingObj,
+    ) -> ResponsesAPIResponse:
+        return model_response
diff --git a/litellm/llms/openai/responses/transformation.py b/litellm/llms/openai/responses/transformation.py
index d67846e1010..fa4fe0274d8 100644
--- a/litellm/llms/openai/responses/transformation.py
+++ b/litellm/llms/openai/responses/transformation.py
@@ -71,7 +71,7 @@ class OpenAIResponsesAPIConfig(BaseResponsesAPIConfig):
             timeout=optional_params.get("timeout"),
         )
 
-    def transform_request(
+    def transform_responses_api_request(
         self,
         model: str,
         input: Union[str, ResponseInputParam],