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" "validating_files", "queued", "running", "succeeded", "failed", "cancelled"
] = "queued" ] = "queued"
if response["state"] == "JOB_STATE_PENDING": if response["state"] == "JOB_STATE_PENDING":
status = "validating_files" status = "queued"
if response["state"] == "JOB_STATE_SUCCEEDED": if response["state"] == "JOB_STATE_SUCCEEDED":
status = "succeeded" status = "succeeded"
if response["state"] == "JOB_STATE_FAILED": if response["state"] == "JOB_STATE_FAILED":