summaryrefslogtreecommitdiffstats
path: root/src/client/openai.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2023-11-02 09:53:54 +0800
committerGitHub <noreply@github.com>2023-11-02 09:53:54 +0800
commit444f4ebe9de8aa68afdf8d39b734a8543b9b5c4b (patch)
treed8b0e16faa0270fda2a6c7afa5fe18c1970b9a93 /src/client/openai.rs
parent5c7bfd92ff3e557477969be9db0638ef0d3d3659 (diff)
downloadaichat-444f4ebe9de8aa68afdf8d39b734a8543b9b5c4b.tar.gz
refactor: improve code quanity (#196)
- rewrite Repl, remove ReplHandler - move ReplyStreamHandler to repl/ and rename it to ReplyHandler - deprecate utils::print_now - refactor session info
Diffstat (limited to 'src/client/openai.rs')
-rw-r--r--src/client/openai.rs11
1 files changed, 8 insertions, 3 deletions
diff --git a/src/client/openai.rs b/src/client/openai.rs
index a82ee9a..98de5a8 100644
--- a/src/client/openai.rs
+++ b/src/client/openai.rs
@@ -1,6 +1,11 @@
use super::{
- ExtraConfig, ModelInfo, OpenAIClient, PromptKind, PromptType, ReplyStreamHandler, SendData,
- TokensCountFactors,
+ ExtraConfig, OpenAIClient, PromptType, SendData,
+};
+
+use crate::{
+ config::{ModelInfo, TokensCountFactors},
+ render::ReplyHandler,
+ utils::PromptKind,
};
use anyhow::{anyhow, bail, Result};
@@ -88,7 +93,7 @@ pub async fn openai_send_message(builder: RequestBuilder) -> Result<String> {
pub async fn openai_send_message_streaming(
builder: RequestBuilder,
- handler: &mut ReplyStreamHandler,
+ handler: &mut ReplyHandler,
) -> Result<()> {
let mut es = builder.eventsource()?;
while let Some(event) = es.next().await {