Lead the user to the AI settings when no provider is set up

policies/WindowsTerminal.admx

@@ -9,6 +9,7 @@
   <supportedOn>
     <definitions>
       <definition name="SUPPORTED_WindowsTerminal_1_21" displayName="$(string.SUPPORTED_WindowsTerminal_1_21)" />
+      <definition name="SUPPORTED_WindowsTerminalCanary_1_23" displayName="$(string.SUPPORTED_WindowsTerminalCanary_1_23)" />
     </definitions>
   </supportedOn>
   <categories>
@@ -24,5 +25,12 @@
         <multiText id="DisabledProfileSources" valueName="DisabledProfileSources" required="true" />
       </elements>
     </policy>
+    <policy name="EnabledLMProviders" class="Both" displayName="$(string.EnabledLMProviders)" explainText="$(string.EnabledLMProvidersText)" presentation="$(presentation.EnabledLMProviders)" key="Software\Policies\Microsoft\Windows Terminal">
+      <parentCategory ref="WindowsTerminal" />
+      <supportedOn ref="SUPPORTED_WindowsTerminalCanary_1_23" />
+      <elements>
+        <multiText id="EnabledLMProviders" valueName="EnabledLMProviders" required="false" />
+      </elements>
+    </policy>
   </policies>
 </policyDefinitions>

policies/en-US/WindowsTerminal.adml

@@ -7,6 +7,7 @@
     <stringTable>
       <string id="WindowsTerminal">Windows Terminal</string>
       <string id="SUPPORTED_WindowsTerminal_1_21">At least Windows Terminal 1.21</string>
+      <string id="SUPPORTED_WindowsTerminalCanary_1_23">At least Windows Terminal Canary 1.23</string>
       <string id="DisabledProfileSources">Disabled Profile Sources</string>
       <string id="DisabledProfileSourcesText">Profiles will not be generated from any sources listed here. Source names can be arbitrary strings. Potential candidates can be found as the "source" property on profile definitions in Windows Terminal's settings.json file.
 
@@ -18,11 +19,25 @@ Common sources are:
 For instance, setting this policy to Windows.Terminal.Wsl will disable the builtin WSL integration of Windows Terminal.
 
 Note: Existing profiles will disappear from Windows Terminal after adding their source to this policy.</string>
+      <string id="EnabledLMProviders">Enabled Language Model/AI Providers</string>
+      <string id="EnabledLMProvidersText">The listed Language Models/AI Providers will be available for use in Terminal Chat.
+
+Enabling the policy but leaving the list empty disallows all providers and therefore disables the Terminal Chat feature completely.
+
+Common providers are:
+- AzureOpenAI
+- OpenAI
+- GitHubCopilot
+
+For instance, setting this policy to GitHubCopilot will allow the use of GitHubCopilot in Terminal Chat.</string>
     </stringTable>
     <presentationTable>
       <presentation id="DisabledProfileSources">
         <multiTextBox refId="DisabledProfileSources">List of disabled sources (one per line)</multiTextBox>
       </presentation>
+      <presentation id="EnabledLMProviders">
+        <multiTextBox refId="EnabledLMProviders">List of enabled Language Model/AI Providers (one per line)</multiTextBox>
+      </presentation>
     </presentationTable>
   </resources>
 </policyDefinitionResources>

src/cascadia/QueryExtension/ExtensionPalette.cpp

@@ -99,17 +99,23 @@ namespace winrt::Microsoft::Terminal::Query::Extension::implementation
         _lmProvider = lmProvider;
         _clearAndInitializeMessages(nullptr, nullptr);
 
-        const auto brandingData = _lmProvider.BrandingData();
-        const auto headerIconPath = brandingData.HeaderIconPath().empty() ? terminalChatLogoPath : brandingData.HeaderIconPath();
+        const auto brandingData = _lmProvider ? _lmProvider.BrandingData() : nullptr;
+        const auto headerIconPath = (!brandingData || brandingData.HeaderIconPath().empty()) ? terminalChatLogoPath : brandingData.HeaderIconPath();
         Windows::Foundation::Uri headerImageSourceUri{ headerIconPath };
         Media::Imaging::BitmapImage headerImageSource{ headerImageSourceUri };
         HeaderIcon().Source(headerImageSource);
 
-        const auto headerText = brandingData.HeaderText().empty() ? RS_(L"IntroText/Text") : brandingData.HeaderText();
+        const auto headerText = (!brandingData || brandingData.HeaderText().empty()) ? RS_(L"IntroText/Text") : brandingData.HeaderText();
         QueryIntro().Text(headerText);
 
-        const auto subheaderText = brandingData.SubheaderText().empty() ? RS_(L"TitleSubheader/Text") : brandingData.SubheaderText();
+        const auto subheaderText = (!brandingData || brandingData.SubheaderText().empty()) ? RS_(L"TitleSubheader/Text") : brandingData.SubheaderText();
         TitleSubheader().Text(subheaderText);
+        _PropertyChangedHandlers(*this, Windows::UI::Xaml::Data::PropertyChangedEventArgs{ L"ProviderExists" });
+    }
+
+    bool ExtensionPalette::ProviderExists() const noexcept
+    {
+        return _lmProvider != nullptr;
     }
 
     void ExtensionPalette::IconPath(const winrt::hstring& iconPath)
@@ -234,9 +240,9 @@ namespace winrt::Microsoft::Terminal::Query::Extension::implementation
             }
         }
 
-        const auto brandingData = _lmProvider.BrandingData();
+        const auto brandingData = _lmProvider ? _lmProvider.BrandingData() : nullptr;
         const auto responseAttribution = response.ResponseAttribution().empty() ? _ProfileName : response.ResponseAttribution();
-        const auto badgeUriPath = _lmProvider ? brandingData.BadgeIconPath() : winrt::hstring{};
+        const auto badgeUriPath = brandingData ? brandingData.BadgeIconPath() : L"";
         const auto responseGroupedMessages = winrt::make<GroupedChatMessages>(time, false, winrt::single_threaded_vector(std::move(messageParts)), responseAttribution, badgeUriPath);
         _messages.Append(responseGroupedMessages);
 
@@ -261,10 +267,12 @@ namespace winrt::Microsoft::Terminal::Query::Extension::implementation
         {
             const auto context = winrt::make<TerminalContext>(_ActiveCommandline);
             _lmProvider.SetContext(std::move(context));
+            _queryBox().Focus(FocusState::Programmatic);
+        }
+        else
+        {
+            SetUpProviderButton().Focus(FocusState::Programmatic);
         }
-
-        // Give the palette focus
-        _queryBox().Focus(FocusState::Programmatic);
     }
 
     void ExtensionPalette::_clearAndInitializeMessages(const Windows::Foundation::IInspectable& /*sender*/,
@@ -459,6 +467,13 @@ namespace winrt::Microsoft::Terminal::Query::Extension::implementation
         }
     }
 
+    void ExtensionPalette::_setUpAIProviderInSettings(const Windows::Foundation::IInspectable& /*sender*/,
+                                                      const Windows::UI::Xaml::RoutedEventArgs& /*args*/)
+    {
+        _SetUpProviderInSettingsRequestedHandlers(nullptr, nullptr);
+        _close();
+    }
+
     // Method Description:
     // - Dismiss the query palette. This will:
     //   * clear all the current text in the input box

src/cascadia/QueryExtension/ExtensionPalette.h

@@ -13,6 +13,7 @@ namespace winrt::Microsoft::Terminal::Query::Extension::implementation
     {
         ExtensionPalette();
         void SetProvider(const Extension::ILMProvider lmProvider);
+        bool ProviderExists() const noexcept;
 
         // We don't use the winrt_property macro here because we just need the setter
         void IconPath(const winrt::hstring& iconPath);
@@ -29,6 +30,7 @@ namespace winrt::Microsoft::Terminal::Query::Extension::implementation
         TYPED_EVENT(ActiveControlInfoRequested, winrt::Microsoft::Terminal::Query::Extension::ExtensionPalette, Windows::Foundation::IInspectable);
         TYPED_EVENT(InputSuggestionRequested, winrt::Microsoft::Terminal::Query::Extension::ExtensionPalette, winrt::hstring);
         TYPED_EVENT(ExportChatHistoryRequested, winrt::Microsoft::Terminal::Query::Extension::ExtensionPalette, winrt::hstring);
+        TYPED_EVENT(SetUpProviderInSettingsRequested, winrt::Microsoft::Terminal::Query::Extension::ExtensionPalette, Windows::Foundation::IInspectable);
 
     private:
         friend struct ExtensionPaletteT<ExtensionPalette>; // for Xaml to bind events
@@ -54,6 +56,7 @@ namespace winrt::Microsoft::Terminal::Query::Extension::implementation
         void _lostFocusHandler(const Windows::Foundation::IInspectable& sender, const Windows::UI::Xaml::RoutedEventArgs& args);
         void _previewKeyDownHandler(const Windows::Foundation::IInspectable& sender,
                                     const Windows::UI::Xaml::Input::KeyRoutedEventArgs& e);
+        void _setUpAIProviderInSettings(const Windows::Foundation::IInspectable& sender, const Windows::UI::Xaml::RoutedEventArgs& args);
 
         void _close();
     };

src/cascadia/QueryExtension/ExtensionPalette.idl

@@ -26,6 +26,7 @@ namespace Microsoft.Terminal.Query.Extension
     {
         ExtensionPalette();
         void SetProvider(ILMProvider lmProvider);
+        Boolean ProviderExists { get; };
 
         String ControlName { get; };
         String QueryBoxPlaceholderText { get; };
@@ -40,5 +41,6 @@ namespace Microsoft.Terminal.Query.Extension
         event Windows.Foundation.TypedEventHandler<ExtensionPalette, IInspectable> ActiveControlInfoRequested;
         event Windows.Foundation.TypedEventHandler<ExtensionPalette, String> InputSuggestionRequested;
         event Windows.Foundation.TypedEventHandler<ExtensionPalette, String> ExportChatHistoryRequested;
+        event Windows.Foundation.TypedEventHandler<ExtensionPalette, IInspectable> SetUpProviderInSettingsRequested;
     }
 }

src/cascadia/QueryExtension/ExtensionPalette.xaml

@@ -8,6 +8,7 @@
              xmlns:d="http://schemas.microsoft.com/expression/blend/2008"
              xmlns:local="using:Microsoft.Terminal.Query.Extension"
              xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006"
+             xmlns:mtu="using:Microsoft.Terminal.UI"
              xmlns:mux="using:Microsoft.UI.Xaml.Controls"
              VerticalAlignment="Stretch"
              AllowFocusOnInteraction="True"
@@ -409,7 +410,26 @@
                      IsSpellCheckEnabled="False"
                      PlaceholderText="{x:Bind QueryBoxPlaceholderText}"
                      Text=""
-                     TextWrapping="Wrap" />
+                     TextWrapping="Wrap"
+                     Visibility="{x:Bind ProviderExists, Mode=OneWay}" />
+            <Grid Grid.Row="1"
+                  HorizontalAlignment="Center"
+                  RowSpacing="8"
+                  Visibility="{x:Bind mtu:Converters.InvertedBooleanToVisibility(ProviderExists), Mode=OneWay}">
+                <Grid.RowDefinitions>
+                    <RowDefinition Height="Auto" />
+                    <RowDefinition Height="Auto" />
+                </Grid.RowDefinitions>
+                <TextBlock x:Uid="SetUpProviderDisclaimer"
+                           Grid.Row="0"
+                           HorizontalAlignment="Center" />
+                <Button x:Name="SetUpProviderButton"
+                        Grid.Row="1"
+                        HorizontalAlignment="Center"
+                        Click="_setUpAIProviderInSettings">
+                    <TextBlock x:Uid="SetUpProviderButton" />
+                </Button>
+            </Grid>
             <TextBlock Grid.Row="2"
                        Margin="20,0,0,16"
                        FontSize="10">

src/cascadia/QueryExtension/Resources/en-US/Resources.resw

@@ -177,6 +177,14 @@
     <value>Assistant</value>
     <comment>A string to represent the section that the chat assistant typed, presented when the user exports the chat history to a file</comment>
   </data>
+  <data name="SetUpProviderDisclaimer.Text" xml:space="preserve">
+    <value>You have not set up an AI provider yet! Set one up in the settings</value>
+    <comment>Disclaimer shown to the user when they open up Terminal Chat without having set up a provider yet.</comment>
+  </data>
+  <data name="SetUpProviderButton.Text" xml:space="preserve">
+    <value>Set up AI provider</value>
+    <comment>Description of the button that sends the user to the settings page where they can set up a provider.</comment>
+  </data>
   <data name="GithubCopilot_HeaderText" xml:space="preserve">
     <value>GitHub Copilot</value>
     <comment>The header for Terminal Chat when GitHub Copilot is the connected service provider</comment>

src/cascadia/TerminalApp/AppActionHandlers.cpp

@@ -662,18 +662,23 @@ namespace winrt::TerminalApp::implementation
     void TerminalPage::_HandleToggleAIChat(const IInspectable& /*sender*/,
                                            const ActionEventArgs& args)
     {
-        if (ExtensionPresenter().Visibility() == Visibility::Collapsed)
-        {
-            _loadQueryExtension();
-            ExtensionPresenter().Visibility(Visibility::Visible);
-            _extensionPalette.Visibility(Visibility::Visible);
-        }
-        else
+        args.Handled(false);
+        // only handle this if the feature is allowed
+        if (WI_IsAnyFlagSet(AIConfig::AllowedLMProviders(), EnabledLMProviders::All))
         {
-            _extensionPalette.Visibility(Visibility::Collapsed);
-            ExtensionPresenter().Visibility(Visibility::Collapsed);
+            if (ExtensionPresenter().Visibility() == Visibility::Collapsed)
+            {
+                _loadQueryExtension();
+                ExtensionPresenter().Visibility(Visibility::Visible);
+                _extensionPalette.Visibility(Visibility::Visible);
+            }
+            else
+            {
+                _extensionPalette.Visibility(Visibility::Collapsed);
+                ExtensionPresenter().Visibility(Visibility::Collapsed);
+            }
+            args.Handled(true);
         }
-        args.Handled(true);
     }
 
     void TerminalPage::_HandleSetColorScheme(const IInspectable& /*sender*/,