Atlas Vector Search 快速入门

1
#include <bson/bson.h>
2
#include <mongoc/mongoc.h>
3
#include <stdio.h>
4
5
int main(void) {
6
  mongoc_client_t *client;
7
  mongoc_collection_t *collection;
8
  bson_error_t error;
9
  char database_name[] = "sample_mflix";
10
  char collection_name[] = "embedded_movies";
11
  char index_name[] = "vector_index";
12
13
  mongoc_init();
14
15
  // Replace the placeholder with your Atlas connection string
16
  client =  mongoc_client_new("<connection-string>");
17
18
  // Connect to your Atlas cluster
19
  collection = mongoc_client_get_collection (client, database_name, collection_name);
20
21
  bson_t cmd;
22
  // Create search index command.
23
  {
24
    char *cmd_str = bson_strdup_printf (
25
        BSON_STR ({
26
                    "createSearchIndexes" : "%s",
27
                    "indexes" : [{
28
                      "definition": {
29
                        "fields": [{
30
                          "type": "vector",
31
                          "path": "plot_embedding",
32
                          "numDimensions": 1536,
33
                          "similarity": "dotProduct",
34
                          "quantization": "scalar"
35
                        }]
36
                      },
37
                      "name": "%s",
38
                      "type": "vectorSearch"
39
                    }]
40
                  }),
41
        collection_name, index_name);
42
    if (!bson_init_from_json(&cmd, cmd_str, -1, &error)) {
43
      printf("Failed to initialize BSON: %s\n", error.message);
44
      bson_free(cmd_str);
45
      return 1;
46
    }
47
    bson_free (cmd_str);
48
  }
49
  if (!mongoc_collection_command_simple (collection, &cmd, NULL /* read_prefs */, NULL /* reply */, &error)) {
50
    bson_destroy (&cmd);
51
    printf ("Failed to run createSearchIndexes: %s", error.message);
52
    return 1;
53
  } else {
54
    printf ("New search index named %s is building.\n", index_name);
55
    bson_destroy (&cmd);
56
  }
57
58
  // Polling for index status
59
  printf("Polling to check if the index is ready. This may take up to a minute.\n");
60
  int queryable = 0;
61
  while (!queryable) {
62
    const char *pipeline_str = "{\"pipeline\": [{\"$listSearchIndexes\": {}}]}";
63
    bson_t pipeline;
64
    if (!bson_init_from_json(&pipeline, pipeline_str, -1, &error)) {
65
      printf("Failed to initialize pipeline BSON: %s\n", error.message);
66
      break; // Exit the loop on error
67
    }
68
    mongoc_cursor_t *cursor = mongoc_collection_aggregate(collection, MONGOC_QUERY_NONE, &pipeline, NULL, NULL);
69
    const bson_t *got;
70
    // Check if the cursor returns any documents
71
    int found_index = 0;
72
    while (mongoc_cursor_next(cursor, &got)) {
73
      bson_iter_t iter;
74
      if (bson_iter_init(&iter, got) && bson_iter_find(&iter, "name")) {
75
        const char *name = bson_iter_utf8(&iter, NULL);
76
        if (strcmp(name, index_name) == 0) {
77
          found_index = 1; // Index found
78
          bson_iter_find(&iter, "queryable");
79
          queryable = bson_iter_bool(&iter);
80
          break; // Exit the loop since we found the index
81
        }
82
      }
83
    }
84
    if (mongoc_cursor_error(cursor, &error)) {
85
      printf("Failed to run $listSearchIndexes: %s\n", error.message);
86
      break; // Exit the loop on error
87
    }
88
    if (!found_index) {
89
      printf("Index %s not found yet. Retrying...\n", index_name);
90
    }
91
    bson_destroy(&pipeline);
92
    mongoc_cursor_destroy(cursor);
93
    sleep(5); // Sleep for 5 seconds before checking again
94
  }
95
  if (queryable) {
96
    printf("%s is ready for querying.\n", index_name);
97
  } else {
98
    printf("Error occurred or index not found.\n");
99
  }
100
101
  // Cleanup
102
  mongoc_collection_destroy(collection);
103
  mongoc_client_destroy(client);
104
  mongoc_cleanup();
105
106
  return 0;
107
}

1
#include <bsoncxx/builder/basic/document.hpp>
2
#include <iostream>
3
#include <mongocxx/client.hpp>
4
#include <mongocxx/instance.hpp>
5
#include <mongocxx/search_index_view.hpp>
6
#include <mongocxx/uri.hpp>
7
#include <thread>
8
9
using bsoncxx::builder::basic::kvp;
10
using bsoncxx::builder::basic::make_array;
11
using bsoncxx::builder::basic::make_document;
12
13
int main() {
14
  try {
15
    mongocxx::instance inst{};
16
17
    // Replace the placeholder with your Atlas connection string
18
    const auto uri = mongocxx::uri{"<connection-string>"};
19
20
    // Connect to your Atlas cluster
21
    mongocxx::client conn{uri};
22
    auto db = conn["sample_mflix"];
23
    auto collection = db["embedded_movies"];
24
25
    auto siv = collection.search_indexes();
26
    std::string name = "vector_index";
27
    auto type = "vectorSearch";
28
    auto definition = make_document(
29
        kvp("fields",
30
            make_array(make_document(
31
                kvp("type", "vector"), kvp("path", "plot_embedding"),
32
                kvp("numDimensions", 1536), kvp("similarity", "dotProduct"), 
33
                kvp("quantization", "scalar")))));
34
    auto model =
35
        mongocxx::search_index_model(name, definition.view()).type(type);
36
    siv.create_one(model);
37
    std::cout << "New search index named " << name << " is building."
38
              << std::endl;
39
40
    // Polling for index status
41
    std::cout << "Polling to check if the index is ready. This may take up to "
42
                 "a minute."
43
              << std::endl;
44
    bool queryable = false;
45
    while (!queryable) {
46
      auto indexes = siv.list();
47
      for (const auto& index : indexes) {
48
        if (index["name"].get_value() == name) {
49
          queryable = index["queryable"].get_bool();
50
        }
51
      }
52
      if (!queryable) {
53
        std::this_thread::sleep_for(std::chrono::seconds(5));
54
      }
55
    }
56
    std::cout << name << " is ready for querying." << std::endl;
57
  } catch (const std::exception& e) {
58
    std::cout << "Exception: " << e.what() << std::endl;
59
  }
60
  return 0;
61
}

1
namespace query_quick_start;
2
3
using MongoDB.Bson;
4
using MongoDB.Driver;
5
using System;
6
using System.Threading;
7
8
public class IndexService
9
{
10
    // Replace the placeholder with your Atlas connection string
11
    private const string MongoConnectionString = "<connection-string>";
12
    public void CreateVectorIndex()
13
    {
14
        try
15
        {
16
            // Connect to your Atlas cluster
17
            var client = new MongoClient(MongoConnectionString);
18
            var database = client.GetDatabase("sample_mflix");
19
            var collection = database.GetCollection<BsonDocument>("embedded_movies");
20
21
            var searchIndexView = collection.SearchIndexes;
22
            var name = "vector_index";
23
            var type = SearchIndexType.VectorSearch;
24
25
            var definition = new BsonDocument
26
            {
27
                { "fields", new BsonArray
28
                    {
29
                        new BsonDocument
30
                        {
31
                            { "type", "vector" },
32
                            { "path", "plot_embedding" },
33
                            { "numDimensions", 1536 },
34
                            { "similarity", "dotProduct" },
35
                            { "quantization", "scalar" }
36
                        }
37
                    }
38
                }
39
            };
40
41
            var model = new CreateSearchIndexModel(name, type, definition);
42
43
            searchIndexView.CreateOne(model);
44
            Console.WriteLine($"New search index named {name} is building.");
45
46
            // Polling for index status
47
            Console.WriteLine("Polling to check if the index is ready. This may take up to a minute.");
48
            bool queryable = false;
49
            while (!queryable)
50
            {
51
                var indexes = searchIndexView.List();
52
                foreach (var index in indexes.ToEnumerable())
53
                {
54
                    if (index["name"] == name)
55
                    {
56
                        queryable = index["queryable"].AsBoolean;
57
                    }
58
                }
59
                if (!queryable)
60
                {
61
                    Thread.Sleep(5000);
62
                }
63
            }
64
            Console.WriteLine($"{name} is ready for querying.");
65
        }
66
        catch (Exception e)
67
        {
68
            Console.WriteLine($"Exception: {e.Message}");
69
        }
70
    }
71
}

using query_quick_start;
var indexService = new IndexService();
indexService.CreateVectorIndex();

1
package main
2
3
import (
4
	"context"
5
	"fmt"
6
	"log"
7
	"time"
8
9
	"go.mongodb.org/mongo-driver/bson"
10
	"go.mongodb.org/mongo-driver/mongo"
11
	"go.mongodb.org/mongo-driver/mongo/options"
12
)
13
14
func main() {
15
	ctx := context.Background()
16
17
	// Replace the placeholder with your Atlas connection string
18
	const uri = "<connectionString>"
19
20
	// Connect to your Atlas cluster
21
	clientOptions := options.Client().ApplyURI(uri)
22
	client, err := mongo.Connect(ctx, clientOptions)
23
	if err != nil {
24
		log.Fatalf("failed to connect to the server: %v", err)
25
	}
26
	defer func() { _ = client.Disconnect(ctx) }()
27
28
	// Set the namespace
29
	coll := client.Database("sample_mflix").Collection("embedded_movies")
30
31
	// Define the index details
32
	type vectorDefinitionField struct {
33
		Type          string `bson:"type"`
34
		Path          string `bson:"path"`
35
		NumDimensions int    `bson:"numDimensions"`
36
		Similarity    string `bson:"similarity"`
37
		Quantization  string `bson:"quantization"`
38
	}
39
40
	type vectorDefinition struct {
41
		Fields []vectorDefinitionField `bson:"fields"`
42
	}
43
44
	indexName := "vector_index"
45
	opts := options.SearchIndexes().SetName(indexName).SetType("vectorSearch")
46
47
	indexModel := mongo.SearchIndexModel{
48
		Definition: vectorDefinition{
49
			Fields: []vectorDefinitionField{{
50
				Type:          "vector",
51
				Path:          "plot_embedding",
52
				NumDimensions: 1536,
53
				Similarity:    "dotProduct",
54
			    Quantization:  "scalar"}},
55
		},
56
		Options: opts,
57
	}
58
59
	// Create the index
60
	searchIndexName, err := coll.SearchIndexes().CreateOne(ctx, indexModel)
61
	if err != nil {
62
		log.Fatalf("failed to create the search index: %v", err)
63
	}
64
	log.Println("New search index named " + searchIndexName + " is building.")
65
66
	// Await the creation of the index.
67
	log.Println("Polling to check if the index is ready. This may take up to a minute.")
68
	searchIndexes := coll.SearchIndexes()
69
	var doc bson.Raw
70
	for doc == nil {
71
		cursor, err := searchIndexes.List(ctx, options.SearchIndexes().SetName(searchIndexName))
72
		if err != nil {
73
			fmt.Errorf("failed to list search indexes: %w", err)
74
		}
75
76
		if !cursor.Next(ctx) {
77
			break
78
		}
79
80
		name := cursor.Current.Lookup("name").StringValue()
81
		queryable := cursor.Current.Lookup("queryable").Boolean()
82
		if name == searchIndexName && queryable {
83
			doc = cursor.Current
84
		} else {
85
			time.Sleep(5 * time.Second)
86
		}
87
	}
88
89
	log.Println(searchIndexName + " is ready for querying.")
90
}

1
import com.mongodb.client.ListSearchIndexesIterable;
2
import com.mongodb.client.MongoClient;
3
import com.mongodb.client.MongoClients;
4
import com.mongodb.client.MongoCollection;
5
import com.mongodb.client.MongoCursor;
6
import com.mongodb.client.MongoDatabase;
7
import com.mongodb.client.model.SearchIndexModel;
8
import com.mongodb.client.model.SearchIndexType;
9
import org.bson.Document;
10
import org.bson.conversions.Bson;
11
12
import java.util.Collections;
13
import java.util.List;
14
15
public class VectorIndex {
16
17
    public static void main(String[] args) {
18
19
        // Replace the placeholder with your Atlas connection string
20
        String uri = "<connectionString>";
21
22
        // Connect to your Atlas cluster
23
        try (MongoClient mongoClient = MongoClients.create(uri)) {
24
25
            // Set the namespace
26
            MongoDatabase database = mongoClient.getDatabase("sample_mflix");
27
            MongoCollection<Document> collection = database.getCollection("embedded_movies");
28
29
            // Define the index details
30
            String indexName = "vector_index";
31
            Bson definition = new Document(
32
                "fields",
33
                Collections.singletonList(
34
                    new Document("type", "vector")
35
                        .append("path", "plot_embedding")
36
                        .append("numDimensions", 1536)
37
                        .append("similarity", "dotProduct")
38
                        .append("quantization", "scalar")));
39
40
            // Define the index model
41
            SearchIndexModel indexModel = new SearchIndexModel(
42
                indexName,
43
                definition,
44
                SearchIndexType.vectorSearch());
45
46
            // Create the index
47
            try {
48
                List<String> result = collection.createSearchIndexes(Collections.singletonList(indexModel));
49
                System.out.println("New search index named " + result.get(0) + " is building.");
50
            } catch (Exception e) {
51
                throw new RuntimeException("Error creating index: " + e);
52
            }
53
54
            // Wait for Atlas to build the index
55
            System.out.println("Polling to check if the index is ready. This may take up to a minute.");
56
57
            ListSearchIndexesIterable<Document> searchIndexes = collection.listSearchIndexes();
58
            Document doc = null;
59
            while (doc == null) {
60
                try (MongoCursor<Document> cursor = searchIndexes.iterator()) {
61
                    if (!cursor.hasNext()) {
62
                        break;
63
                    }
64
                    Document current = cursor.next();
65
                    String name = current.getString("name");
66
                    // When the index completes building, it becomes `queryable`
67
                    boolean queryable = current.getBoolean("queryable");
68
                    if (name.equals(indexName) && queryable) {
69
                        doc = current;
70
                    } else {
71
                        Thread.sleep(500);
72
                    }
73
                } catch (Exception e) {
74
                    throw new RuntimeException("Failed to list search indexes: " + e);
75
                }
76
            }
77
            System.out.println(indexName + " is ready for querying.");
78
79
        } catch (Exception e) {
80
            throw new RuntimeException("Error connecting to MongoDB: " + e);
81
        }
82
    }
83
}

1
import com.mongodb.MongoException
2
import com.mongodb.client.model.SearchIndexModel
3
import com.mongodb.client.model.SearchIndexType
4
import com.mongodb.kotlin.client.coroutine.MongoClient
5
import kotlinx.coroutines.delay
6
import kotlinx.coroutines.flow.toList
7
import org.bson.Document
8
import kotlinx.coroutines.runBlocking
9
10
fun main() {
11
    // Replace the placeholder with your MongoDB deployment's connection string
12
    val uri = "<connection-string>"
13
    val mongoClient = MongoClient.create(uri)
14
    val database = mongoClient.getDatabase("sample_mflix")
15
    val collection = database.getCollection<Document>("embedded_movies")
16
    val indexName = "vector_index"
17
    val searchIndexModel = SearchIndexModel(
18
        indexName,
19
        Document(
20
            "fields",
21
            listOf(
22
                Document("type", "vector")
23
                    .append("path", "plot_embedding")
24
                    .append("numDimensions", 1536)
25
                    .append("similarity", "dotProduct")
26
                    .append("quantization", "scalar")
27
            )
28
        ),
29
        SearchIndexType.vectorSearch()
30
    )
31
32
    runBlocking {
33
        try {
34
            collection.createSearchIndexes(listOf(searchIndexModel))
35
                .collect { result ->
36
                    println("New search index named $result is building.")
37
                }
38
39
            // Polling to check if the index is queryable
40
            println("Polling to check if the index is ready. This may take up to a minute.")
41
            var isQueryable = false
42
            while (!isQueryable) {
43
                delay(5000L) // Poll every 5 seconds
44
45
                val indexes = collection.listSearchIndexes().toList()
46
                isQueryable = indexes.any { index ->
47
                    index.getString("name") == indexName && index.getBoolean("queryable")
48
                }
49
                if (isQueryable) {
50
                    println("$indexName is ready for querying.")
51
                }
52
            }
53
        } catch (me: MongoException) {
54
            System.err.println("An error occurred: $me")
55
        }
56
    }
57
    mongoClient.close()
58
}

1
import com.mongodb.MongoException
2
import com.mongodb.client.model.SearchIndexModel
3
import com.mongodb.client.model.SearchIndexType
4
import com.mongodb.kotlin.client.MongoClient
5
import org.bson.Document
6
7
fun main() {
8
    // Replace the placeholder with your MongoDB deployment's connection string
9
    val uri = "<connection-string>"
10
    val mongoClient = MongoClient.create(uri)
11
    val database = mongoClient.getDatabase("sample_mflix")
12
    val collection = database.getCollection<Document>("embedded_movies")
13
    val indexName = "vector_index"
14
    val searchIndexModel = SearchIndexModel(
15
        indexName,
16
        Document(
17
            "fields",
18
            listOf(
19
                Document("type", "vector")
20
                    .append("path", "plot_embedding")
21
                    .append("numDimensions", 1536)
22
                    .append("similarity", "dotProduct")
23
            )
24
        ),
25
        SearchIndexType.vectorSearch()
26
    )
27
28
    try {
29
        val result = collection.createSearchIndexes(
30
            listOf(searchIndexModel)
31
        )
32
        println("New search index named ${result.get(0)} is building.")
33
34
        // Polling to check if the index is queryable
35
        println("Polling to check if the index is ready. This may take up to a minute.")
36
        var isQueryable = false
37
        while (!isQueryable) {
38
            val results = collection.listSearchIndexes()
39
            results.forEach { result ->
40
                if (result.getString("name") == indexName && result.getBoolean("queryable")) {
41
                    println("$indexName is ready for querying.")
42
                    isQueryable = true
43
                } else {
44
                    Thread.sleep(5000) // Poll every 5 seconds
45
                }
46
            }
47
        }
48
    } catch (me: MongoException) {
49
        System.err.println("An error occurred: $me")
50
    }
51
    mongoClient.close()
52
}

1
const { MongoClient } = require("mongodb");
2
3
// connect to your Atlas deployment
4
const uri =  "<connectionString>";
5
6
const client = new MongoClient(uri);
7
8
async function run() {
9
   try {
10
     const database = client.db("sample_mflix");
11
     const collection = database.collection("embedded_movies");
12
    
13
     // define your Atlas Vector Search index
14
     const index = {
15
         name: "vector_index",
16
         type: "vectorSearch",
17
         definition: {
18
           "fields": [
19
             {
20
               "type": "vector",
21
               "numDimensions": 1536,
22
               "path": "plot_embedding",
23
               "similarity": "dotProduct",
24
               "quantization": "scalar"
25
             }
26
           ]
27
         }
28
     }
29
30
     // run the helper method
31
     const result = await collection.createSearchIndex(index);
32
     console.log(`New search index named ${result} is building.`);
33
34
     // wait for the index to be ready to query
35
     console.log("Polling to check if the index is ready. This may take up to a minute.")
36
     let isQueryable = false;
37
     while (!isQueryable) {
38
       const cursor = collection.listSearchIndexes();
39
       for await (const index of cursor) {
40
         if (index.name === result) {
41
           if (index.queryable) {
42
             console.log(`${result} is ready for querying.`);
43
             isQueryable = true;
44
           } else {
45
             await new Promise(resolve => setTimeout(resolve, 5000));
46
           }
47
         }
48
       }
49
     }
50
   } finally {
51
     await client.close();
52
   }
53
}
54
run().catch(console.dir);

1
<?php
2
3
require 'vendor/autoload.php';
4
5
// Replace the placeholder with your Atlas connection string
6
$uri = "<connection-string>";
7
$client = new MongoDB\Client($uri);
8
$collection = $client->sample_mflix->embedded_movies;
9
$indexName = "vector_index";
10
try {
11
    $result = $collection->createSearchIndexes(
12
        [[
13
            'name' => $indexName,
14
            'type' => 'vectorSearch',
15
            'definition' => [
16
                'fields' => [[
17
                    'type' => 'vector',
18
                    'path' => 'plot_embedding',
19
                    'numDimensions' => 1536,
20
                    'similarity' => 'dotProduct',
21
                    'quantization' => 'scalar'
22
                ]]
23
            ],
24
        ]]
25
    );
26
    echo "New search index named $result[0] is building." .PHP_EOL;
27
28
    // Polling for the index to become queryable
29
    echo "Polling to check if the index is ready. This may take up to a minute." .PHP_EOL;
30
    $isIndexQueryable = false;
31
    while (!$isIndexQueryable) {
32
        // List the search indexes
33
        $searchIndexes = $collection->listSearchIndexes();
34
        // Check if the index is present and queryable
35
        foreach ($searchIndexes as $index) {
36
            if ($index->name === $indexName) {
37
                $isIndexQueryable = $index->queryable;
38
            }
39
        }
40
        if (!$isIndexQueryable) {
41
            sleep(5); // Wait for 5 seconds before polling again
42
        }
43
    }
44
    echo "$indexName is ready for querying." .PHP_EOL;
45
}
46
catch (MongoDB\Driver\Exception\Exception $e) {
47
    print 'Error creating the index: ' .$e->getMessage() .PHP_EOL;
48
}

1
from pymongo.mongo_client import MongoClient
2
from pymongo.operations import SearchIndexModel
3
import time
4
5
# Connect to your Atlas deployment
6
uri = "<connectionString>"
7
client = MongoClient(uri)
8
9
# Access your database and collection
10
database = client["sample_mflix"]
11
collection = database["embedded_movies"]
12
13
# Create your index model, then create the search index
14
search_index_model = SearchIndexModel(
15
  definition={
16
    "fields": [
17
      {
18
        "type": "vector",
19
        "path": "plot_embedding",
20
        "numDimensions": 1536,
21
        "similarity": "dotProduct",
22
        "quantization": "scalar"
23
      }
24
    ]
25
  },
26
  name="vector_index",
27
  type="vectorSearch",
28
)
29
30
result = collection.create_search_index(model=search_index_model)
31
print("New search index named " + result + " is building.")
32
33
# Wait for initial sync to complete
34
print("Polling to check if the index is ready. This may take up to a minute.")
35
predicate=None
36
if predicate is None:
37
  predicate = lambda index: index.get("queryable") is True
38
39
while True:
40
  indices = list(collection.list_search_indexes(result))
41
  if len(indices) and predicate(indices[0]):
42
    break
43
  time.sleep(5)
44
print(result + " is ready for querying.")
45
46
client.close()

1
import org.mongodb.scala._
2
import org.mongodb.scala.model._
3
import com.mongodb.client.model.SearchIndexType
4
5
class VectorIndex {
6
  def createIndex(): Unit = {
7
    val collection =
8
      MongoClient("<connection-string>")
9
        .getDatabase("sample_mflix")
10
        .getCollection("embedded_movies")
11
    val indexName = "vector_index"
12
    val indexNameAsOption = Option(indexName)
13
    val indexDefinition = Document(
14
      "fields" -> List(
15
        Document(
16
          "type" -> "vector",
17
          "path" -> "plot_embedding",
18
          "numDimensions" -> 1536,
19
          "similarity" -> "dotProduct",
20
          "quantization" -> "scalar"
21
        )
22
      )
23
    )
24
    val indexType = Option(SearchIndexType.vectorSearch())
25
    val indexModel: SearchIndexModel = SearchIndexModel(
26
      indexNameAsOption, indexDefinition, indexType
27
    )
28
    collection.createSearchIndexes(List(indexModel)).foreach { doc => println(s"New search index named $doc is building.") }
29
    Thread.sleep(2000)
30
    println("Polling to check if the index is ready. This may take up to a minute.")
31
    var indexReady = false
32
    while (!indexReady) {
33
      val searchIndexes = collection.listSearchIndexes()
34
      searchIndexes.foreach { current =>
35
        val document = current.toBsonDocument()
36
        val name = document.get("name").asString().getValue
37
        val queryable = document.get("queryable").asBoolean().getValue
38
        if (name == indexName && queryable) {
39
          indexReady = true
40
        }
41
      }
42
      if (!indexReady) {
43
        Thread.sleep(5000)
44
      }
45
    }
46
    println(s"$indexName is ready for querying.")
47
  }
48
}

1
#include <bson/bson.h>
2
#include <mongoc/mongoc.h>
3
#include <stdio.h>
4
5
int main(void) {
6
  mongoc_client_t *client;
7
  mongoc_collection_t *collection;
8
  bson_error_t error;
9
  char database_name[] = "sample_mflix";
10
  char collection_name[] = "embedded_movies";
11
  char index_name[] = "vector_index";
12
13
  mongoc_init();
14
15
  // Replace the placeholder with your Atlas connection string
16
  client =  mongoc_client_new("<connection-string>");
17
18
  // Connect to your Atlas cluster
19
  collection = mongoc_client_get_collection (client, database_name, collection_name);
20
21
  bson_t cmd;
22
  // Create search index command.
23
  {
24
    char *cmd_str = bson_strdup_printf (
25
        BSON_STR ({
26
                    "createSearchIndexes" : "%s",
27
                    "indexes" : [{
28
                      "definition": {
29
                        "fields": [{
30
                          "type": "vector",
31
                          "path": "plot_embedding",
32
                          "numDimensions": 1536,
33
                          "similarity": "dotProduct",
34
                          "quantization": "scalar"
35
                        }]
36
                      },
37
                      "name": "%s",
38
                      "type": "vectorSearch"
39
                    }]
40
                  }),
41
        collection_name, index_name);
42
    if (!bson_init_from_json(&cmd, cmd_str, -1, &error)) {
43
      printf("Failed to initialize BSON: %s\n", error.message);
44
      bson_free(cmd_str);
45
      return 1;
46
    }
47
    bson_free (cmd_str);
48
  }
49
  if (!mongoc_collection_command_simple (collection, &cmd, NULL /* read_prefs */, NULL /* reply */, &error)) {
50
    bson_destroy (&cmd);
51
    printf ("Failed to run createSearchIndexes: %s", error.message);
52
    return 1;
53
  } else {
54
    printf ("New search index named %s is building.\n", index_name);
55
    bson_destroy (&cmd);
56
  }
57
58
  // Polling for index status
59
  printf("Polling to check if the index is ready. This may take up to a minute.\n");
60
  int queryable = 0;
61
  while (!queryable) {
62
    const char *pipeline_str = "{\"pipeline\": [{\"$listSearchIndexes\": {}}]}";
63
    bson_t pipeline;
64
    if (!bson_init_from_json(&pipeline, pipeline_str, -1, &error)) {
65
      printf("Failed to initialize pipeline BSON: %s\n", error.message);
66
      break; // Exit the loop on error
67
    }
68
    mongoc_cursor_t *cursor = mongoc_collection_aggregate(collection, MONGOC_QUERY_NONE, &pipeline, NULL, NULL);
69
    const bson_t *got;
70
    // Check if the cursor returns any documents
71
    int found_index = 0;
72
    while (mongoc_cursor_next(cursor, &got)) {
73
      bson_iter_t iter;
74
      if (bson_iter_init(&iter, got) && bson_iter_find(&iter, "name")) {
75
        const char *name = bson_iter_utf8(&iter, NULL);
76
        if (strcmp(name, index_name) == 0) {
77
          found_index = 1; // Index found
78
          bson_iter_find(&iter, "queryable");
79
          queryable = bson_iter_bool(&iter);
80
          break; // Exit the loop since we found the index
81
        }
82
      }
83
    }
84
    if (mongoc_cursor_error(cursor, &error)) {
85
      printf("Failed to run $listSearchIndexes: %s\n", error.message);
86
      break; // Exit the loop on error
87
    }
88
    if (!found_index) {
89
      printf("Index %s not found yet. Retrying...\n", index_name);
90
    }
91
    bson_destroy(&pipeline);
92
    mongoc_cursor_destroy(cursor);
93
    sleep(5); // Sleep for 5 seconds before checking again
94
  }
95
  if (queryable) {
96
    printf("%s is ready for querying.\n", index_name);
97
  } else {
98
    printf("Error occurred or index not found.\n");
99
  }
100
101
  // Cleanup
102
  mongoc_collection_destroy(collection);
103
  mongoc_client_destroy(client);
104
  mongoc_cleanup();
105
106
  return 0;
107
}

1
#include <bsoncxx/builder/basic/document.hpp>
2
#include <iostream>
3
#include <mongocxx/client.hpp>
4
#include <mongocxx/instance.hpp>
5
#include <mongocxx/search_index_view.hpp>
6
#include <mongocxx/uri.hpp>
7
#include <thread>
8
9
using bsoncxx::builder::basic::kvp;
10
using bsoncxx::builder::basic::make_array;
11
using bsoncxx::builder::basic::make_document;
12
13
int main() {
14
  try {
15
    mongocxx::instance inst{};
16
17
    // Replace the placeholder with your Atlas connection string
18
    const auto uri = mongocxx::uri{"<connection-string>"};
19
20
    // Connect to your Atlas cluster
21
    mongocxx::client conn{uri};
22
    auto db = conn["sample_mflix"];
23
    auto collection = db["embedded_movies"];
24
25
    auto siv = collection.search_indexes();
26
    std::string name = "vector_index";
27
    auto type = "vectorSearch";
28
    auto definition = make_document(
29
        kvp("fields",
30
            make_array(make_document(
31
                kvp("type", "vector"), kvp("path", "plot_embedding"),
32
                kvp("numDimensions", 1536), kvp("similarity", "dotProduct"), 
33
                kvp("quantization", "scalar")))));
34
    auto model =
35
        mongocxx::search_index_model(name, definition.view()).type(type);
36
    siv.create_one(model);
37
    std::cout << "New search index named " << name << " is building."
38
              << std::endl;
39
40
    // Polling for index status
41
    std::cout << "Polling to check if the index is ready. This may take up to "
42
                 "a minute."
43
              << std::endl;
44
    bool queryable = false;
45
    while (!queryable) {
46
      auto indexes = siv.list();
47
      for (const auto& index : indexes) {
48
        if (index["name"].get_value() == name) {
49
          queryable = index["queryable"].get_bool();
50
        }
51
      }
52
      if (!queryable) {
53
        std::this_thread::sleep_for(std::chrono::seconds(5));
54
      }
55
    }
56
    std::cout << name << " is ready for querying." << std::endl;
57
  } catch (const std::exception& e) {
58
    std::cout << "Exception: " << e.what() << std::endl;
59
  }
60
  return 0;
61
}