Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Feat(integration test): MongoDB #126

Merged
merged 15 commits into from
Dec 3, 2024
Merged

Feat(integration test): MongoDB #126

Changes from 1 commit
Commits
Show all changes
15 commits
Select commit Hold shift + click to select a range
7e438f3
feat: add integration test for mongo
marieaurore123 Nov 26, 2024
5f8333d
style: cargo fmt
marieaurore123 Nov 26, 2024
d1e465c
fix: failing test
marieaurore123 Nov 26, 2024
f104e92
fix: make PR requested changes
marieaurore123 Dec 2, 2024
f051e64
Merge branch 'main' into feat(integration-tests)/mongodb
marieaurore123 Dec 2, 2024
d9c86ee
refactor(rig-mongodb): use embeddings overhaul in integration test
marieaurore123 Dec 2, 2024
f71fd5d
fix: error caused by container coming out of scope
marieaurore123 Dec 2, 2024
a29e37e
fix: failing test
marieaurore123 Dec 2, 2024
a0db873
fix: rename mongodb example for uniqueness
marieaurore123 Dec 2, 2024
5757187
refactor: cleanup integration test
marieaurore123 Dec 2, 2024
a29bb2c
fix: remove unnecessary clone
marieaurore123 Dec 2, 2024
9effe21
refactor: rename FakeDefinition to Definition
marieaurore123 Dec 3, 2024
7e56301
fix: add sleep
marieaurore123 Dec 3, 2024
6183634
Merge branch 'main' into feat(integration-tests)/mongodb
marieaurore123 Dec 3, 2024
24c92c9
fix: rename struct to Word
marieaurore123 Dec 3, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
fix: add sleep
  • Loading branch information
@marieaurore123
marieaurore123 committed Dec 3, 2024
commit 7e56301678fa89adce5a010fb5bd50911b5be221
3 changes: 3 additions & 0 deletions rig-mongodb/tests/integration_tests.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@ use testcontainers::{
runners::AsyncRunner,
GenericImage, ImageExt,
};
use tokio::time::{sleep, Duration};

#[derive(Embed, Clone, serde::Deserialize, serde::Serialize, Debug, PartialEq)]
struct Definition {
Expand Down Expand Up @@ -70,6 +71,8 @@ async fn vector_search_test() {
.await
.unwrap();

sleep(Duration::from_secs(5)).await;

// Query the index
let results = index
.top_n::<serde_json::Value>("What is a linglingdong?", 1)
Expand Down
Loading