phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-31 07:03:55 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Clean up some more usage of direct OpenAI types

Browse source
This commit is contained in:
Ben Browning 2025-04-08 09:10:52 -04:00
parent 92fdf6d0c9
commit 5bc5fed6df
4 changed files with 10 additions and 64 deletions
Download patch file Download diff file Expand all files Collapse all files

3
llama_stack/providers/utils/inference/openai_compat.py
View file

@ -9,7 +9,6 @@ import warnings
from typing import Any, AsyncGenerator, Dict, Iterable, List, Optional, Union
from openai import AsyncStream
from openai.types.chat import ChatCompletion as OpenAIChatCompletion
from openai.types.chat import (
ChatCompletionAssistantMessageParam as OpenAIChatCompletionAssistantMessage,
)
@ -55,7 +54,6 @@ from openai.types.chat.chat_completion_content_part_image_param import (
from openai.types.chat.chat_completion_message_tool_call_param import (
Function as OpenAIFunction,
)
from openai.types.completion import Completion as OpenAICompletion
from pydantic import BaseModel
from llama_stack.apis.common.content_types import (
@ -85,6 +83,7 @@ from llama_stack.apis.inference import (
TopPSamplingStrategy,
UserMessage,
)
from llama_stack.apis.inference.inference import OpenAIChatCompletion, OpenAICompletion
from llama_stack.models.llama.datatypes import (
BuiltinTool,
StopReason,