From f0dba005542706edd5e53d4506290e9a574c8f3a Mon Sep 17 00:00:00 2001
From: Logan <logan.markewich@live.com>
Date: Tue, 12 Mar 2024 16:44:43 -0600
Subject: [PATCH] use proper retriever kwargs (#11874)
---
.../llama_index/indices/managed/llama_cloud/retriever.py | 5 ++++-
.../llama-index-indices-managed-llama-cloud/pyproject.toml | 2 +-
2 files changed, 5 insertions(+), 2 deletions(-)
diff --git a/llama-index-integrations/indices/llama-index-indices-managed-llama-cloud/llama_index/indices/managed/llama_cloud/retriever.py b/llama-index-integrations/indices/llama-index-indices-managed-llama-cloud/llama_index/indices/managed/llama_cloud/retriever.py
index 8390c84c49..a8af0f6858 100644
--- a/llama-index-integrations/indices/llama-index-indices-managed-llama-cloud/llama_index/indices/managed/llama_cloud/retriever.py
+++ b/llama-index-integrations/indices/llama-index-indices-managed-llama-cloud/llama_index/indices/managed/llama_cloud/retriever.py
@@ -43,7 +43,10 @@ class LlamaCloudRetriever(BaseRetriever):
self._alpha = alpha or OMIT
self._search_filters = search_filters or OMIT
- super().__init__(**kwargs)
+ super().__init__(
+ callback_manager=kwargs.get("callback_manager", None),
+ verbose=kwargs.get("verbose", False),
+ )
def _result_nodes_to_node_with_score(
self, result_nodes: List[TextNodeWithScore]
diff --git a/llama-index-integrations/indices/llama-index-indices-managed-llama-cloud/pyproject.toml b/llama-index-integrations/indices/llama-index-indices-managed-llama-cloud/pyproject.toml
index 65c5805db8..4ec183892b 100644
--- a/llama-index-integrations/indices/llama-index-indices-managed-llama-cloud/pyproject.toml
+++ b/llama-index-integrations/indices/llama-index-indices-managed-llama-cloud/pyproject.toml
@@ -30,7 +30,7 @@ exclude = ["**/BUILD"]
license = "MIT"
name = "llama-index-indices-managed-llama-cloud"
readme = "README.md"
-version = "0.1.3"
+version = "0.1.4"
[tool.poetry.dependencies]
python = ">=3.8.1,<4.0"
--
GitLab