tailwindcss + wip auth

src/retrieval/graph.rs

@@ -1,4 +1,4 @@
-use surrealdb::{engine::remote::ws::Client, Error, Surreal};
+use surrealdb::{engine::any::Any, Error, Surreal};
 use tracing::debug;
 
 use crate::storage::types::{knowledge_entity::KnowledgeEntity, StoredObject};
@@ -33,7 +33,7 @@ use crate::storage::types::{knowledge_entity::KnowledgeEntity, StoredObject};
 pub async fn find_entities_by_source_ids<T>(
     source_id: Vec<String>,
     table_name: String,
-    db_client: &Surreal<Client>,
+    db_client: &Surreal<Any>,
 ) -> Result<Vec<T>, Error>
 where
     T: for<'de> serde::Deserialize<'de>,
@@ -50,7 +50,7 @@ where
 
 /// Find entities by their relationship to the id
 pub async fn find_entities_by_relationship_by_id(
-    db_client: &Surreal<Client>,
+    db_client: &Surreal<Any>,
     entity_id: String,
 ) -> Result<Vec<KnowledgeEntity>, Error> {
     let query = format!(
@@ -65,7 +65,7 @@ pub async fn find_entities_by_relationship_by_id(
 
 /// Get a specific KnowledgeEntity by its id
 pub async fn get_entity_by_id(
-    db_client: &Surreal<Client>,
+    db_client: &Surreal<Any>,
     entity_id: &str,
 ) -> Result<Option<KnowledgeEntity>, Error> {
     db_client

src/retrieval/mod.rs

@@ -11,7 +11,7 @@ use crate::{
 };
 use futures::future::{try_join, try_join_all};
 use std::collections::HashMap;
-use surrealdb::{engine::remote::ws::Client, Surreal};
+use surrealdb::{engine::any::Any, Surreal};
 
 /// Performs a comprehensive knowledge entity retrieval using multiple search strategies
 /// to find the most relevant entities for a given query.
@@ -34,7 +34,7 @@ use surrealdb::{engine::remote::ws::Client, Surreal};
 /// * `Result<Vec<KnowledgeEntity>, ProcessingError>` - A deduplicated vector of relevant
 ///   knowledge entities, or an error if the retrieval process fails
 pub async fn combined_knowledge_entity_retrieval(
-    db_client: &Surreal<Client>,
+    db_client: &Surreal<Any>,
     openai_client: &async_openai::Client<async_openai::config::OpenAIConfig>,
     query: &str,
 ) -> Result<Vec<KnowledgeEntity>, ProcessingError> {

src/retrieval/vector.rs

@@ -1,4 +1,4 @@
-use surrealdb::{engine::remote::ws::Client, Surreal};
+use surrealdb::{engine::any::Any, Surreal};
 
 use crate::{error::ProcessingError, utils::embedding::generate_embedding};
 
@@ -25,7 +25,7 @@ use crate::{error::ProcessingError, utils::embedding::generate_embedding};
 pub async fn find_items_by_vector_similarity<T>(
     take: u8,
     input_text: &str,
-    db_client: &Surreal<Client>,
+    db_client: &Surreal<Any>,
     table: String,
     openai_client: &async_openai::Client<async_openai::config::OpenAIConfig>,
 ) -> Result<Vec<T>, ProcessingError>