reasoning-deployment-service 0.3.4__py3-none-any.whl → 0.3.5__py3-none-any.whl

reasoning_deployment_service/runner.py

@@ -1,6 +1,7 @@
-import sys
+import sys, os
 import argparse
 from pathlib import Path
+from dotenv import load_dotenv
 from reasoning_deployment_service import ReasoningEngineDeploymentService
 from reasoning_deployment_service.gui_editor import GUIEditor
 from reasoning_deployment_service.cli_editor import CLIRunner
@@ -27,7 +28,7 @@ class Runner:
         # --- Load root_agent dynamically ---
         root_agent = Runner._load_agent(args.agent_path)
 
-        # --- Check env config ---
+        # --- Require config files ---
         if not Path(".env.agent").exists() or not Path("aix_agent.yaml").exists():
             print("Missing .env.agent or aix_agent.yaml.")
             print("Options:\n 5) Generate placeholder config\n q) Quit")
@@ -37,6 +38,11 @@ class Runner:
                 svc._check_required_files_exist()
             sys.exit(0)
 
+        # --- Sanity check env vars ---
+        if not Runner._check_proper_configuration():
+            print("Error: Missing required environment variables in .env.agent")
+            sys.exit(1)
+
         # --- Run mode or menu ---
         if args.mode:
             Runner._dispatch(args.mode, root_agent)
@@ -61,11 +67,12 @@ class Runner:
 
     @staticmethod
     def _load_agent(agent_path_arg: str):
-        """Load root_agent either from --agent-path or your_agent_import.py."""
+        """Load root_agent either from --agent-path or your_agent_import.py, bootstrapping if needed."""
         sys.path.insert(0, str(Path.cwd()))  # ensure project root on sys.path
+        import importlib
 
+        # Case 1: explicit --agent-path (power user / devs only)
         if agent_path_arg:
-            import importlib
             if ":" in agent_path_arg:
                 module_path, attr = agent_path_arg.split(":")
             else:
@@ -73,12 +80,43 @@ class Runner:
             module = importlib.import_module(module_path)
             return getattr(module, attr)
 
-        if Path("your_agent_import.py").exists():
+        # Case 2: shim already exists
+        shim = Path("your_agent_import.py")
+        if shim.exists():
             from your_agent_import import root_agent
             return root_agent
 
-        print("Error: No agent path provided and your_agent_import.py not found.")
-        sys.exit(1)
+        # Case 3: bootstrap (first run)
+        print("No agent path configured. Let's set it up once.")
+        agent_dir = input("Enter the directory where your root_agent (agent.py) lives: ").strip()
+        agent_dir = agent_dir.rstrip("/")
+
+        if not Path(agent_dir, "agent.py").exists():
+            print(f"Error: {agent_dir}/agent.py not found")
+            sys.exit(1)
+
+        import_path = agent_dir.replace("/", ".") + ".agent"
+        shim.write_text(f"from {import_path} import root_agent\n")
+        print(f"Created {shim} pointing to {agent_dir}/agent.py")
+
+        from your_agent_import import root_agent
+        return root_agent
+
+
+    @staticmethod
+    def _check_proper_configuration():
+        """Ensure required environment variables are set in .env.agent."""
+        required_vars = ['PROJECT_ID', 'PROJECT_NUMBER', 'PROJECT_LOCATION', 'AGENT_SPACE_ENGINE']
+        load_dotenv(dotenv_path=".env.agent")
+
+        ok = True
+        for var in required_vars:
+            dev_var = f"DEV_{var}"
+            prod_var = f"PROD_{var}"
+            if not (Path(".env.agent").exists() and (os.getenv(dev_var) or os.getenv(prod_var))):
+                print(f"Missing: {dev_var} or {prod_var}")
+                ok = False
+        return ok
 
     @staticmethod
     def _dispatch(mode, root_agent):

{reasoning_deployment_service-0.3.4.dist-info → reasoning_deployment_service-0.3.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reasoning-deployment-service
-Version: 0.3.4
+Version: 0.3.5
 Summary: Deployment helper for Vertex AI Reasoning Engines & Agent Spaces
 Author-email: Sergio Estrada <sergio.estrada@accenture.com>
 License: Apache-2.0

{reasoning_deployment_service-0.3.4.dist-info → reasoning_deployment_service-0.3.5.dist-info}/RECORD

@@ -1,6 +1,6 @@
 reasoning_deployment_service/__init__.py,sha256=xDuKt9gGviQiTV6vXBdkBvygnlAOIrwnUjVaMGZy0L4,670
 reasoning_deployment_service/reasoning_deployment_service.py,sha256=0QHU2IqqojwFI2wPJ0izrskiHiwBfxdBEB9I_YxYbSA,29133
-reasoning_deployment_service/runner.py,sha256=Vv8LYj4E8TQ0vmXoQXpe4Sm0Og-C1OD6JSFd6i6JWkk,3874
+reasoning_deployment_service/runner.py,sha256=e2Uu4DdRWuCHQI4H1x8Q6s1eMbBNoWFkBTnAaspK1tk,5413
 reasoning_deployment_service/cli_editor/__init__.py,sha256=bN8NPkw8riB92pj2lAwJZuEMOQIO_RRuge0ehnJTW1I,118
 reasoning_deployment_service/cli_editor/api_client.py,sha256=PUXPWUslFOksKMrcKjJSK_gVJkTI0bO_bqtCEYeM85g,27897
 reasoning_deployment_service/cli_editor/cli_runner.py,sha256=w8jv2ep6TiYfpc_KnlvDOJ9hsCSuZcix4Rv4GQcZO4g,15729
@@ -21,8 +21,8 @@ reasoning_deployment_service/gui_editor/src/ui/authorization_view.py,sha256=uiyN
 reasoning_deployment_service/gui_editor/src/ui/reasoning_engine_view.py,sha256=tCvSPEf4dW0NRdAqfs3yT5Pa873gYeLzCMMIt2r2T4o,14644
 reasoning_deployment_service/gui_editor/src/ui/reasoning_engines_view.py,sha256=IRjFlBbY98usAZa0roOonjvWQOsF6NBW4bBg_k8KnKI,7860
 reasoning_deployment_service/gui_editor/src/ui/ui_components.py,sha256=HdQHy-oSZ3GobQ3FNdH7y_w3ANbFiuf2rMoflAmff0A,55366
-reasoning_deployment_service-0.3.4.dist-info/METADATA,sha256=SGmPBaX-F-ReutdQgR8vi4fdLKjI9W-r6O2nDrQpTbU,4991
-reasoning_deployment_service-0.3.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-reasoning_deployment_service-0.3.4.dist-info/entry_points.txt,sha256=onGKjR5ONTtRv3aqEtK863iw9Ty1kLcjfZlsplkRZrA,84
-reasoning_deployment_service-0.3.4.dist-info/top_level.txt,sha256=GKuQS1xHUYLZbatw9DmcYdBxxLhWhhGkV4FmFxgKdp0,29
-reasoning_deployment_service-0.3.4.dist-info/RECORD,,
+reasoning_deployment_service-0.3.5.dist-info/METADATA,sha256=t6j6T_kXXQuzStSR53_2h1WUfB_wDyvxOLT27Gmpr6s,4991
+reasoning_deployment_service-0.3.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+reasoning_deployment_service-0.3.5.dist-info/entry_points.txt,sha256=onGKjR5ONTtRv3aqEtK863iw9Ty1kLcjfZlsplkRZrA,84
+reasoning_deployment_service-0.3.5.dist-info/top_level.txt,sha256=GKuQS1xHUYLZbatw9DmcYdBxxLhWhhGkV4FmFxgKdp0,29
+reasoning_deployment_service-0.3.5.dist-info/RECORD,,