Implement create collection.

ChromaDB.Client.Tests/ChromaDBAuthTests.cs

@@ -11,6 +11,6 @@ public async Task NoAuth()
 	{
 		using var httpClient = new ChromaDBHttpClient(ConfigurationOptions);
 		var client = new ChromaDBClient(ConfigurationOptions, httpClient);
-		await Assert.ThatAsync(client.Heartbeat, Throws.Nothing);	
+		await Assert.ThatAsync(client.Heartbeat, Throws.Nothing);
 	}
 }

ChromaDB.Client.Tests/ChromaDBCollectionTests.cs

@@ -0,0 +1,55 @@
+using ChromaDB.Client.Models.Requests;
+using ChromaDB.Client.Services.Implementations;
+using NUnit.Framework;
+
+namespace ChromaDB.Client.Tests;
+
+[TestFixture]
+public class ChromaDBCollectionTests : ChromaDBTestsBase
+{
+	[Test]
+	public async Task CreateCollectionWithoutMetadata()
+	{
+		using var httpClient = new ChromaDBHttpClient(ConfigurationOptions);
+		var client = new ChromaDBClient(ConfigurationOptions, httpClient);
+		var result = await client.CreateCollection(new DBCreateCollectionRequest()
+		{
+			Name = $"collection{Random.Shared.Next()}",
+		});
+		Assert.That(result.Success, Is.True);
+	}
+
+	[Test]
+	public async Task CreateCollectionWithMetadata()
+	{
+		using var httpClient = new ChromaDBHttpClient(ConfigurationOptions);
+		var client = new ChromaDBClient(ConfigurationOptions, httpClient);
+		var result = await client.CreateCollection(new DBCreateCollectionRequest()
+		{
+			Name = $"collection{Random.Shared.Next()}",
+			Metadata = new Dictionary<string, object>()
+			{
+				{ "test", "foo" },
+				{ "test2", 10 },
+			},
+		});
+		Assert.That(result.Success, Is.True);
+	}
+
+	[Test]
+	public async Task CreateCollectionAlreadyExists()
+	{
+		using var httpClient = new ChromaDBHttpClient(ConfigurationOptions);
+		var client = new ChromaDBClient(ConfigurationOptions, httpClient);
+		await client.CreateCollection(new DBCreateCollectionRequest()
+		{
+			Name = $"collection_exists",
+		});
+		var result = await client.CreateCollection(new DBCreateCollectionRequest()
+		{
+			Name = $"collection_exists",
+		});
+		Assert.That(result.Success, Is.False);
+		Assert.That(result.ReasonPhrase, Is.Not.Null.And.Not.Empty);
+	}
+}

ChromaDB.Client/Models/Collection.cs

@@ -9,6 +9,7 @@ namespace ChromaDB.Client.Models;
 [ChromaGetRoute(Endpoint = "collections?tenant={tenant}&database={database}", Source = typeof(Collection), ResponseType = typeof(List<Collection>))]
 [ChromaPostRoute(Endpoint = "collections/{collection_id}/get", Source = typeof(Collection), RequestType = typeof(CollectionGetRequest), ResponseType = typeof(CollectionEntriesResponse))]
 [ChromaPostRoute(Endpoint = "collections/{collection_id}/query", Source = typeof(Collection), RequestType = typeof(CollectionQueryRequest), ResponseType = typeof(CollectionEntriesQueryResponse))]
+[ChromaPostRoute(Endpoint = "collections", Source = typeof(Collection), RequestType = typeof(DBCreateCollectionRequest), ResponseType = typeof(Collection))]
 public class Collection
 {
 	[JsonPropertyName("id")]

ChromaDB.Client/Models/Requests/DBCreateCollectionRequest.cs

@@ -0,0 +1,12 @@
+using System.Text.Json.Serialization;
+
+namespace ChromaDB.Client.Models.Requests;
+
+public class DBCreateCollectionRequest
+{
+	[JsonPropertyName("name")]
+	public required string Name { get; init; }
+
+	[JsonPropertyName("metadata")]
+	public IDictionary<string, object>? Metadata { get; init; }
+}

ChromaDB.Client/Services/Implementations/ChromaDBClient.cs

@@ -51,4 +51,14 @@ public async Task<BaseResponse<Heartbeat>> Heartbeat()
 	{
 		return await _httpClient.Get<Heartbeat, Heartbeat>(new RequestQueryParams());
 	}
+
+	public async Task<BaseResponse<Collection>> CreateCollection(DBCreateCollectionRequest request, string? tenant = null, string? database = null)
+	{
+		tenant = tenant is not null and not [] ? tenant : _currentTenant.Name;
+		database = database is not null and not [] ? database : _currentDatabase.Name;
+		RequestQueryParams requestParams = new RequestQueryParams()
+			.Add("{tenant}", tenant)
+			.Add("{database}", database);
+		return await _httpClient.Post<Collection, DBCreateCollectionRequest, Collection>(request, requestParams);
+	}
 }

ChromaDB.Client/Services/Interfaces/IChromaDBClient.cs

@@ -1,4 +1,5 @@
 using ChromaDB.Client.Models;
+using ChromaDB.Client.Models.Requests;
 
 namespace ChromaDB.Client.Services.Interfaces;
 
@@ -7,4 +8,5 @@ public interface IChromaDBClient
 	Task<BaseResponse<Collection>> GetCollectionByName(string name, string? tenant = null, string? database = null);
 	Task<BaseResponse<List<Collection>>> GetCollections(string? tenant = null, string? database = null);
 	Task<BaseResponse<Heartbeat>> Heartbeat();
+	Task<BaseResponse<Collection>> CreateCollection(DBCreateCollectionRequest request, string? tenant = null, string? database = null);
 }

README.md

@@ -9,7 +9,8 @@ With ChromaDB.Client, you can easily connect to a ChromaDB instance, create and
 ## Features
 
 - [x] Basic connection and authentication (partially done)
-- [ ] Collection creation and deletion
+- [x] Collection creation (partially done)
+- [ ] Collection deletion
 - [x] Collection retrieval and modification (partially done)
 - [ ] Document insertion, deletion, and update
 - [ ] Document retrieval by ID or filter

Samples/ChromaDB.Client.Sample/Program.cs

@@ -9,6 +9,11 @@
 using IChromaDBHttpClient httpClient = new ChromaDBHttpClient(configOptions);
 IChromaDBClient client = new ChromaDBClient(configOptions, httpClient);
 
+BaseResponse<Collection> createCollectionResponse = await client.CreateCollection(new DBCreateCollectionRequest()
+{
+	Name = "string5",
+}, database: "test", tenant: "nedeljko");
+
 BaseResponse<List<Collection>> collections = await client.GetCollections(database: "test", tenant: "nedeljko");
 
 BaseResponse<Collection> collection1 = await client.GetCollectionByName("string5", database: "test", tenant: "nedeljko");
@@ -18,7 +23,7 @@
 BaseResponse<List<CollectionEntry>> getResponse = await string5Client.Get(new CollectionGetRequest()
 {
 	Ids = ["340a36ad-c38a-406c-be38-250174aee5a4"],
-	Include = ["metadatas", "documents", "embeddings"]
+	Include = ["metadatas", "documents", "embeddings"],
 });
 
 BaseResponse<CollectionEntriesQueryResponse> queryResponse = await string5Client.Query(new CollectionQueryRequest()