Fix: correctly pass api_base and add --max-run-time CLI argument in main_full_repo

cover_agent/main_full_repo.py

@@ -28,7 +28,7 @@ async def run():
     async with context_helper.start_server():
         print("LSP server initialized.")
 
-        ai_caller = AICaller(model=args.model)
+        ai_caller = AICaller(model=args.model,api_base=args.api_base, max_tokens=8192)
 
         # main loop for analyzing test files
         for test_file in test_files:

cover_agent/utils.py

@@ -331,6 +331,12 @@ def parse_args_full_repo():
         type=str,
         default="main",
     )
+    parser.add_argument(
+        "--max-run-time",
+        type=int,
+        default=30,
+        help="Maximum time (in seconds) allowed for test execution. Overrides the value in configuration.toml if provided. Defaults to 30 seconds.",
+    )
     return parser.parse_args()