mirror of
https://github.com/tiennm99/litellm.git
synced 2026-08-07 16:21:45 +00:00
Add cachePoint support for assistant and tool messages in Bedrock
This commit is contained in:
@@ -25,6 +25,7 @@ from litellm.llms.base_llm.chat.transformation import BaseConfig, BaseLLMExcepti
|
||||
from litellm.types.llms.bedrock import *
|
||||
from litellm.types.llms.openai import (
|
||||
AllMessageValues,
|
||||
ChatCompletionAssistantMessage,
|
||||
ChatCompletionRedactedThinkingBlock,
|
||||
ChatCompletionResponseMessage,
|
||||
ChatCompletionSystemMessage,
|
||||
@@ -505,6 +506,7 @@ class AmazonConverseConfig(BaseConfig):
|
||||
OpenAIMessageContentListBlock,
|
||||
ChatCompletionUserMessage,
|
||||
ChatCompletionSystemMessage,
|
||||
ChatCompletionAssistantMessage,
|
||||
],
|
||||
block_type: Literal["system"],
|
||||
) -> Optional[SystemContentBlock]:
|
||||
@@ -517,6 +519,7 @@ class AmazonConverseConfig(BaseConfig):
|
||||
OpenAIMessageContentListBlock,
|
||||
ChatCompletionUserMessage,
|
||||
ChatCompletionSystemMessage,
|
||||
ChatCompletionAssistantMessage,
|
||||
],
|
||||
block_type: Literal["content_block"],
|
||||
) -> Optional[ContentBlock]:
|
||||
@@ -528,6 +531,7 @@ class AmazonConverseConfig(BaseConfig):
|
||||
OpenAIMessageContentListBlock,
|
||||
ChatCompletionUserMessage,
|
||||
ChatCompletionSystemMessage,
|
||||
ChatCompletionAssistantMessage,
|
||||
],
|
||||
block_type: Literal["system", "content_block"],
|
||||
) -> Optional[Union[SystemContentBlock, ContentBlock]]:
|
||||
|
||||
Reference in New Issue
Block a user