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

linting fixes for transforming responses API

Browse source
This commit is contained in:
Ishaan Jaff 2025-04-18 19:38:32 -07:00
parent 5c7297229f
commit 760b6d5d40
1 changed files with 39 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

44
litellm/responses/litellm_completion_transformation/transformation.py
View file

@ -31,10 +31,12 @@ from litellm.types.llms.openai import (
) )
from litellm.types.responses.main import ( from litellm.types.responses.main import (
GenericResponseOutputItem, GenericResponseOutputItem,
GenericResponseOutputItemContentAnnotation,
OutputFunctionToolCall, OutputFunctionToolCall,
OutputText, OutputText,
) )
from litellm.types.utils import ( from litellm.types.utils import (
ChatCompletionAnnotation,
ChatCompletionMessageToolCall, ChatCompletionMessageToolCall,
Choices, Choices,
Function, Function,
@ -329,7 +331,7 @@ class LiteLLMCompletionResponsesConfig:
@staticmethod @staticmethod
def _transform_responses_api_content_to_chat_completion_content( def _transform_responses_api_content_to_chat_completion_content(
content: Any, content: Any,
) -> Union[str, List[Dict[str, Any]]]: ) -> Union[str, List[Union[str, Dict[str, Any]]]]:
""" """
Transform a Responses API content into a Chat Completion content Transform a Responses API content into a Chat Completion content
""" """
@ -337,7 +339,7 @@ class LiteLLMCompletionResponsesConfig:
if isinstance(content, str): if isinstance(content, str):
return content return content
elif isinstance(content, list): elif isinstance(content, list):
content_list = [] content_list: List[Union[str, Dict[str, Any]]] = []
for item in content: for item in content:
if isinstance(item, str): if isinstance(item, str):
content_list.append(item) content_list.append(item)
@ -392,7 +394,7 @@ class LiteLLMCompletionResponsesConfig:
name=tool["name"], name=tool["name"],
description=tool.get("description") or "", description=tool.get("description") or "",
parameters=tool.get("parameters", {}), parameters=tool.get("parameters", {}),
strict=tool.get("strict", None), strict=tool.get("strict", False),
), ),
) )
) )
@ -425,6 +427,7 @@ class LiteLLMCompletionResponsesConfig:
name=function_definition.name or "", name=function_definition.name or "",
arguments=function_definition.get("arguments") or "", arguments=function_definition.get("arguments") or "",
call_id=tool.id or "", call_id=tool.id or "",
id=tool.id or "",
type="function_call", # critical this is "function_call" to work with tools like openai codex type="function_call", # critical this is "function_call" to work with tools like openai codex
status=function_definition.get("status") or "completed", status=function_definition.get("status") or "completed",
) )
@ -536,8 +539,8 @@ class LiteLLMCompletionResponsesConfig:
] ]
] = [] ] = []
output_items = responses_api_output.output output_items = responses_api_output.output
for output_item in output_items: for _output_item in output_items:
output_item = dict(output_item) output_item: dict = dict(_output_item)
if output_item.get("type") == "function_call": if output_item.get("type") == "function_call":
# handle function call output # handle function call output
messages.append( messages.append(
@ -577,8 +580,39 @@ class LiteLLMCompletionResponsesConfig:
return OutputText( return OutputText(
type="output_text", type="output_text",
text=message.content, text=message.content,
annotations=LiteLLMCompletionResponsesConfig._transform_chat_completion_annotations_to_response_output_annotations(
annotations=message.annotations
),
) )
@staticmethod
def _transform_chat_completion_annotations_to_response_output_annotations(
annotations: Optional[List[ChatCompletionAnnotation]],
) -> List[GenericResponseOutputItemContentAnnotation]:
response_output_annotations: List[
GenericResponseOutputItemContentAnnotation
] = []
if annotations is None:
return response_output_annotations
for annotation in annotations:
annotation_type = annotation.get("type")
if annotation_type == "url_citation" and "url_citation" in annotation:
url_citation = annotation["url_citation"]
response_output_annotations.append(
GenericResponseOutputItemContentAnnotation(
type=annotation_type,
start_index=url_citation.get("start_index"),
end_index=url_citation.get("end_index"),
url=url_citation.get("url"),
title=url_citation.get("title"),
)
)
# Handle other annotation types here
return response_output_annotations
@staticmethod @staticmethod
def _transform_chat_completion_usage_to_responses_usage( def _transform_chat_completion_usage_to_responses_usage(
chat_completion_response: ModelResponse, chat_completion_response: ModelResponse,