mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-12 12:06:04 +00:00
clean
This commit is contained in:
parent
a720dbb942
commit
0a5799e0ce
1 changed files with 4 additions and 11 deletions
|
|
@ -9,6 +9,7 @@ import base64
|
||||||
import io
|
import io
|
||||||
import json
|
import json
|
||||||
import re
|
import re
|
||||||
|
from typing import Any
|
||||||
|
|
||||||
import httpx
|
import httpx
|
||||||
from PIL import Image as PIL_Image
|
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(
|
def interleaved_content_as_str(
|
||||||
content: InterleavedContent
|
content: Any,
|
||||||
| str
|
|
||||||
| list[OpenAIChatCompletionContentPartTextParam | OpenAIChatCompletionContentPartImageParam | OpenAIFile]
|
|
||||||
| list[OpenAIChatCompletionContentPartTextParam]
|
|
||||||
| None,
|
|
||||||
sep: str = " ",
|
sep: str = " ",
|
||||||
) -> str:
|
) -> str:
|
||||||
if content is None:
|
if content is None:
|
||||||
|
|
@ -91,13 +88,9 @@ def interleaved_content_as_str(
|
||||||
def _process(c) -> str:
|
def _process(c) -> str:
|
||||||
if isinstance(c, str):
|
if isinstance(c, str):
|
||||||
return c
|
return c
|
||||||
elif isinstance(c, ImageContentItem):
|
elif isinstance(c, TextContentItem) or isinstance(c, OpenAIChatCompletionContentPartTextParam):
|
||||||
return "<image>"
|
|
||||||
elif isinstance(c, TextContentItem):
|
|
||||||
return c.text
|
return c.text
|
||||||
elif isinstance(c, OpenAIChatCompletionContentPartTextParam):
|
elif isinstance(c, ImageContentItem) or isinstance(c, OpenAIChatCompletionContentPartImageParam):
|
||||||
return c.text
|
|
||||||
elif isinstance(c, OpenAIChatCompletionContentPartImageParam):
|
|
||||||
return "<image>"
|
return "<image>"
|
||||||
elif isinstance(c, OpenAIFile):
|
elif isinstance(c, OpenAIFile):
|
||||||
return "<file>"
|
return "<file>"
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue