Skip to content

Commit b0452d1

Browse files
committed
fix dedicated
1 parent 1387205 commit b0452d1

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

.github/workflows/ci-astra-dev.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
name: AstraDEV
1+
name: AstraDev
22

33
on:
44
#push:

astra-db-java/src/test/java/com/datastax/astra/test/integration/dev_vectorize/AstraDevVectorizeHuggingFaceDedicatedITTest.java

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,6 @@ public String getRegion() {
4646
@Test
4747
public void testHuggingFaceDedicated() {
4848
String providerName = System.getenv("EMBEDDING_PROVIDER");
49-
String huggingFaceKey = System.getenv("EMBEDDING_API_KEY");
5049
String collectionName = "collection_hf_dedicated";
5150
FindEmbeddingProvidersResult result = getDatabaseAdmin().findEmbeddingProviders();
5251
assertThat(result.getEmbeddingProviders().get(providerName)).isNotNull();
@@ -58,14 +57,14 @@ public void testHuggingFaceDedicated() {
5857
params.put("endpointName", System.getenv("HUGGINGFACEDED_ENDPOINTNAME"));
5958
params.put("regionName", System.getenv("HUGGINGFACEDED_REGIONNAME"));
6059
params.put("cloudName", System.getenv("HUGGINGFACEDED_CLOUDNAME"));
61-
builder.vectorize(providerName, null, huggingFaceKey, params);
60+
builder.vectorize(providerName, null, null, params);
6261
Collection<Document> collection = getDatabase()
6362
.createCollection(collectionName, builder.build(), new CommandOptions<>());
6463
assertThat(getDatabase()
6564
.collectionExists(collectionName)).isTrue();
6665

6766
// Test Collection
68-
testCollection(collection, new EmbeddingAPIKeyHeaderProvider(huggingFaceKey));
67+
testCollection(collection, new EmbeddingAPIKeyHeaderProvider( System.getenv("EMBEDDING_API_KEY")));
6968

7069
// Drop Collection
7170
getDatabase().dropCollection(collectionName);

0 commit comments

Comments
 (0)