aljaz/goose
1
0
Fork 0
mirror of https://github.com/aljazceru/goose.git synced 2026-02-21 14:34:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix: Set include_usage=true for OpenAI streaming (#3441)

Browse Source
This commit is contained in:
Jack Amadeo
2025-07-15 16:53:41 -04:00
committed by GitHub
parent 70f41d10df
commit f4e3d06f9e
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
crates/goose/src/providers/openai.rs
View File

@@ -3,7 +3,7 @@ use async_stream::try_stream;
use async_trait::async_trait;
use futures::TryStreamExt;
use reqwest::{Client, Response};
use serde_json::Value;
use serde_json::{json, Value};
use std::collections::HashMap;
use std::io;
use std::time::Duration;
@@ -254,6 +254,9 @@ impl Provider for OpenAiProvider {
let mut payload =
create_request(&self.model, system, messages, tools, &ImageFormat::OpenAi)?;
payload["stream"] = serde_json::Value::Bool(true);
payload["stream_options"] = json!({
"include_usage": true,
});
let response = handle_status_openai_compat(self.post(payload.clone()).await?).await?;