ai-space/goose
3
0
Fork 0
mirror of https://github.com/aaif-goose/goose.git synced 2026-07-03 14:10:03 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix: check context length exceeded phrases on lowercased response (#771)

Browse Source
This commit is contained in:
Salman Mohammed
2025-01-25 23:24:49 -05:00
committed by GitHub
parent cb0049f163
commit d771cc125f
1 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/goose/src/providers/databricks.rs
+11 -9
View File
@@ -163,15 +163,17 @@ impl DatabricksProvider {
}
StatusCode::BAD_REQUEST => {
// Databricks provides a generic 'error' but also includes 'external_model_message' which is provider specific
// we try our best to extract the error message from the payload
let payload_str = serde_json::to_string(&payload).unwrap_or_default();
if payload_str.contains("too long")
|| payload_str.contains("context length")
|| payload_str.contains("context_length_exceeded")
|| payload_str.contains("reduce the length")
|| payload_str.contains("token count")
|| payload_str.contains("exceeds")
{
// We try to extract the error message from the payload and check for phrases that indicate context length exceeded
let payload_str = serde_json::to_string(&payload).unwrap_or_default().to_lowercase();
let check_phrases = [
"too long",
"context length",
"context_length_exceeded",
"reduce the length",
"token count",
"exceeds",
];
if check_phrases.iter().any(|c| payload_str.contains(c)) {
return Err(ProviderError::ContextLengthExceeded(payload_str));
}