mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-06-17 19:05:13 +02:00
IAM tech spec: Auth and access management current state and proposed
changes. Workspace support: - Support for separate workspaces - Addition of workspace CLI support for test purposes - Massive test update - Remove many 'user' references in services - workspace now provides the same separation - Update API
This commit is contained in:
parent
48da6c5f8b
commit
594deba73e
347 changed files with 6788 additions and 5540 deletions
|
|
@ -99,7 +99,6 @@ class TestDocumentRagIntegration:
|
|||
# Act
|
||||
result = await document_rag.query(
|
||||
query=query,
|
||||
user=user,
|
||||
collection=collection,
|
||||
doc_limit=doc_limit
|
||||
)
|
||||
|
|
@ -110,7 +109,6 @@ class TestDocumentRagIntegration:
|
|||
mock_doc_embeddings_client.query.assert_called_once_with(
|
||||
vector=[[0.1, 0.2, 0.3, 0.4, 0.5], [0.6, 0.7, 0.8, 0.9, 1.0]],
|
||||
limit=doc_limit,
|
||||
user=user,
|
||||
collection=collection
|
||||
)
|
||||
|
||||
|
|
@ -278,14 +276,12 @@ class TestDocumentRagIntegration:
|
|||
# Act
|
||||
await document_rag.query(
|
||||
f"query from {user} in {collection}",
|
||||
user=user,
|
||||
collection=collection
|
||||
)
|
||||
|
||||
# Assert
|
||||
mock_doc_embeddings_client.query.assert_called_once()
|
||||
call_args = mock_doc_embeddings_client.query.call_args
|
||||
assert call_args.kwargs['user'] == user
|
||||
assert call_args.kwargs['collection'] == collection
|
||||
|
||||
@pytest.mark.asyncio
|
||||
|
|
@ -353,6 +349,5 @@ class TestDocumentRagIntegration:
|
|||
# Assert
|
||||
mock_doc_embeddings_client.query.assert_called_once()
|
||||
call_args = mock_doc_embeddings_client.query.call_args
|
||||
assert call_args.kwargs['user'] == "trustgraph"
|
||||
assert call_args.kwargs['collection'] == "default"
|
||||
assert call_args.kwargs['limit'] == 20
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue