Begin working with kalosm and generating embeddings in the morning.

Author: flusterIO

apps/fluster/src-tauri/src/features/ai/ai_client.rs

@@ -1,8 +1,10 @@
 use crate::core::types::errors::errors::FlusterResult;
 
-use super::ai_providers::{ai_provider::AiProvider, ollama_provider::OllamaProvider};
+use super::ai_providers::ai_provider::AiProvider;
 
-pub struct AiClient {}
+pub struct AiClient {
+    pub provider: dyn AiProvider,
+}
 
 // FIXME: Add an onboarding screen and a field to the HealthReport struct that installs the text embedding model if it is not installed already.
 impl AiClient {

apps/fluster/src-tauri/src/features/ai/ai_providers/ai_provider.rs

@@ -1,17 +1,37 @@
 use kalosm::language::*;
 
-pub async fn get_embeddings(dir_path: &str) -> FlusterResult<()> {
-    let f = DocumentFolder::new(dir_path).unwrap();
-    Ok(())
+use crate::{
+    core::types::errors::errors::{FlusterError, FlusterResult},
+    features::ai::data::{
+        traits::ai_provider::AiProvider, types::embeddings_result::EmbeddingResult,
+    },
+};
+
+pub struct LocalAiClient {}
+
+impl AiProvider for LocalAiClient {
+    fn get_text_embeddings(&self, dir_path: &str) -> FlusterResult<EmbeddingResult> {
+        let f = DocumentFolder::new(dir_path).unwrap();
+        println!("DocumentFolder: {:?}", f);
+        Err(FlusterError::NotImplemented)
+    }
+
+    // pub async fn get_embeddings(dir_path: &str) -> FlusterResult<()> {
+    //     let f = DocumentFolder::new(dir_path).unwrap();
+    //     Ok(())
+    // }
 }
 
 #[cfg(test)]
 mod tests {
+
     use super::*;
 
     #[tokio::test]
     async fn gets_embeddings() {
-        let res = get_embeddings("/Users/bigsexy/Desktop/notes/content/").await;
+        let res = LocalAiClient {}
+            .get_text_embeddings("/Users/bigsexy/Desktop/notes/content/")
+            .await;
         assert!(res.is_ok(), "Get's embeddings without throwing an error.");
         // assert_eq!(result, 4);
     }

apps/fluster/src-tauri/src/features/ai/ai_providers/local_ai_provider.rs

@@ -1,2 +1 @@
-pub mod ai_provider;
 pub mod local_ai_provider;

apps/fluster/src-tauri/src/features/ai/ai_providers/mod.rs

@@ -0,0 +1,2 @@
+pub mod traits;
+pub mod types;

apps/fluster/src-tauri/src/features/ai/data/mod.rs

@@ -0,0 +1,8 @@
+use crate::{
+    core::types::errors::errors::FlusterResult,
+    features::ai::data::types::embeddings_result::EmbeddingResult,
+};
+
+pub trait AiProvider {
+    fn get_text_embeddings(&self, dir_path: &str) -> FlusterResult<EmbeddingResult>;
+}

apps/fluster/src-tauri/src/features/ai/data/traits/ai_provider.rs

@@ -0,0 +1,2 @@
+pub mod ai_provider;
+pub mod embeddings_result;

apps/fluster/src-tauri/src/features/ai/data/traits/mod.rs

@@ -0,0 +1,6 @@
+use std::collections::HashMap;
+
+pub type EmbeddingVec = Vec<f32>;
+
+/// A map of the file system path and the embedding vector.
+pub type EmbeddingResult = HashMap<String, EmbeddingVec>;

apps/fluster/src-tauri/src/features/ai/data/types/embeddings_result.rs

@@ -0,0 +1 @@
+pub mod embeddings_result;

apps/fluster/src-tauri/src/features/ai/data/types/mod.rs

@@ -1,3 +1,4 @@
 pub mod ai_client;
 pub mod ai_providers;
+pub mod data;
 pub mod static_defaults;