fix(llms): update default AI21 model to j2, as j1 being deprecated (#2108)

the j1-* models are marked as [Legacy] in the docs and are expected to
be deprecated in 2023-06-01 according to
https://docs.ai21.com/docs/jurassic-1-models-legacy

ensured `tests/integration_tests/llms/test_ai21.py` pass.

empirically observed that `j2-jumbo-instruct` works better the
`j2-jumbo` in various simple agent chains, as also expected given the
prompt templates are mostly zero shot.

Co-authored-by: Michael Gokhman <michaelg@ai21.com>
searx
Michael Gokhman 1 year ago committed by GitHub
parent aff33d52c5
commit 5f34dffedc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -29,10 +29,10 @@ class AI21(LLM, BaseModel):
.. code-block:: python
from langchain.llms import AI21
ai21 = AI21(model="j1-jumbo")
ai21 = AI21(model="j2-jumbo-instruct")
"""
model: str = "j1-jumbo"
model: str = "j2-jumbo-instruct"
"""Model name to use."""
temperature: float = 0.7

Loading…
Cancel
Save