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

update test

Browse Source
This commit is contained in:
Chester Curme 2024-11-06 10:02:00 -05:00
parent 8662fd8c7d
commit c8d96ad346
1 changed files with 24 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
libs/partners/anthropic/tests/integration_tests/test_chat_models.py
View File

@ -547,6 +547,30 @@ def test_get_num_tokens_from_messages() -> None:
"""Get weather report for a city""" """Get weather report for a city"""
return "Sunny" return "Sunny"
messages = [
HumanMessage(content="What is the weather in SF?"),
AIMessage(
content=[
{"text": "Let's see.", "type": "text"},
{
"id": "toolu_01V6d6W32QGGSmQm4BT98EKk",
"input": {"location": "SF"},
"name": "get_weather",
"type": "tool_use",
},
],
tool_calls=[
{
"name": "get_weather",
"args": {"location": "SF"},
"id": "toolu_01V6d6W32QGGSmQm4BT98EKk",
"type": "tool_call",
},
],
),
ToolMessage(content="Sunny", tool_call_id="toolu_01V6d6W32QGGSmQm4BT98EKk"),
]
## via init ## via init
formatted_tool = convert_to_anthropic_tool(get_weather) formatted_tool = convert_to_anthropic_tool(get_weather)
llm = ChatAnthropic( llm = ChatAnthropic(