Upload alpha_factory/run.py

alpha_factory/run.py

@@ -41,10 +41,16 @@ async def setup_models(interactive: bool = False, hf_token: str = None) -> Model
     return manager
 
 
-async def run_pipeline(config, manager: ModelManager | None = None):
-    """Run the full pipeline under a single event loop."""
+async def run_pipeline(config, ollama_url: str = "http://localhost:11434", manager: ModelManager | None = None):
+    """Run the full pipeline under a single event loop.
+    
+    Args:
+        config: Pipeline configuration
+        ollama_url: Base URL for Ollama (default: http://localhost:11434)
+        manager: Optional ModelManager for LLM mode (None for proven mode)
+    """
     # Update LLM config with Ollama URL
-    config.llm.base_url = f"{args.ollama_url}/v1"
+    config.llm.base_url = f"{ollama_url}/v1"
 
     # Create pipeline
     pipeline = AlphaPipeline(config)
@@ -118,7 +124,7 @@ def main():
     try:
         if args.proven:
             console.print("  [green]Proven template mode — no LLM model discovery needed[/]")
-            asyncio.run(run_pipeline(config, manager=None))
+            asyncio.run(run_pipeline(config, ollama_url=args.ollama_url, manager=None))
         else:
             # Discover and select models, then run pipeline
             async def _discover_and_run():
@@ -126,7 +132,7 @@ def main():
                     interactive=args.interactive,
                     hf_token=hf_token,
                 )
-                return await run_pipeline(config, manager=manager)
+                return await run_pipeline(config, ollama_url=args.ollama_url, manager=manager)
             asyncio.run(_discover_and_run())
     except KeyboardInterrupt:
         console.print("\n[yellow]Interrupted by user[/]")