|
|
@ -1,10 +1,10 @@
|
|
|
|
import pytest
|
|
|
|
import pytest
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
from langchain.schema.runnable.base import RunnableLambda
|
|
|
|
from langchain.schema.runnable.utils import (
|
|
|
|
from langchain.schema.runnable.utils import (
|
|
|
|
get_lambda_source,
|
|
|
|
get_lambda_source,
|
|
|
|
indent_lines_after_first,
|
|
|
|
indent_lines_after_first,
|
|
|
|
)
|
|
|
|
)
|
|
|
|
from langchain.schema.runnable.base import RunnableLambda
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# Test get_lambda_source function
|
|
|
|
# Test get_lambda_source function
|
|
|
@ -21,7 +21,6 @@ def test_get_lambda_source(func, expected_source):
|
|
|
|
assert source == expected_source
|
|
|
|
assert source == expected_source
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
|
|
@pytest.mark.parametrize(
|
|
|
|
"text,prefix,expected_output",
|
|
|
|
"text,prefix,expected_output",
|
|
|
|
[
|
|
|
|
[
|
|
|
|