github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-19 13:23:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

BUGFIX: anthropic models on bedrock (#13629)

Browse Source 
Introduced in #13403
This commit is contained in:
Erick Friis 2023-11-21 17:40:29 -08:00 committed by GitHub
parent 04bddbaba4
commit 131db4ba68
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
libs/langchain/tests/unit_tests/chat_models/test_bedrock.py
View File

@ -1,8 +1,10 @@
"""Test Anthropic Chat API wrapper.""" """Test Anthropic Chat API wrapper."""
from typing import List from typing import List
from unittest.mock import MagicMock
import pytest import pytest
from langchain.chat_models import BedrockChat
from langchain.chat_models.meta import convert_messages_to_prompt_llama from langchain.chat_models.meta import convert_messages_to_prompt_llama
from langchain.schema import AIMessage, BaseMessage, HumanMessage, SystemMessage from langchain.schema import AIMessage, BaseMessage, HumanMessage, SystemMessage
@ -28,3 +30,19 @@ from langchain.schema import AIMessage, BaseMessage, HumanMessage, SystemMessage
def test_formatting(messages: List[BaseMessage], expected: str) -> None: def test_formatting(messages: List[BaseMessage], expected: str) -> None:
result = convert_messages_to_prompt_llama(messages) result = convert_messages_to_prompt_llama(messages)
assert result == expected assert result == expected
def test_anthropic_bedrock() -> None:
client = MagicMock()
respbody = MagicMock(
read=MagicMock(
return_value=MagicMock(
decode=MagicMock(return_value=b'{"completion":"Hi back"}')
)
)
)
client.invoke_model.return_value = {"body": respbody}
model = BedrockChat(model_id="anthropic.claude-v2", client=client)
# should not throw an error
model.invoke("hello there")