mirror of
https://github.com/hwchase17/langchain
synced 2024-11-08 07:10:35 +00:00
f6fdabd20b
- Description: The aviary integration has changed url link. This PR provide fix for those changes and also it makes providing the input URL optional to the API (since they can be set via env variables). - Issue: N/A - Dependencies: N/A - Twitter handle: N/A --------- Signed-off-by: Kourosh Hakhamaneshi <kourosh@anyscale.com>
12 lines
267 B
Python
12 lines
267 B
Python
"""Test Anyscale API wrapper."""
|
|
|
|
from langchain.llms.aviary import Aviary
|
|
|
|
|
|
def test_aviary_call() -> None:
|
|
"""Test valid call to Anyscale."""
|
|
llm = Aviary()
|
|
output = llm("Say bar:")
|
|
print(f"llm answer:\n{output}")
|
|
assert isinstance(output, str)
|