From b7275b1781dcdbb5f0654a4ad399372af3fdaeda Mon Sep 17 00:00:00 2001
From: Sean Shi <sean.shi@scale.com>
Date: Wed, 19 Jul 2023 17:10:13 -0700
Subject: [PATCH] add field to GetFineTuneResponse

---
 clients/python/llmengine/data_types.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/clients/python/llmengine/data_types.py b/clients/python/llmengine/data_types.py
index 97ebb009..ffcd4f5c 100644
--- a/clients/python/llmengine/data_types.py
+++ b/clients/python/llmengine/data_types.py
@@ -381,6 +381,16 @@ class GetFineTuneResponse(BaseModel):
     The ID of the FineTune.
     """
 
+    fine_tuned_model: Optional[str] = Field(
+        default=None,
+        description="Name of the resulting fine-tuned model. This can be plugged into the "
+        "Completion API once the fine-tune is complete",
+    )
+    """
+    The name of the resulting fine-tuned model. This can be plugged into the Completion API
+    once the fine-tune is complete.
+    """
+
     status: BatchJobStatus = Field(..., description="Status of the requested job.")
     """
     The status of the FineTune job.