Yehowshua/ZIm
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

History mostly working

Browse source 
Co-authored-by: Antonio Scandurra <me@as-cii.com>
This commit is contained in:
Conrad Irwin 2025-08-18 11:24:31 -06:00
parent 4b1a48e4de
commit fc076e84ca
9 changed files with 95 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

5
crates/language_model/src/fake_provider.rs
View file

@ -102,6 +102,8 @@ pub struct FakeLanguageModel {
impl Default for FakeLanguageModel {
fn default() -> Self {
dbg!("default......");
eprintln!("{}", std::backtrace::Backtrace::force_capture());
Self {
provider_id: LanguageModelProviderId::from("fake".to_string()),
provider_name: LanguageModelProviderName::from("Fake".to_string()),
@ -149,12 +151,14 @@ impl FakeLanguageModel {
}
pub fn end_completion_stream(&self, request: &LanguageModelRequest) {
dbg!("remove...");
self.current_completion_txs
.lock()
.retain(|(req, _)| req != request);
}
pub fn send_last_completion_stream_text_chunk(&self, chunk: impl Into<String>) {
dbg!("read...");
self.send_completion_stream_text_chunk(self.pending_completions().last().unwrap(), chunk);
}
@ -223,6 +227,7 @@ impl LanguageModel for FakeLanguageModel {
>,
> {
let (tx, rx) = mpsc::unbounded();
dbg!("insert...");
self.current_completion_txs.lock().push((request, tx));
async move { Ok(rx.map(Ok).boxed()) }.boxed()
}