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

add base model to EmbeddingProvider, not yet leveraged for truncation

Browse source
This commit is contained in:
KCaverly 2023-10-22 15:00:09 +02:00
parent d1dec8314a
commit 2b780ee7b2
5 changed files with 57 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

10
crates/semantic_index/src/semantic_index_tests.rs
View file

@ -4,8 +4,11 @@ use crate::{
semantic_index_settings::SemanticIndexSettings,
FileToEmbed, JobHandle, SearchResult, SemanticIndex, EMBEDDING_QUEUE_FLUSH_TIMEOUT,
};
use ai::embedding::{Embedding, EmbeddingProvider};
use ai::providers::dummy::DummyEmbeddingProvider;
use ai::providers::dummy::{DummyEmbeddingProvider, DummyLanguageModel};
use ai::{
embedding::{Embedding, EmbeddingProvider},
models::LanguageModel,
};
use anyhow::Result;
use async_trait::async_trait;
use gpui::{executor::Deterministic, Task, TestAppContext};
@ -1282,6 +1285,9 @@ impl FakeEmbeddingProvider {
#[async_trait]
impl EmbeddingProvider for FakeEmbeddingProvider {
fn base_model(&self) -> Box<dyn LanguageModel> {
Box::new(DummyLanguageModel {})
}
fn is_authenticated(&self) -> bool {
true
}