From 4a26c0d54fe1138d43ff9f77e5dbceb98c2091cf Mon Sep 17 00:00:00 2001 From: Dinesh Yeduguru Date: Fri, 1 Nov 2024 14:41:29 -0700 Subject: [PATCH] precommit failures fixed --- llama_stack/apis/datasets/datasets.py | 4 ++-- llama_stack/apis/scoring_functions/scoring_functions.py | 2 +- llama_stack/distribution/server/server.py | 6 +++--- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/llama_stack/apis/datasets/datasets.py b/llama_stack/apis/datasets/datasets.py index f5efc7cbc..1695c888b 100644 --- a/llama_stack/apis/datasets/datasets.py +++ b/llama_stack/apis/datasets/datasets.py @@ -10,10 +10,10 @@ from llama_models.llama3.api.datatypes import URL from llama_models.schema_utils import json_schema_type, webmethod -from llama_stack.apis.common.type_system import ParamType - from pydantic import BaseModel, Field +from llama_stack.apis.common.type_system import ParamType + @json_schema_type class DatasetDef(BaseModel): diff --git a/llama_stack/apis/scoring_functions/scoring_functions.py b/llama_stack/apis/scoring_functions/scoring_functions.py index d5c65e738..d0a9cc597 100644 --- a/llama_stack/apis/scoring_functions/scoring_functions.py +++ b/llama_stack/apis/scoring_functions/scoring_functions.py @@ -7,9 +7,9 @@ from typing import Any, Dict, List, Literal, Optional, Protocol, runtime_checkable from llama_models.schema_utils import json_schema_type, webmethod +from pydantic import BaseModel, Field from llama_stack.apis.common.type_system import ParamType -from pydantic import BaseModel, Field @json_schema_type diff --git a/llama_stack/distribution/server/server.py b/llama_stack/distribution/server/server.py index 44e88e2ea..2a20597f8 100644 --- a/llama_stack/distribution/server/server.py +++ b/llama_stack/distribution/server/server.py @@ -22,6 +22,9 @@ import yaml from fastapi import Body, FastAPI, HTTPException, Request, Response from fastapi.exceptions import RequestValidationError from fastapi.responses import JSONResponse, StreamingResponse +from pydantic import BaseModel, ValidationError +from termcolor import cprint +from typing_extensions import Annotated from llama_stack.distribution.distribution import ( builtin_automatically_routed_apis, @@ -36,9 +39,6 @@ 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.request_headers import set_request_provider_data from llama_stack.distribution.resolver import resolve_impls