Merge branch 'chore/openai-model-update' of github.com:explosion/spac…

…y-llm into chore/openai-model-update
explosion · Nov 8, 2023 · 34bf1aa · 34bf1aa
2 parents f5ceda6 + 638ec20
commit 34bf1aa
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/spacy_llm/models/rest/openai/registry.py b/spacy_llm/models/rest/openai/registry.py
@@ -151,7 +151,6 @@ def openai_gpt_3_5_v2(
         "gpt-3.5-turbo-0613",
         "gpt-3.5-turbo-0613-16k",
         "gpt-3.5-turbo-instruct",
-        "gpt-3.5-turbo-1106",
     ] = "gpt-3.5-turbo",  # noqa: F722,F821
     strict: bool = OpenAI.DEFAULT_STRICT,
     max_tries: int = OpenAI.DEFAULT_MAX_TRIES,
@@ -162,8 +161,7 @@ def openai_gpt_3_5_v2(
 
     config (Dict[Any, Any]): LLM config passed on to the model's initialization.
     name (Literal[
-        "gpt-3.5-turbo", "gpt-3.5-turbo-16k", "gpt-3.5-turbo-0613", "gpt-3.5-turbo-0613-16k", "gpt-3.5-turbo-instruct",
-        "gpt-3.5-turbo-1106"
+        "gpt-3.5-turbo", "gpt-3.5-turbo-16k", "gpt-3.5-turbo-0613", "gpt-3.5-turbo-0613-16k", "gpt-3.5-turbo-instruct"
     ]): Model to use.
     RETURNS (Callable[[Iterable[str]], Iterable[str]]]): OpenAI instance for 'gpt-3.5' model
 

diff --git a/spacy_llm/tests/test_combinations.py b/spacy_llm/tests/test_combinations.py
@@ -12,7 +12,7 @@
 @pytest.mark.skipif(has_langchain is False, reason="LangChain is not installed")
 @pytest.mark.parametrize(
     "model",
-    ["langchain.OpenAI.v1", "spacy.GPT-3-5.v2", "spacy.GPT-4.v2"],
+    ["langchain.OpenAI.v1", "spacy.GPT-3-5.v3", "spacy.GPT-4.v3"],
     ids=["langchain", "rest-openai", "rest-openai"],
 )
 @pytest.mark.parametrize(