add correct image

SciPhi-AI · Aug 22, 2024 · 4ed4d29 · 4ed4d29
1 parent 206a453
commit 4ed4d29
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/py/cli/utils/docker_utils.py b/py/cli/utils/docker_utils.py
@@ -124,6 +124,9 @@ def run_docker_serve(
 
         config = R2RConfig.from_toml(R2RBuilder.CONFIG_OPTIONS[config_name])
 
+    if config.parsing_config.provider == "unstructured" and not image:
+        image = "ragtoriches/prod_unstructured"
+
     completion_provider = config.completion.provider
     completion_model = config.completion.generation_config.model
     completion_model_provider = completion_model.split("/")[0]