llama-index

1
"""Init file."""
2

3
from llama_index.legacy.embeddings.adapter import (
4
    AdapterEmbeddingModel,
5
    LinearAdapterEmbeddingModel,
6
)
7
from llama_index.legacy.embeddings.anyscale import AnyscaleEmbedding
8
from llama_index.legacy.embeddings.azure_openai import AzureOpenAIEmbedding
9
from llama_index.legacy.embeddings.base import BaseEmbedding, SimilarityMode
10
from llama_index.legacy.embeddings.bedrock import BedrockEmbedding
11
from llama_index.legacy.embeddings.clarifai import ClarifaiEmbedding
12
from llama_index.legacy.embeddings.clip import ClipEmbedding
13
from llama_index.legacy.embeddings.cohereai import CohereEmbedding
14
from llama_index.legacy.embeddings.dashscope import (
15
    DashScopeBatchTextEmbeddingModels,
16
    DashScopeEmbedding,
17
    DashScopeMultiModalEmbeddingModels,
18
    DashScopeTextEmbeddingModels,
19
    DashScopeTextEmbeddingType,
20
)
21
from llama_index.legacy.embeddings.elasticsearch import (
22
    ElasticsearchEmbedding,
23
    ElasticsearchEmbeddings,
24
)
25
from llama_index.legacy.embeddings.fastembed import FastEmbedEmbedding
26
from llama_index.legacy.embeddings.gemini import GeminiEmbedding
27
from llama_index.legacy.embeddings.google import GoogleUnivSentEncoderEmbedding
28
from llama_index.legacy.embeddings.google_palm import GooglePaLMEmbedding
29
from llama_index.legacy.embeddings.gradient import GradientEmbedding
30
from llama_index.legacy.embeddings.huggingface import (
31
    HuggingFaceEmbedding,
32
    HuggingFaceInferenceAPIEmbedding,
33
    HuggingFaceInferenceAPIEmbeddings,
34
)
35
from llama_index.legacy.embeddings.huggingface_optimum import OptimumEmbedding
36
from llama_index.legacy.embeddings.huggingface_utils import (
37
    DEFAULT_HUGGINGFACE_EMBEDDING_MODEL,
38
)
39
from llama_index.legacy.embeddings.instructor import InstructorEmbedding
40
from llama_index.legacy.embeddings.langchain import LangchainEmbedding
41
from llama_index.legacy.embeddings.llm_rails import (
42
    LLMRailsEmbedding,
43
    LLMRailsEmbeddings,
44
)
45
from llama_index.legacy.embeddings.mistralai import MistralAIEmbedding
46
from llama_index.legacy.embeddings.nomic import NomicEmbedding
47
from llama_index.legacy.embeddings.ollama_embedding import OllamaEmbedding
48
from llama_index.legacy.embeddings.openai import OpenAIEmbedding
49
from llama_index.legacy.embeddings.pooling import Pooling
50
from llama_index.legacy.embeddings.sagemaker_embedding_endpoint import (
51
    SageMakerEmbedding,
52
)
53
from llama_index.legacy.embeddings.text_embeddings_inference import (
54
    TextEmbeddingsInference,
55
)
56
from llama_index.legacy.embeddings.together import TogetherEmbedding
57
from llama_index.legacy.embeddings.utils import resolve_embed_model
58
from llama_index.legacy.embeddings.voyageai import VoyageEmbedding
59

60
__all__ = [
61
    "AdapterEmbeddingModel",
62
    "BedrockEmbedding",
63
    "ClarifaiEmbedding",
64
    "ClipEmbedding",
65
    "CohereEmbedding",
66
    "BaseEmbedding",
67
    "DEFAULT_HUGGINGFACE_EMBEDDING_MODEL",
68
    "ElasticsearchEmbedding",
69
    "FastEmbedEmbedding",
70
    "GoogleUnivSentEncoderEmbedding",
71
    "GradientEmbedding",
72
    "HuggingFaceInferenceAPIEmbedding",
73
    "HuggingFaceEmbedding",
74
    "InstructorEmbedding",
75
    "LangchainEmbedding",
76
    "LinearAdapterEmbeddingModel",
77
    "LLMRailsEmbedding",
78
    "MistralAIEmbedding",
79
    "OpenAIEmbedding",
80
    "AzureOpenAIEmbedding",
81
    "AnyscaleEmbedding",
82
    "OptimumEmbedding",
83
    "Pooling",
84
    "SageMakerEmbedding",
85
    "GooglePaLMEmbedding",
86
    "SimilarityMode",
87
    "TextEmbeddingsInference",
88
    "TogetherEmbedding",
89
    "resolve_embed_model",
90
    "NomicEmbedding",
91
    # Deprecated, kept for backwards compatibility
92
    "LLMRailsEmbeddings",
93
    "ElasticsearchEmbeddings",
94
    "HuggingFaceInferenceAPIEmbeddings",
95
    "VoyageEmbedding",
96
    "OllamaEmbedding",
97
    "GeminiEmbedding",
98
    "DashScopeEmbedding",
99
    "DashScopeTextEmbeddingModels",
100
    "DashScopeTextEmbeddingType",
101
    "DashScopeBatchTextEmbeddingModels",
102
    "DashScopeMultiModalEmbeddingModels",
103
]
104