summaryrefslogtreecommitdiffstats
path: root/src/client/localai.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/client/localai.rs')
-rw-r--r--src/client/localai.rs4
1 files changed, 3 insertions, 1 deletions
diff --git a/src/client/localai.rs b/src/client/localai.rs
index d438388..796b574 100644
--- a/src/client/localai.rs
+++ b/src/client/localai.rs
@@ -1,5 +1,7 @@
use super::openai::{openai_build_body, OPENAI_TOKENS_COUNT_FACTORS};
-use super::{ExtraConfig, LocalAIClient, ModelInfo, PromptKind, PromptType, SendData};
+use super::{ExtraConfig, LocalAIClient, PromptType, SendData};
+
+use crate::{config::ModelInfo, utils::PromptKind};
use anyhow::Result;
use async_trait::async_trait;