From 31165916b05dc247e9e3817b910f0887a5af8c42 Mon Sep 17 00:00:00 2001
From: FlorentLvr <48032461+FlorentLvr@users.noreply.github.com>
Date: Fri, 3 Nov 2023 17:24:54 +0100
Subject: [PATCH] Update naas_chat_plugin.py

---
 naas_drivers/tools/naas_chat_plugin.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/naas_drivers/tools/naas_chat_plugin.py b/naas_drivers/tools/naas_chat_plugin.py
index 9239d8d6..ffc44af4 100644
--- a/naas_drivers/tools/naas_chat_plugin.py
+++ b/naas_drivers/tools/naas_chat_plugin.py
@@ -46,8 +46,9 @@ def check_tokens(self, prompt, model, limit=0.2):
         """
         # Check if the model is in the MODELS dictionary
         if model not in list(MODELS.keys()):
-            print(f"⛔ Model {model} not found. Must be one of {list(MODELS.keys())}")
-            return None, None
+            print(f"⛔ Model {model} not found. Default model: 'gpt-3.5-turbo'")
+            model = 'gpt-3.5-turbo'
+            
         # Get max tokens
         max_tokens = MODELS.get(model)