tenant.openagents/omega

1use crate::{LanguageModelProviderId, LanguageModelProviderName};
2
3pub const ANTHROPIC_PROVIDER_ID: LanguageModelProviderId =
4    LanguageModelProviderId::new("anthropic");
5pub const ANTHROPIC_PROVIDER_NAME: LanguageModelProviderName =
6    LanguageModelProviderName::new("Anthropic");
7
8pub const OPEN_AI_PROVIDER_ID: LanguageModelProviderId = LanguageModelProviderId::new("openai");
9pub const OPEN_AI_PROVIDER_NAME: LanguageModelProviderName =
10    LanguageModelProviderName::new("OpenAI");
11
12pub const GOOGLE_PROVIDER_ID: LanguageModelProviderId = LanguageModelProviderId::new("google");
13pub const GOOGLE_PROVIDER_NAME: LanguageModelProviderName =
14    LanguageModelProviderName::new("Google AI");
15
16pub const X_AI_PROVIDER_ID: LanguageModelProviderId = LanguageModelProviderId::new("x_ai");
17pub const X_AI_PROVIDER_NAME: LanguageModelProviderName = LanguageModelProviderName::new("xAI");
18
19pub const ZED_CLOUD_PROVIDER_ID: LanguageModelProviderId = LanguageModelProviderId::new("zed.dev");
20pub const ZED_CLOUD_PROVIDER_NAME: LanguageModelProviderName =
21    LanguageModelProviderName::new("Zed");
22
23pub fn provider_name_for_id(provider_id: &LanguageModelProviderId) -> LanguageModelProviderName {
24    if provider_id == &OPEN_AI_PROVIDER_ID {
25        OPEN_AI_PROVIDER_NAME
26    } else if provider_id == &ANTHROPIC_PROVIDER_ID {
27        ANTHROPIC_PROVIDER_NAME
28    } else if provider_id == &GOOGLE_PROVIDER_ID {
29        GOOGLE_PROVIDER_NAME
30    } else if provider_id == &X_AI_PROVIDER_ID {
31        X_AI_PROVIDER_NAME
32    } else if provider_id == &ZED_CLOUD_PROVIDER_ID {
33        ZED_CLOUD_PROVIDER_NAME
34    } else {
35        LanguageModelProviderName(provider_id.0.clone())
36    }
37}
38