[ENH]: (Rust client): fix database resolution (#5692)

Author: codetheweb

rust/api-types/src/user_identity.rs

@@ -1,9 +1,10 @@
 use serde::{Deserialize, Serialize};
+use std::collections::HashSet;
 
 #[derive(Serialize, Deserialize, Debug)]
 #[cfg_attr(feature = "utoipa", derive(utoipa::ToSchema))]
 pub struct GetUserIdentityResponse {
     pub user_id: String,
     pub tenant: String,
-    pub databases: Vec<String>,
+    pub databases: HashSet<String>,
 }

rust/chroma/src/client/chroma_http_client.rs

@@ -349,7 +349,7 @@ impl ChromaHttpClient {
             ));
         }
 
-        let database_name = identity.databases.first().ok_or_else(|| {
+        let database_name = identity.databases.into_iter().next().ok_or_else(|| {
             ChromaClientError::CouldNotResolveDatabaseId(
                 "Client has access to no databases".to_string(),
             )

rust/frontend/src/auth/mod.rs

@@ -1,3 +1,4 @@
+use std::collections::HashSet;
 use std::fmt::{Display, Formatter};
 use std::future::{ready, Future};
 use std::pin::Pin;
@@ -122,7 +123,7 @@ fn default_identity() -> GetUserIdentityResponse {
     GetUserIdentityResponse {
         user_id: String::new(),
         tenant: "default_tenant".to_string(),
-        databases: vec!["default_database".to_string()],
+        databases: HashSet::from(["default_database".to_string()]),
     }
 }