stage tmp changes

This commit is contained in:
Xi Yan 2024-09-20 15:33:31 -07:00
parent abe312c092
commit 2dc14cba2c
2 changed files with 99 additions and 3 deletions

View file

@ -5,6 +5,7 @@
# the root directory of this source tree.
import asyncio
import importlib
import inspect
import json
import signal
@ -35,9 +36,6 @@ from fastapi import Body, FastAPI, HTTPException, Request, Response
from fastapi.exceptions import RequestValidationError
from fastapi.responses import JSONResponse, StreamingResponse
from fastapi.routing import APIRoute
from pydantic import BaseModel, ValidationError
from termcolor import cprint
from typing_extensions import Annotated
from llama_stack.providers.utils.telemetry.tracing import (
end_trace,
@ -45,6 +43,9 @@ from llama_stack.providers.utils.telemetry.tracing import (
SpanStatus,
start_trace,
)
from pydantic import BaseModel, ValidationError
from termcolor import cprint
from typing_extensions import Annotated
from llama_stack.distribution.datatypes import * # noqa: F403
from llama_stack.distribution.distribution import api_endpoints, api_providers