.

Author: penguine-ip

deepeval/cli/generate/command.py

@@ -1,11 +1,10 @@
+import sys
 from pathlib import Path
-from typing import List, Optional
+from typing import Any, List, Optional
 
 import typer
 from rich import print
 
-# Synthesizer chain (ChromaDB etc.) is deferred to inside generate_command
-# so unrelated CLI commands (e.g. ``deepeval test run``) don't pay for it.
 from deepeval.cli.generate.utils import (
     FileType,
     GenerationMethod,
@@ -20,6 +19,25 @@
 )
 
 
+# Lazy module-level attrs: ``Synthesizer`` and ``ContextConstructionConfig``
+# materialize on first access (PEP 562) so unrelated CLI commands like
+# ``deepeval test run`` don't pay for the synthesizer chain at startup.
+# Tests still see them as module attributes so ``monkeypatch.setattr(
+# generate_cli, "Synthesizer", _Fake)`` works.
+def __getattr__(name: str) -> Any:
+    if name == "Synthesizer":
+        from deepeval.synthesizer import Synthesizer
+
+        globals()["Synthesizer"] = Synthesizer
+        return Synthesizer
+    if name == "ContextConstructionConfig":
+        from deepeval.synthesizer.config import ContextConstructionConfig
+
+        globals()["ContextConstructionConfig"] = ContextConstructionConfig
+        return ContextConstructionConfig
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+
+
 def generate_command(
     method: GenerationMethod = typer.Option(
         ...,
@@ -186,8 +204,12 @@ def generate_command(
     ),
 ):
     """Generate synthetic goldens with the golden synthesizer."""
-    from deepeval.synthesizer import Synthesizer
-    from deepeval.synthesizer.config import ContextConstructionConfig
+    # Go through the module so test monkeypatches stick. Direct
+    # ``from deepeval.synthesizer import Synthesizer`` would always
+    # fetch the real class and ignore patched module attrs.
+    _self = sys.modules[__name__]
+    Synthesizer = _self.Synthesizer
+    ContextConstructionConfig = _self.ContextConstructionConfig
 
     document_paths = None
     contexts = None