add tool role for ChatCompletionMessage

This commit is contained in:
hansel
2024-07-17 10:03:54 +08:00
parent 7e10e3a293
commit 4a11c78393
6 changed files with 12 additions and 0 deletions

View File

@@ -101,6 +101,7 @@ pub enum MessageRole {
system,
assistant,
function,
tool,
}
#[derive(Debug, Deserialize, Clone, PartialEq, Eq)]
@@ -158,6 +159,8 @@ pub struct ChatCompletionMessage {
pub name: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub tool_calls: Option<Vec<ToolCall>>,
#[serde(skip_serializing_if = "Option::is_none")]
pub tool_call_id: Option<String>,
}
#[derive(Debug, Deserialize, Serialize)]