fix dedicated

datastax · Jul 10, 2024 · b0452d1 · b0452d1
1 parent 1387205
commit b0452d1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/.github/workflows/ci-astra-dev.yml b/.github/workflows/ci-astra-dev.yml
@@ -1,4 +1,4 @@
-name: AstraDEV
+name: AstraDev
 
 on:
   #push:

diff --git a/...tax/astra/test/integration/dev_vectorize/AstraDevVectorizeHuggingFaceDedicatedITTest.java b/...tax/astra/test/integration/dev_vectorize/AstraDevVectorizeHuggingFaceDedicatedITTest.java
@@ -46,7 +46,6 @@ public String getRegion() {
     @Test
     public void testHuggingFaceDedicated() {
         String providerName   = System.getenv("EMBEDDING_PROVIDER");
-        String huggingFaceKey = System.getenv("EMBEDDING_API_KEY");
         String collectionName = "collection_hf_dedicated";
         FindEmbeddingProvidersResult result = getDatabaseAdmin().findEmbeddingProviders();
         assertThat(result.getEmbeddingProviders().get(providerName)).isNotNull();
@@ -58,14 +57,14 @@ public void testHuggingFaceDedicated() {
         params.put("endpointName",  System.getenv("HUGGINGFACEDED_ENDPOINTNAME"));
         params.put("regionName", System.getenv("HUGGINGFACEDED_REGIONNAME"));
         params.put("cloudName",  System.getenv("HUGGINGFACEDED_CLOUDNAME"));
-        builder.vectorize(providerName, null, huggingFaceKey, params);
+        builder.vectorize(providerName, null, null, params);
         Collection<Document> collection = getDatabase()
                 .createCollection(collectionName, builder.build(), new CommandOptions<>());
         assertThat(getDatabase()
                 .collectionExists(collectionName)).isTrue();
 
         // Test Collection
-        testCollection(collection, new EmbeddingAPIKeyHeaderProvider(huggingFaceKey));
+        testCollection(collection, new EmbeddingAPIKeyHeaderProvider( System.getenv("EMBEDDING_API_KEY")));
 
         // Drop Collection
         getDatabase().dropCollection(collectionName);