29bfb56739
- Added support for DeepSeek as a new language model provider in Zed Assistant - Implemented streaming API support for real-time responses from DeepSeek models. - Added a configuration UI for DeepSeek API key management and settings. - Updated documentation with detailed setup instructions for DeepSeek integration. - Added DeepSeek-specific icons and model definitions for seamless integration into the Zed UI. - Integrated DeepSeek into the language model registry, making it available alongside other providers like OpenAI and Anthropic. Release Notes: - Added support for DeepSeek to the Assistant. --------- Co-authored-by: Marshall Bowers <git@maxdeviant.com>
87 lines
2.4 KiB
Rust
87 lines
2.4 KiB
Rust
use serde::{Deserialize, Serialize};
|
|
use std::fmt::{self, Display};
|
|
|
|
#[derive(Clone, Copy, Serialize, Deserialize, Debug, Eq, PartialEq, Hash)]
|
|
#[serde(rename_all = "lowercase")]
|
|
pub enum Role {
|
|
User,
|
|
Assistant,
|
|
System,
|
|
}
|
|
|
|
impl Role {
|
|
pub fn from_proto(role: i32) -> Role {
|
|
match proto::LanguageModelRole::from_i32(role) {
|
|
Some(proto::LanguageModelRole::LanguageModelUser) => Role::User,
|
|
Some(proto::LanguageModelRole::LanguageModelAssistant) => Role::Assistant,
|
|
Some(proto::LanguageModelRole::LanguageModelSystem) => Role::System,
|
|
None => Role::User,
|
|
}
|
|
}
|
|
|
|
pub fn to_proto(&self) -> proto::LanguageModelRole {
|
|
match self {
|
|
Role::User => proto::LanguageModelRole::LanguageModelUser,
|
|
Role::Assistant => proto::LanguageModelRole::LanguageModelAssistant,
|
|
Role::System => proto::LanguageModelRole::LanguageModelSystem,
|
|
}
|
|
}
|
|
|
|
pub fn cycle(self) -> Role {
|
|
match self {
|
|
Role::User => Role::Assistant,
|
|
Role::Assistant => Role::System,
|
|
Role::System => Role::User,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Display for Role {
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> std::fmt::Result {
|
|
match self {
|
|
Role::User => write!(f, "user"),
|
|
Role::Assistant => write!(f, "assistant"),
|
|
Role::System => write!(f, "system"),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl From<Role> for ollama::Role {
|
|
fn from(val: Role) -> Self {
|
|
match val {
|
|
Role::User => ollama::Role::User,
|
|
Role::Assistant => ollama::Role::Assistant,
|
|
Role::System => ollama::Role::System,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl From<Role> for open_ai::Role {
|
|
fn from(val: Role) -> Self {
|
|
match val {
|
|
Role::User => open_ai::Role::User,
|
|
Role::Assistant => open_ai::Role::Assistant,
|
|
Role::System => open_ai::Role::System,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl From<Role> for deepseek::Role {
|
|
fn from(val: Role) -> Self {
|
|
match val {
|
|
Role::User => deepseek::Role::User,
|
|
Role::Assistant => deepseek::Role::Assistant,
|
|
Role::System => deepseek::Role::System,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl From<Role> for lmstudio::Role {
|
|
fn from(val: Role) -> Self {
|
|
match val {
|
|
Role::User => lmstudio::Role::User,
|
|
Role::Assistant => lmstudio::Role::Assistant,
|
|
Role::System => lmstudio::Role::System,
|
|
}
|
|
}
|
|
}
|