Skip to content

Instantly share code, notes, and snippets.

@AutoScrape123TX
AutoScrape123TX / QdrantVectorPrevNextNodePostprocessor.py
Last active July 19, 2024 21:59 — forked from ShawonAshraf/prevnext.py
PrevNextNodePostprocessor tailored to QdrantVectorStore for TextNode relationships
from typing import Dict, List, Optional
from llama_index.core.schema import NodeRelationship, NodeWithScore, QueryBundle
from llama_index.core.bridge.pydantic import Field, validator
from llama_index.core.postprocessor.types import BaseNodePostprocessor
from llama_index.vector_stores.qdrant import QdrantVectorStore
def qdrant_vector_store_get_forward_nodes(
node_with_score: NodeWithScore, num_nodes: int, vector_store: QdrantVectorStore
) -> Dict[str, NodeWithScore]: