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.db_path = milvus_vec_db_path
|
||||
yield index
|
||||
# Proper cleanup: close the async client
|
||||
|
||||
await client.close()
|
||||
|
||||
|
||||
|
|
|
@ -48,7 +48,7 @@ async def mock_milvus_client() -> MagicMock:
|
|||
# Mock async insert operation
|
||||
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(
|
||||
return_value=[
|
||||
[
|
||||
|
@ -87,13 +87,10 @@ async def mock_milvus_client() -> MagicMock:
|
|||
]
|
||||
)
|
||||
|
||||
# Mock async hybrid_search operation
|
||||
client.hybrid_search = AsyncMock(return_value=[])
|
||||
|
||||
# Mock async delete operation
|
||||
client.delete = AsyncMock(return_value=None)
|
||||
|
||||
# Mock async close operation
|
||||
client.close = AsyncMock(return_value=None)
|
||||
|
||||
return client
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue