gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: fix model names and test names

Browse source
This commit is contained in:
Andrej Milicevic 2025-11-26 13:35:56 +01:00
parent 02b9fa485c
commit 700362a233
2 changed files with 8 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
.github/workflows/test_llms.yml vendored
View file

@ -98,11 +98,11 @@ jobs:
python-version: '3.11.x' python-version: '3.11.x'
extra-dependencies: "aws" extra-dependencies: "aws"
- name: Run Bedrock API Key Test - name: Run Bedrock API Key Simple Example
env: env:
LLM_PROVIDER: "bedrock" LLM_PROVIDER: "bedrock"
LLM_API_KEY: ${{ secrets.BEDROCK_API_KEY }} LLM_API_KEY: ${{ secrets.BEDROCK_API_KEY }}
LLM_MODEL: "eu.amazon.nova-lite-v1:0" LLM_MODEL: "eu.anthropic.claude-sonnet-4-5-20250929-v1:0"
LLM_MAX_TOKENS: "16384" LLM_MAX_TOKENS: "16384"
AWS_REGION_NAME: "eu-west-1" AWS_REGION_NAME: "eu-west-1"
EMBEDDING_PROVIDER: "bedrock" EMBEDDING_PROVIDER: "bedrock"
@ -125,10 +125,10 @@ jobs:
python-version: '3.11.x' python-version: '3.11.x'
extra-dependencies: "aws" extra-dependencies: "aws"
- name: Run Bedrock API Key Test - name: Run Bedrock AWS Credentials Simple Example
env: env:
LLM_PROVIDER: "bedrock" LLM_PROVIDER: "bedrock"
LLM_MODEL: "eu.amazon.nova-lite-v1:0" LLM_MODEL: "eu.anthropic.claude-sonnet-4-5-20250929-v1:0"
LLM_MAX_TOKENS: "16384" LLM_MAX_TOKENS: "16384"
AWS_REGION_NAME: "eu-west-1" AWS_REGION_NAME: "eu-west-1"
AWS_ACCESS_KEY_ID: ${{ secrets.AWS_ACCESS_KEY_ID }} AWS_ACCESS_KEY_ID: ${{ secrets.AWS_ACCESS_KEY_ID }}
@ -162,10 +162,11 @@ jobs:
aws_secret_access_key = ${{ secrets.AWS_SECRET_ACCESS_KEY }} aws_secret_access_key = ${{ secrets.AWS_SECRET_ACCESS_KEY }}
EOF EOF
- name: Run Bedrock AWS Profile Test - name: Run Bedrock AWS Profile Simple Example
env: env:
LLM_PROVIDER: "bedrock" LLM_PROVIDER: "bedrock"
LLM_MODEL: "eu.amazon.nova-lite-v1:0" LLM_MODEL: "eu.anthropic.claude-sonnet-4-5-20250929-v1:0"
LLM_MAX_TOKENS: "16384"
AWS_PROFILE_NAME: "bedrock-test" AWS_PROFILE_NAME: "bedrock-test"
AWS_REGION_NAME: "eu-west-1" AWS_REGION_NAME: "eu-west-1"
EMBEDDING_PROVIDER: "bedrock" EMBEDDING_PROVIDER: "bedrock"

2
cognee/modules/settings/get_settings.py
View file

@ -164,7 +164,7 @@ def get_settings() -> SettingsDict:
], ],
"bedrock": [ "bedrock": [
{ {
"value": "eu.anthropic.claude-haiku-4-5-20251001-v1:0", "value": "eu.anthropic.claude-sonnet-4-5-20250929-v1:0",
"label": "Claude 4.5 Sonnet", "label": "Claude 4.5 Sonnet",
}, },
{ {