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

Update URL type to avoid string-ifying and creating complexity

Browse source
This commit is contained in:
Ashwin Bharambe 2024-12-17 22:48:47 -08:00
parent 75e72cf2fc
commit f1d6cb22d7
3 changed files with 18 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

7
llama_stack/apis/common/content_types.py
View file

@ -11,15 +11,10 @@ from llama_models.schema_utils import json_schema_type, register_schema
from pydantic import BaseModel, Field, model_validator
@json_schema_type(
schema={"type": "string", "format": "uri", "pattern": "^(https?://|file://|data:)"}
)
@json_schema_type
class URL(BaseModel):
uri: str
def __str__(self) -> str:
return self.uri
class _URLOrData(BaseModel):
url: Optional[URL] = None