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

memory tests fix

Browse source
This commit is contained in:
Xi Yan 2024-12-26 17:27:01 -08:00
parent 3f86c19150
commit 3b1f20ac00
3 changed files with 21 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

12
llama_stack/providers/tests/memory/test_memory.py
View file

@ -8,14 +8,18 @@ import uuid
import pytest
from llama_stack.apis.memory import * # noqa: F403
from llama_stack.distribution.datatypes import * # noqa: F403
from llama_stack.apis.memory_banks.memory_banks import VectorMemoryBankParams
from llama_stack.apis.memory import MemoryBankDocument, QueryDocumentsResponse
from llama_stack.apis.memory_banks import (
MemoryBank,
MemoryBanks,
VectorMemoryBankParams,
)
# How to run this test:
#
# pytest llama_stack/providers/tests/memory/test_memory.py
# -m "meta_reference"
# -m "sentence_transformers" --env EMBEDDING_DIMENSION=384
# -v -s --tb=short --disable-warnings