Paillat-dev/viralfactory
1
0
Fork 0
mirror of https://github.com/Paillat-dev/viralfactory.git synced 2026-01-02 01:06:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

🎨 Format code

Browse Source
This commit is contained in:
Paillat
2024-07-10 20:55:21 +02:00
parent de3c5d5a02
commit 034deca568
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/engines/LLMEngine/OpenaiLLMEngine.py
View File

@@ -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
)