andras/great-ai
andras/great-ai
Watch
1
0
Fork
You've already forked great-ai
 0
Code Issues Pull requests 2 Projects Releases 4 Packages 1 Wiki Activity Actions

Store model versions globally

Browse source
This commit is contained in:
Andras Schmelczer 2022-07-04 09:27:43 +02:00
commit d971ea054f
1 changed files with 41 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/great_ai/great_ai/deploy/great_ai.py
View file

@ -25,6 +25,7 @@ from ..helper import (
snake_case_to_text,
use_http_exceptions,
)
from ..models import model_versions
from ..parameters import automatically_decorate_parameters
from ..tracing.tracing_context import TracingContext
from ..views import ApiMetadata, CacheStatistics, HealthCheckResponse, Trace
@ -168,9 +169,11 @@ class GreatAI(Generic[T]):
@property
def version(self) -> str:
return (
f"{self._version}+{get_function_metadata_store(self._func).model_versions}"
)
flat_model_versions = ".".join(f"{k}-v{v}" for k, v in model_versions)
if flat_model_versions:
flat_model_versions = f"+{flat_model_versions}"
return f"{self._version}{flat_model_versions}"
@property
def documentation(self) -> str: