assistant_context: Fix `thread_summary_model` not getting used in Text Threads (#38859)

Umesh Yadav created 1 month ago

Closes https://github.com/zed-industries/zed/issues/37472

Release Notes:

- Fixed an issue in Text Threads where it was using `default_model` even
in case `thread_summary_model` was set.

---------

Signed-off-by: Umesh Yadav <[e-mail]>

Change summary

crates/assistant_context/src/assistant_context.rs       |  2 
crates/assistant_context/src/assistant_context_tests.rs | 13 +++++-----
2 files changed, 7 insertions(+), 8 deletions(-)

Detailed changes

crates/assistant_context/src/assistant_context.rs

@@ -2669,7 +2669,7 @@ impl AssistantContext {
     }
 
     pub fn summarize(&mut self, mut replace_old: bool, cx: &mut Context<Self>) {
-        let Some(model) = LanguageModelRegistry::read_global(cx).default_model() else {
+        let Some(model) = LanguageModelRegistry::read_global(cx).thread_summary_model() else {
             return;
         };
 

crates/assistant_context/src/assistant_context_tests.rs

@@ -1329,13 +1329,12 @@ fn setup_context_editor_with_fake_model(
     cx.update(|cx| {
         init_test(cx);
         LanguageModelRegistry::global(cx).update(cx, |registry, cx| {
-            registry.set_default_model(
-                Some(ConfiguredModel {
-                    provider: fake_provider.clone(),
-                    model: fake_model.clone(),
-                }),
-                cx,
-            )
+            let configured_model = ConfiguredModel {
+                provider: fake_provider.clone(),
+                model: fake_model.clone(),
+            };
+            registry.set_default_model(Some(configured_model.clone()), cx);
+            registry.set_thread_summary_model(Some(configured_model), cx);
         })
     });