From 0a5799e0ceee07c9d067f799620080768defb4cf Mon Sep 17 00:00:00 2001 From: Swapna Lekkala Date: Sun, 12 Oct 2025 07:20:40 -0700 Subject: [PATCH] clean --- .../providers/utils/inference/prompt_adapter.py | 15 ++++----------- 1 file changed, 4 insertions(+), 11 deletions(-) diff --git a/llama_stack/providers/utils/inference/prompt_adapter.py b/llama_stack/providers/utils/inference/prompt_adapter.py index f02e5aeb6..d06b7454d 100644 --- a/llama_stack/providers/utils/inference/prompt_adapter.py +++ b/llama_stack/providers/utils/inference/prompt_adapter.py @@ -9,6 +9,7 @@ import base64 import io import json import re +from typing import Any import httpx from PIL import Image as PIL_Image @@ -78,11 +79,7 @@ def decode_assistant_message(content: str, stop_reason: StopReason) -> RawMessag def interleaved_content_as_str( - content: InterleavedContent - | str - | list[OpenAIChatCompletionContentPartTextParam | OpenAIChatCompletionContentPartImageParam | OpenAIFile] - | list[OpenAIChatCompletionContentPartTextParam] - | None, + content: Any, sep: str = " ", ) -> str: if content is None: @@ -91,13 +88,9 @@ def interleaved_content_as_str( def _process(c) -> str: if isinstance(c, str): return c - elif isinstance(c, ImageContentItem): - return "" - elif isinstance(c, TextContentItem): + elif isinstance(c, TextContentItem) or isinstance(c, OpenAIChatCompletionContentPartTextParam): return c.text - elif isinstance(c, OpenAIChatCompletionContentPartTextParam): - return c.text - elif isinstance(c, OpenAIChatCompletionContentPartImageParam): + elif isinstance(c, ImageContentItem) or isinstance(c, OpenAIChatCompletionContentPartImageParam): return "" elif isinstance(c, OpenAIFile): return ""