Skip to content

Commit 58e915b

Browse files
committed
use getattr for .func access to satisfy ty type checker
1 parent 5c206ca commit 58e915b

1 file changed

Lines changed: 6 additions & 3 deletions

File tree

backend/tests/test_langchain_tools.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -131,7 +131,8 @@ def test_retrieve_city_state_laws_returns_joined_docs(mock_rag_class):
131131
"""Test that RAG results are joined with newlines."""
132132
mock_rag_class.return_value.search.return_value = "Doc1 content\nDoc2 content"
133133

134-
result = retrieve_city_state_laws.func( # type: ignore[unresolved-attribute]
134+
_func = getattr(retrieve_city_state_laws, "func")
135+
result = _func(
135136
query="eviction notice",
136137
state=UsaState("or"),
137138
city=OregonCity("portland"),
@@ -146,7 +147,8 @@ def test_retrieve_city_state_laws_empty_results(mock_rag_class):
146147
"""Test behavior when RAG returns no documents."""
147148
mock_rag_class.return_value.search.return_value = ""
148149

149-
result = retrieve_city_state_laws.func( # type: ignore[unresolved-attribute]
150+
_func = getattr(retrieve_city_state_laws, "func")
151+
result = _func(
150152
query="obscure law",
151153
state=UsaState("or"),
152154
runtime=MagicMock(),
@@ -174,6 +176,7 @@ def test_generate_letter_empty_string(mock_get_stream_writer):
174176
mock_writer = MagicMock()
175177
mock_get_stream_writer.return_value = mock_writer
176178

177-
result = generate_letter.func(letter="") # type: ignore[unresolved-attribute]
179+
_func = getattr(generate_letter, "func")
180+
result = _func(letter="")
178181
mock_writer.assert_called_once_with({"type": "letter", "content": ""})
179182
assert result == "Letter generated successfully."

0 commit comments

Comments
 (0)