Update Llama 3.1 70b to Llama 3.3 70b (#9)

* Update Llama 3.1 to Llama 3.3 * Bump to version 0.4.0
langchain-ai · Dec 16, 2024 · ab7a632 · ab7a632
1 parent 761ed2a
commit ab7a632
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/libs/cerebras/README.md b/libs/cerebras/README.md
@@ -39,7 +39,7 @@ from langchain_core.prompts import ChatPromptTemplate
 from langchain_cerebras import ChatCerebras
 
 chat = ChatCerebras(
-    model="llama3.1-70b",
+    model="llama-3.3-70b",
 )
 
 system = "You are an expert on animals who must answer questions in a manner that a 5 year old can understand."

diff --git a/libs/cerebras/langchain_cerebras/chat_models.py b/libs/cerebras/langchain_cerebras/chat_models.py
@@ -53,7 +53,7 @@ class ChatCerebras(BaseChatOpenAI):
             from langchain_cerebras import ChatCerebras
 
             llm = ChatCerebras(
-                model="llama3.1-70b",
+                model="llama-3.3-70b",
                 temperature=0,
                 max_tokens=None,
                 timeout=None,
@@ -80,7 +80,7 @@ class ChatCerebras(BaseChatOpenAI):
                 content='The translation of "I love programming" to French is:\n\n"J\'adore programmer."',
                 response_metadata={
                     'token_usage': {'completion_tokens': 20, 'prompt_tokens': 32, 'total_tokens': 52},
-                    'model_name': 'llama3.1-70b',
+                    'model_name': 'llama-3.3-70b',
                     'system_fingerprint': 'fp_679dff74c0',
                     'finish_reason': 'stop',
                 },
@@ -115,7 +115,7 @@ class ChatCerebras(BaseChatOpenAI):
             content='ore' id='run-3f9dc84e-208f-48da-b15d-e552b6759c24'
             content=' programmer' id='run-3f9dc84e-208f-48da-b15d-e552b6759c24'
             content='."' id='run-3f9dc84e-208f-48da-b15d-e552b6759c24'
-            content='' response_metadata={'finish_reason': 'stop', 'model_name': 'llama3.1-70b', 'system_fingerprint': 'fp_679dff74c0'} id='run-3f9dc84e-208f-48da-b15d-e552b6759c24'
+            content='' response_metadata={'finish_reason': 'stop', 'model_name': 'llama-3.3-70b', 'system_fingerprint': 'fp_679dff74c0'} id='run-3f9dc84e-208f-48da-b15d-e552b6759c24'
 
     Async:
         .. code-block:: python
@@ -134,7 +134,7 @@ class ChatCerebras(BaseChatOpenAI):
                 content='The translation of "I love programming" to French is:\n\n"J\'adore programmer."',
                 response_metadata={
                     'token_usage': {'completion_tokens': 20, 'prompt_tokens': 32, 'total_tokens': 52},
-                    'model_name': 'llama3.1-70b',
+                    'model_name': 'llama-3.3-70b',
                     'system_fingerprint': 'fp_679dff74c0',
                     'finish_reason': 'stop',
                 },
@@ -146,7 +146,7 @@ class ChatCerebras(BaseChatOpenAI):
 
             from langchain_core.pydantic_v1 import BaseModel, Field
 
-            llm = ChatCerebras(model="llama3.1-70b")
+            llm = ChatCerebras(model="llama-3.3-70b")
 
             class GetWeather(BaseModel):
                 '''Get the current weather in a given location'''

diff --git a/libs/cerebras/pyproject.toml b/libs/cerebras/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langchain-cerebras"
-version = "0.3.0"
+version = "0.4.0"
 description = "An integration package connecting Cerebras and LangChain"
 authors = []
 readme = "README.md"

diff --git a/libs/cerebras/tests/integration_tests/test_base_standard.py b/libs/cerebras/tests/integration_tests/test_base_standard.py
@@ -16,7 +16,7 @@ def chat_model_class(self) -> Type[BaseChatModel]:
 
     @property
     def chat_model_params(self) -> dict:
-        return {"model": "llama3.1-70b"}
+        return {"model": "llama-3.3-70b"}
 
     @pytest.mark.xfail(reason=("Array input not supported"))
     def test_tool_message_histories_list_content(self, model: BaseChatModel) -> None: