Fixed _get_finetuning_job result structure (#103)

premAI-io · Apr 18, 2024 · bba7968 · bba7968
1 parent b0b0b22
commit bba7968
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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),