From 53ccd2f772991f10b58f3bbe17979ce306b2ae60 Mon Sep 17 00:00:00 2001 From: deedy5 <65482418+deedy5@users.noreply.github.com> Date: Wed, 14 Aug 2024 13:12:58 +0300 Subject: [PATCH] [chat]: docstring - update "llama-3-70b" to "llama-3.1-70b" --- README.md | 2 +- duckduckgo_search/duckduckgo_search.py | 2 +- duckduckgo_search/duckduckgo_search_async.py | 2 +- tests/test_duckduckgo_search.py | 2 +- tests/test_duckduckgo_search_async.py | 2 +- 5 files changed, 5 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index 68a71a6..e57231e 100755 --- 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 index 8212963..196275e 100644 --- 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 index 7a004f3..cbc6bc8 100644 --- 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 index 8c5fa5b..0fd35a8 100644 --- 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 index b5d0223..505c632 100644 --- 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