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

(Feat) Add logging for POST v1/fine_tuning/jobs (#7426)

Browse source 
* init commit ft jobs logging

* add ft logging

* add logging for FineTuningJob

* simple FT Job create test
This commit is contained in:
Ishaan Jaff 2024-12-26 08:58:47 -08:00 • committed by GitHub
parent 2dcde8ce2b
commit f7316f517a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 45 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/fine_tuning/main.py
View file

@ -19,12 +19,16 @@ import httpx
import litellm
from litellm._logging import verbose_logger
from litellm.llms.azure.fine_tuning.handler import AzureOpenAIFineTuningAPI
from litellm.llms.openai.fine_tuning.handler import FineTuningJob, OpenAIFineTuningAPI
from litellm.llms.openai.fine_tuning.handler import OpenAIFineTuningAPI
from litellm.llms.vertex_ai.fine_tuning.handler import VertexFineTuningAPI
from litellm.secret_managers.main import get_secret_str
from litellm.types.llms.openai import FineTuningJobCreate, Hyperparameters
from litellm.types.llms.openai import (
FineTuningJob,
FineTuningJobCreate,
Hyperparameters,
)
from litellm.types.router import *
from litellm.utils import supports_httpx_timeout
from litellm.utils import client, supports_httpx_timeout
####### ENVIRONMENT VARIABLES ###################
openai_fine_tuning_apis_instance = OpenAIFineTuningAPI()
@ -33,6 +37,7 @@ vertex_fine_tuning_apis_instance = VertexFineTuningAPI()
#################################################
@client
async def acreate_fine_tuning_job(
model: str,
training_file: str,
@ -86,6 +91,7 @@ async def acreate_fine_tuning_job(
raise e
@client
def create_fine_tuning_job(
model: str,
training_file: str,