wx44wx/pipecat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

aws: use AWS prefix for all services

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2025-05-06 14:52:29 -07:00
parent ce1a72850b
commit a8405649d0
6 changed files with 74 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tests/test_context_aggregators.py
View File

@@ -41,9 +41,9 @@ from pipecat.services.anthropic.llm import (
AnthropicUserContextAggregator,
)
from pipecat.services.aws.llm import (
BedrockAssistantContextAggregator,
BedrockLLMContext,
BedrockUserContextAggregator,
AWSBedrockAssistantContextAggregator,
AWSBedrockLLMContext,
AWSBedrockUserContextAggregator,
)
from pipecat.services.google.llm import (
GoogleAssistantContextAggregator,
@@ -714,11 +714,11 @@ class TestAnthropicAssistantContextAggregator(
#
class TestBedrockUserContextAggregator(
class TestAWSBedrockUserContextAggregator(
BaseTestUserContextAggregator, unittest.IsolatedAsyncioTestCase
):
CONTEXT_CLASS = BedrockLLMContext
AGGREGATOR_CLASS = BedrockUserContextAggregator
CONTEXT_CLASS = AWSBedrockLLMContext
AGGREGATOR_CLASS = AWSBedrockUserContextAggregator
def check_message_multi_content(
self, context: OpenAILLMContext, content_index: int, index: int, content: str
@@ -727,11 +727,11 @@ class TestBedrockUserContextAggregator(
assert messages["content"][index]["text"] == content
class TestBedrockAssistantContextAggregator(
class TestAWSBedrockAssistantContextAggregator(
BaseTestAssistantContextAggreagator, unittest.IsolatedAsyncioTestCase
):
CONTEXT_CLASS = BedrockLLMContext
AGGREGATOR_CLASS = BedrockAssistantContextAggregator
CONTEXT_CLASS = AWSBedrockLLMContext
AGGREGATOR_CLASS = AWSBedrockAssistantContextAggregator
EXPECTED_CONTEXT_FRAMES = [OpenAILLMContextFrame, OpenAILLMContextAssistantTimestampFrame]
def check_message_multi_content(