hey guys! I'm trying to use Milvus

hey guys! I'm trying to use Milvus vector store with hybrid retrieval and I having issues with BGEM3F embedding model:

Plain Text

pip install FlagEmbedding

I create my vector store:

Plain Text

        vector_store = MilvusVectorStore(
            collection_name=app_settings.milvus_collection,
            dim=app_settings.vector_dim,
            overwrite=app_settings.overwrite,
            token=app_settings.milvus_token,
            uri=app_settings.milvus_uri,
            enable_sparse=app_settings.enable_sparse,
            hybrid_ranker=app_settings.hybrid_ranker,
            hybrid_ranker_params=app_settings.hybrid_ranker_params,            
        )

and then I have this in my terminal:

Plain Text

DEBUG:pymilvus.milvus_client.milvus_client:Created new connection using: d28ca022b4884f4ab606408820a88944
WARNING:llama_index.vector_stores.milvus.base:Sparse embedding function is not provided, using default.
INFO:datasets:PyTorch version 2.4.0 available.
CRITICAL:llama_index.vector_stores.milvus.utils:Cannot import BGEM3FlagModel from FlagEmbedding. It seems it is not installed. Please install it using:
pip install FlagEmbedding

but in fact it has already been installed

10 comments

LLogan M

Are you running in a notebook? You'll have to restart that if so

LLogan M

This is the codeblock raising this error, if you wanted to try it on its own

Plain Text

        try:
            from FlagEmbedding import BGEM3FlagModel

            self.model = BGEM3FlagModel("BAAI/bge-m3", use_fp16=False)
        except Exception as ImportError:
            error_info = (
                "Cannot import BGEM3FlagModel from FlagEmbedding. It seems it is not installed. "
                "Please install it using:\n"
                "pip install FlagEmbedding\n"
            )
            logger.fatal(error_info)
            sys.exit(1)

YYamil Abraham

I'm running it in a python file

LLogan M

not sure. Try running the above import and see if it actually works I guess? Maybe you need pip install -U BGEM3FlagModel

YYamil Abraham

I will, thanks!

YYamil Abraham

it doesn't work 😕

Attachment

LLogan M

oh whoops, I suck at typing lol

LLogan M

pip install -U FlagEmbedding

LLogan M

BGEM3FlagModel is the import 😅

YYamil Abraham

haha, that one ran well but same message as above

CRITICAL:llama_index.vector_stores.milvus.utils:Cannot import BGEM3FlagModel from FlagEmbedding. It seems it is not installed. Please install it using:
pip install FlagEmbedding

Add a reply

Find answers from the community

hey guys! I'm trying to use Milvus