fix translating response

This commit is contained in:
Ishaan Jaff 2024-08-02 18:13:59 -07:00
parent 945c719ed9
commit 0055816f48

View file

@ -47,7 +47,7 @@ class VertexFineTuningAPI(VertexLLM):
"validating_files", "queued", "running", "succeeded", "failed", "cancelled"
] = "queued"
if response["state"] == "JOB_STATE_PENDING":
status = "validating_files"
status = "queued"
if response["state"] == "JOB_STATE_SUCCEEDED":
status = "succeeded"
if response["state"] == "JOB_STATE_FAILED":