summaryrefslogtreecommitdiffstats
path: root/src/config/input.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/config/input.rs')
-rw-r--r--src/config/input.rs7
1 files changed, 7 insertions, 0 deletions
diff --git a/src/config/input.rs b/src/config/input.rs
index e468f19..482b1d5 100644
--- a/src/config/input.rs
+++ b/src/config/input.rs
@@ -244,6 +244,13 @@ impl Input {
init_client(&self.config, Some(self.role().model().clone()))
}
+ pub async fn fetch_chat_text(&self) -> Result<String> {
+ let client = self.create_client()?;
+ let text = client.chat_completions(self.clone()).await?.text;
+ let text = strip_think_tag(&text).to_string();
+ Ok(text)
+ }
+
pub fn prepare_completion_data(
&self,
model: &Model,