From c9eda14e4bc899a39722fa6a6d7979a467442a52 Mon Sep 17 00:00:00 2001
From: Simon Suo <simonsdsuo@gmail.com>
Date: Fri, 29 Mar 2024 14:59:35 -0700
Subject: [PATCH] Fix bug where user specified llm is not respected in fallback
logic (#12403)
wip
---
.../llama_index/core/node_parser/relational/base_element.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/llama-index-core/llama_index/core/node_parser/relational/base_element.py b/llama-index-core/llama_index/core/node_parser/relational/base_element.py
index c3697b0b21..738a780576 100644
--- a/llama-index-core/llama_index/core/node_parser/relational/base_element.py
+++ b/llama-index-core/llama_index/core/node_parser/relational/base_element.py
@@ -174,7 +174,7 @@ class BaseElementNodeParser(NodeParser):
except ValidationError:
# There was a pydantic validation error, so we will run with text completion
# fill in the summary and leave other fields blank
- query_engine = index.as_query_engine()
+ query_engine = index.as_query_engine(llm=llm)
response_txt = await query_engine.aquery(summary_query_str)
return TableOutput(summary=str(response_txt), columns=[])
--
GitLab