llama-index

1
from typing import Any
2

3
from packaging import version
4

5

6
def _import_pinecone() -> Any:
7
    """
8
    Try to import pinecone module. If it's not already installed, instruct user how to install.
9
    """
10
    try:
11
        import pinecone
12
    except ImportError as e:
13
        raise ImportError(
14
            "Could not import pinecone python package. "
15
            "Please install it with `pip install pinecone-client`."
16
        ) from e
17
    return pinecone
18

19

20
def _is_pinecone_v3() -> bool:
21
    """
22
    Check whether the pinecone client is >= 3.0.0.
23
    """
24
    pinecone = _import_pinecone()
25
    pinecone_client_version = pinecone.__version__
26
    if version.parse(pinecone_client_version) >= version.parse(
27
        "3.0.0"
28
    ):  # Will not work with .dev versions, e.g. "3.0.0.dev8"
29
        return True
30
    return False
31