Improve long product-data generation runs

This commit is contained in:
wuyang6
2026-05-11 20:26:06 +08:00
parent 4c29ce21f1
commit c4427bbd07
14 changed files with 57 additions and 6 deletions
+8
View File
@@ -38,6 +38,12 @@ def main() -> None:
'--api-key',
default=os.environ.get('OPENAI_API_KEY', ''),
)
parser.add_argument(
'--timeout-seconds',
type=float,
default=float(os.environ.get('OPENAI_TIMEOUT_SECONDS', '3600')),
help='Timeout for each model API request in seconds.',
)
parser.add_argument(
'--session-dir',
default=str(DEFAULT_AGENT_SESSION_DIR),
@@ -58,6 +64,7 @@ def main() -> None:
model=args.model,
base_url=args.base_url,
api_key=args.api_key,
timeout_seconds=args.timeout_seconds,
allow_shell=args.allow_shell,
allow_write=args.allow_write,
session_directory=Path(args.session_dir),
@@ -69,6 +76,7 @@ def main() -> None:
print(f' cwd : {state.cwd}')
print(f' model : {state.model}')
print(f' base-url : {state.base_url}')
print(f' timeout : {args.timeout_seconds:g}s')
print(f' sessions : {state.session_directory}')
print(f' shell : {"on" if state.allow_shell else "off"}')
print(f' write : {"on" if state.allow_write else "off"}')