diff --git a/tests/logging_callback_tests/test_unit_tests_init_callbacks.py b/tests/logging_callback_tests/test_unit_tests_init_callbacks.py index 0226fd66fd..ffb54dd99c 100644 --- a/tests/logging_callback_tests/test_unit_tests_init_callbacks.py +++ b/tests/logging_callback_tests/test_unit_tests_init_callbacks.py @@ -102,7 +102,7 @@ async def use_callback_in_llm_call( elif callback == "openmeter": # it's currently handled in jank way, TODO: fix openmete and then actually run it's test return - elif callback == "bitbucket": + elif callback == "bitbucket" or callback == "gitlab": # Set up mock bitbucket configuration required for initialization litellm.global_bitbucket_config = { "workspace": "test-workspace", @@ -110,6 +110,13 @@ async def use_callback_in_llm_call( "access_token": "test-token", "branch": "main" } + litellm.global_gitlab_config = { + "project": "a/b/", + "access_token": "your-access-token", + "base_url": "gitlab url", + "prompts_path": "src/prompts", # folder to point to, defaults to root + "branch":"main" # optional, defaults to main + } # Mock BitBucket HTTP calls to prevent actual API requests import httpx from unittest.mock import MagicMock