Alex J. Chan commited on
Commit
51f6814
·
2 Parent(s): 92af62b d572225

Merge pull request #9 from mag-/fix_parameter_parsing

Browse files
Files changed (1) hide show
  1. src/proxy_lite/cli.py +9 -9
src/proxy_lite/cli.py CHANGED
@@ -12,9 +12,9 @@ from proxy_lite.logger import logger
12
 
13
  def update_config_from_env(config: RunnerConfig) -> RunnerConfig:
14
  if os.getenv("PROXY_LITE_API_BASE"):
15
- config.solver.client.api_base = os.getenv("PROXY_LITE_API_BASE")
16
  if os.getenv("PROXY_LITE_MODEL"):
17
- config.solver.client.model_id = os.getenv("PROXY_LITE_MODEL")
18
  if os.getenv("PROXY_LITE_VIEWPORT_WIDTH"):
19
  config.environment.viewport_width = int(os.getenv("PROXY_LITE_VIEWPORT_WIDTH"))
20
  if os.getenv("PROXY_LITE_VIEWPORT_HEIGHT"):
@@ -31,9 +31,9 @@ def do_command(args):
31
  config = update_config_from_env(config)
32
  # Update config from command-line arguments
33
  if args.api_base:
34
- config.solver.client.api_base = args.api_base
35
  if args.model:
36
- config.solver.client.model_id = args.model
37
  if args.homepage:
38
  config.environment.homepage = args.homepage
39
  if args.viewport_width:
@@ -68,33 +68,33 @@ def main():
68
  )
69
  parser.add_argument(
70
  "--model",
71
- type=Optional[str],
72
  default=None,
73
  help="The model to use.",
74
  )
75
  parser.add_argument(
76
  "--api_base",
77
- type=Optional[str],
78
  default=None,
79
  help="The API base URL to use.",
80
  )
81
  # New option for setting a homepage URL:
82
  parser.add_argument(
83
  "--homepage",
84
- type=Optional[str],
85
  default=None,
86
  help="The homepage URL to use.",
87
  )
88
  # New viewport controls:
89
  parser.add_argument(
90
  "--viewport-width",
91
- type=Optional[int],
92
  default=None,
93
  help="Viewport width in pixels.",
94
  )
95
  parser.add_argument(
96
  "--viewport-height",
97
- type=Optional[int],
98
  default=None,
99
  help="Viewport height in pixels.",
100
  )
 
12
 
13
  def update_config_from_env(config: RunnerConfig) -> RunnerConfig:
14
  if os.getenv("PROXY_LITE_API_BASE"):
15
+ config.solver.agent.client.api_base = os.getenv("PROXY_LITE_API_BASE")
16
  if os.getenv("PROXY_LITE_MODEL"):
17
+ config.solver.agent.client.model_id = os.getenv("PROXY_LITE_MODEL")
18
  if os.getenv("PROXY_LITE_VIEWPORT_WIDTH"):
19
  config.environment.viewport_width = int(os.getenv("PROXY_LITE_VIEWPORT_WIDTH"))
20
  if os.getenv("PROXY_LITE_VIEWPORT_HEIGHT"):
 
31
  config = update_config_from_env(config)
32
  # Update config from command-line arguments
33
  if args.api_base:
34
+ config.solver.agent.client.api_base = args.api_base
35
  if args.model:
36
+ config.solver.agent.client.model_id = args.model
37
  if args.homepage:
38
  config.environment.homepage = args.homepage
39
  if args.viewport_width:
 
68
  )
69
  parser.add_argument(
70
  "--model",
71
+ type=str,
72
  default=None,
73
  help="The model to use.",
74
  )
75
  parser.add_argument(
76
  "--api_base",
77
+ type=str,
78
  default=None,
79
  help="The API base URL to use.",
80
  )
81
  # New option for setting a homepage URL:
82
  parser.add_argument(
83
  "--homepage",
84
+ type=str,
85
  default=None,
86
  help="The homepage URL to use.",
87
  )
88
  # New viewport controls:
89
  parser.add_argument(
90
  "--viewport-width",
91
+ type=int,
92
  default=None,
93
  help="Viewport width in pixels.",
94
  )
95
  parser.add_argument(
96
  "--viewport-height",
97
+ type=int,
98
  default=None,
99
  help="Viewport height in pixels.",
100
  )