mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-10-04 04:04:14 +00:00
chore: remove irrelevant comments
Signed-off-by: Mustafa Elbehery <melbeher@redhat.com>
This commit is contained in:
parent
5a6c20313d
commit
e7444c1d9b
2 changed files with 2 additions and 5 deletions
|
@ -182,7 +182,7 @@ async def milvus_vec_index(milvus_vec_db_path, embedding_dimension):
|
||||||
index = MilvusIndex(client, name, consistency_level="Strong")
|
index = MilvusIndex(client, name, consistency_level="Strong")
|
||||||
index.db_path = milvus_vec_db_path
|
index.db_path = milvus_vec_db_path
|
||||||
yield index
|
yield index
|
||||||
# Proper cleanup: close the async client
|
|
||||||
await client.close()
|
await client.close()
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -48,7 +48,7 @@ async def mock_milvus_client() -> MagicMock:
|
||||||
# Mock async insert operation
|
# Mock async insert operation
|
||||||
client.insert = AsyncMock(return_value={"insert_count": 10})
|
client.insert = AsyncMock(return_value={"insert_count": 10})
|
||||||
|
|
||||||
# Mock async search operation - return mock results (data should be dict, not JSON string)
|
# Mock async search operation
|
||||||
client.search = AsyncMock(
|
client.search = AsyncMock(
|
||||||
return_value=[
|
return_value=[
|
||||||
[
|
[
|
||||||
|
@ -87,13 +87,10 @@ async def mock_milvus_client() -> MagicMock:
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
|
|
||||||
# Mock async hybrid_search operation
|
|
||||||
client.hybrid_search = AsyncMock(return_value=[])
|
client.hybrid_search = AsyncMock(return_value=[])
|
||||||
|
|
||||||
# Mock async delete operation
|
|
||||||
client.delete = AsyncMock(return_value=None)
|
client.delete = AsyncMock(return_value=None)
|
||||||
|
|
||||||
# Mock async close operation
|
|
||||||
client.close = AsyncMock(return_value=None)
|
client.close = AsyncMock(return_value=None)
|
||||||
|
|
||||||
return client
|
return client
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue