Skip to content

Commit 5ee3321

Browse files
author
Dongri Jin
committed
Fix message
1 parent a0c319c commit 5ee3321

File tree

2 files changed

+14
-3
lines changed

2 files changed

+14
-3
lines changed

examples/function_call_role.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -92,10 +92,10 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
9292
},
9393
chat_completion::ChatCompletionMessage {
9494
role: chat_completion::MessageRole::function,
95-
content: ({
95+
content: {
9696
let price = get_coin_price(&coin).await;
9797
format!("{{\"price\": {}}}", price)
98-
}),
98+
},
9999
name: Some(String::from("get_coin_price")),
100100
function_call: None,
101101
},

src/v1/chat_completion.rs

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,10 +62,21 @@ pub struct ChatCompletionMessage {
6262
pub function_call: Option<FunctionCall>,
6363
}
6464

65+
#[derive(Debug, Serialize, Deserialize)]
66+
pub struct ChatCompletionMessageForResponse {
67+
pub role: MessageRole,
68+
#[serde(skip_serializing_if = "Option::is_none")]
69+
pub content: Option<String>,
70+
#[serde(skip_serializing_if = "Option::is_none")]
71+
pub name: Option<String>,
72+
#[serde(skip_serializing_if = "Option::is_none")]
73+
pub function_call: Option<FunctionCall>,
74+
}
75+
6576
#[derive(Debug, Deserialize)]
6677
pub struct ChatCompletionChoice {
6778
pub index: i64,
68-
pub message: ChatCompletionMessage,
79+
pub message: ChatCompletionMessageForResponse,
6980
pub finish_reason: FinishReason,
7081
}
7182

0 commit comments

Comments
 (0)