feat: add mlflow 3 support (#6)

* feat: add MLflow 3.x LoggedModel ID import support

MLflow 3.x uses LoggedModel IDs (format: m-<32 hex chars>) in model
  version source fields instead of the original artifact_path. This
  causes import failures because the exported artifacts use the original
  path while the source field references the LoggedModel ID.
                                                                                                                                                                              Changes:
  - Add _is_logged_model_id() to detect MLflow 3.x LoggedModel IDs
  - Add _resolve_logged_model_path() to resolve IDs to actual artifact                                                                                                            paths using mlflow.log-model.history tag or directory scanning
  - Update _extract_model_path() in both import modules to handle                                                                                                                 LoggedModel IDs and pass input_dir for resolution
  - Fix bug in import_model.py where "artifacts" pattern incorrectly                                                                                                             matched "mlflow-artifacts:" URL scheme (changed to "/artifacts/")

* fix: formatting for the unit testing script

* fix: black formatting for prompts changes not related to the changes made in this PR

* fix: fixed pre-commit issues

Author: brayanjuls

docs/bulk.md

@@ -388,26 +388,26 @@ cat experiment-names.csv
 /Users/me@mycompany.com,/Users/you@mycompany.com
 /Users/foo@mycompany.com,/Users/bar@mycompany.com
 ```
- 
+
 ## Prompts
- 
+
 Export/import prompts from the MLflow Prompt Registry (MLflow 2.21.0+).
- 
-**Notes:** 
+
+**Notes:**
 * Prompt Registry support requires MLflow 2.21.0 or higher. The export/import will be skipped with a warning message if the MLflow version doesn't support prompts.
 * **Version handling**: MLflow 3.0+ uses efficient pagination to export all prompt versions. MLflow 2.21-2.x uses iterative discovery with early stopping (stops after 3 consecutive missing versions).
- 
+
 ### Export prompts
- 
+
 Export prompts from the MLflow Prompt Registry to a directory.
- 
+
 Source: [export_prompts.py](mlflow_migration/bulk/export_prompts.py).
- 
+
 #### Usage
- 
+
 ```
 export-prompts --help
- 
+
 Options:
   --output-dir TEXT        Output directory.  [required]
   --prompts TEXT           Prompt names: 'all' for all prompts, comma-delimited
@@ -416,16 +416,16 @@ Options:
                           [required]
   --use-threads BOOLEAN    Use multithreading for export.  [default: False]
 ```
- 
+
 #### Examples
- 
+
 ##### Export all prompts
 ```
 export-prompts \
   --output-dir out/prompts \
   --prompts all
 ```
- 
+
 ##### Export specific prompts
 ```
 export-prompts \
@@ -445,30 +445,30 @@ where `my-prompts.txt` is:
 my-prompt-1
 my-prompt-2
 ```
- 
+
 ### Import prompts
- 
+
 Import prompts to the MLflow Prompt Registry from a directory.
- 
+
 Source: [import_prompts.py](mlflow_migration/bulk/import_prompts.py).
- 
+
 #### Usage
- 
+
 ```
 import-prompts --help
- 
+
 Options:
   --input-dir TEXT         Input directory containing exported prompts.  [required]
   --delete-prompt BOOLEAN  Delete existing prompt before importing.  [default: False]
   --use-threads BOOLEAN    Use multithreading for import.  [default: False]
 ```
- 
+
 #### Examples
- 
+
 ```
 import-prompts --input-dir out/prompts
 ```
- 
-**Notes:** 
+
+**Notes:**
 * Prompts are imported with their original names and version numbers are preserved.
 * All versions of each prompt are exported and imported to maintain complete version history.

mlflow_migration/bulk/export_all.py

@@ -83,17 +83,19 @@ def export_all(
             output_dir=os.path.join(output_dir, "prompts"),
             prompt_names=None,  # Export all prompts
             use_threads=use_threads,
-            mlflow_client=mlflow_client
+            mlflow_client=mlflow_client,
         )
         # Log if unsupported but don't fail
         if res_prompts and "unsupported" in res_prompts:
-            _logger.warning(f"Prompts not supported in MLflow {res_prompts.get('mlflow_version')}")
+            _logger.warning(
+                f"Prompts not supported in MLflow {res_prompts.get('mlflow_version')}"
+            )
         elif res_prompts and "error" in res_prompts:
             _logger.warning(f"Failed to export prompts: {res_prompts['error']}")
     except Exception as e:
         _logger.warning(f"Failed to export prompts: {e}")
         res_prompts = {"error": str(e)}
-    
+
     duration = round(time.time() - start_time, 1)
     info_attr = {
         "options": {
@@ -108,8 +110,8 @@ def export_all(
             "duration": duration,
             "models": res_models,
             "experiments": res_exps,
-            "prompts": res_prompts
-        }
+            "prompts": res_prompts,
+        },
     }
     io_utils.write_export_file(output_dir, "manifest.json", __file__, {}, info_attr)
     _logger.info(f"Duration for entire tracking server export: {duration} seconds")