phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add transform_request for OpenAI responses API

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-11 16:33:26 -07:00
parent c7e8534df4
commit 0f8de3d0a5
3 changed files with 48 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

26
litellm/llms/openai/responses/transformation.py
View file

@ -1,9 +1,14 @@
from typing import Optional
from typing import Optional, Union
import litellm
from litellm.llms.base_llm.responses.transformation import BaseResponsesAPIConfig
from litellm.secret_managers.main import get_secret_str
from litellm.types.llms.openai import ResponsesAPIRequestParams
from litellm.types.llms.openai import (
ResponseInputParam,
ResponsesAPIOptionalRequestParams,
ResponsesAPIRequestParams,
)
from litellm.types.router import GenericLiteLLMParams
class OpenAIResponsesAPIConfig(BaseResponsesAPIConfig):
@ -41,9 +46,9 @@ class OpenAIResponsesAPIConfig(BaseResponsesAPIConfig):
optional_params: dict,
model: str,
drop_params: bool,
) -> ResponsesAPIRequestParams:
) -> ResponsesAPIOptionalRequestParams:
return ResponsesAPIRequestParams(
return ResponsesAPIOptionalRequestParams(
include=optional_params.get("include"),
instructions=optional_params.get("instructions"),
max_output_tokens=optional_params.get("max_output_tokens"),
@ -66,6 +71,18 @@ class OpenAIResponsesAPIConfig(BaseResponsesAPIConfig):
timeout=optional_params.get("timeout"),
)
def transform_request(
self,
model: str,
input: Union[str, ResponseInputParam],
response_api_optional_request_params: ResponsesAPIOptionalRequestParams,
litellm_params: GenericLiteLLMParams,
headers: dict,
) -> ResponsesAPIRequestParams:
return ResponsesAPIRequestParams(
model=model, input=input, **response_api_optional_request_params
)
def validate_environment(
self,
headers: dict,
@ -89,7 +106,6 @@ class OpenAIResponsesAPIConfig(BaseResponsesAPIConfig):
self,
api_base: Optional[str],
model: str,
optional_params: dict,
stream: Optional[bool] = None,
) -> str:
"""