Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions app/assets/bundled/svg/kimi.svg
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
8 changes: 8 additions & 0 deletions app/src/ai/llms.rs
Original file line number Diff line number Diff line change
Expand Up @@ -186,11 +186,19 @@ pub fn model_leading_icon(llm: &LLMInfo, flags: ModelIconFlags) -> Icon {
Icon::Aws
} else if flags.is_using_gemini_enterprise {
Icon::GeminiEnterpriseAgentPlatform
} else if is_kimi_model(llm) {
Icon::KimiLogo
} else {
llm.provider.icon().unwrap_or(Icon::Agent)
}
}

/// Kimi is Fireworks-hosted, which the server reports as `LLMProvider::Unknown`,
/// so the provider enum can't carry its logo.
pub fn is_kimi_model(llm: &LLMInfo) -> bool {
llm.id.as_str().starts_with("kimi-")
}

/// Key for cached LLM metadata in user preferences.
///
/// Note: this key used to store a single [`AvailableLLMs`]
Expand Down
4 changes: 3 additions & 1 deletion app/src/terminal/view/ambient_agent/model_selector.rs
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ use crate::ai::custom_model_routers::is_custom_router_id;
use crate::ai::execution_profiles::model_menu_items::is_auto;
use crate::ai::harness_availability::{HarnessAvailabilityEvent, HarnessAvailabilityModel};
use crate::ai::harness_display::icon_for as harness_icon_for;
use crate::ai::llms::{LLMId, LLMPreferences, LLMPreferencesEvent};
use crate::ai::llms::{LLMId, LLMPreferences, LLMPreferencesEvent, is_kimi_model};
use crate::editor::{
EditorView, Event as EditorEvent, PropagateAndNoOpEscapeKey, PropagateAndNoOpNavigationKeys,
SingleLineEditorOptions, TextOptions,
Expand Down Expand Up @@ -509,6 +509,8 @@ impl ModelSelector {
let display_name = llm.menu_display_name();
let leading_icon = if is_custom_router_id(llm.id.as_str()) {
Icon::Dataflow
} else if is_kimi_model(llm) {
Icon::KimiLogo
} else {
llm.provider.icon().unwrap_or(Icon::Agent)
};
Expand Down
2 changes: 2 additions & 0 deletions crates/warp_core/src/ui/icons.rs
Original file line number Diff line number Diff line change
Expand Up @@ -275,6 +275,7 @@ pub enum Icon {
ClaudeLogo,
GeminiLogo,
GrokLogo,
KimiLogo,
OpenAILogo,
XLogo,
AmpLogo,
Expand Down Expand Up @@ -623,6 +624,7 @@ impl From<Icon> for &'static str {
Icon::ClaudeLogo => "bundled/svg/claude.svg",
Icon::GeminiLogo => "bundled/svg/gemini_cli.svg",
Icon::GrokLogo => "bundled/svg/grok.svg",
Icon::KimiLogo => "bundled/svg/kimi.svg",
Icon::OpenAILogo => "bundled/svg/openai.svg",
Icon::XLogo => "bundled/svg/x-logo.svg",
Icon::AmpLogo => "bundled/svg/amp.svg",
Expand Down
Loading