Skip to content

Commit 21fe408

Browse files
Add usage information to the run object (64bit#195)
1 parent 48d18e0 commit 21fe408

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

async-openai/src/types/run.rs

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,10 @@ pub struct RunObject {
5353
/// The list of [File](https://platform.openai.com/docs/api-reference/files) IDs the [assistant](/docs/api-reference/assistants) used for this run.
5454
pub file_ids: Vec<String>,
5555

56+
/// Usage information for the [run](https://platform.openai.com/docs/api-reference/runs/step-object#runs/object-usage).
57+
/// Usage information is available when the run reaches a terminal state.
58+
pub usage: Option<RunUsage>,
59+
5660
pub metadata: Option<HashMap<String, serde_json::Value>>,
5761
}
5862

@@ -107,6 +111,14 @@ pub enum LastErrorCode {
107111
RateLimitExceeded,
108112
}
109113

114+
#[derive(Clone, Serialize, Debug, Deserialize, PartialEq)]
115+
#[serde(rename_all = "snake_case")]
116+
pub struct RunUsage {
117+
pub completion_tokens: usize,
118+
pub prompt_tokens: usize,
119+
pub total_tokens: usize,
120+
}
121+
110122
#[derive(Clone, Serialize, Default, Debug, Deserialize, Builder, PartialEq)]
111123
#[builder(name = "CreateRunRequestArgs")]
112124
#[builder(pattern = "mutable")]

0 commit comments

Comments
 (0)