Skip to content

Commit ffc1403

Browse files
authored
(fix) Tool message type added (#28)
1 parent bc68ad9 commit ffc1403

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

backend/src/common/types/chat_request.rs

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,10 @@ pub enum ChatCompletionRequestMessage{
7171
#[serde(skip_serializing_if = "Option::is_none")]
7272
name: Option<String>,
7373
},
74+
Tool {
75+
content: String,
76+
tool_call_id: String
77+
},
7478
}
7579

7680
#[derive(Debug, Deserialize, Serialize, Clone)]
@@ -125,6 +129,7 @@ impl ChatCompletionRequestMessage {
125129
ChatCompletionRequestMessage::System { content, .. } => content,
126130
ChatCompletionRequestMessage::User { content, .. } => content,
127131
ChatCompletionRequestMessage::Assistant { content, .. } => content,
132+
ChatCompletionRequestMessage::Tool { content, .. } => content,
128133
}
129134
}
130135

@@ -134,6 +139,7 @@ impl ChatCompletionRequestMessage {
134139
ChatCompletionRequestMessage::System { name, .. } => name.as_deref(),
135140
ChatCompletionRequestMessage::User { name, .. } => name.as_deref(),
136141
ChatCompletionRequestMessage::Assistant { name, .. } => name.as_deref(),
142+
_ => None
137143
}
138144
}
139145

@@ -143,6 +149,7 @@ impl ChatCompletionRequestMessage {
143149
ChatCompletionRequestMessage::System { .. } => "system",
144150
ChatCompletionRequestMessage::User { .. } => "user",
145151
ChatCompletionRequestMessage::Assistant { .. } => "assistant",
152+
ChatCompletionRequestMessage::Tool { .. } => "tool",
146153
}
147154
}
148155

0 commit comments

Comments
 (0)