mirror of
https://github.com/hwchase17/langchain
synced 2024-11-08 07:10:35 +00:00
5124c1e0d9
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>
11 lines
249 B
Python
11 lines
249 B
Python
"""Test Anyscale API wrapper."""
|
|
|
|
from langchain.llms.aviary import Aviary
|
|
|
|
|
|
def test_aviary_call() -> None:
|
|
"""Test valid call to Anyscale."""
|
|
llm = Aviary(model="test/model")
|
|
output = llm("Say bar:")
|
|
assert isinstance(output, str)
|