Update poetry lock (#1209)

* Update the version of llama_index used to fix transient openai errors

* Update poetry.lock file

* Make `local` mode the default mode by default
This commit is contained in:
lopagela
2023-11-11 22:44:19 +01:00
committed by GitHub
parent a22969ad1f
commit a579c9bdc5
14 changed files with 313 additions and 268 deletions

View File

@@ -44,12 +44,12 @@ class BatchedChromaVectorStore(ChromaVectorStore):
)
self.chroma_client = chroma_client
def add(self, nodes: list[BaseNode]) -> list[str]:
def add(self, nodes: list[BaseNode], **add_kwargs: Any) -> list[str]:
"""Add nodes to index, batching the insertion to avoid issues.
Args:
nodes: List[BaseNode]: list of nodes with embeddings
add_kwargs: _
"""
if not self.chroma_client:
raise ValueError("Client not initialized")

View File

@@ -46,8 +46,11 @@ class VectorStoreComponent:
"make_this_parameterizable_per_api_call"
) # TODO
self.vector_store = BatchedChromaVectorStore(
chroma_client=chroma_client, chroma_collection=chroma_collection
self.vector_store = typing.cast(
VectorStore,
BatchedChromaVectorStore(
chroma_client=chroma_client, chroma_collection=chroma_collection
),
)
@staticmethod