refactor(mistralai): simplify the workflow integration

Author: louis-sanna-dev

pyproject.toml

@@ -29,6 +29,9 @@ agents = [
     "griffe >=1.7.3,<2.0",
     "authlib >=1.5.2,<2.0",
 ]
+workflows = [
+    "mistralai-workflows; python_version >= '3.12'",
+]
 
 [project.urls]
 Repository = "https://github.com/mistralai/client-python.git"
@@ -58,6 +61,9 @@ default-groups = [
     "lint",
 ]
 
+[tool.uv.sources]
+mistralai-workflows = { path = "../dashboard/workflow_sdk" }
+
 [tool.setuptools.package-data]
 "*" = ["py.typed", "src/mistralai/py.typed"]
 

src/mistralai/workflows/__init__.py

@@ -1,33 +1,6 @@
-import importlib.util
-import os
-import sys
-from types import ModuleType
-
-# Find and import the installed mistral-workflows package
-bridge_path = os.path.abspath(__file__)
-sys.modules.pop(__name__, None)  # Remove bridge to avoid recursion
-
 try:
-    # Search for the installed package (skip this bridge file)
-    for path in sys.path:
-        if not path:
-            continue
-        target = os.path.join(path, "mistralai", "workflows", "__init__.py")
-        if os.path.exists(target) and os.path.abspath(target) != bridge_path:
-            spec = importlib.util.spec_from_file_location("mistralai.workflows", target)
-            if spec and spec.loader:
-                workflows: ModuleType = importlib.util.module_from_spec(spec)
-                sys.modules["mistralai.workflows"] = workflows
-                spec.loader.exec_module(workflows)
-                globals().update(
-                    {name: getattr(workflows, name) for name in workflows.__all__}
-                )
-                break
-    else:
-        raise ImportError(
-            "mistral-workflows not installed. Run: pip install mistral-workflows"
-        )
-finally:
-    current_module = sys.modules.get("mistralai.workflows") or sys.modules.get(__name__)
-    if current_module is not None:
-        sys.modules[__name__] = current_module
+    from mistralai_workflows import * # type: ignore
+except ImportError:
+    raise ImportError(
+        "mistralai-workflows not installed. Run: pip install 'mistralai[workflows]'"
+    )