diff options
Diffstat (limited to 'src/client')
| -rw-r--r-- | src/client/bedrock.rs | 4 | ||||
| -rw-r--r-- | src/client/qianwen.rs | 2 |
2 files changed, 3 insertions, 3 deletions
diff --git a/src/client/bedrock.rs b/src/client/bedrock.rs index d6ab696..6abd939 100644 --- a/src/client/bedrock.rs +++ b/src/client/bedrock.rs @@ -141,7 +141,7 @@ async fn send_message( match model_category { ModelCategory::Anthropic => claude_extract_completion(&data), ModelCategory::MetaLlama3 => llama_extract_completion(&data), - ModelCategory::Mistral => mistral_extrat_completion(&data), + ModelCategory::Mistral => mistral_extract_completion(&data), } } @@ -283,7 +283,7 @@ fn llama_extract_completion(data: &Value) -> Result<(String, CompletionDetails)> Ok((text.to_string(), details)) } -fn mistral_extrat_completion(data: &Value) -> Result<(String, CompletionDetails)> { +fn mistral_extract_completion(data: &Value) -> Result<(String, CompletionDetails)> { let text = data["outputs"][0]["text"] .as_str() .ok_or_else(|| anyhow!("Invalid response data: {data}"))?; diff --git a/src/client/qianwen.rs b/src/client/qianwen.rs index 76d7436..3fa17e6 100644 --- a/src/client/qianwen.rs +++ b/src/client/qianwen.rs @@ -210,7 +210,7 @@ fn extract_completion_text(data: &Value, is_vl: bool) -> Result<(String, Complet Ok((text.to_string(), details)) } -/// Patch messsages, upload embedded images to oss +/// Patch messages, upload embedded images to oss async fn patch_messages(model: &str, api_key: &str, messages: &mut Vec<Message>) -> Result<()> { for message in messages { if let MessageContent::Array(list) = message.content.borrow_mut() { |