From 77f22d3f98280dc3682d9ae98222bdc7142f6752 Mon Sep 17 00:00:00 2001 From: Brian L Date: Mon, 22 Apr 2024 12:04:38 +1000 Subject: [PATCH] Fix databricks vector search metadata (#12999) --- .../llama_index/vector_stores/databricks/base.py | 3 ++- .../llama-index-vector-stores-databricks/pyproject.toml | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/llama-index-integrations/vector_stores/llama-index-vector-stores-databricks/llama_index/vector_stores/databricks/base.py b/llama-index-integrations/vector_stores/llama-index-vector-stores-databricks/llama_index/vector_stores/databricks/base.py index bd7f5ae3282187..e5cb92b4abf49e 100644 --- a/llama-index-integrations/vector_stores/llama-index-vector-stores-databricks/llama_index/vector_stores/databricks/base.py +++ b/llama-index-integrations/vector_stores/llama-index-vector-stores-databricks/llama_index/vector_stores/databricks/base.py @@ -223,7 +223,8 @@ def add( metadata_columns = self.columns or [] # explicitly record doc_id as metadata (for delete) - metadata_columns.append("doc_id") + if "doc_id" not in metadata_columns: + metadata_columns.append("doc_id") entry = { self._primary_key: node_id, diff --git a/llama-index-integrations/vector_stores/llama-index-vector-stores-databricks/pyproject.toml b/llama-index-integrations/vector_stores/llama-index-vector-stores-databricks/pyproject.toml index 0b45006cde64b7..97c3b4aaf0cd73 100644 --- a/llama-index-integrations/vector_stores/llama-index-vector-stores-databricks/pyproject.toml +++ b/llama-index-integrations/vector_stores/llama-index-vector-stores-databricks/pyproject.toml @@ -26,7 +26,7 @@ description = "llama-index vector_stores databricks vector search integration" license = "MIT" name = "llama-index-vector-stores-databricks" readme = "README.md" -version = "0.1.1" +version = "0.1.2" [tool.poetry.dependencies] python = ">=3.8.1,<3.12"