diff --git a/cover_agent/main_full_repo.py b/cover_agent/main_full_repo.py
index d54298e9..620a627e 100644
--- a/cover_agent/main_full_repo.py
+++ b/cover_agent/main_full_repo.py
@@ -28,7 +28,8 @@ async def run():
     async with context_helper.start_server():
         print("LSP server initialized.")
 
-        ai_caller = AICaller(model=args.model)
+        api_base = getattr(args, 'api_base', None)
+        ai_caller = AICaller(model=args.model, api_base=api_base, max_tokens=8192)
 
         # main loop for analyzing test files
         for test_file in test_files:
diff --git a/cover_agent/utils.py b/cover_agent/utils.py
index 85a3ac8d..2bf017e5 100644
--- a/cover_agent/utils.py
+++ b/cover_agent/utils.py
@@ -332,6 +332,17 @@ 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 sec",
+)
+    parser.add_argument(
+    "--record-mode",
+    action="store_true",
+    help="Enable record mode for LLM responses. Default: False.",
+)
     parser.add_argument(
         "--record-mode",
         action="store_true",
@@ -343,6 +354,7 @@ def parse_args_full_repo():
         default=False,
         help="Suppress all generated log files (HTML, logs, DB files).",
     )
+
     return parser.parse_args()