Skip to content

Commit 4f5b666

Browse files
authored
chore: Fix langchain_core test that used a new Sequence type (#1697)
1 parent 88ce2e3 commit 4f5b666

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

tests/pytest/test_chat.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
import sys
44
from datetime import datetime
5-
from typing import Sequence, Union, cast, get_args, get_origin
5+
from typing import Union, cast, get_args, get_origin
66

77
import pytest
88

@@ -390,6 +390,9 @@ def test_as_google_message():
390390

391391
def test_as_langchain_message():
392392
from langchain_core.language_models.base import LanguageModelInput
393+
from langchain_core.language_models.base import (
394+
Sequence as LangchainSequence, # pyright: ignore[reportPrivateImportUsage]
395+
)
393396
from langchain_core.language_models.chat_models import BaseChatModel
394397
from langchain_core.messages import (
395398
AIMessage,
@@ -404,7 +407,12 @@ def test_as_langchain_message():
404407
assert BaseChatModel.invoke.__annotations__["input"] == "LanguageModelInput"
405408
assert BaseChatModel.stream.__annotations__["input"] == "LanguageModelInput"
406409

407-
assert is_type_in_union(Sequence[MessageLikeRepresentation], LanguageModelInput)
410+
assert is_type_in_union(
411+
# Use `LangchainSequence` instead of `Sequence` to avoid incorrect comparison
412+
# between `typing.Sequence` and `collections.abc.Sequence`
413+
LangchainSequence[MessageLikeRepresentation],
414+
LanguageModelInput,
415+
)
408416
assert is_type_in_union(BaseMessage, MessageLikeRepresentation)
409417

410418
assert issubclass(AIMessage, BaseMessage)

0 commit comments

Comments
 (0)