feat(ui/model_dashboard.tsx): show if model is config or db model

This commit is contained in:
Krrish Dholakia 2024-05-07 18:29:14 -07:00
parent a325bf2fb8
commit fbcda918de
4 changed files with 42 additions and 8 deletions

View file

@ -1,6 +1,6 @@
from typing import List, Optional, Union, Dict, Tuple, Literal
import httpx
from pydantic import BaseModel, validator
from pydantic import BaseModel, validator, Field
from .completion import CompletionRequest
from .embedding import EmbeddingRequest
import uuid, enum
@ -70,6 +70,9 @@ class ModelInfo(BaseModel):
id: Optional[
str
] # Allow id to be optional on input, but it will always be present as a str in the model instance
db_model: bool = (
False # used for proxy - to separate models which are stored in the db vs. config.
)
def __init__(self, id: Optional[Union[str, int]] = None, **params):
if id is None: