Improve long product-data generation runs
This commit is contained in:
@@ -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"}')
|
||||
|
||||
Reference in New Issue
Block a user