github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-14 08:56:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
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>
This commit is contained in:
Michael Gokhman 2023-03-28 23:07:05 +03:00 committed by GitHub
parent aff33d52c5
commit 5f34dffedc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
langchain/llms/ai21.py
View File

@ -29,10 +29,10 @@ class AI21(LLM, BaseModel):
.. code-block:: python .. code-block:: python
from langchain.llms import AI21 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.""" """Model name to use."""
temperature: float = 0.7 temperature: float = 0.7