vexor 0.20.0__py3-none-any.whl → 0.21.1__py3-none-any.whl

vexor/__init__.py

@@ -15,7 +15,7 @@ __all__ = [
     "set_data_dir",
 ]
 
-__version__ = "0.20.0"
+__version__ = "0.21.1"
 
 
 def get_version() -> str:

vexor/api.py

@@ -42,6 +42,8 @@ class RuntimeSettings:
     model_name: str
     batch_size: int
     embed_concurrency: int
+    extract_concurrency: int
+    extract_backend: str
     base_url: str | None
     api_key: str | None
     local_cuda: bool
@@ -90,6 +92,8 @@ def search(
     model: str | None = None,
     batch_size: int | None = None,
     embed_concurrency: int | None = None,
+    extract_concurrency: int | None = None,
+    extract_backend: str | None = None,
     base_url: str | None = None,
     api_key: str | None = None,
     local_cuda: bool | None = None,
@@ -121,6 +125,8 @@ def search(
         model=model,
         batch_size=batch_size,
         embed_concurrency=embed_concurrency,
+        extract_concurrency=extract_concurrency,
+        extract_backend=extract_backend,
         base_url=base_url,
         api_key=api_key,
         local_cuda=local_cuda,
@@ -141,6 +147,8 @@ def search(
         model_name=settings.model_name,
         batch_size=settings.batch_size,
         embed_concurrency=settings.embed_concurrency,
+        extract_concurrency=settings.extract_concurrency,
+        extract_backend=settings.extract_backend,
         provider=settings.provider,
         base_url=settings.base_url,
         api_key=settings.api_key,
@@ -170,6 +178,8 @@ def index(
     model: str | None = None,
     batch_size: int | None = None,
     embed_concurrency: int | None = None,
+    extract_concurrency: int | None = None,
+    extract_backend: str | None = None,
     base_url: str | None = None,
     api_key: str | None = None,
     local_cuda: bool | None = None,
@@ -190,6 +200,8 @@ def index(
         model=model,
         batch_size=batch_size,
         embed_concurrency=embed_concurrency,
+        extract_concurrency=extract_concurrency,
+        extract_backend=extract_backend,
         base_url=base_url,
         api_key=api_key,
         local_cuda=local_cuda,
@@ -208,6 +220,8 @@ def index(
         model_name=settings.model_name,
         batch_size=settings.batch_size,
         embed_concurrency=settings.embed_concurrency,
+        extract_concurrency=settings.extract_concurrency,
+        extract_backend=settings.extract_backend,
         provider=settings.provider,
         base_url=settings.base_url,
         api_key=settings.api_key,
@@ -282,6 +296,8 @@ def _resolve_settings(
     model: str | None,
     batch_size: int | None,
     embed_concurrency: int | None,
+    extract_concurrency: int | None,
+    extract_backend: str | None,
     base_url: str | None,
     api_key: str | None,
     local_cuda: bool | None,
@@ -313,11 +329,21 @@ def _resolve_settings(
     embed_value = (
         embed_concurrency if embed_concurrency is not None else config.embed_concurrency
     )
+    extract_value = (
+        extract_concurrency
+        if extract_concurrency is not None
+        else config.extract_concurrency
+    )
+    extract_backend_value = (
+        extract_backend if extract_backend is not None else config.extract_backend
+    )
     return RuntimeSettings(
         provider=provider_value,
         model_name=model_name,
         batch_size=batch_value,
         embed_concurrency=embed_value,
+        extract_concurrency=extract_value,
+        extract_backend=extract_backend_value,
         base_url=base_url if base_url is not None else config.base_url,
         api_key=api_key if api_key is not None else config.api_key,
         local_cuda=bool(local_cuda if local_cuda is not None else config.local_cuda),