llama-stack version alpha -> v1

llama_stack/apis/common/content_types.py

@@ -64,11 +64,13 @@ InterleavedContent = register_schema(
 )
 
 
+@json_schema_type
 class TextDelta(BaseModel):
     type: Literal["text"] = "text"
     text: str
 
 
+@json_schema_type
 class ImageDelta(BaseModel):
     type: Literal["image"] = "image"
     data: bytes

llama_stack/apis/memory_banks/memory_banks.py

@@ -15,7 +15,7 @@ from typing import (
     Union,
 )
 
-from llama_models.schema_utils import json_schema_type, webmethod
+from llama_models.schema_utils import json_schema_type, register_schema, webmethod
 
 from pydantic import BaseModel, Field
 
@@ -113,15 +113,18 @@ class GraphMemoryBank(MemoryBankResourceMixin):
     memory_bank_type: Literal[MemoryBankType.graph.value] = MemoryBankType.graph.value
 
 
-MemoryBank = Annotated[
-    Union[
-        VectorMemoryBank,
-        KeyValueMemoryBank,
-        KeywordMemoryBank,
-        GraphMemoryBank,
+MemoryBank = register_schema(
+    Annotated[
+        Union[
+            VectorMemoryBank,
+            KeyValueMemoryBank,
+            KeywordMemoryBank,
+            GraphMemoryBank,
+        ],
+        Field(discriminator="memory_bank_type"),
     ],
-    Field(discriminator="memory_bank_type"),
-]
+    name="MemoryBank",
+)
 
 
 class MemoryBankInput(BaseModel):

llama_stack/apis/version.py

@@ -4,4 +4,4 @@
 # This source code is licensed under the terms described in the LICENSE file in
 # the root directory of this source tree.
 
-LLAMA_STACK_API_VERSION = "alpha"
+LLAMA_STACK_API_VERSION = "v1"