From 034deca5689f68893fdda98481c7b09c85ce3612 Mon Sep 17 00:00:00 2001 From: Paillat Date: Wed, 10 Jul 2024 20:55:21 +0200 Subject: [PATCH] :art: Format code --- src/engines/LLMEngine/OpenaiLLMEngine.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/engines/LLMEngine/OpenaiLLMEngine.py b/src/engines/LLMEngine/OpenaiLLMEngine.py index 6a198ee..92ee8d0 100644 --- a/src/engines/LLMEngine/OpenaiLLMEngine.py +++ b/src/engines/LLMEngine/OpenaiLLMEngine.py @@ -9,7 +9,7 @@ from .BaseLLMEngine import BaseLLMEngine OPENAI_POSSIBLE_MODELS = [ # Theese shall be the openai models supporting force_json "gpt-3.5-turbo-0125", "gpt-4-turbo", - "gpt-4o" + "gpt-4o", ] @@ -109,4 +109,8 @@ class OpenaiLLMEngine(BaseLLMEngine): @property def supports_vision(self) -> bool: - return True if self.model in ["gpt-4-turbo-preview", "gpt-4-turbo", "gpt-4o"] else False + return ( + True + if self.model in ["gpt-4-turbo-preview", "gpt-4-turbo", "gpt-4o"] + else False + )