abelworkflow 1.0.0-rc.1 → 1.0.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1,238 +0,0 @@
1
- #!/usr/bin/env python3
2
- """Cross-platform bootstrap entrypoint for Prompt Enhancer."""
3
-
4
- import os
5
- import re
6
- import shutil
7
- import subprocess
8
- import sys
9
- from pathlib import Path
10
- from typing import List, Optional
11
-
12
- from _dotenv import load_dotenv
13
-
14
-
15
- ROOT_DIR = Path(__file__).resolve().parent.parent
16
- REQ_FILE = ROOT_DIR / "requirements.txt"
17
- CLI_PY = ROOT_DIR / "scripts" / "enhance.py"
18
-
19
-
20
- def debug_enabled() -> bool:
21
- value = os.environ.get("PE_DEBUG", "")
22
- return value.strip().lower() in {"1", "true", "yes", "on"}
23
-
24
-
25
- def venv_dir() -> Path:
26
- configured = os.environ.get("PROMPT_ENHANCER_VENV_DIR")
27
- if not configured:
28
- return ROOT_DIR / ".venv"
29
- dir_path = Path(configured).expanduser()
30
- if dir_path.is_absolute():
31
- return dir_path
32
- return ROOT_DIR / dir_path
33
-
34
-
35
- def venv_python() -> Optional[Path]:
36
- dir_path = venv_dir()
37
- candidates = []
38
- if sys.platform == "win32":
39
- candidates.extend(
40
- [
41
- dir_path / "Scripts" / "python.exe",
42
- dir_path / "Scripts" / "python",
43
- ]
44
- )
45
- candidates.append(dir_path / "bin" / "python")
46
- for candidate in candidates:
47
- if candidate.is_file():
48
- return candidate
49
- return None
50
-
51
-
52
- def python_spec() -> Optional[str]:
53
- for name in ("PROMPT_ENHANCER_PYTHON", "AGENTS_SKILLS_PYTHON"):
54
- value = os.environ.get(name)
55
- if value:
56
- return value
57
- return None
58
-
59
-
60
- def has_uv() -> bool:
61
- return shutil.which("uv") is not None
62
-
63
-
64
- def find_system_python() -> Optional[str]:
65
- env_python = os.environ.get("AGENTS_SKILLS_PYTHON")
66
- if env_python and Path(env_python).is_file():
67
- return env_python
68
- if sys.executable and Path(sys.executable).is_file():
69
- return sys.executable
70
- for command in ("python3", "python"):
71
- found = shutil.which(command)
72
- if found:
73
- return found
74
- return None
75
-
76
-
77
- def create_venv() -> None:
78
- dir_path = venv_dir()
79
- if has_uv():
80
- command = ["uv", "venv"]
81
- spec = python_spec()
82
- if spec:
83
- command.extend(["--python", spec])
84
- command.append(str(dir_path))
85
- subprocess.run(command, check=True)
86
- return
87
- python_bin = find_system_python()
88
- if not python_bin:
89
- print("Error: No usable uv or python found. Cannot create virtual environment.", file=sys.stderr)
90
- sys.exit(1)
91
- subprocess.run([python_bin, "-m", "venv", str(dir_path)], check=True)
92
-
93
-
94
- def first_non_empty(*values: str) -> str:
95
- for value in values:
96
- stripped = value.strip()
97
- if stripped:
98
- return stripped
99
- return ""
100
-
101
-
102
- def argv_value(flag: str, argv: Optional[List[str]] = None) -> str:
103
- argv = sys.argv[1:] if argv is None else argv
104
- for index, token in enumerate(argv):
105
- if token == flag:
106
- if index + 1 < len(argv):
107
- return argv[index + 1].strip()
108
- return ""
109
- if token.startswith(f"{flag}="):
110
- return token.split("=", 1)[1].strip()
111
- return ""
112
-
113
-
114
- def resolved_provider(argv: Optional[List[str]] = None) -> Optional[str]:
115
- argv = sys.argv[1:] if argv is None else argv
116
- api_url = first_non_empty(argv_value("--url", argv), os.environ.get("PE_API_URL", ""))
117
- api_key = first_non_empty(argv_value("--api-key", argv), os.environ.get("PE_API_KEY", ""))
118
- model = first_non_empty(argv_value("--model", argv), os.environ.get("PE_MODEL", ""))
119
-
120
- if api_url and api_key and model:
121
- return "openai"
122
- return None
123
-
124
-
125
- def required_modules() -> List[str]:
126
- return ["openai"] if resolved_provider() == "openai" else []
127
-
128
-
129
- def install_targets(modules: List[str]) -> List[str]:
130
- if not REQ_FILE.is_file():
131
- return modules
132
-
133
- requirements = {}
134
- for raw_line in REQ_FILE.read_text(encoding="utf-8").splitlines():
135
- line = raw_line.split("#", 1)[0].strip()
136
- if not line:
137
- continue
138
- package = re.split(r"[<>=!~\\[; ]", line, maxsplit=1)[0].strip().lower()
139
- if package:
140
- requirements[package] = line
141
-
142
- return [requirements.get(module.lower(), module) for module in modules]
143
-
144
-
145
- def has_required_modules(python_bin: str, modules: List[str]) -> bool:
146
- if not modules:
147
- return True
148
- check = subprocess.run(
149
- [python_bin, "-c", "; ".join(f"import {module}" for module in modules)],
150
- stdout=subprocess.DEVNULL,
151
- stderr=subprocess.DEVNULL,
152
- )
153
- return check.returncode == 0
154
-
155
-
156
- def install_deps(python_bin: Path, modules: Optional[List[str]] = None) -> None:
157
- modules = required_modules() if modules is None else modules
158
- if not modules:
159
- return
160
- targets = install_targets(modules)
161
- if has_required_modules(str(python_bin), modules):
162
- return
163
- try:
164
- kwargs = {
165
- "check": True,
166
- "stdout": subprocess.DEVNULL,
167
- "stderr": subprocess.PIPE,
168
- "text": True,
169
- }
170
- if has_uv():
171
- subprocess.run(["uv", "pip", "install", "--python", str(python_bin), *targets], **kwargs)
172
- return
173
- subprocess.run([str(python_bin), "-m", "pip", "install", *targets], **kwargs)
174
- except subprocess.CalledProcessError as exc:
175
- if debug_enabled():
176
- details = (exc.stderr or "").strip()
177
- if details:
178
- print(details, file=sys.stderr)
179
- print(
180
- "Warning: Failed to install prompt enhancer dependencies; continuing with existing environment.",
181
- file=sys.stderr,
182
- )
183
-
184
-
185
- def validate_venv_dir() -> None:
186
- dir_path = venv_dir()
187
- if dir_path.exists() and not dir_path.is_dir():
188
- print(f"Error: {dir_path} exists but is not a directory.", file=sys.stderr)
189
- sys.exit(1)
190
- if dir_path.is_dir() and not (dir_path / "pyvenv.cfg").exists() and venv_python() is None:
191
- print(f"Error: {dir_path} exists but is not a valid venv.", file=sys.stderr)
192
- sys.exit(1)
193
-
194
-
195
- def should_passthrough_cli() -> bool:
196
- return len(sys.argv) < 2
197
-
198
-
199
- def passthrough_python() -> Optional[str]:
200
- if sys.executable and Path(sys.executable).is_file():
201
- return sys.executable
202
- return find_system_python()
203
-
204
-
205
- def runtime_python() -> str:
206
- modules = required_modules()
207
- current_python = passthrough_python()
208
- if current_python and has_required_modules(current_python, modules):
209
- return current_python
210
-
211
- validate_venv_dir()
212
- python_bin = venv_python()
213
- if python_bin is None:
214
- create_venv()
215
- python_bin = venv_python()
216
- if python_bin is None:
217
- print("Error: Failed to locate python in venv after creation.", file=sys.stderr)
218
- sys.exit(1)
219
- install_deps(python_bin, modules)
220
- return str(python_bin)
221
-
222
-
223
- def main() -> None:
224
- if should_passthrough_cli():
225
- python_bin = passthrough_python()
226
- if not python_bin:
227
- print("Error: No usable python found for CLI usage output.", file=sys.stderr)
228
- sys.exit(1)
229
- result = subprocess.run([python_bin, str(CLI_PY)] + sys.argv[1:])
230
- sys.exit(result.returncode)
231
- load_dotenv()
232
- python_bin = runtime_python()
233
- result = subprocess.run([python_bin, str(CLI_PY)] + sys.argv[1:])
234
- sys.exit(result.returncode)
235
-
236
-
237
- if __name__ == "__main__":
238
- main()