Skip to content

Commit 27e9cb6

Browse files
committed
Pass linting test
1 parent f89993d commit 27e9cb6

File tree

1 file changed

+6
-3
lines changed
  • libs/core/langchain_core/messages

1 file changed

+6
-3
lines changed

libs/core/langchain_core/messages/utils.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -939,22 +939,25 @@ def list_token_counter(messages: Sequence[BaseMessage]) -> int:
939939
msg = f"Unrecognized {strategy=}. Supported strategies are 'last' and 'first'."
940940
raise ValueError(msg)
941941

942+
942943
@overload
943944
def convert_to_openai_messages(
944-
messages: MessageLikeRepresentation,
945+
messages: BaseMessage | dict | str,
945946
*,
946947
text_format: Literal["string", "block"] = "string",
947948
) -> dict: ...
948949

950+
949951
@overload
950952
def convert_to_openai_messages(
951-
messages: Sequence[MessageLikeRepresentation],
953+
messages: Sequence[BaseMessage | list[str] | tuple[str, str] | dict[str, Any]],
952954
*,
953955
text_format: Literal["string", "block"] = "string",
954956
) -> list[dict]: ...
955957

958+
956959
def convert_to_openai_messages(
957-
messages: Union[MessageLikeRepresentation, Sequence[MessageLikeRepresentation]],
960+
messages: Union[BaseMessage, dict, str, Sequence[MessageLikeRepresentation]],
958961
*,
959962
text_format: Literal["string", "block"] = "string",
960963
) -> Union[dict, list[dict]]:

0 commit comments

Comments
 (0)