pyconverters-openai_vision 0.5.2__tar.gz → 0.5.3__tar.gz

{pyconverters_openai_vision-0.5.2 → pyconverters_openai_vision-0.5.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pyconverters-openai_vision
-Version: 0.5.2
+Version: 0.5.3
 Summary: OpenAIVision converter
 Home-page: https://kairntech.com/
 Author: Olivier Terrier

{pyconverters_openai_vision-0.5.2 → pyconverters_openai_vision-0.5.3}/pyconverters_openai_vision/__init__.py

@@ -1,2 +1,2 @@
 """OpenAIVision converter"""
-__version__ = "0.5.2"
+__version__ = "0.5.3"

{pyconverters_openai_vision-0.5.2 → pyconverters_openai_vision-0.5.3}/pyconverters_openai_vision/openai_utils.py

@@ -9,6 +9,11 @@ from strenum import StrEnum
 
 logger = Logger("pymultirole")
 DEFAULT_CHAT_GPT_MODEL = "gpt-4o-mini"
+DEEPINFRA_VISION_MODELS = [
+    "vision",
+    "llava",
+    "pixtral"
+]
 
 
 # Now use default retry with backoff of openai api
@@ -18,6 +23,13 @@ def openai_chat_completion(prefix, **kwargs):
     return response
 
 
+def is_vision_model(model):
+    for m in DEEPINFRA_VISION_MODELS:
+        if m in model.lower():
+            return True
+    return False
+
+
 def openai_list_models(prefix, **kwargs):
     def sort_by_created(x):
         if 'created' in x:
@@ -63,7 +75,7 @@ def openai_list_models(prefix, **kwargs):
             mods = list(
                 {m['model_name'] for m in mods if m['task'] == 'text-generation' and m['status'] == 'running'})
             deepinfra_models.update({m: m for m in mods})
-        models = [m for m in deepinfra_models.keys() if 'vision' in m.lower()]
+        models = [m for m in deepinfra_models.keys() if is_vision_model(m)]
     elif prefix.startswith("AZURE"):
         models = comma_separated_to_list(os.getenv(prefix + "OPENAI_DEPLOYMENT_ID", None))
     else:

{pyconverters_openai_vision-0.5.2 → pyconverters_openai_vision-0.5.3}/pyproject.toml

@@ -45,7 +45,7 @@ requires-python = ">=3.8"
 [tool.flit.entrypoints."pyconverters.plugins"]
 openai_vision = "pyconverters_openai_vision.openai_vision:OpenAIVisionConverter"
 deepinfra_openai_vision = "pyconverters_openai_vision.openai_vision:DeepInfraOpenAIVisionConverter"
-azure_openai_vision = "pyconverters_openai_vision.openai_vision:AzureOpenAIVisionConverter"
+# azure_openai_vision = "pyconverters_openai_vision.openai_vision:AzureOpenAIVisionConverter"
 
 
 [tool.flit.metadata.requires-extra]

{pyconverters_openai_vision-0.5.2 → pyconverters_openai_vision-0.5.3}/setup.py

@@ -37,15 +37,13 @@ extras_require = \
           'flask==2.1.3']}
 
 entry_points = \
-{'pyconverters.plugins': ['azure_openai_vision = '
-                          'pyconverters_openai_vision.openai_vision:AzureOpenAIVisionConverter',
-                          'deepinfra_openai_vision = '
+{'pyconverters.plugins': ['deepinfra_openai_vision = '
                           'pyconverters_openai_vision.openai_vision:DeepInfraOpenAIVisionConverter',
                           'openai_vision = '
                           'pyconverters_openai_vision.openai_vision:OpenAIVisionConverter']}
 
 setup(name='pyconverters-openai_vision',
-      version='0.5.2',
+      version='0.5.3',
       description='OpenAIVision converter',
       author='Olivier Terrier',
       author_email='olivier.terrier@kairntech.com',