Spaces:
Runtime error
Runtime error
File size: 1,422 Bytes
8a58cf3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
"""Default query for GPTListIndex."""
from typing import List, Optional
from gpt_index.data_structs.data_structs import IndexList, Node
from gpt_index.indices.query.base import BaseGPTIndexQuery
from gpt_index.indices.query.embedding_utils import SimilarityTracker
from gpt_index.indices.query.schema import QueryBundle
class BaseGPTListIndexQuery(BaseGPTIndexQuery[IndexList]):
"""GPTListIndex query.
Arguments are shared among subclasses.
Args:
text_qa_template (Optional[QuestionAnswerPrompt]): A Question Answering Prompt
(see :ref:`Prompt-Templates`).
refine_template (Optional[RefinePrompt]): A Refinement Prompt
(see :ref:`Prompt-Templates`).
"""
class GPTListIndexQuery(BaseGPTListIndexQuery):
"""GPTListIndex query.
The default query mode for GPTListIndex, which traverses
each node in sequence and synthesizes a response across all nodes
(with an optional keyword filter).
Set when `mode="default"` in `query` method of `GPTListIndex`.
.. code-block:: python
response = index.query("<query_str>", mode="default")
See BaseGPTListIndexQuery for arguments.
"""
def _get_nodes_for_response(
self,
query_bundle: QueryBundle,
similarity_tracker: Optional[SimilarityTracker] = None,
) -> List[Node]:
"""Get nodes for response."""
return self.index_struct.nodes
|