FlashrankRerank from LangChain initialization error

Question

I am getting this error while trying intiliaze FlashrankRerank

PydanticUserError: FlashrankRerank is not fully defined; you should define Ranker, then call FlashrankRerank.model_rebuild()

This is my code

compressor = FlashrankRerank()
compression_retriever = ContextualCompressionRetriever(
    base_compressor=compressor, base_retriever=retriever
)

I tried compressor = FlashrankRerank.model_rebuild() But got this error PydanticUndefinedAnnotation: name 'Ranker' is not defined

Welcome to Stack Overflow. Please include the full traceback error. — ewokx
– ewokx, Commented Jan 27 at 7:42

Bugra · Accepted Answer · 2025-01-27 11:21:03Z

1

You have to initialize model that you want to use with Ranker function in flashrank, you can define like:

from flashrank import Ranker 

ranker = Ranker(model_name="ms-marco-MiniLM-L-12-v2")

then fill it like:

compressor = FlashrankRerank(model="ms-marco-MiniLM-L-12-v2", top_n=3)

compression_retriever = ContextualCompressionRetriever(
base_compressor=compressor, base_retriever=retriever)

edited Jan 27 at 11:21

answered Jan 27 at 11:19

Bugra

112 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

FlashrankRerank from LangChain initialization error

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related