phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

Add simple validation for RemoteProviderConfig

Browse source
This commit is contained in:
Ashwin Bharambe 2024-08-09 15:15:20 -07:00
parent 898cd5b352
commit 2cd8b2ff5b
1 changed files with 8 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

9
llama_toolchain/distribution/datatypes.py
View file

@ -9,7 +9,7 @@ from typing import Any, Dict, List, Optional
from llama_models.schema_utils import json_schema_type
from pydantic import BaseModel, Field
from pydantic import BaseModel, Field, validator
@json_schema_type
@ -62,6 +62,13 @@ class RemoteProviderConfig(BaseModel):
..., description="API key, if needed, for the provider"
)
@validator("base_url")
@classmethod
def validate_base_url(cls, base_url: str) -> str:
if not base_url.startswith("http"):
raise ValueError(f"URL must start with http: {base_url}")
return base_url
@json_schema_type
class RemoteProviderSpec(ProviderSpec):