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

llama-stack version alpha -> v1

Browse source
This commit is contained in:
Ashwin Bharambe 2025-01-15 05:58:09 -08:00
parent a51c8b4efc
commit b78e6675ea
8 changed files with 390 additions and 407 deletions
Download patch file Download diff file Expand all files Collapse all files

21
llama_stack/apis/memory_banks/memory_banks.py
View file

@ -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):