phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-30 07:39:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

linter fix

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-11-05 11:39:57 -08:00
parent 5567243553
commit 4c8e2c3d47
1 changed files with 8 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

14
llama_stack/providers/utils/kvstore/config.py
View file

@ -4,12 +4,12 @@
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.
import re
from enum import Enum
from typing import Literal, Optional, Union
from pydantic import BaseModel, Field, field_validator
from typing_extensions import Annotated
import re
from llama_stack.distribution.utils.config_dirs import RUNTIME_BASE_DIR
@ -54,17 +54,19 @@ class PostgresKVStoreConfig(CommonConfig):
password: Optional[str] = None
table_name: str = "llamastack_kvstore"
@field_validator('table_name')
def validate_table_name(cls, v: str) -> str:
@field_validator("table_name")
def validate_table_name(self, v: str) -> str:
# PostgreSQL identifiers rules:
# - Must start with a letter or underscore
# - Can contain letters, numbers, and underscores
# - Maximum length is 63 bytes
pattern = r'^[a-zA-Z_][a-zA-Z0-9_]*$'
pattern = r"^[a-zA-Z_][a-zA-Z0-9_]*$"
if not re.match(pattern, v):
raise ValueError('Invalid table name. Must start with letter or underscore and contain only letters, numbers, and underscores')
raise ValueError(
"Invalid table name. Must start with letter or underscore and contain only letters, numbers, and underscores"
)
if len(v) > 63:
raise ValueError('Table name must be less than 63 characters')
raise ValueError("Table name must be less than 63 characters")
return v