phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-03 19:57:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

thanks ruff

Browse source
This commit is contained in:
Matthew Farrellee 2025-09-25 07:13:21 -04:00
parent 466ef6f490
commit 22dba64f09
1 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

7
llama_stack/providers/remote/inference/databricks/databricks.py
View file

@ -23,6 +23,8 @@ from llama_stack.apis.inference import (
Inference, Inference,
LogProbConfig, LogProbConfig,
Message, Message,
Model,
ModelType,
OpenAICompletion, OpenAICompletion,
ResponseFormat, ResponseFormat,
SamplingParams, SamplingParams,
@ -31,15 +33,12 @@ from llama_stack.apis.inference import (
ToolConfig, ToolConfig,
ToolDefinition, ToolDefinition,
ToolPromptFormat, ToolPromptFormat,
Model,
ModelType,
) )
from llama_stack.log import get_logger
from llama_stack.providers.utils.inference.openai_mixin import OpenAIMixin from llama_stack.providers.utils.inference.openai_mixin import OpenAIMixin
from .config import DatabricksImplConfig from .config import DatabricksImplConfig
from llama_stack.log import get_logger
logger = get_logger(name=__name__, category="inference::databricks") logger = get_logger(name=__name__, category="inference::databricks")