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

Merge branch 'main' into enable-vector-stores-files-api-tests

Browse source
This commit is contained in:
Francisco Arceo 2025-07-30 20:32:48 -04:00 • committed by GitHub
commit 87780534b0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
40 changed files with 4156 additions and 846 deletions
Download patch file Download diff file Expand all files Collapse all files

7
llama_stack/providers/remote/vector_io/weaviate/weaviate.py
View file

@ -14,6 +14,7 @@ from weaviate.classes.init import Auth
from weaviate.classes.query import Filter
from llama_stack.apis.common.content_types import InterleavedContent
from llama_stack.apis.common.errors import VectorStoreNotFoundError
from llama_stack.apis.files.files import Files
from llama_stack.apis.vector_dbs import VectorDB
from llama_stack.apis.vector_io import Chunk, QueryChunksResponse, VectorIO
@ -259,7 +260,7 @@ class WeaviateVectorIOAdapter(
vector_db = await self.vector_db_store.get_vector_db(sanitized_collection_name)
if not vector_db:
raise ValueError(f"Vector DB {sanitized_collection_name} not found")
raise VectorStoreNotFoundError(vector_db_id)
client = self._get_client()
if not client.collections.exists(vector_db.identifier):
@ -282,7 +283,7 @@ class WeaviateVectorIOAdapter(
sanitized_collection_name = sanitize_collection_name(vector_db_id, weaviate_format=True)
index = await self._get_and_cache_vector_db_index(sanitized_collection_name)
if not index:
raise ValueError(f"Vector DB {sanitized_collection_name} not found")
raise VectorStoreNotFoundError(vector_db_id)
await index.insert_chunks(chunks)
@ -295,7 +296,7 @@ class WeaviateVectorIOAdapter(
sanitized_collection_name = sanitize_collection_name(vector_db_id, weaviate_format=True)
index = await self._get_and_cache_vector_db_index(sanitized_collection_name)
if not index:
raise ValueError(f"Vector DB {sanitized_collection_name} not found")
raise VectorStoreNotFoundError(vector_db_id)
return await index.query_chunks(query, params)