Precommit fix

BlackHC · Dec 27, 2023 · f55f1f8 · f55f1f8
1 parent 1d53176
commit f55f1f8
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/llm_hyperparameters/__init__.py b/llm_hyperparameters/__init__.py
@@ -1 +1 @@
-__version__ = '2.1.1'
+__version__ = "2.1.1"
diff --git a/llm_strategy/__init__.py b/llm_strategy/__init__.py
@@ -4,4 +4,4 @@
 )
 from llm_strategy.llm_strategy import llm_dataclass, llm_strategy  # type: ignore
 
-__version__ = '2.1.1'
+__version__ = "2.1.1"
diff --git a/llm_strategy/llm_function.py b/llm_strategy/llm_function.py
@@ -454,8 +454,8 @@ def query(self, language_model_or_chat_chain, schema):  # noqa: C901
             # Check if the language model is of type "openai" and extend model args with a response format in that case
             model_dict = model.dict()
             if "openai" in model_dict["_type"] and model_dict.get("model_name") in (
-                    "gpt-4-1106-preview",
-                    "gpt-3.5-turbo-1106",
+                "gpt-4-1106-preview",
+                "gpt-3.5-turbo-1106",
             ):
                 model_args = dict(response_format=dict(type="json_object"))
             else: