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

Remove model restriction

Browse Source
This commit is contained in:
Adithya Suresh
2025-04-04 06:17:40 +00:00
committed by Aleix Conchillo Flaqué
parent 6d6bffcabd
commit 1bfc53dd39
1 changed files with 1 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/pipecat/services/aws/llm.py
View File

@@ -561,18 +561,7 @@ class BedrockLLMService(LLMService):
"additional_model_request_fields": params.additional_model_request_fields if isinstance(params.additional_model_request_fields, dict) else {}, "additional_model_request_fields": params.additional_model_request_fields if isinstance(params.additional_model_request_fields, dict) else {},
} }
# Determine model provider from model ID logger.info(f"Using AWS Bedrock model: {model}")
self.model_provider = self._get_model_provider(model)
logger.info(f"Using AWS Bedrock model: {model} from provider: {self.model_provider}")
def _get_model_provider(self, model: str) -> str:
"""Determine the model provider from the model ID"""
if "anthropic." in model:
return "anthropic"
elif "amazon." in model:
return "amazon"
else:
raise ValueError(f"Unsupported model: {model}. Only Anthropic Claude and Amazon Nova model families are supported.")
def can_generate_metrics(self) -> bool: def can_generate_metrics(self) -> bool:
return True return True