From 06aca55be6d714170dd7ad9bae64d204306ed679 Mon Sep 17 00:00:00 2001
From: Aymeric Roucher <69208727+aymeric-roucher@users.noreply.github.com>
Date: Wed, 15 Jan 2025 14:03:41 +0100
Subject: [PATCH] Fix import from ChatMessage in test_monitoring (#202)

---
 tests/test_monitoring.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_monitoring.py b/tests/test_monitoring.py
index e55afb4..bd8b148 100644
--- a/tests/test_monitoring.py
+++ b/tests/test_monitoring.py
@@ -22,7 +22,7 @@ from smolagents import (
     ToolCallingAgent,
     stream_to_gradio,
 )
-from huggingface_hub import (
+from smolagents.models import (
     ChatMessage,
     ChatMessageToolCall,
     ChatMessageToolCallDefinition,