phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-12 04:50:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

chore: pre-commit changes

Browse source
This commit is contained in:
sarthakdeshpande 2025-03-09 17:42:46 +05:30
parent 501467846d
commit 73cb368900
1 changed files with 2 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_stack/providers/inline/vector_io/faiss/faiss.py
View file

@ -4,11 +4,11 @@
# This source code is licensed under the terms described in the LICENSE file in # This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree. # the root directory of this source tree.
import asyncio
import base64 import base64
import io import io
import json import json
import logging import logging
import asyncio
from typing import Any, Dict, List, Optional from typing import Any, Dict, List, Optional
import faiss import faiss
@ -100,9 +100,7 @@ class FaissIndex(EmbeddingIndex):
await self._save_index() await self._save_index()
async def query(self, embedding: NDArray, k: int, score_threshold: float) -> QueryChunksResponse: async def query(self, embedding: NDArray, k: int, score_threshold: float) -> QueryChunksResponse:
distances, indices = await asyncio.to_thread( distances, indices = await asyncio.to_thread(self.index.search, embedding.reshape(1, -1).astype(np.float32), k)
self.index.search, embedding.reshape(1, -1).astype(np.float32), k
)
chunks = [] chunks = []
scores = [] scores = []