pyconverters-openai_vision 0.5.2__py3-none-any.whl → 0.5.3__py3-none-any.whl

pyconverters_openai_vision/__init__.py

@@ -1,2 +1,2 @@
 """OpenAIVision converter"""
-__version__ = "0.5.2"
+__version__ = "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.dist-info → pyconverters_openai_vision-0.5.3.dist-info}/METADATA

@@ -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.3.dist-info/RECORD

@@ -0,0 +1,7 @@
+pyconverters_openai_vision/__init__.py,sha256=3OKw8raUFiHJX-jh8L1RvKEFeD7DZW5-8hpnp21EK-A,51
+pyconverters_openai_vision/openai_utils.py,sha256=HRJ6sJg88en66gkQbOpQKh7cbwtfoAwVLNh7JQSA9ps,5014
+pyconverters_openai_vision/openai_vision.py,sha256=MoEyjYCZ75jWiMonaQdR1erpFBpWgYEiqabp9D8102Y,9562
+pyconverters_openai_vision-0.5.3.dist-info/entry_points.txt,sha256=-DS1gRUTf08Fjb79S_8sqCaqxBifC3q3EJZqXXdcf7Q,197
+pyconverters_openai_vision-0.5.3.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
+pyconverters_openai_vision-0.5.3.dist-info/METADATA,sha256=PJWwNjeSChCjjJQhgWxR1RIn9jA7-HvcPvKHWT7sJjk,2635
+pyconverters_openai_vision-0.5.3.dist-info/RECORD,,

{pyconverters_openai_vision-0.5.2.dist-info → pyconverters_openai_vision-0.5.3.dist-info}/entry_points.txt

@@ -1,5 +1,4 @@
 [pyconverters.plugins]
-azure_openai_vision=pyconverters_openai_vision.openai_vision:AzureOpenAIVisionConverter
 deepinfra_openai_vision=pyconverters_openai_vision.openai_vision:DeepInfraOpenAIVisionConverter
 openai_vision=pyconverters_openai_vision.openai_vision:OpenAIVisionConverter
 

pyconverters_openai_vision-0.5.2.dist-info/RECORD

@@ -1,7 +0,0 @@
-pyconverters_openai_vision/__init__.py,sha256=59Qc-bsfVK9kVVX4vnt2PQF_YysgYAQ-T0KEABdUD64,51
-pyconverters_openai_vision/openai_utils.py,sha256=YNOcbeh1sJKeRvkaMh6FnixrUcOfF4rJIGKG5fmnUDo,4806
-pyconverters_openai_vision/openai_vision.py,sha256=MoEyjYCZ75jWiMonaQdR1erpFBpWgYEiqabp9D8102Y,9562
-pyconverters_openai_vision-0.5.2.dist-info/entry_points.txt,sha256=BeU_sWrRTfIdBnkWCBALpYNOrYOYIlRjQdpR2XQhkSU,285
-pyconverters_openai_vision-0.5.2.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
-pyconverters_openai_vision-0.5.2.dist-info/METADATA,sha256=d2SvC6t_svBjtYRBo-JLShoD_9fVg5ODMvlI9pDULRo,2635
-pyconverters_openai_vision-0.5.2.dist-info/RECORD,,