From bba796823939de4816cfa3d1bd2ae2e1feff0072 Mon Sep 17 00:00:00 2001
From: Alberto Mordonini <alberto@premai.io>
Date: Thu, 18 Apr 2024 17:03:10 +0200
Subject: [PATCH] Fixed _get_finetuning_job result structure (#103)

---
 prem_utils/connectors/cohere.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/prem_utils/connectors/cohere.py b/prem_utils/connectors/cohere.py
index 02bad2d..d926f5b 100644
--- a/prem_utils/connectors/cohere.py
+++ b/prem_utils/connectors/cohere.py
@@ -180,7 +180,7 @@ def list_jobs(self) -> list[dict[str, any]]:
     def _get_finetuning_job(self, model: CustomModel) -> dict[str, any]:
         return {
             "id": model.id,
-            "fine_tuned_model": model.model_id,
+            "model": model.model_id,
             "created_at": int(model.created_at.timestamp()),
             "finished_at": int(model.completed_at.timestamp()) if model.completed_at else None,
             "status": self._parse_job_status(model.status),