pub async fn query<D: DeserializeOwned, P: Into<PartitionKeys>, C: ToString>(
    collection_name: C,
    pk: P,
    query: String,
    max_count: i32
) -> Result<Vec<D>, CosmosErrorStruct>