apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-20 15:38:09 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #1450 from JGalego/feat/bedrock-update

Browse source 
feat(bedrock): Temporary AWS credentials via env vars + supported models update
This commit is contained in:
Alexander Wu 2024-10-20 13:58:03 +08:00 • committed by GitHub
commit 8f34c746a7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 178 additions and 60 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/metagpt/provider/test_bedrock_api.py
View file

@ -3,7 +3,7 @@ import json
import pytest
from metagpt.provider.bedrock.utils import (
NOT_SUUPORT_STREAM_MODELS,
NOT_SUPPORT_STREAM_MODELS,
SUPPORT_STREAM_MODELS,
)
from metagpt.provider.bedrock_api import BedrockLLM
@ -14,7 +14,7 @@ from tests.metagpt.provider.req_resp_const import (
)
# all available model from bedrock
models = SUPPORT_STREAM_MODELS | NOT_SUUPORT_STREAM_MODELS
models = SUPPORT_STREAM_MODELS | NOT_SUPPORT_STREAM_MODELS
messages = [{"role": "user", "content": "Hi!"}]
usage = {
"prompt_tokens": 1000000,