28 lines
797 B
YAML
28 lines
797 B
YAML
name: test | bedrock | api key
|
|
|
|
on:
|
|
workflow_call:
|
|
|
|
jobs:
|
|
test-bedrock-api-key:
|
|
name: Run Bedrock API Key Test
|
|
runs-on: ubuntu-22.04
|
|
steps:
|
|
- name: Check out repository
|
|
uses: actions/checkout@v4
|
|
|
|
- name: Cognee Setup
|
|
uses: ./.github/actions/cognee_setup
|
|
with:
|
|
python-version: '3.11.x'
|
|
|
|
- name: Run Bedrock API Key Test
|
|
env:
|
|
LLM_PROVIDER: "bedrock"
|
|
LLM_API_KEY: ${{ secrets.BEDROCK_API_KEY }}
|
|
LLM_MODEL: "us.anthropic.claude-3-5-sonnet-20241022-v2:0"
|
|
AWS_REGION_NAME: "us-east-1"
|
|
EMBEDDING_PROVIDER: "bedrock"
|
|
EMBEDDING_MODEL: "amazon.titan-embed-text-v1"
|
|
EMBEDDING_DIMENSIONS: "1536"
|
|
run: poetry run python ./examples/python/simple_example.py
|