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

anthropic: pydantic mypy plugin (#29144)

Browse Source
This commit is contained in:
Erick Friis 2025-01-13 15:32:40 -08:00 committed by GitHub
parent cdf3a17e55
commit c55af44711
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 14 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
libs/partners/anthropic/pyproject.toml
View File

@ -13,6 +13,7 @@ license = "MIT"
[tool.mypy] [tool.mypy]
disallow_untyped_defs = "True" disallow_untyped_defs = "True"
plugins = ['pydantic.mypy']
[tool.poetry.urls] [tool.poetry.urls]
"Source Code" = "https://github.com/langchain-ai/langchain/tree/master/libs/partners/anthropic" "Source Code" = "https://github.com/langchain-ai/langchain/tree/master/libs/partners/anthropic"

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

@ -153,7 +153,7 @@ async def test_abatch_tags() -> None:
async def test_async_tool_use() -> None: async def test_async_tool_use() -> None:
llm = ChatAnthropic( # type: ignore[call-arg] llm = ChatAnthropic(
model=MODEL_NAME, model=MODEL_NAME,
) )
@ -249,7 +249,7 @@ def test_system_invoke() -> None:
def test_anthropic_call() -> None: def test_anthropic_call() -> None:
"""Test valid call to anthropic.""" """Test valid call to anthropic."""
chat = ChatAnthropic(model=MODEL_NAME) # type: ignore[call-arg] chat = ChatAnthropic(model=MODEL_NAME)
message = HumanMessage(content="Hello") message = HumanMessage(content="Hello")
response = chat.invoke([message]) response = chat.invoke([message])
assert isinstance(response, AIMessage) assert isinstance(response, AIMessage)
@ -258,7 +258,7 @@ def test_anthropic_call() -> None:
def test_anthropic_generate() -> None: def test_anthropic_generate() -> None:
"""Test generate method of anthropic.""" """Test generate method of anthropic."""
chat = ChatAnthropic(model=MODEL_NAME) # type: ignore[call-arg] chat = ChatAnthropic(model=MODEL_NAME)
chat_messages: List[List[BaseMessage]] = [ chat_messages: List[List[BaseMessage]] = [
[HumanMessage(content="How many toes do dogs have?")] [HumanMessage(content="How many toes do dogs have?")]
] ]
@ -274,7 +274,7 @@ def test_anthropic_generate() -> None:
def test_anthropic_streaming() -> None: def test_anthropic_streaming() -> None:
"""Test streaming tokens from anthropic.""" """Test streaming tokens from anthropic."""
chat = ChatAnthropic(model=MODEL_NAME) # type: ignore[call-arg] chat = ChatAnthropic(model=MODEL_NAME)
message = HumanMessage(content="Hello") message = HumanMessage(content="Hello")
response = chat.stream([message]) response = chat.stream([message])
for token in response: for token in response:
@ -286,7 +286,7 @@ def test_anthropic_streaming_callback() -> None:
"""Test that streaming correctly invokes on_llm_new_token callback.""" """Test that streaming correctly invokes on_llm_new_token callback."""
callback_handler = FakeCallbackHandler() callback_handler = FakeCallbackHandler()
callback_manager = CallbackManager([callback_handler]) callback_manager = CallbackManager([callback_handler])
chat = ChatAnthropic( # type: ignore[call-arg] chat = ChatAnthropic(
model=MODEL_NAME, model=MODEL_NAME,
callback_manager=callback_manager, callback_manager=callback_manager,
verbose=True, verbose=True,
@ -302,7 +302,7 @@ async def test_anthropic_async_streaming_callback() -> None:
"""Test that streaming correctly invokes on_llm_new_token callback.""" """Test that streaming correctly invokes on_llm_new_token callback."""
callback_handler = FakeCallbackHandler() callback_handler = FakeCallbackHandler()
callback_manager = CallbackManager([callback_handler]) callback_manager = CallbackManager([callback_handler])
chat = ChatAnthropic( # type: ignore[call-arg] chat = ChatAnthropic(
model=MODEL_NAME, model=MODEL_NAME,
callback_manager=callback_manager, callback_manager=callback_manager,
verbose=True, verbose=True,
@ -318,7 +318,7 @@ async def test_anthropic_async_streaming_callback() -> None:
def test_anthropic_multimodal() -> None: def test_anthropic_multimodal() -> None:
"""Test that multimodal inputs are handled correctly.""" """Test that multimodal inputs are handled correctly."""
chat = ChatAnthropic(model=MODEL_NAME) # type: ignore[call-arg] chat = ChatAnthropic(model=MODEL_NAME)
messages: list[BaseMessage] = [ messages: list[BaseMessage] = [
HumanMessage( HumanMessage(
content=[ content=[
@ -369,7 +369,7 @@ async def test_astreaming() -> None:
def test_tool_use() -> None: def test_tool_use() -> None:
llm = ChatAnthropic(model=MODEL_NAME) # type: ignore[call-arg] llm = ChatAnthropic(model=MODEL_NAME)
llm_with_tools = llm.bind_tools( llm_with_tools = llm.bind_tools(
[ [
{ {
@ -452,7 +452,7 @@ def test_anthropic_with_empty_text_block() -> None:
"""Type the given letter.""" """Type the given letter."""
return "OK" return "OK"
model = ChatAnthropic(model="claude-3-opus-20240229", temperature=0).bind_tools( # type: ignore[call-arg] model = ChatAnthropic(model="claude-3-opus-20240229", temperature=0).bind_tools(
[type_letter] [type_letter]
) )
@ -490,7 +490,7 @@ def test_anthropic_with_empty_text_block() -> None:
def test_with_structured_output() -> None: def test_with_structured_output() -> None:
llm = ChatAnthropic( # type: ignore[call-arg] llm = ChatAnthropic(
model="claude-3-opus-20240229", model="claude-3-opus-20240229",
) )
@ -510,7 +510,7 @@ def test_with_structured_output() -> None:
def test_get_num_tokens_from_messages() -> None: def test_get_num_tokens_from_messages() -> None:
llm = ChatAnthropic(model="claude-3-5-sonnet-20241022") # type: ignore[call-arg] llm = ChatAnthropic(model="claude-3-5-sonnet-20241022")
# Test simple case # Test simple case
messages = [ messages = [
@ -571,7 +571,7 @@ class GetWeather(BaseModel):
@pytest.mark.parametrize("tool_choice", ["GetWeather", "auto", "any"]) @pytest.mark.parametrize("tool_choice", ["GetWeather", "auto", "any"])
def test_anthropic_bind_tools_tool_choice(tool_choice: str) -> None: def test_anthropic_bind_tools_tool_choice(tool_choice: str) -> None:
chat_model = ChatAnthropic( # type: ignore[call-arg] chat_model = ChatAnthropic(
model=MODEL_NAME, model=MODEL_NAME,
) )
chat_model_with_tools = chat_model.bind_tools([GetWeather], tool_choice=tool_choice) chat_model_with_tools = chat_model.bind_tools([GetWeather], tool_choice=tool_choice)
@ -583,7 +583,7 @@ def test_pdf_document_input() -> None:
url = "https://www.w3.org/WAI/ER/tests/xhtml/testfiles/resources/pdf/dummy.pdf" url = "https://www.w3.org/WAI/ER/tests/xhtml/testfiles/resources/pdf/dummy.pdf"
data = b64encode(requests.get(url).content).decode() data = b64encode(requests.get(url).content).decode()
result = ChatAnthropic(model_name=MODEL_NAME).invoke( # type: ignore[call-arg] result = ChatAnthropic(model=MODEL_NAME).invoke(
[ [
HumanMessage( HumanMessage(
[ [