phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-11 20:40:40 +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
# the root directory of this source tree.
import asyncio
import base64
import io
import json
import logging
import asyncio
from typing import Any, Dict, List, Optional
import faiss
@ -100,9 +100,7 @@ class FaissIndex(EmbeddingIndex):
await self._save_index()
async def query(self, embedding: NDArray, k: int, score_threshold: float) -> QueryChunksResponse:
distances, indices = await asyncio.to_thread(
self.index.search, embedding.reshape(1, -1).astype(np.float32), k
)
distances, indices = await asyncio.to_thread(self.index.search, embedding.reshape(1, -1).astype(np.float32), k)
chunks = []
scores = []