Fixed Anthropic API by creating OpenAIChatMessage

This commit is contained in:
Silas Marvin
2024-06-09 01:57:57 -07:00
parent 501e9435ae
commit 01fcc11394
2 changed files with 9 additions and 4 deletions

View File

@@ -36,7 +36,6 @@ pub enum ValidModel {
pub struct ChatMessage {
pub role: String,
pub content: String,
pub tool_calls: Option<Value>, // This is to be compatible with Mistral
}
impl ChatMessage {
@@ -44,7 +43,7 @@ impl ChatMessage {
Self {
role,
content,
tool_calls: None,
// tool_calls: None,
}
}
}

View File

@@ -1,7 +1,7 @@
use std::collections::HashMap;
use anyhow::Context;
use serde::Deserialize;
use serde::{Deserialize, Serialize};
use serde_json::{json, Value};
use tracing::instrument;
@@ -71,9 +71,15 @@ struct OpenAICompletionsResponse {
pub other: HashMap<String, Value>,
}
#[derive(Debug, Deserialize, Serialize)]
pub struct OpenAIChatMessage {
pub role: String,
pub content: String,
}
#[derive(Deserialize)]
pub struct OpenAIChatChoices {
pub message: ChatMessage,
pub message: OpenAIChatMessage,
}
#[derive(Deserialize)]