[chat]: docstring - update "llama-3-70b" to "llama-3.1-70b"

deedy5 · Aug 14, 2024 · 53ccd2f · 53ccd2f
1 parent ecfc9ad
commit 53ccd2f
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -230,7 +230,7 @@ def chat(self, keywords: str, model: str = "gpt-4o-mini", timeout: int = 30) ->
 
     Args:
         keywords (str): The initial message or question to send to the AI.
-        model (str): The model to use: "gpt-4o-mini", "claude-3-haiku", "llama-3-70b", "mixtral-8x7b".
+        model (str): The model to use: "gpt-4o-mini", "claude-3-haiku", "llama-3.1-70b", "mixtral-8x7b".
             Defaults to "gpt-4o-mini".
         timeout (int): Timeout value for the HTTP client. Defaults to 30.
 

diff --git a/duckduckgo_search/duckduckgo_search.py b/duckduckgo_search/duckduckgo_search.py
@@ -138,7 +138,7 @@ def chat(self, keywords: str, model: str = "gpt-4o-mini", timeout: int = 30) ->
 
         Args:
             keywords (str): The initial message or question to send to the AI.
-            model (str): The model to use: "gpt-4o-mini", "claude-3-haiku", "llama-3-70b", "mixtral-8x7b".
+            model (str): The model to use: "gpt-4o-mini", "claude-3-haiku", "llama-3.1-70b", "mixtral-8x7b".
                 Defaults to "gpt-4o-mini".
             timeout (int): Timeout value for the HTTP client. Defaults to 20.
 

diff --git a/duckduckgo_search/duckduckgo_search_async.py b/duckduckgo_search/duckduckgo_search_async.py
@@ -41,7 +41,7 @@ async def achat(self, keywords: str, model: str = "gpt-4o-mini", timeout: int =
 
         Args:
             keywords (str): The initial message or question to send to the AI.
-            model (str): The model to use: "gpt-4o-mini", "claude-3-haiku", "llama-3-70b", "mixtral-8x7b".
+            model (str): The model to use: "gpt-4o-mini", "claude-3-haiku", "llama-3.1-70b", "mixtral-8x7b".
                 Defaults to "gpt-4o-mini".
             timeout (int): Timeout value for the HTTP client. Defaults to 30.
 

diff --git a/tests/test_duckduckgo_search.py b/tests/test_duckduckgo_search.py
@@ -15,7 +15,7 @@ def test_context_manager():
         assert 20 <= len(results) <= 30
 
 
-@pytest.mark.parametrize("model", ["gpt-4o-mini", "claude-3-haiku", "llama-3-70b", "mixtral-8x7b"])
+@pytest.mark.parametrize("model", ["gpt-4o-mini", "claude-3-haiku", "llama-3.1-70b", "mixtral-8x7b"])
 def test_chat(model):
     results = DDGS().chat("cat", model=model)
     assert  len(results) >= 1

diff --git a/tests/test_duckduckgo_search_async.py b/tests/test_duckduckgo_search_async.py
@@ -16,7 +16,7 @@ async def test_context_manager():
 
 
 @pytest.mark.asyncio
-@pytest.mark.parametrize("model", ["gpt-4o-mini", "claude-3-haiku", "llama-3-70b", "mixtral-8x7b"])
+@pytest.mark.parametrize("model", ["gpt-4o-mini", "claude-3-haiku", "llama-3.1-70b", "mixtral-8x7b"])
 async def test_chat(model):
     results = await AsyncDDGS().achat("cat", model=model)
     assert  len(results) >= 1