langchain/tests/integration_tests/llms
Daniel King de6e6c764e
Add MosaicML inference endpoints (#4607)
# Add MosaicML inference endpoints
This PR adds support in langchain for MosaicML inference endpoints. We
both serve a select few open source models, and allow customers to
deploy their own models using our inference service. Docs are here
(https://docs.mosaicml.com/en/latest/inference.html), and sign up form
is here (https://forms.mosaicml.com/demo?utm_source=langchain). I'm not
intimately familiar with the details of langchain, or the contribution
process, so please let me know if there is anything that needs fixing or
this is the wrong way to submit a new integration, thanks!

I'm also not sure what the procedure is for integration tests. I have
tested locally with my api key.

## Who can review?
@hwchase17

---------

Co-authored-by: Harrison Chase <hw.chase.17@gmail.com>
2023-05-23 15:59:08 -07:00
..
__init__.py initial commit 2022-10-24 14:51:15 -07:00
test_ai21.py Harrison/cohere experimental (#638) 2023-01-17 22:28:55 -08:00
test_aleph_alpha.py add aleph alpha llm (#1207) 2023-02-22 10:37:36 -08:00
test_anthropic.py Callbacks Refactor [base] (#3256) 2023-04-30 11:14:09 -07:00
test_anyscale.py Add LLMs support for Anyscale Service (#4350) 2023-05-11 00:39:59 -07:00
test_banana.py Add Writer, Banana, Modal, StochasticAI (#1270) 2023-02-24 06:58:58 -08:00
test_cerebrium.py Add GooseAI, CerebriumAI, Petals, ForefrontAI (#981) 2023-02-13 21:20:19 -08:00
test_cohere.py Harrison/llm saving (#331) 2022-12-13 06:46:01 -08:00
test_forefrontai.py Add GooseAI, CerebriumAI, Petals, ForefrontAI (#981) 2023-02-13 21:20:19 -08:00
test_google_palm.py Add ChatModel, LLM, and Embeddings for Google's PaLM APIs (#3575) 2023-05-01 15:23:16 -07:00
test_gooseai.py Add GooseAI, CerebriumAI, Petals, ForefrontAI (#981) 2023-02-13 21:20:19 -08:00
test_gpt4all.py [Breaking] Migrate GPT4All to use PyGPT4All (#3934) 2023-05-01 20:42:45 -07:00
test_huggingface_endpoint.py Add summarization task type for HuggingFace APIs (#4721) 2023-05-15 16:26:17 -07:00
test_huggingface_hub.py Add summarization task type for HuggingFace APIs (#4721) 2023-05-15 16:26:17 -07:00
test_huggingface_pipeline.py Add summarization task type for HuggingFace APIs (#4721) 2023-05-15 16:26:17 -07:00
test_llamacpp.py Callbacks Refactor [base] (#3256) 2023-04-30 11:14:09 -07:00
test_manifest.py Harrison/fix lint (#138) 2022-11-14 08:55:59 -08:00
test_modal.py Add Writer, Banana, Modal, StochasticAI (#1270) 2023-02-24 06:58:58 -08:00
test_mosaicml.py Add MosaicML inference endpoints (#4607) 2023-05-23 15:59:08 -07:00
test_nlpcloud.py Harrison/llm saving (#331) 2022-12-13 06:46:01 -08:00
test_openai.py Add 'get_token_ids' method (#4784) 2023-05-22 13:17:26 +00:00
test_openlm.py Add OpenLM LLM multi-provider (#4993) 2023-05-22 18:09:53 -07:00
test_petals.py Add GooseAI, CerebriumAI, Petals, ForefrontAI (#981) 2023-02-13 21:20:19 -08:00
test_pipelineai.py Add PipelineAI LLM integration (#3644) 2023-04-27 08:22:26 -07:00
test_predictionguard.py Harrison/prediction guard (#3490) 2023-04-24 22:27:22 -07:00
test_promptlayer_openai.py Harrison/llm integrations (#1039) 2023-02-13 22:06:25 -08:00
test_propmptlayer_openai_chat.py Harrison/prompt layer (#1547) 2023-03-08 21:24:27 -08:00
test_replicate.py Add replicate take 2 (#2077) 2023-03-28 11:56:57 -07:00
test_rwkv.py Add support for rwkv (#2422) 2023-04-06 14:41:06 -07:00
test_self_hosted_llm.py Add summarization task type for HuggingFace APIs (#4721) 2023-05-15 16:26:17 -07:00
test_stochasticai.py Add Writer, Banana, Modal, StochasticAI (#1270) 2023-02-24 06:58:58 -08:00
test_writer.py Add Writer, Banana, Modal, StochasticAI (#1270) 2023-02-24 06:58:58 -08:00
utils.py Harrison/improve cache (#368) 2022-12-18 16:22:42 -05:00