Yesterday17andMarshall Bowers
9e17018416
Partially fixes #4321, since Azure OpenAI API can be converted to OpenAI API. Release Notes: - Added `assistant.openai_api_url` setting to allow OpenAI API URL to be configured. --------- Co-authored-by: Marshall Bowers <elliott.codes@gmail.com>
103 lines
2.9 KiB
Rust
103 lines
2.9 KiB
Rust
use anyhow;
|
|
use gpui::Pixels;
|
|
use schemars::JsonSchema;
|
|
use serde::{Deserialize, Serialize};
|
|
use settings::Settings;
|
|
|
|
#[derive(Clone, Debug, Serialize, Deserialize, JsonSchema, PartialEq)]
|
|
pub enum OpenAiModel {
|
|
#[serde(rename = "gpt-3.5-turbo-0613")]
|
|
ThreePointFiveTurbo,
|
|
#[serde(rename = "gpt-4-0613")]
|
|
Four,
|
|
#[serde(rename = "gpt-4-1106-preview")]
|
|
FourTurbo,
|
|
}
|
|
|
|
impl OpenAiModel {
|
|
pub fn full_name(&self) -> &'static str {
|
|
match self {
|
|
OpenAiModel::ThreePointFiveTurbo => "gpt-3.5-turbo-0613",
|
|
OpenAiModel::Four => "gpt-4-0613",
|
|
OpenAiModel::FourTurbo => "gpt-4-1106-preview",
|
|
}
|
|
}
|
|
|
|
pub fn short_name(&self) -> &'static str {
|
|
match self {
|
|
OpenAiModel::ThreePointFiveTurbo => "gpt-3.5-turbo",
|
|
OpenAiModel::Four => "gpt-4",
|
|
OpenAiModel::FourTurbo => "gpt-4-turbo",
|
|
}
|
|
}
|
|
|
|
pub fn cycle(&self) -> Self {
|
|
match self {
|
|
OpenAiModel::ThreePointFiveTurbo => OpenAiModel::Four,
|
|
OpenAiModel::Four => OpenAiModel::FourTurbo,
|
|
OpenAiModel::FourTurbo => OpenAiModel::ThreePointFiveTurbo,
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Debug, Serialize, Deserialize, JsonSchema)]
|
|
#[serde(rename_all = "snake_case")]
|
|
pub enum AssistantDockPosition {
|
|
Left,
|
|
Right,
|
|
Bottom,
|
|
}
|
|
|
|
#[derive(Deserialize, Debug)]
|
|
pub struct AssistantSettings {
|
|
pub button: bool,
|
|
pub dock: AssistantDockPosition,
|
|
pub default_width: Pixels,
|
|
pub default_height: Pixels,
|
|
pub default_open_ai_model: OpenAiModel,
|
|
pub openai_api_url: String,
|
|
}
|
|
|
|
/// Assistant panel settings
|
|
#[derive(Clone, Default, Serialize, Deserialize, JsonSchema, Debug)]
|
|
pub struct AssistantSettingsContent {
|
|
/// Whether to show the assistant panel button in the status bar.
|
|
///
|
|
/// Default: true
|
|
pub button: Option<bool>,
|
|
/// Where to dock the assistant.
|
|
///
|
|
/// Default: right
|
|
pub dock: Option<AssistantDockPosition>,
|
|
/// Default width in pixels when the assistant is docked to the left or right.
|
|
///
|
|
/// Default: 640
|
|
pub default_width: Option<f32>,
|
|
/// Default height in pixels when the assistant is docked to the bottom.
|
|
///
|
|
/// Default: 320
|
|
pub default_height: Option<f32>,
|
|
/// The default OpenAI model to use when starting new conversations.
|
|
///
|
|
/// Default: gpt-4-1106-preview
|
|
pub default_open_ai_model: Option<OpenAiModel>,
|
|
/// OpenAI API base URL to use when starting new conversations.
|
|
///
|
|
/// Default: https://api.openai.com/v1
|
|
pub openai_api_url: Option<String>,
|
|
}
|
|
|
|
impl Settings for AssistantSettings {
|
|
const KEY: Option<&'static str> = Some("assistant");
|
|
|
|
type FileContent = AssistantSettingsContent;
|
|
|
|
fn load(
|
|
default_value: &Self::FileContent,
|
|
user_values: &[&Self::FileContent],
|
|
_: &mut gpui::AppContext,
|
|
) -> anyhow::Result<Self> {
|
|
Self::load_via_json_merge(default_value, user_values)
|
|
}
|
|
}
|