fix: change visibility of certain modules and traits

0xPlaygrounds · Nov 15, 2024 · 007b00a · 007b00a
1 parent 2dada67
commit 007b00a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/rig-lancedb/src/utils/deserializer.rs b/rig-lancedb/src/utils/deserializer.rs
@@ -27,7 +27,7 @@ fn arrow_to_rig_error(e: ArrowError) -> VectorStoreError {
 
 /// Trait used to deserialize data returned from LanceDB queries into a serde_json::Value vector.
 /// Data returned by LanceDB is a vector of `RecordBatch` items.
-pub trait RecordBatchDeserializer {
+pub(crate) trait RecordBatchDeserializer {
     fn deserialize(&self) -> Result<Vec<serde_json::Value>, VectorStoreError>;
 }
 

diff --git a/rig-lancedb/src/utils/mod.rs b/rig-lancedb/src/utils/mod.rs
@@ -1,4 +1,4 @@
-pub mod deserializer;
+mod deserializer;
 
 use deserializer::RecordBatchDeserializer;
 use futures::TryStreamExt;
@@ -10,7 +10,7 @@ use crate::lancedb_to_rig_error;
 
 /// Trait that facilitates the conversion of columnar data returned by a lanceDb query to serde_json::Value.
 /// Used whenever a lanceDb table is queried.
-pub trait QueryToJson {
+pub(crate) trait QueryToJson {
     async fn execute_query(&self) -> Result<Vec<serde_json::Value>, VectorStoreError>;
 }