Skip to content

Help with Query Embeddings Serialization (using stored Query embeddings in RAG pipeline) #13049

Closed Answered by gody10
gody10 asked this question in Q&A
Discussion options

You must be logged in to vote

Answer

I found the answer to my question. The queries can be serialized through the QueryBundle class. You just have to add the following Code before retrieving the nodes. The embeddings_query_df is a DataFrame that contains the embeddings for all the queries.

#Creating Query Bundle Object
    query_bundle = QueryBundle(
        query_str=row['query'],
        embedding=embeddings_query_df[i],
    )

   retrieved_nodes = vector_retriever.retrieve(query_bundle)

Replies: 2 comments

Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
0 replies
Answer selected by gody10
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
1 participant