chore: fix clippy

This commit is contained in:
yavens 2025-04-10 19:59:44 -04:00
parent 0abd7b8c76
commit 4f023046d5
2 changed files with 13 additions and 21 deletions

View File

@ -1,33 +1,29 @@
use crate::completion::{CompletionError, CompletionRequest};
use crate::message::{ToolCall, ToolFunction};
use crate::providers::cohere::completion::{AssistantContent, BilledUnits, Message, Usage};
use crate::providers::cohere::completion::Usage;
use crate::providers::cohere::CompletionModel;
use crate::streaming::{RawStreamingChoice, StreamingCompletionModel};
use crate::{json_utils, streaming};
use async_stream::stream;
use futures::StreamExt;
use serde::Deserialize;
use serde_json::{json, Value};
use std::collections::HashMap;
use std::future::Future;
use serde_json::json;
#[derive(Debug, Deserialize)]
#[serde(rename_all = "kebab-case", tag = "type")]
pub enum StreamingEvent {
enum StreamingEvent {
MessageStart,
ContentStart,
ContentDelta { delta: Option<Delta> },
ContentEnd,
ToolPlan { delta: Option<Delta> },
ToolPlan,
ToolCallStart { delta: Option<Delta> },
ToolCallDelta { delta: Option<Delta> },
ToolCallEnd { delta: Option<Delta> },
ToolCallEnd,
MessageEnd { delta: Option<MessageEndDelta> },
}
#[derive(Debug, Deserialize)]
struct MessageContentDelta {
r#type: Option<String>,
text: Option<String>,
}
@ -40,14 +36,12 @@ struct MessageToolFunctionDelta {
#[derive(Debug, Deserialize)]
struct MessageToolCallDelta {
id: Option<String>,
r#type: Option<String>,
function: Option<MessageToolFunctionDelta>,
}
#[derive(Debug, Deserialize)]
struct MessageDelta {
content: Option<MessageContentDelta>,
tool_plan: Option<String>,
tool_calls: Option<MessageToolCallDelta>,
}
@ -58,7 +52,6 @@ struct Delta {
#[derive(Debug, Deserialize)]
struct MessageEndDelta {
finish_reason: Option<String>,
usage: Option<Usage>,
}
@ -121,7 +114,7 @@ impl StreamingCompletionModel for CompletionModel {
};
let event = {
let result = serde_json::from_str::<StreamingEvent>(&line);
let result = serde_json::from_str::<StreamingEvent>(line);
let Ok(event) = result else {
continue;

View File

@ -17,7 +17,6 @@ use crate::message::AssistantContent;
use crate::OneOrMany;
use futures::{Stream, StreamExt};
use std::boxed::Box;
use std::fmt::{Display, Formatter};
use std::future::Future;
use std::pin::Pin;
use std::task::{Context, Poll};