Fix-319: Correctly pass api_base and add --max-run-time CLI argument in main_full_repo (#336)

* Pass api_base to AICaller

* Modify unit test

Author: andsimakov

cover_agent/main_full_repo.py

@@ -27,7 +27,8 @@ async def run():
         print("LSP server initialized.")
 
         generate_log_files = not args.suppress_log_files
-        ai_caller = AICaller(model=args.model, generate_log_files=generate_log_files)
+        api_base = getattr(args, "api_base", "")
+        ai_caller = AICaller(model=args.model, api_base=api_base, generate_log_files=generate_log_files)
 
         # main loop for analyzing test files
         for test_file in test_files:

tests/test_ai_caller.py

@@ -17,7 +17,7 @@ def ai_caller(self):
         """
         Fixture to create an instance of AICaller for testing.
         """
-        return AICaller("test-model", "test-api", enable_retry=False)
+        return AICaller(model="test-model", api_base="test-api", enable_retry=False)
 
     @patch("cover_agent.ai_caller.AICaller.call_model")
     def test_call_model_simplified(self, mock_call_model):
@@ -105,6 +105,7 @@ def test_call_model_api_base(self, mock_completion, ai_caller):
                 "usage": {"prompt_tokens": 2, "completion_tokens": 10},
             }
             response, prompt_tokens, response_tokens = ai_caller.call_model(prompt)
+            assert ai_caller.api_base == "test-api"
             assert response == "response"
             assert prompt_tokens == 2
             assert response_tokens == 10