Add llm model dir accessor

spyglass-search · Nov 26, 2024 · eb06c40 · eb06c40
1 parent 6e59ed6
commit eb06c40
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/crates/shared/src/config.rs b/crates/shared/src/config.rs
@@ -359,6 +359,10 @@ impl Config {
         self.model_dir().join("embeddings")
     }
 
+    pub fn llm_model_dir(&self) -> PathBuf {
+        self.model_dir().join("llm")
+    }
+
     pub fn prefs_dir() -> PathBuf {
         let proj_dirs = ProjectDirs::from("com", "athlabs", &Config::app_identifier())
             .expect("Unable to find a suitable settings directory");

diff --git a/crates/spyglass-searcher/src/lib.rs b/crates/spyglass-searcher/src/lib.rs
@@ -127,7 +127,7 @@ pub trait WriteTrait {
 
 type SearcherResult<T> = Result<T, SearchError>;
 
-#[derive(Clone, Serialize)]
+#[derive(Clone, Serialize, Debug)]
 pub struct RetrievedDocument {
     pub doc_id: String,
     pub domain: String,