Add AI21 LLMs (#99)

Integrate AI21 /complete API into langchain, to allow access to Jurassic
models.
This commit is contained in:
tomeras91
2022-11-10 18:12:28 +02:00
committed by GitHub
parent 2179ea3103
commit d8734ce5ad
2 changed files with 142 additions and 0 deletions

View File

@@ -0,0 +1,10 @@
"""Test AI21 API wrapper."""
from langchain.llms.ai21 import AI21
def test_ai21_call() -> None:
"""Test valid call to ai21."""
llm = AI21(maxTokens=10)
output = llm("Say foo:")
assert isinstance(output, str)