Skip to content

Commit f06f7e9

Browse files
clunclun
committed
Add Bedrock
1 parent f75d84c commit f06f7e9

File tree

2 files changed

+112
-247
lines changed

2 files changed

+112
-247
lines changed

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

Lines changed: 112 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,7 @@ jobs:
8282
# run: |
8383
# cd astra-db-java
8484
# mvn test -Dtest=com.datastax.astra.test.integration.dev.*Test
85+
8586
AWS_OPEN_AI:
8687
needs: AWS_US_WEST2_CORE
8788
runs-on: ubuntu-latest
@@ -260,8 +261,9 @@ jobs:
260261

261262

262263

263-
AWS_HF_DEDICATED:
264-
needs: AWS_HUGGING_FACE
264+
265+
GCP_OPEN_AI:
266+
needs: GCP_EUROPE_WEST4
265267
runs-on: ubuntu-latest
266268
steps:
267269
- name: Checkout repository
@@ -274,19 +276,14 @@ jobs:
274276
- name: Maven Test
275277
env:
276278
ASTRA_DB_APPLICATION_TOKEN_DEV: ${{ secrets.ASTRA_DB_APPLICATION_TOKEN_DEV }}
277-
ASTRA_CLOUD_PROVIDER_DEV: AWS
278-
ASTRA_CLOUD_REGION_DEV: us-west-2
279-
EMBEDDING_PROVIDER: huggingfaceDedicated
280-
EMBEDDING_API_KEY: ${{ secrets.HUGGINGFACEDED_API_KEY }}
281-
HUGGINGFACEDED_DIMENSION: ${{ vars.HUGGINGFACEDED_DIMENSION }}
282-
HUGGINGFACEDED_ENDPOINTNAME: ${{ vars.HUGGINGFACEDED_ENDPOINTNAME }}
283-
HUGGINGFACEDED_REGIONNAME: ${{ vars.HUGGINGFACEDED_REGIONNAME }}
284-
HUGGINGFACEDED_CLOUDNAME: ${{ vars.HUGGINGFACEDED_CLOUDNAME }}
279+
ASTRA_CLOUD_PROVIDER_DEV: GCP
280+
ASTRA_CLOUD_REGION_DEV: europe-west4
281+
EMBEDDING_PROVIDER: openai
282+
EMBEDDING_API_KEY: ${{ secrets.OPENAI_API_KEY }}
285283
run: |
286284
cd astra-db-java
287-
mvn test -Dtest=com.datastax.astra.test.integration.dev_vectorize.AstraDevVectorizeHuggingFaceDedicatedITTest
288-
289-
GCP_OPEN_AI:
285+
mvn test -Dtest=com.datastax.astra.test.integration.dev_vectorize.AstraDevVectorizeEmbeddingApiKeyITTest
286+
GCP_AZURE_OPEN_AI:
290287
needs: GCP_EUROPE_WEST4
291288
runs-on: ubuntu-latest
292289
steps:
@@ -302,12 +299,12 @@ jobs:
302299
ASTRA_DB_APPLICATION_TOKEN_DEV: ${{ secrets.ASTRA_DB_APPLICATION_TOKEN_DEV }}
303300
ASTRA_CLOUD_PROVIDER_DEV: AWS
304301
ASTRA_CLOUD_REGION_DEV: us-west-2
305-
EMBEDDING_PROVIDER: openai
306-
EMBEDDING_API_KEY: ${{ secrets.OPENAI_API_KEY }}
302+
EMBEDDING_PROVIDER: azureOpenAI
303+
EMBEDDING_API_KEY: ${{ secrets.AZURE_OPENAI_API_KEY }}
307304
run: |
308305
cd astra-db-java
309306
mvn test -Dtest=com.datastax.astra.test.integration.dev_vectorize.AstraDevVectorizeEmbeddingApiKeyITTest
310-
GCP_AZURE_OPEN_AI:
307+
GCP_HUGGING_FACE:
311308
needs: GCP_OPEN_AI
312309
runs-on: ubuntu-latest
313310
steps:
@@ -321,14 +318,15 @@ jobs:
321318
- name: Maven Test
322319
env:
323320
ASTRA_DB_APPLICATION_TOKEN_DEV: ${{ secrets.ASTRA_DB_APPLICATION_TOKEN_DEV }}
324-
ASTRA_CLOUD_PROVIDER_DEV: AWS
325-
ASTRA_CLOUD_REGION_DEV: us-west-2
326-
EMBEDDING_PROVIDER: azureOpenAI
327-
EMBEDDING_API_KEY: ${{ secrets.AZURE_OPENAI_API_KEY }}
321+
ASTRA_CLOUD_PROVIDER_DEV: GCP
322+
ASTRA_CLOUD_REGION_DEV: europe-west4
323+
EMBEDDING_PROVIDER: huggingface
324+
EMBEDDING_API_KEY: ${{ secrets.HF_API_KEY }}
328325
run: |
329326
cd astra-db-java
330-
mvn test -Dtest=com.datastax.astra.test.integration.dev_vectorize.AstraDevVectorizeEmbeddingApiKeyITTest
331-
GCP_HUGGING_FACE:
327+
mvn test -Dtest=com.datastax.astra.test.integration.dev_vectorize.AstraDevVectorizeEmbeddingApiKeyITTest
328+
329+
GCP_MISTRAL_AI:
332330
needs: GCP_AZURE_OPEN_AI
333331
runs-on: ubuntu-latest
334332
steps:
@@ -342,10 +340,97 @@ jobs:
342340
- name: Maven Test
343341
env:
344342
ASTRA_DB_APPLICATION_TOKEN_DEV: ${{ secrets.ASTRA_DB_APPLICATION_TOKEN_DEV }}
345-
ASTRA_CLOUD_PROVIDER_DEV: AWS
346-
ASTRA_CLOUD_REGION_DEV: us-west-2
347-
EMBEDDING_PROVIDER: huggingface
348-
EMBEDDING_API_KEY: ${{ secrets.HF_API_KEY }}
343+
ASTRA_CLOUD_PROVIDER_DEV: GCP
344+
ASTRA_CLOUD_REGION_DEV: europe-west4
345+
EMBEDDING_PROVIDER: mistral
346+
EMBEDDING_API_KEY: ${{ secrets.MISTRAL_API_KEY }}
349347
run: |
350348
cd astra-db-java
351-
mvn test -Dtest=com.datastax.astra.test.integration.dev_vectorize.AstraDevVectorizeEmbeddingApiKeyITTest
349+
mvn test -Dtest=com.datastax.astra.test.integration.dev_vectorize.AstraDevVectorizeEmbeddingApiKeyITTest
350+
GCP_VOYAGE_AI:
351+
needs: GCP_OPEN_AI
352+
runs-on: ubuntu-latest
353+
steps:
354+
- name: Checkout repository
355+
uses: actions/checkout@v2
356+
- name: Set up JDK
357+
uses: actions/setup-java@v2
358+
with:
359+
distribution: 'adopt'
360+
java-version: 11
361+
- name: Maven Test
362+
env:
363+
ASTRA_DB_APPLICATION_TOKEN_DEV: ${{ secrets.ASTRA_DB_APPLICATION_TOKEN_DEV }}
364+
ASTRA_CLOUD_PROVIDER_DEV: GCP
365+
ASTRA_CLOUD_REGION_DEV: europe-west4
366+
EMBEDDING_PROVIDER: voyageAI
367+
EMBEDDING_API_KEY: ${{ secrets.VOYAGE_API_KEY }}
368+
run: |
369+
cd astra-db-java
370+
mvn test -Dtest=com.datastax.astra.test.integration.dev_vectorize.AstraDevVectorizeEmbeddingApiKeyITTest
371+
GCP_UPSTAGE_AI:
372+
needs: GCP_OPEN_AI
373+
runs-on: ubuntu-latest
374+
steps:
375+
- name: Checkout repository
376+
uses: actions/checkout@v2
377+
- name: Set up JDK
378+
uses: actions/setup-java@v2
379+
with:
380+
distribution: 'adopt'
381+
java-version: 11
382+
- name: Maven Test
383+
env:
384+
ASTRA_DB_APPLICATION_TOKEN_DEV: ${{ secrets.ASTRA_DB_APPLICATION_TOKEN_DEV }}
385+
ASTRA_CLOUD_PROVIDER_DEV: GCP
386+
ASTRA_CLOUD_REGION_DEV: europe-west4
387+
EMBEDDING_PROVIDER: upstageAI
388+
EMBEDDING_API_KEY: ${{ secrets.UPSTAGE_API_KEY }}
389+
run: |
390+
cd astra-db-java
391+
mvn test -Dtest=com.datastax.astra.test.integration.dev_vectorize.AstraDevVectorizeEmbeddingApiKeyITTest
392+
393+
394+
395+
GCP_JINA_AI:
396+
needs: GCP_UPSTAGE_AI
397+
runs-on: ubuntu-latest
398+
steps:
399+
- name: Checkout repository
400+
uses: actions/checkout@v2
401+
- name: Set up JDK
402+
uses: actions/setup-java@v2
403+
with:
404+
distribution: 'adopt'
405+
java-version: 11
406+
- name: Maven Test
407+
env:
408+
ASTRA_DB_APPLICATION_TOKEN_DEV: ${{ secrets.ASTRA_DB_APPLICATION_TOKEN_DEV }}
409+
ASTRA_CLOUD_PROVIDER_DEV: GCP
410+
ASTRA_CLOUD_REGION_DEV: europe-west4
411+
EMBEDDING_PROVIDER: jinaAI
412+
EMBEDDING_API_KEY: ${{ secrets.JINA_API_KEY }}
413+
run: |
414+
cd astra-db-java
415+
mvn test -Dtest=com.datastax.astra.test.integration.dev_vectorize.AstraDevVectorizeEmbeddingApiKeyITTest
416+
GCP_NVIDIA:
417+
needs: GCP_VOYAGE_AI
418+
runs-on: ubuntu-latest
419+
steps:
420+
- name: Checkout repository
421+
uses: actions/checkout@v2
422+
- name: Set up JDK
423+
uses: actions/setup-java@v2
424+
with:
425+
distribution: 'adopt'
426+
java-version: 11
427+
- name: Maven Test
428+
env:
429+
ASTRA_DB_APPLICATION_TOKEN_DEV: ${{ secrets.ASTRA_DB_APPLICATION_TOKEN_DEV }}
430+
ASTRA_CLOUD_PROVIDER_DEV: GCP
431+
ASTRA_CLOUD_REGION_DEV: europe-west4
432+
EMBEDDING_PROVIDER: nvidia
433+
EMBEDDING_API_KEY: nvidia # required for github actions, not used
434+
run: |
435+
cd astra-db-java
436+
mvn test -Dtest=com.datastax.astra.test.integration.dev_vectorize.AstraDevVectorizeEmbeddingApiKeyITTest

0 commit comments

Comments
 (0)