mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-05-17 11:25:12 +02:00
Fixing tests
This commit is contained in:
parent
fe82b844c6
commit
61a100dec9
1 changed files with 1 additions and 1 deletions
|
|
@ -102,7 +102,7 @@ class TestDocumentRagIntegration:
|
||||||
mock_embeddings_client.embed.assert_called_once_with([query])
|
mock_embeddings_client.embed.assert_called_once_with([query])
|
||||||
|
|
||||||
mock_doc_embeddings_client.query.assert_called_once_with(
|
mock_doc_embeddings_client.query.assert_called_once_with(
|
||||||
[[0.1, 0.2, 0.3, 0.4, 0.5], [0.6, 0.7, 0.8, 0.9, 1.0]],
|
vector=[[0.1, 0.2, 0.3, 0.4, 0.5], [0.6, 0.7, 0.8, 0.9, 1.0]],
|
||||||
limit=doc_limit,
|
limit=doc_limit,
|
||||||
user=user,
|
user=user,
|
||||||
collection=collection
|
collection=collection
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue