fix: add feature flag to examples in mongodb and lancedb crates

0xPlaygrounds · Oct 18, 2024 · 3688b78 · 3688b78
1 parent 3bb2231
commit 3688b78
Show file tree

Hide file tree

Showing 7 changed files with 27 additions and 8 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -3,5 +3,4 @@ resolver = "2"
 members = [
     "rig-core", "rig-core/rig-core-derive",
     "rig-mongodb",
-    "rig-lancedb"
-]
+    "rig-lancedb"]
diff --git a/rig-lancedb/Cargo.toml b/rig-lancedb/Cargo.toml
@@ -5,7 +5,7 @@ edition = "2021"
 
 [dependencies]
 lancedb = "0.10.0"
-rig-core = { path = "../rig-core", version = "0.2.1", features = ["derive"] }
+rig-core = { path = "../rig-core", version = "0.2.1" }
 arrow-array = "52.2.0"
 serde_json = "1.0.128"
 serde = "1.0.210"
@@ -14,3 +14,19 @@ futures = "0.3.30"
 [dev-dependencies]
 tokio = "1.40.0"
 anyhow = "1.0.89"
+
+[[example]]
+name = "fixtures"
+required-features = ["rig-core/derive"]
+
+[[example]]
+name = "vector_search_local_ann"
+required-features = ["rig-core/derive"]
+
+[[example]]
+name = "vector_search_local_enn"
+required-features = ["rig-core/derive"]
+
+[[example]]
+name = "vector_search_s3_ann"
+required-features = ["rig-core/derive"]
diff --git a/rig-lancedb/examples/fixtures/lib.rs → rig-lancedb/examples/fixtures/main.rs b/rig-lancedb/examples/fixtures/lib.rs → rig-lancedb/examples/fixtures/main.rs
@@ -2,7 +2,7 @@ use std::sync::Arc;
 
 use arrow_array::{types::Float64Type, ArrayRef, FixedSizeListArray, RecordBatch, StringArray};
 use lancedb::arrow::arrow_schema::{DataType, Field, Fields, Schema};
-use rig::embeddings::embedding::Embedding;
+use rig::embeddings::Embedding;
 use rig::{Embeddable, OneOrMany};
 use serde::Deserialize;
 

diff --git a/rig-lancedb/examples/vector_search_local_ann.rs b/rig-lancedb/examples/vector_search_local_ann.rs
@@ -10,7 +10,7 @@ use rig::{
 };
 use rig_lancedb::{LanceDbVectorStore, SearchParams};
 
-#[path = "./fixtures/lib.rs"]
+#[path = "./fixtures/main.rs"]
 mod fixture;
 
 #[tokio::main]

diff --git a/rig-lancedb/examples/vector_search_local_enn.rs b/rig-lancedb/examples/vector_search_local_enn.rs
@@ -9,7 +9,7 @@ use rig::{
 };
 use rig_lancedb::{LanceDbVectorStore, SearchParams};
 
-#[path = "./fixtures/lib.rs"]
+#[path = "./fixtures/main.rs"]
 mod fixture;
 
 #[tokio::main]

diff --git a/rig-lancedb/examples/vector_search_s3_ann.rs b/rig-lancedb/examples/vector_search_s3_ann.rs
@@ -10,7 +10,7 @@ use rig::{
 };
 use rig_lancedb::{LanceDbVectorStore, SearchParams};
 
-#[path = "./fixtures/lib.rs"]
+#[path = "./fixtures/main.rs"]
 mod fixture;
 
 // Note: see docs to deploy LanceDB on other cloud providers such as google and azure.

diff --git a/rig-mongodb/Cargo.toml b/rig-mongodb/Cargo.toml
@@ -12,7 +12,7 @@ repository = "https://github.com/0xPlaygrounds/rig"
 [dependencies]
 futures = "0.3.30"
 mongodb = "2.8.2"
-rig-core = { path = "../rig-core", version = "0.2.1", features = ["derive"] }
+rig-core = { path = "../rig-core", version = "0.2.1" }
 
 serde = { version = "1.0.203", features = ["derive"] }
 serde_json = "1.0.117"
@@ -21,3 +21,7 @@ tracing = "0.1.40"
 [dev-dependencies]
 anyhow = "1.0.86"
 tokio = { version = "1.38.0", features = ["macros"] }
+
+[[example]]
+name = "vector_search_mongodb"
+required-features = ["rig-core/derive"]