Merge pull request #7 from SilasMarvin/silas-fix-anthropic-api

Fixed Anthropic API by creating OpenAIChatMessage
This commit is contained in:
Silas Marvin 2024-06-09 01:59:32 -07:00 committed by GitHub
commit f84b5fe2be
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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)]