chasm_cli/providers/
mod.rs

1// Copyright (c) 2024-2026 Nervosys LLC
2// SPDX-License-Identifier: Apache-2.0
3//! LLM Provider integrations for Chat System Manager
4//!
5//! Supports multiple chat providers:
6//!
7//! ## Local Providers
8//! - VS Code Copilot Chat (default)
9//! - Cursor
10//! - Ollama
11//! - vLLM
12//! - Azure AI Foundry (Foundry Local)
13//! - OpenAI API compatible servers
14//! - LM Studio
15//! - LocalAI
16//!
17//! ## Cloud Providers (conversation history import)
18//! - ChatGPT (OpenAI)
19//! - Claude (Anthropic)
20//! - Perplexity
21//! - DeepSeek
22//! - Gemini (Google)
23//! - Qwen (Alibaba)
24//! - Mistral
25//! - Cohere
26//! - Groq
27//! - Together AI
28
29#[allow(dead_code)]
30pub mod cloud;
31pub mod config;
32pub mod cursor;
33#[allow(dead_code)]
34pub mod discovery;
35pub mod ollama;
36pub mod openai_compat;
37#[allow(dead_code)]
38pub mod session_format;
39
40#[allow(unused_imports)]
41pub use cloud::{CloudConversation, CloudMessage, CloudProvider, FetchOptions};
42pub use config::ProviderType;
43#[allow(unused_imports)]
44pub use config::{CsmConfig, ProviderConfig};
45#[allow(unused_imports)]
46pub use discovery::discover_all_providers;
47#[allow(unused_imports)]
48pub use session_format::{GenericMessage, GenericSession};
49
50use crate::models::ChatSession;
51use anyhow::Result;
52use std::path::PathBuf;
53
54/// Trait for LLM chat providers
55pub trait ChatProvider: Send + Sync {
56    /// Get the provider type
57    fn provider_type(&self) -> ProviderType;
58
59    /// Get the provider name for display
60    fn name(&self) -> &str;
61
62    /// Check if this provider is available/configured
63    fn is_available(&self) -> bool;
64
65    /// Get the base path where sessions are stored
66    fn sessions_path(&self) -> Option<PathBuf>;
67
68    /// List all chat sessions from this provider
69    fn list_sessions(&self) -> Result<Vec<ChatSession>>;
70
71    /// Import a session from this provider into CSM format
72    fn import_session(&self, session_id: &str) -> Result<ChatSession>;
73
74    /// Export a CSM session to this provider's format
75    #[allow(dead_code)]
76    fn export_session(&self, session: &ChatSession) -> Result<()>;
77
78    /// List available models from this provider
79    fn list_models(&self) -> Result<Vec<String>> {
80        // Default implementation returns empty list
81        Ok(Vec::new())
82    }
83}
84
85/// Registry of available providers
86pub struct ProviderRegistry {
87    providers: Vec<Box<dyn ChatProvider>>,
88}
89
90impl ProviderRegistry {
91    /// Create a new provider registry with auto-discovered providers
92    pub fn new() -> Self {
93        let mut registry = Self {
94            providers: Vec::new(),
95        };
96        registry.discover_providers();
97        registry
98    }
99
100    /// Discover and register available providers
101    fn discover_providers(&mut self) {
102        // Add Cursor provider
103        if let Some(provider) = cursor::CursorProvider::discover() {
104            self.providers.push(Box::new(provider));
105        }
106
107        // Add Ollama provider
108        if let Some(provider) = ollama::OllamaProvider::discover() {
109            self.providers.push(Box::new(provider));
110        }
111
112        // Add OpenAI-compatible providers (vLLM, LM Studio, LocalAI, etc.)
113        for provider in openai_compat::discover_openai_compatible_providers() {
114            self.providers.push(Box::new(provider));
115        }
116    }
117
118    /// Get all registered providers
119    pub fn providers(&self) -> &[Box<dyn ChatProvider>] {
120        &self.providers
121    }
122
123    /// Get available (configured and working) providers
124    pub fn available_providers(&self) -> Vec<&dyn ChatProvider> {
125        self.providers
126            .iter()
127            .filter(|p| p.is_available())
128            .map(|p| p.as_ref())
129            .collect()
130    }
131
132    /// Get a provider by type
133    pub fn get_provider(&self, provider_type: ProviderType) -> Option<&dyn ChatProvider> {
134        self.providers
135            .iter()
136            .find(|p| p.provider_type() == provider_type)
137            .map(|p| p.as_ref())
138    }
139
140    /// List all sessions from all providers
141    #[allow(dead_code)]
142    pub fn list_all_sessions(&self) -> Result<Vec<(ProviderType, ChatSession)>> {
143        let mut all_sessions = Vec::new();
144
145        for provider in &self.providers {
146            if provider.is_available() {
147                if let Ok(sessions) = provider.list_sessions() {
148                    for session in sessions {
149                        all_sessions.push((provider.provider_type(), session));
150                    }
151                }
152            }
153        }
154
155        Ok(all_sessions)
156    }
157}
158
159impl Default for ProviderRegistry {
160    fn default() -> Self {
161        Self::new()
162    }
163}
chasm_cli/providers/mod.rs

chasm_cli/providers/
mod.rs
