Pass endpoint URL to openai

This commit is contained in:
shamoon
2025-11-13 09:58:45 -08:00
parent 9043563df3
commit fb883869ec
3 changed files with 5 additions and 2 deletions

View File

@@ -47,11 +47,13 @@ def test_get_llm_openai(mock_ai_config, mock_openai_llm):
mock_ai_config.llm_backend = "openai"
mock_ai_config.llm_model = "test_model"
mock_ai_config.llm_api_key = "test_api_key"
mock_ai_config.llm_endpoint = "http://test-url"
client = AIClient()
mock_openai_llm.assert_called_once_with(
model="test_model",
api_base="http://test-url",
api_key="test_api_key",
)
assert client.llm == mock_openai_llm.return_value