diff --git a/backend-agent/llm.py b/backend-agent/llm.py
index bc55f26..7d04adc 100644
--- a/backend-agent/llm.py
+++ b/backend-agent/llm.py
@@ -30,6 +30,7 @@
     'aicore-mistralai':
     [
         'mistralai--mistral-large-instruct',
+        'mistralai--mistral-medium-instruct',
         'mistralai--mistral-small-instruct',
     ],
     'aicore-opensource':
@@ -43,7 +44,6 @@
         'amazon--nova-pro',
         'amazon--nova-premier',
         'anthropic--claude-3-haiku',
-        'anthropic--claude-3-sonnet',
         'anthropic--claude-3-opus',
         'anthropic--claude-3.5-sonnet',
         'anthropic--claude-3.7-sonnet',