Skip to content

Commit 26425f4

Browse files
authored
fix: detect context length errors in GCP Vertex AI provider (aaif-goose#6976)
Signed-off-by: rabi <ramishra@redhat.com>
1 parent 8b0920b commit 26425f4

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

crates/goose/src/providers/gcpvertexai.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@ use crate::providers::formats::gcpvertexai::{
2626
ModelProvider, RequestContext, DEFAULT_MODEL, KNOWN_MODELS,
2727
};
2828
use crate::providers::gcpauth::GcpAuth;
29+
use crate::providers::openai_compatible::map_http_error_to_provider_error;
2930
use crate::providers::retry::RetryConfig;
3031
use crate::providers::utils::RequestLog;
3132
use crate::session_context::SESSION_ID_HEADER;
@@ -351,9 +352,8 @@ impl GcpVertexAIProvider {
351352
)));
352353
} else {
353354
let response_text = response.text().await.unwrap_or_default();
354-
return Err(ProviderError::RequestFailed(format!(
355-
"Request failed with status {status}: {response_text}"
356-
)));
355+
let payload = serde_json::from_str::<Value>(&response_text).ok();
356+
return Err(map_http_error_to_provider_error(status, payload));
357357
}
358358
}
359359
}

0 commit comments

Comments
 (0)