From 3e3f9e3f0cbe746a13fef002adab1cb380c78f12 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Mon, 29 Jul 2024 18:59:55 -0700
Subject: [PATCH] add test_create_fine_tune_job

---
 litellm/tests/test_fine_tuning_api.py | 47 +++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)
 create mode 100644 litellm/tests/test_fine_tuning_api.py

diff --git a/litellm/tests/test_fine_tuning_api.py b/litellm/tests/test_fine_tuning_api.py
new file mode 100644
index 000000000..eec81d007
--- /dev/null
+++ b/litellm/tests/test_fine_tuning_api.py
@@ -0,0 +1,47 @@
+import os
+import sys
+import traceback
+
+import pytest
+
+sys.path.insert(
+    0, os.path.abspath("../..")
+)  # Adds the parent directory to the system path
+from openai import APITimeoutError as Timeout
+
+import litellm
+
+litellm.num_retries = 0
+import logging
+
+from litellm import create_fine_tuning_job
+from litellm._logging import verbose_logger
+
+
+def test_create_fine_tune_job():
+    verbose_logger.setLevel(logging.DEBUG)
+    file_name = "openai_batch_completions.jsonl"
+    _current_dir = os.path.dirname(os.path.abspath(__file__))
+    file_path = os.path.join(_current_dir, file_name)
+
+    file_obj = litellm.create_file(
+        file=open(file_path, "rb"),
+        purpose="fine-tune",
+        custom_llm_provider="openai",
+    )
+    print("Response from creating file=", file_obj)
+
+    response = litellm.create_fine_tuning_job(
+        model="gpt-3.5-turbo",
+        training_file=file_obj.id,
+    )
+
+    print("response from litellm.create_fine_tuning_job=", response)
+
+    assert response.id is not None
+    assert response.model == "gpt-3.5-turbo"
+
+    # delete file
+
+    # cancel ft job
+    pass