fixed truncation error in fake language model

KCaverly created 2 years ago

Change summary

crates/ai/src/auth.rs                        | 2 +-
crates/ai/src/test.rs                        | 4 ++++
crates/semantic_index/src/embedding_queue.rs | 2 +-
crates/semantic_index/src/semantic_index.rs  | 5 ++++-
4 files changed, 10 insertions(+), 3 deletions(-)

Detailed changes

crates/ai/src/auth.rs

@@ -1,6 +1,6 @@
 use gpui::AppContext;
 
-#[derive(Clone)]
+#[derive(Clone, Debug)]
 pub enum ProviderCredential {
     Credentials { api_key: String },
     NoCredentials,

crates/ai/src/test.rs

@@ -29,6 +29,10 @@ impl LanguageModel for FakeLanguageModel {
         length: usize,
         direction: TruncationDirection,
     ) -> anyhow::Result<String> {
+        if length > self.count_tokens(content)? {
+            return anyhow::Ok(content.to_string());
+        }
+
         anyhow::Ok(match direction {
             TruncationDirection::End => content.chars().collect::<Vec<char>>()[..length]
                 .into_iter()

crates/semantic_index/src/embedding_queue.rs

@@ -69,7 +69,7 @@ impl EmbeddingQueue {
     }
 
     pub fn set_credential(&mut self, credential: ProviderCredential) {
-        self.provider_credential = credential
+        self.provider_credential = credential;
     }
 
     pub fn push(&mut self, file: FileToEmbed) {

crates/semantic_index/src/semantic_index.rs

@@ -291,7 +291,6 @@ impl SemanticIndex {
         }
 
         self.embedding_queue.lock().set_credential(credential);
-
         self.is_authenticated()
     }
 
@@ -299,6 +298,7 @@ impl SemanticIndex {
         let credential = &self.provider_credential;
         match credential {
             &ProviderCredential::Credentials { .. } => true,
+            &ProviderCredential::NotNeeded => true,
             _ => false,
         }
     }
@@ -1020,11 +1020,14 @@ impl SemanticIndex {
         cx: &mut ModelContext<Self>,
     ) -> Task<Result<()>> {
         if !self.is_authenticated() {
+            println!("Authenticating");
             if !self.authenticate(cx) {
                 return Task::ready(Err(anyhow!("user is not authenticated")));
             }
         }
 
+        println!("SHOULD NOW BE AUTHENTICATED");
+
         if !self.projects.contains_key(&project.downgrade()) {
             let subscription = cx.subscribe(&project, |this, project, event, cx| match event {
                 project::Event::WorktreeAdded | project::Event::WorktreeRemoved(_) => {