video-image-production-build/tests/support/fake_models.py

from __future__ import annotations

from typing import Any, Sequence

from langchain_core.language_models.fake_chat_models import FakeMessagesListChatModel
from langchain_core.tools import BaseTool


class ToolAwareFakeChatModel(FakeMessagesListChatModel):
    """FakeMessagesListChatModel 加上 create_agent 所需的 bind_tools。"""

    def bind_tools(
        self,
        tools: Sequence[BaseTool | dict[str, Any]],
        *,
        tool_choice: str | None = None,
        **kwargs: Any,
    ) -> ToolAwareFakeChatModel:
        del tools, tool_choice, kwargs
        return self