3cdf5ce947
Adds a new `agent.model_parameters` setting that allows the user to
specify a custom temperature for a provider AND/OR model:
```json5
"model_parameters": [
// To set parameters for all requests to OpenAI models:
{
"provider": "openai",
"temperature": 0.5
},
// To set parameters for all requests in general:
{
"temperature": 0
},
// To set parameters for a specific provider and model:
{
"provider": "zed.dev",
"model": "claude-3-7-sonnet-latest",
"temperature": 1.0
}
],
```
Release Notes:
- agent: Allow customizing temperature by provider/model
---------
Co-authored-by: Max Brunsfeld <maxbrunsfeld@gmail.com>
Co-authored-by: Marshall Bowers <git@maxdeviant.com>
77 lines
2 KiB
Rust
77 lines
2 KiB
Rust
use std::sync::Arc;
|
|
|
|
use collections::IndexMap;
|
|
use gpui::SharedString;
|
|
use schemars::JsonSchema;
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
pub mod builtin_profiles {
|
|
use super::AgentProfileId;
|
|
|
|
pub const WRITE: &str = "write";
|
|
pub const ASK: &str = "ask";
|
|
pub const MINIMAL: &str = "minimal";
|
|
|
|
pub fn is_builtin(profile_id: &AgentProfileId) -> bool {
|
|
profile_id.as_str() == WRITE || profile_id.as_str() == ASK || profile_id.as_str() == MINIMAL
|
|
}
|
|
}
|
|
|
|
#[derive(Default)]
|
|
pub struct GroupedAgentProfiles {
|
|
pub builtin: IndexMap<AgentProfileId, AgentProfile>,
|
|
pub custom: IndexMap<AgentProfileId, AgentProfile>,
|
|
}
|
|
|
|
impl GroupedAgentProfiles {
|
|
pub fn from_settings(settings: &crate::AssistantSettings) -> Self {
|
|
let mut builtin = IndexMap::default();
|
|
let mut custom = IndexMap::default();
|
|
|
|
for (profile_id, profile) in settings.profiles.clone() {
|
|
if builtin_profiles::is_builtin(&profile_id) {
|
|
builtin.insert(profile_id, profile);
|
|
} else {
|
|
custom.insert(profile_id, profile);
|
|
}
|
|
}
|
|
|
|
Self { builtin, custom }
|
|
}
|
|
}
|
|
|
|
#[derive(Debug, PartialEq, Eq, Hash, Clone, Serialize, Deserialize, JsonSchema)]
|
|
pub struct AgentProfileId(pub Arc<str>);
|
|
|
|
impl AgentProfileId {
|
|
pub fn as_str(&self) -> &str {
|
|
&self.0
|
|
}
|
|
}
|
|
|
|
impl std::fmt::Display for AgentProfileId {
|
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
write!(f, "{}", self.0)
|
|
}
|
|
}
|
|
|
|
impl Default for AgentProfileId {
|
|
fn default() -> Self {
|
|
Self("write".into())
|
|
}
|
|
}
|
|
|
|
/// A profile for the Zed Agent that controls its behavior.
|
|
#[derive(Debug, Clone)]
|
|
pub struct AgentProfile {
|
|
/// The name of the profile.
|
|
pub name: SharedString,
|
|
pub tools: IndexMap<Arc<str>, bool>,
|
|
pub enable_all_context_servers: bool,
|
|
pub context_servers: IndexMap<Arc<str>, ContextServerPreset>,
|
|
}
|
|
|
|
#[derive(Debug, Clone, Default)]
|
|
pub struct ContextServerPreset {
|
|
pub tools: IndexMap<Arc<str>, bool>,
|
|
}
|