Skip to main content
Loading navigation
Back to prompt library

Construct LlamaIndex Somatic Mutation Query Engine

Linked challenge: Somatic Mutation Dosage RAG and Metastatic Tropism Agent using LlamaIndex

Prompt source

implementation7 lines2 sectionsNo variables0 checklist items
from llamaindex.core import VectorStoreIndex, SimpleDirectoryReader, StorageContext
from llamaindex.core.query_engine import RetrieverQueryEngine

# Build LlamaIndex vector store over somatic mutation dosage cohort dataset
def build_tropism_query_engine(documents):
    index = VectorStoreIndex.from_documents(documents)
    query_engine = index.as_query_engine(similarity_top_k=5)
    return query_engine