fix(zsh): filter provider list to show only llm providers (#2098)

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
Co-authored-by: Tushar Mathur <tusharmath@gmail.com>

Author: 3 people

crates/forge_domain/src/provider.rs

@@ -4,13 +4,15 @@ use std::collections::HashMap;
 use derive_more::{AsRef, Deref, From};
 use schemars::JsonSchema;
 use serde::{Deserialize, Serialize};
-use strum_macros::Display;
+use strum_macros::{Display, EnumString};
 use url::Url;
 
 use crate::{ApiKey, AuthCredential, AuthDetails, Model, Template};
 
 /// Distinguishes between different categories of providers
-#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, Serialize, Deserialize, Display, Default)]
+#[derive(
+    Debug, Clone, Copy, PartialEq, Eq, Hash, Serialize, Deserialize, Display, EnumString, Default,
+)]
 #[serde(rename_all = "snake_case")]
 #[strum(serialize_all = "snake_case")]
 pub enum ProviderType {

crates/forge_main/src/cli.rs

@@ -281,7 +281,12 @@ pub enum ListCommand {
 
     /// List available API providers.
     #[command(alias = "providers")]
-    Provider,
+    Provider {
+        /// Filter providers by type (e.g., llm, context_engine). Can be
+        /// specified multiple times.
+        #[arg(long = "type", short = 't')]
+        types: Vec<forge_domain::ProviderType>,
+    },
 
     /// List available models.
     #[command(alias = "models")]
@@ -574,7 +579,12 @@ pub enum ProviderCommand {
     },
 
     /// List available providers.
-    List,
+    List {
+        /// Filter providers by type (e.g., llm, context_engine). Can be
+        /// specified multiple times.
+        #[arg(long = "type", short = 't')]
+        types: Vec<forge_domain::ProviderType>,
+    },
 }
 
 /// Group of Commit-related commands

crates/forge_main/src/ui.rs

@@ -383,8 +383,8 @@ impl<A: API + 'static, F: Fn() -> A + Send + Sync> UI<A, F> {
                     ListCommand::Agent => {
                         self.on_show_agents(porcelain).await?;
                     }
-                    ListCommand::Provider => {
-                        self.on_show_providers(porcelain).await?;
+                    ListCommand::Provider { types } => {
+                        self.on_show_providers(porcelain, types).await?;
                     }
                     ListCommand::Model => {
                         self.on_show_models(porcelain).await?;
@@ -756,8 +756,9 @@ impl<A: API + 'static, F: Fn() -> A + Send + Sync> UI<A, F> {
             ProviderCommand::Logout { provider } => {
                 self.handle_provider_logout(provider.as_ref()).await?;
             }
-            ProviderCommand::List => {
-                self.on_show_providers(provider_group.porcelain).await?;
+            ProviderCommand::List { types } => {
+                self.on_show_providers(provider_group.porcelain, types)
+                    .await?;
             }
         }
 
@@ -977,8 +978,17 @@ impl<A: API + 'static, F: Fn() -> A + Send + Sync> UI<A, F> {
     }
 
     /// Lists all the providers
-    async fn on_show_providers(&mut self, porcelain: bool) -> anyhow::Result<()> {
-        let providers = self.api.get_providers().await?;
+    async fn on_show_providers(
+        &mut self,
+        porcelain: bool,
+        types: Vec<forge_domain::ProviderType>,
+    ) -> anyhow::Result<()> {
+        let mut providers = self.api.get_providers().await?;
+
+        // Filter by type if specified
+        if !types.is_empty() {
+            providers.retain(|p| types.contains(p.provider_type()));
+        }
 
         if providers.is_empty() {
             return Ok(());

shell-plugin/lib/actions/config.zsh

@@ -80,11 +80,12 @@ function _forge_action_agent() {
 function _forge_action_provider() {
     echo
     local selected
-    selected=$(_forge_select_provider)
+    # Only show LLM providers (exclude context_engine and other non-LLM types)
+    selected=$(_forge_select_provider "" "" "llm")
 
     if [[ -n "$selected" ]]; then
         # Extract the second field (provider ID) from the selected line
-        # Format: "DisplayName  provider_id  host  status"
+        # Format: "DisplayName  provider_id  host  type  status"
         local provider_id=$(echo "$selected" | awk '{print $2}')
         # Always use config set - it will handle authentication if needed
         _forge_exec config set provider "$provider_id"

shell-plugin/lib/actions/provider.zsh

@@ -3,13 +3,21 @@
 # Provider selection helper
 
 # Helper function to select a provider from the list
-# Usage: _forge_select_provider [filter_status] [current_provider]
+# Usage: _forge_select_provider [filter_status] [current_provider] [filter_type]
 # Returns: selected provider line (via stdout)
 function _forge_select_provider() {
     local filter_status="${1:-}"
     local current_provider="${2:-}"
+    local filter_type="${3:-}"
     local output
-    output=$($_FORGE_BIN list provider --porcelain 2>/dev/null)
+    
+    # Build the command with type filter if specified
+    local cmd="$_FORGE_BIN list provider --porcelain"
+    if [[ -n "$filter_type" ]]; then
+        cmd="$cmd --type=$filter_type"
+    fi
+    
+    output=$(eval "$cmd" 2>/dev/null)
 
     if [[ -z "$output" ]]; then
         _forge_log error "No providers available"