Skip to content

Commit 00c96ec

Browse files
authored
Fix: use "generation" as default response_mode for EmptyIndex (run-llama#6534)
* ignore pyrightconfig.json * use "generation" as default response_mode for EmptyIndex
1 parent 90a0016 commit 00c96ec

File tree

2 files changed

+14
-1
lines changed

2 files changed

+14
-1
lines changed

.gitignore

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -143,4 +143,7 @@ modules/
143143

144144
# pipenv
145145
Pipfile
146-
Pipfile.lock
146+
Pipfile.lock
147+
148+
# pyright
149+
pyrightconfig.json

llama_index/indices/empty/base.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
from llama_index.data_structs.node import Node
1212
from llama_index.indices.base import BaseIndex
1313
from llama_index.indices.base_retriever import BaseRetriever
14+
from llama_index.indices.query.base import BaseQueryEngine
1415
from llama_index.indices.service_context import ServiceContext
1516
from llama_index.storage.docstore.types import RefDocInfo
1617

@@ -48,6 +49,15 @@ def as_retriever(self, **kwargs: Any) -> BaseRetriever:
4849

4950
return EmptyIndexRetriever(self)
5051

52+
def as_query_engine(self, **kwargs: Any) -> BaseQueryEngine:
53+
if "response_mode" not in kwargs:
54+
kwargs["response_mode"] = "generation"
55+
else:
56+
if kwargs["response_mode"] != "generation":
57+
raise ValueError("EmptyIndex only supports response_mode=generation.")
58+
59+
return super().as_query_engine(**kwargs)
60+
5161
def _build_index_from_nodes(self, nodes: Sequence[Node]) -> EmptyIndexStruct:
5262
"""Build the index from documents.
5363

0 commit comments

Comments
 (0)