From 5458150ed3203cf13b0371efa2c791ac696cee93 Mon Sep 17 00:00:00 2001 From: sigoden Date: Thu, 23 May 2024 19:28:56 +0800 Subject: fix: json stream parser and refine client modules (#538) --- src/client/openai.rs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/client/openai.rs') diff --git a/src/client/openai.rs b/src/client/openai.rs index c8275cc..881365f 100644 --- a/src/client/openai.rs +++ b/src/client/openai.rs @@ -1,6 +1,6 @@ use super::{ catch_error, message::*, sse_stream, Client, CompletionOutput, ExtraConfig, Model, ModelData, - ModelPatches, OpenAIClient, PromptAction, PromptKind, SendData, SsMmessage, SseHandler, + ModelPatches, OpenAIClient, PromptAction, PromptKind, SendData, SseHandler, SseMmessage, ToolCall, }; @@ -71,7 +71,7 @@ pub async fn openai_send_message_streaming( let mut function_name = String::new(); let mut function_arguments = String::new(); let mut function_id = String::new(); - let handle = |message: SsMmessage| -> Result { + let handle = |message: SseMmessage| -> Result { if message.data == "[DONE]" { if !function_name.is_empty() { handler.tool_call(ToolCall::new( -- cgit v1.2.3