Fix regression that caused Anthropic custom models to error (#15329)

/cc: @bennetbo 

Release Notes:

- N/A

Co-authored-by: Nathan <nathan@zed.dev>
This commit is contained in:
Antonio Scandurra 2024-07-27 14:45:18 +02:00 committed by GitHub
parent 4bd935b409
commit 70e895a8c7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 4 additions and 14 deletions

View File

@ -100,21 +100,13 @@ impl From<Role> for String {
#[derive(Debug, Serialize)]
pub struct Request {
#[serde(serialize_with = "serialize_request_model")]
pub model: Model,
pub model: String,
pub messages: Vec<RequestMessage>,
pub stream: bool,
pub system: String,
pub max_tokens: u32,
}
fn serialize_request_model<S>(model: &Model, serializer: S) -> Result<S::Ok, S::Error>
where
S: serde::Serializer,
{
serializer.serialize_str(&model.id())
}
#[derive(Serialize, Deserialize, Debug, Eq, PartialEq)]
pub struct RequestMessage {
pub role: Role,

View File

@ -4697,8 +4697,6 @@ async fn complete_with_anthropic(
session: UserSession,
api_key: Arc<str>,
) -> Result<()> {
let model = anthropic::Model::from_id(&request.model)?;
let mut system_message = String::new();
let messages = request
.messages
@ -4734,7 +4732,7 @@ async fn complete_with_anthropic(
anthropic::ANTHROPIC_API_URL,
&api_key,
anthropic::Request {
model,
model: request.model,
messages,
stream: true,
system: system_message,

View File

@ -27,7 +27,7 @@ pub struct LanguageModelCompletionProvider {
const MAX_CONCURRENT_COMPLETION_REQUESTS: usize = 4;
pub struct LanguageModelCompletionResponse {
pub inner: BoxStream<'static, Result<String>>,
inner: BoxStream<'static, Result<String>>,
_lock: SemaphoreGuardArc,
}

View File

@ -174,7 +174,7 @@ impl AnthropicModel {
}
Request {
model: self.model.clone(),
model: self.model.id().to_string(),
messages: request
.messages
.iter()