From 25b3cfb14681cbc0e3bef821e45b77102b5e344e Mon Sep 17 00:00:00 2001 From: Fabian Fischer Date: Fri, 6 Dec 2024 06:52:50 +0100 Subject: [PATCH] feat: add amazon nova inference profiles in us (#79) --- src/api/models/bedrock.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/src/api/models/bedrock.py b/src/api/models/bedrock.py index 0782324..8e32196 100644 --- a/src/api/models/bedrock.py +++ b/src/api/models/bedrock.py @@ -312,6 +312,25 @@ class BedrockModel(BaseChatModel): "tool_call": True, "stream_tool_call": True, }, + # Amazon Nova models - AWS's proprietary large language models + "us.amazon.nova-lite-v1:0": { + "system": True, # Supports system prompts for context setting + "multimodal": True, # Capable of processing both text and images + "tool_call": True, + "stream_tool_call": True, + }, + "us.amazon.nova-micro-v1:0": { + "system": True, # Supports system prompts for context setting + "multimodal": False, # Text-only model, no image processing capabilities + "tool_call": True, + "stream_tool_call": True, + }, + "us.amazon.nova-pro-v1:0": { + "system": True, # Supports system prompts for context setting + "multimodal": True, # Capable of processing both text and images + "tool_call": True, + "stream_tool_call": True, + }, } def list_models(self) -> list[str]: