From 2c5feddad50833058657ed7fa00a984c15650cbd Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <aider@paulg.org>
Date: Tue, 19 Nov 2024 10:04:41 -0800
Subject: [PATCH] test: Add imports for ModelSettings and MODEL_SETTINGS

---
 tests/basic/test_models.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tests/basic/test_models.py b/tests/basic/test_models.py
index f9e79e16a..9c1035d84 100644
--- a/tests/basic/test_models.py
+++ b/tests/basic/test_models.py
@@ -1,9 +1,11 @@
 import unittest
-from unittest.mock import ANY, MagicMock, patch
+from unittest.mock import ANY, MagicMock, patch, call
 
 from aider.models import (
     Model,
     ModelInfoManager,
+    ModelSettings,
+    MODEL_SETTINGS,
     sanity_check_model,
     sanity_check_models,
 )