Class DummyLLMWithPipe

tests/test_generate_answer_node.py:59–68

DummyLLM that supports the pipe '|' operator. When used in a chain with a PromptTemplate, the pipe operator returns self, simulating chain composition.

class DummyLLMWithPipe:
    """DummyLLM that supports the pipe '|' operator.
    When used in a chain with a PromptTemplate, the pipe operator returns self,
    simulating chain composition."""

    def __or__(self, other):
        return self

    def __call__(self, *args, **kwargs):
        return {"content": "script single-chunk answer"}


@pytest.fixture