mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-07-02 04:28:46 +00:00
small cleanup and updating metadata inclusion in MemoryToolRuntimeImpl.query() to be more flexible in the future
Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
This commit is contained in:
parent
0cc625560d
commit
7ed916dbb3
3 changed files with 17 additions and 23 deletions
|
@ -56,7 +56,7 @@ class TestRagQuery:
|
|||
|
||||
assert result is not None
|
||||
expected_metadata_string = (
|
||||
"Metadata: {'key1': 'value1', 'document_id': 'doc1', 'chunk_id': 'chunk1', 'source': 'test_source'}"
|
||||
"Metadata: {'chunk_id': 'chunk1', 'document_id': 'doc1', 'source': 'test_source', 'key1': 'value1'}"
|
||||
)
|
||||
assert expected_metadata_string in result.content[1].text
|
||||
assert result.content is not None
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue