langchain/libs/community/tests/integration_tests/llms
Prince Canuma 1f9f4d8742
community[minor]: Add support for MLX models (chat & llm) (#18152)
**Description:** This PR adds support for MLX models both chat (i.e.,
instruct) and llm (i.e., pretrained) types/
**Dependencies:** mlx, mlx_lm, transformers
**Twitter handle:** @Prince_Canuma

---------

Co-authored-by: Bagatur <baskaryan@gmail.com>
Co-authored-by: Bagatur <22008038+baskaryan@users.noreply.github.com>
2024-04-09 14:17:07 +00:00
..
__init__.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_ai21.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_aleph_alpha.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_anthropic.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_anyscale.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_arcee.py infra: add print rule to ruff (#16221) 2024-02-09 16:13:30 -08:00
test_aviary.py infra: add print rule to ruff (#16221) 2024-02-09 16:13:30 -08:00
test_azure_openai.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_azureml_endpoint.py infra: add -p to mkdir in lint steps (#17013) 2024-02-05 11:22:06 -08:00
test_baichuan.py community: Add Baichuan LLM to community (#16724) 2024-01-29 20:08:24 -08:00
test_banana.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_baseten.py community: refactor Baseten integration with new API endpoints & docs (#15017) 2023-12-22 12:46:24 -08:00
test_beam.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_bedrock.py infra: add -p to mkdir in lint steps (#17013) 2024-02-05 11:22:06 -08:00
test_bigdl_llm.py community[minor]: migrate bigdl-llm to ipex-llm (#19518) 2024-03-27 20:12:59 -07:00
test_bittensor.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_cerebriumai.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_chatglm.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_clarifai.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_cloudflare_workersai.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_cohere.py community[patch]: Make cohere_api_key a SecretStr (#12188) 2024-03-01 20:27:53 +00:00
test_confident.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_ctransformers.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_deepinfra.py community[minor]: DeepInfra support for chat models (#16380) 2024-01-22 11:22:17 -08:00
test_deepsparse.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_edenai.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_fireworks.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_forefrontai.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_friendli.py community[minor]: Integration for Friendli LLM and ChatFriendli ChatModel. (#17913) 2024-03-08 02:20:47 +00:00
test_google_palm.py google-genai[patch], community[patch]: Added support for new Google GenerativeAI models (#14530) 2023-12-14 20:56:46 -08:00
test_gooseai.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_gpt4all.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_gradient_ai.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_huggingface_endpoint.py Community: Fuse HuggingFace Endpoint-related classes into one (#17254) 2024-02-19 10:33:15 -08:00
test_huggingface_hub.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_huggingface_pipeline.py community[minor]: Add openvino backend support (#11591) 2024-03-01 10:04:24 -08:00
test_huggingface_text_gen_inference.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_ipex_llm.py community[minor]: migrate bigdl-llm to ipex-llm (#19518) 2024-03-27 20:12:59 -07:00
test_konko.py community[minor]: Adding Konko Completion endpoint (#15570) 2024-01-23 18:22:32 -08:00
test_layerup_security.py community[minor]: add Layerup Security integration (#19787) 2024-04-01 23:49:00 +00:00
test_llamacpp.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_llamafile.py community[minor]: Adds Llamafile as an LLM (#17431) 2024-02-14 11:15:24 -08:00
test_manifest.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_minimax.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_mlx_pipeline.py community[minor]: Add support for MLX models (chat & llm) (#18152) 2024-04-09 14:17:07 +00:00
test_modal.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_mosaicml.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_nlpcloud.py infra: add print rule to ruff (#16221) 2024-02-09 16:13:30 -08:00
test_octoai_endpoint.py infra: add print rule to ruff (#16221) 2024-02-09 16:13:30 -08:00
test_opaqueprompts.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_openai.py openai[minor]: implement langchain-openai package (#15503) 2024-01-05 15:03:28 -08:00
test_openllm.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_openlm.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_pai_eas_endpoint.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_petals.py infra: add print rule to ruff (#16221) 2024-02-09 16:13:30 -08:00
test_pipelineai.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_predictionguard.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_promptlayer_openai.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_propmptlayer_openai_chat.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_qianfan_endpoint.py community:qianfan endpoint support init params & remove useless params definietion (#15381) 2024-01-01 13:12:31 -08:00
test_replicate.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_rwkv.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_self_hosted_llm.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_sparkllm.py community[minor]: Add SparkLLM to community (#17702) 2024-02-20 11:23:47 -08:00
test_stochasticai.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_symblai_nebula.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_titan_takeoff_pro.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_titan_takeoff.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_together.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_tongyi.py infra: add print rule to ruff (#16221) 2024-02-09 16:13:30 -08:00
test_vertexai.py Update Vertex AI to include Gemini (#14670) 2023-12-13 10:45:02 -08:00
test_volcengine_maas.py infra: add print rule to ruff (#16221) 2024-02-09 16:13:30 -08:00
test_watsonxllm.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_weight_only_quantization.py community[minor]: weight only quantization with intel-extension-for-transformers. (#14504) 2024-04-03 16:21:34 +00:00
test_writer.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_xinference.py infra: add print rule to ruff (#16221) 2024-02-09 16:13:30 -08:00
test_yuan2.py community[patch]: fix yuan2 errors in LLMs (#19004) 2024-03-28 14:37:44 -07:00
utils.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00