langchain/tests/integration_tests/llms
M Waleed Kadous 5124c1e0d9
Add aviary support (#5661)
Aviary is an open source toolkit for evaluating and deploying open
source LLMs. You can find out more about it on
[http://github.com/ray-project/aviary). You can try it out at
[http://aviary.anyscale.com](aviary.anyscale.com).

This code adds support for Aviary in LangChain. To minimize
dependencies, it connects directly to the HTTP endpoint.

The current implementation is not accelerated and uses the default
implementation of `predict` and `generate`.

It includes a test and a simple example. 

@hwchase17 and @agola11 could you have a look at this?

---------

Co-authored-by: Dev 2049 <dev.dev2049@gmail.com>
2023-06-05 16:28:42 -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_aviary.py Add aviary support (#5661) 2023-06-05 16:28:42 -07:00
test_banana.py Add Writer, Banana, Modal, StochasticAI (#1270) 2023-02-24 06:58:58 -08:00
test_beam.py Beam (#4996) 2023-05-24 01:25:18 -07: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_ctransformers.py Add C Transformers for GGML Models (#5218) 2023-05-25 13:42:44 -07: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 update (#5404) 2023-05-29 07:14:59 -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_vertexai.py Harrison/vertex (#5049) 2023-05-24 15:51:12 -07: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