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

View File

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