From c79217dd75163893c1d14e93affec17bab92440c Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Wed, 22 Jan 2025 10:04:58 -0800 Subject: [PATCH] fix: Add missing imports for FinishReasonLength and sanity_check_messages --- tests/basic/test_coder.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/basic/test_coder.py b/tests/basic/test_coder.py index 146ddcb6c..50d111f38 100644 --- a/tests/basic/test_coder.py +++ b/tests/basic/test_coder.py @@ -7,12 +7,13 @@ from unittest.mock import MagicMock, patch import git from aider.coders import Coder -from aider.coders.base_coder import UnknownEditFormat +from aider.coders.base_coder import UnknownEditFormat, FinishReasonLength from aider.dump import dump # noqa: F401 from aider.io import InputOutput from aider.models import Model from aider.repo import GitRepo from aider.utils import GitTemporaryDirectory +from aider.sendchat import sanity_check_messages class TestCoder(unittest.TestCase):