Antonio ScandurraandNathan
d6bdaa8a91
In this pull request, we change the zed.dev protocol so that we pass the
raw JSON for the specified provider directly to our server. This avoids
the need to define a protobuf message that's a superset of all these
formats.
@bennetbo: We also changed the settings for available_models under
zed.dev to be a flat format, because the nesting seemed too confusing.
Can you help us upgrade the local provider configuration to be
consistent with this? We do whatever we need to do when parsing the
settings to make this simple for users, even if it's a bit more complex
on our end. We want to use versioning to avoid breaking existing users,
but need to keep making progress.
```json
"zed.dev": {
"available_models": [
{
"provider": "anthropic",
"name": "some-newly-released-model-we-havent-added",
"max_tokens": 200000
}
]
}
```
Release Notes:
- N/A
---------
Co-authored-by: Nathan <nathan@zed.dev>
68 lines
1.9 KiB
Rust
68 lines
1.9 KiB
Rust
use serde::{Deserialize, Serialize};
|
|
use std::fmt::{self, Display};
|
|
|
|
#[derive(Clone, Copy, Serialize, Deserialize, Debug, Eq, PartialEq)]
|
|
#[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,
|
|
}
|
|
}
|
|
}
|