2024-04-10 17:01:19 +00:00
|
|
|
from langchain_community.embeddings import __all__, _module_lookup
|
2023-12-11 21:53:30 +00:00
|
|
|
|
|
|
|
EXPECTED_ALL = [
|
|
|
|
"OpenAIEmbeddings",
|
2024-03-29 00:53:53 +00:00
|
|
|
"AnyscaleEmbeddings",
|
2023-12-11 21:53:30 +00:00
|
|
|
"AzureOpenAIEmbeddings",
|
2024-01-26 20:57:26 +00:00
|
|
|
"BaichuanTextEmbeddings",
|
2023-12-11 21:53:30 +00:00
|
|
|
"ClarifaiEmbeddings",
|
|
|
|
"CohereEmbeddings",
|
|
|
|
"DatabricksEmbeddings",
|
|
|
|
"ElasticsearchEmbeddings",
|
|
|
|
"FastEmbedEmbeddings",
|
|
|
|
"HuggingFaceEmbeddings",
|
|
|
|
"HuggingFaceInferenceAPIEmbeddings",
|
|
|
|
"InfinityEmbeddings",
|
2024-02-22 00:33:13 +00:00
|
|
|
"InfinityEmbeddingsLocal",
|
2023-12-11 21:53:30 +00:00
|
|
|
"GradientEmbeddings",
|
|
|
|
"JinaEmbeddings",
|
2024-02-26 20:16:37 +00:00
|
|
|
"LaserEmbeddings",
|
2023-12-11 21:53:30 +00:00
|
|
|
"LlamaCppEmbeddings",
|
2024-03-01 21:49:18 +00:00
|
|
|
"LlamafileEmbeddings",
|
2024-01-01 23:24:50 +00:00
|
|
|
"LLMRailsEmbeddings",
|
2023-12-11 21:53:30 +00:00
|
|
|
"HuggingFaceHubEmbeddings",
|
|
|
|
"MlflowAIGatewayEmbeddings",
|
|
|
|
"MlflowEmbeddings",
|
2024-01-22 19:38:11 +00:00
|
|
|
"MlflowCohereEmbeddings",
|
2023-12-11 21:53:30 +00:00
|
|
|
"ModelScopeEmbeddings",
|
|
|
|
"TensorflowHubEmbeddings",
|
|
|
|
"SagemakerEndpointEmbeddings",
|
|
|
|
"HuggingFaceInstructEmbeddings",
|
|
|
|
"MosaicMLInstructorEmbeddings",
|
|
|
|
"SelfHostedEmbeddings",
|
|
|
|
"SelfHostedHuggingFaceEmbeddings",
|
|
|
|
"SelfHostedHuggingFaceInstructEmbeddings",
|
|
|
|
"FakeEmbeddings",
|
|
|
|
"DeterministicFakeEmbedding",
|
|
|
|
"AlephAlphaAsymmetricSemanticEmbedding",
|
|
|
|
"AlephAlphaSymmetricSemanticEmbedding",
|
|
|
|
"SentenceTransformerEmbeddings",
|
|
|
|
"GooglePalmEmbeddings",
|
|
|
|
"MiniMaxEmbeddings",
|
|
|
|
"VertexAIEmbeddings",
|
|
|
|
"BedrockEmbeddings",
|
|
|
|
"DeepInfraEmbeddings",
|
|
|
|
"EdenAiEmbeddings",
|
|
|
|
"DashScopeEmbeddings",
|
|
|
|
"EmbaasEmbeddings",
|
|
|
|
"OctoAIEmbeddings",
|
|
|
|
"SpacyEmbeddings",
|
|
|
|
"NLPCloudEmbeddings",
|
|
|
|
"GPT4AllEmbeddings",
|
2024-03-25 23:08:37 +00:00
|
|
|
"GigaChatEmbeddings",
|
2023-12-11 21:53:30 +00:00
|
|
|
"XinferenceEmbeddings",
|
|
|
|
"LocalAIEmbeddings",
|
|
|
|
"AwaEmbeddings",
|
|
|
|
"HuggingFaceBgeEmbeddings",
|
|
|
|
"ErnieEmbeddings",
|
|
|
|
"JavelinAIGatewayEmbeddings",
|
|
|
|
"OllamaEmbeddings",
|
|
|
|
"QianfanEmbeddingsEndpoint",
|
|
|
|
"JohnSnowLabsEmbeddings",
|
|
|
|
"VoyageEmbeddings",
|
|
|
|
"BookendEmbeddings",
|
2024-01-01 22:37:35 +00:00
|
|
|
"VolcanoEmbeddings",
|
community[minor]: Add OCI Generative AI integration (#16548)
<!-- Thank you for contributing to LangChain!
Please title your PR "<package>: <description>", where <package> is
whichever of langchain, community, core, experimental, etc. is being
modified.
Replace this entire comment with:
- **Description:** Adding Oracle Cloud Infrastructure Generative AI
integration. Oracle Cloud Infrastructure (OCI) Generative AI is a fully
managed service that provides a set of state-of-the-art, customizable
large language models (LLMs) that cover a wide range of use cases, and
which is available through a single API. Using the OCI Generative AI
service you can access ready-to-use pretrained models, or create and
host your own fine-tuned custom models based on your own data on
dedicated AI clusters.
https://docs.oracle.com/en-us/iaas/Content/generative-ai/home.htm
- **Issue:** None,
- **Dependencies:** OCI Python SDK,
- **Twitter handle:** we announce bigger features on Twitter. If your PR
gets announced, and you'd like a mention, we'll gladly shout you out!
Please make sure your PR is passing linting and testing before
submitting. Run `make format`, `make lint` and `make test` from the root
of the package you've modified to check this locally.
Passed
See contribution guidelines for more information on how to write/run
tests, lint, etc: https://python.langchain.com/docs/contributing/
If you're adding a new integration, please include:
1. a test for the integration, preferably unit tests that do not rely on
network access,
2. an example notebook showing its use. It lives in
`docs/docs/integrations` directory.
we provide unit tests. However, we cannot provide integration tests due
to Oracle policies that prohibit public sharing of api keys.
If no one reviews your PR within a few days, please @-mention one of
@baskaryan, @eyurtsev, @hwchase17.
-->
---------
Co-authored-by: Arthur Cheng <arthur.cheng@oracle.com>
Co-authored-by: Bagatur <baskaryan@gmail.com>
2024-01-25 02:23:50 +00:00
|
|
|
"OCIGenAIEmbeddings",
|
2024-02-15 19:01:24 +00:00
|
|
|
"QuantizedBiEncoderEmbeddings",
|
2024-02-14 21:25:42 +00:00
|
|
|
"NeMoEmbeddings",
|
2024-02-20 19:22:27 +00:00
|
|
|
"SparkLLMTextEmbeddings",
|
2024-03-27 07:22:06 +00:00
|
|
|
"QuantizedBgeEmbeddings",
|
2024-03-26 01:37:19 +00:00
|
|
|
"PremAIEmbeddings",
|
2024-03-26 00:05:57 +00:00
|
|
|
"YandexGPTEmbeddings",
|
2024-03-29 08:34:51 +00:00
|
|
|
"OpenVINOEmbeddings",
|
|
|
|
"OpenVINOBgeEmbeddings",
|
2024-03-29 16:36:05 +00:00
|
|
|
"SolarEmbeddings",
|
2023-12-11 21:53:30 +00:00
|
|
|
]
|
|
|
|
|
|
|
|
|
|
|
|
def test_all_imports() -> None:
|
|
|
|
assert set(__all__) == set(EXPECTED_ALL)
|
2024-04-10 17:01:19 +00:00
|
|
|
assert set(__all__) == set(_module_lookup.keys())
|