fix incorrect parameters in VertexAIIndex (#16080)

run-llama · Sep 18, 2024 · 4ef2cdc · 4ef2cdc
1 parent f9760a6
commit 4ef2cdc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/...indices/llama-index-indices-managed-vertexai/llama_index/indices/managed/vertexai/base.py b/...indices/llama-index-indices-managed-vertexai/llama_index/indices/managed/vertexai/base.py
@@ -104,8 +104,8 @@ def import_files(
 
         with telemetry.tool_context_manager(self._user_agent):
             return rag.import_files(
-                corpus=self.corpus_name,
-                uris=uris,
+                self.corpus_name,
+                paths=uris,
                 chunk_size=chunk_size,
                 chunk_overlap=chunk_overlap,
                 timeout=timeout,

diff --git a/llama-index-integrations/indices/llama-index-indices-managed-vertexai/pyproject.toml b/llama-index-integrations/indices/llama-index-indices-managed-vertexai/pyproject.toml
@@ -30,12 +30,11 @@ exclude = ["**/BUILD"]
 license = "MIT"
 name = "llama-index-indices-managed-vertexai"
 readme = "README.md"
-version = "0.1.0"
+version = "0.1.1"
 
 [tool.poetry.dependencies]
 python = ">=3.8.1,<4.0"
-llamaindex-py-client = "^0.1.19"
-google-cloud-aiplatform = "^1.53.0"
+google-cloud-aiplatform = ">=1.53.0"
 llama-index-core = "^0.11.0"
 
 [tool.poetry.group.dev.dependencies]