phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 19:24:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add test for ft endpoints on azure

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-31 15:30:26 -07:00
parent 91c787a77b
commit 7cded198c2
4 changed files with 74 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

13
litellm/proxy/fine_tuning_endpoints/endpoints.py
View file

@ -40,11 +40,20 @@ fine_tuning_config = None
def set_fine_tuning_config(config):
global fine_tuning_config
if not isinstance(config, list):
raise ValueError("invalid fine_tuning config, expected a list is not a list")
for element in config:
if isinstance(element, dict):
for key, value in element.items():
if isinstance(value, str) and value.startswith("os.environ/"):
element[key] = litellm.get_secret(value)
fine_tuning_config = config
# Function to search for specific custom_llm_provider and return its configuration
def get_provider_config(
def get_fine_tuning_provider_config(
custom_llm_provider: str,
):
global fine_tuning_config
@ -122,7 +131,7 @@ async def create_fine_tuning_job(
)
# get configs for custom_llm_provider
llm_provider_config = get_provider_config(
llm_provider_config = get_fine_tuning_provider_config(
custom_llm_provider=fine_tuning_request.custom_llm_provider,
)