1use std::sync::Arc;
  2
  3use ::settings::{Settings, SettingsStore};
  4use client::{Client, UserStore};
  5use collections::HashSet;
  6use gpui::{App, Context, Entity};
  7use language_model::{LanguageModelProviderId, LanguageModelRegistry};
  8use provider::deepseek::DeepSeekLanguageModelProvider;
  9
 10mod api_key;
 11pub mod provider;
 12mod settings;
 13pub mod ui;
 14
 15use crate::provider::anthropic::AnthropicLanguageModelProvider;
 16use crate::provider::bedrock::BedrockLanguageModelProvider;
 17use crate::provider::cloud::CloudLanguageModelProvider;
 18use crate::provider::copilot_chat::CopilotChatLanguageModelProvider;
 19use crate::provider::google::GoogleLanguageModelProvider;
 20use crate::provider::lmstudio::LmStudioLanguageModelProvider;
 21use crate::provider::mistral::MistralLanguageModelProvider;
 22use crate::provider::ollama::OllamaLanguageModelProvider;
 23use crate::provider::open_ai::OpenAiLanguageModelProvider;
 24use crate::provider::open_ai_compatible::OpenAiCompatibleLanguageModelProvider;
 25use crate::provider::open_router::OpenRouterLanguageModelProvider;
 26use crate::provider::vercel::VercelLanguageModelProvider;
 27use crate::provider::x_ai::XAiLanguageModelProvider;
 28pub use crate::settings::*;
 29
 30pub fn init(user_store: Entity<UserStore>, client: Arc<Client>, cx: &mut App) {
 31    crate::settings::init_settings(cx);
 32    let registry = LanguageModelRegistry::global(cx);
 33    registry.update(cx, |registry, cx| {
 34        register_language_model_providers(registry, user_store, client.clone(), cx);
 35    });
 36
 37    let mut openai_compatible_providers = AllLanguageModelSettings::get_global(cx)
 38        .openai_compatible
 39        .keys()
 40        .cloned()
 41        .collect::<HashSet<_>>();
 42
 43    registry.update(cx, |registry, cx| {
 44        register_openai_compatible_providers(
 45            registry,
 46            &HashSet::default(),
 47            &openai_compatible_providers,
 48            client.clone(),
 49            cx,
 50        );
 51    });
 52    cx.observe_global::<SettingsStore>(move |cx| {
 53        let openai_compatible_providers_new = AllLanguageModelSettings::get_global(cx)
 54            .openai_compatible
 55            .keys()
 56            .cloned()
 57            .collect::<HashSet<_>>();
 58        if openai_compatible_providers_new != openai_compatible_providers {
 59            registry.update(cx, |registry, cx| {
 60                register_openai_compatible_providers(
 61                    registry,
 62                    &openai_compatible_providers,
 63                    &openai_compatible_providers_new,
 64                    client.clone(),
 65                    cx,
 66                );
 67            });
 68            openai_compatible_providers = openai_compatible_providers_new;
 69        }
 70    })
 71    .detach();
 72}
 73
 74fn register_openai_compatible_providers(
 75    registry: &mut LanguageModelRegistry,
 76    old: &HashSet<Arc<str>>,
 77    new: &HashSet<Arc<str>>,
 78    client: Arc<Client>,
 79    cx: &mut Context<LanguageModelRegistry>,
 80) {
 81    for provider_id in old {
 82        if !new.contains(provider_id) {
 83            registry.unregister_provider(LanguageModelProviderId::from(provider_id.clone()), cx);
 84        }
 85    }
 86
 87    for provider_id in new {
 88        if !old.contains(provider_id) {
 89            registry.register_provider(
 90                OpenAiCompatibleLanguageModelProvider::new(
 91                    provider_id.clone(),
 92                    client.http_client(),
 93                    cx,
 94                ),
 95                cx,
 96            );
 97        }
 98    }
 99}
100
101fn register_language_model_providers(
102    registry: &mut LanguageModelRegistry,
103    user_store: Entity<UserStore>,
104    client: Arc<Client>,
105    cx: &mut Context<LanguageModelRegistry>,
106) {
107    registry.register_provider(
108        CloudLanguageModelProvider::new(user_store, client.clone(), cx),
109        cx,
110    );
111
112    registry.register_provider(
113        AnthropicLanguageModelProvider::new(client.http_client(), cx),
114        cx,
115    );
116    registry.register_provider(
117        OpenAiLanguageModelProvider::new(client.http_client(), cx),
118        cx,
119    );
120    registry.register_provider(
121        OllamaLanguageModelProvider::new(client.http_client(), cx),
122        cx,
123    );
124    registry.register_provider(
125        LmStudioLanguageModelProvider::new(client.http_client(), cx),
126        cx,
127    );
128    registry.register_provider(
129        DeepSeekLanguageModelProvider::new(client.http_client(), cx),
130        cx,
131    );
132    registry.register_provider(
133        GoogleLanguageModelProvider::new(client.http_client(), cx),
134        cx,
135    );
136    registry.register_provider(
137        MistralLanguageModelProvider::new(client.http_client(), cx),
138        cx,
139    );
140    registry.register_provider(
141        BedrockLanguageModelProvider::new(client.http_client(), cx),
142        cx,
143    );
144    registry.register_provider(
145        OpenRouterLanguageModelProvider::new(client.http_client(), cx),
146        cx,
147    );
148    registry.register_provider(
149        VercelLanguageModelProvider::new(client.http_client(), cx),
150        cx,
151    );
152    registry.register_provider(XAiLanguageModelProvider::new(client.http_client(), cx), cx);
153    registry.register_provider(CopilotChatLanguageModelProvider::new(cx), cx);
154}