Fix ray-project/Aviary integration (#6607)

- 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>
This commit is contained in:
kourosh hakhamaneshi
2023-06-23 14:49:53 -07:00
committed by GitHub
parent dbe1d029ec
commit f6fdabd20b
2 changed files with 103 additions and 56 deletions

View File

@@ -5,6 +5,7 @@ from langchain.llms.aviary import Aviary
def test_aviary_call() -> None:
"""Test valid call to Anyscale."""
llm = Aviary(model="test/model")
llm = Aviary()
output = llm("Say bar:")
print(f"llm answer:\n{output}")
assert isinstance(output, str)