027f055841
This PR updates the casing of "OpenAI" when used in Rust identifiers to match the [Rust naming guidelines](https://rust-lang.github.io/api-guidelines/naming.html): > In `UpperCamelCase`, acronyms and contractions of compound words count as one word: use `Uuid` rather than `UUID`, `Usize` rather than `USize` or `Stdin` rather than `StdIn`. Release Notes: - N/A
130 lines
3.1 KiB
Rust
130 lines
3.1 KiB
Rust
pub mod assistant_panel;
|
|
mod assistant_settings;
|
|
mod codegen;
|
|
mod prompts;
|
|
mod streaming_diff;
|
|
|
|
use ai::providers::open_ai::Role;
|
|
use anyhow::Result;
|
|
pub use assistant_panel::AssistantPanel;
|
|
use assistant_settings::OpenAiModel;
|
|
use chrono::{DateTime, Local};
|
|
use collections::HashMap;
|
|
use fs::Fs;
|
|
use futures::StreamExt;
|
|
use gpui::{actions, AppContext, SharedString};
|
|
use regex::Regex;
|
|
use serde::{Deserialize, Serialize};
|
|
use std::{cmp::Reverse, ffi::OsStr, path::PathBuf, sync::Arc};
|
|
use util::paths::CONVERSATIONS_DIR;
|
|
|
|
actions!(
|
|
assistant,
|
|
[
|
|
NewConversation,
|
|
Assist,
|
|
Split,
|
|
CycleMessageRole,
|
|
QuoteSelection,
|
|
ToggleFocus,
|
|
ResetKey,
|
|
InlineAssist,
|
|
ToggleIncludeConversation,
|
|
ToggleRetrieveContext,
|
|
]
|
|
);
|
|
|
|
#[derive(
|
|
Copy, Clone, Debug, Default, Eq, PartialEq, PartialOrd, Ord, Hash, Serialize, Deserialize,
|
|
)]
|
|
struct MessageId(usize);
|
|
|
|
#[derive(Clone, Debug, Serialize, Deserialize)]
|
|
struct MessageMetadata {
|
|
role: Role,
|
|
sent_at: DateTime<Local>,
|
|
status: MessageStatus,
|
|
}
|
|
|
|
#[derive(Clone, Debug, Serialize, Deserialize)]
|
|
enum MessageStatus {
|
|
Pending,
|
|
Done,
|
|
Error(SharedString),
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize)]
|
|
struct SavedMessage {
|
|
id: MessageId,
|
|
start: usize,
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize)]
|
|
struct SavedConversation {
|
|
id: Option<String>,
|
|
zed: String,
|
|
version: String,
|
|
text: String,
|
|
messages: Vec<SavedMessage>,
|
|
message_metadata: HashMap<MessageId, MessageMetadata>,
|
|
summary: String,
|
|
model: OpenAiModel,
|
|
}
|
|
|
|
impl SavedConversation {
|
|
const VERSION: &'static str = "0.1.0";
|
|
}
|
|
|
|
struct SavedConversationMetadata {
|
|
title: String,
|
|
path: PathBuf,
|
|
mtime: chrono::DateTime<chrono::Local>,
|
|
}
|
|
|
|
impl SavedConversationMetadata {
|
|
pub async fn list(fs: Arc<dyn Fs>) -> Result<Vec<Self>> {
|
|
fs.create_dir(&CONVERSATIONS_DIR).await?;
|
|
|
|
let mut paths = fs.read_dir(&CONVERSATIONS_DIR).await?;
|
|
let mut conversations = Vec::<SavedConversationMetadata>::new();
|
|
while let Some(path) = paths.next().await {
|
|
let path = path?;
|
|
if path.extension() != Some(OsStr::new("json")) {
|
|
continue;
|
|
}
|
|
|
|
let pattern = r" - \d+.zed.json$";
|
|
let re = Regex::new(pattern).unwrap();
|
|
|
|
let metadata = fs.metadata(&path).await?;
|
|
if let Some((file_name, metadata)) = path
|
|
.file_name()
|
|
.and_then(|name| name.to_str())
|
|
.zip(metadata)
|
|
{
|
|
let title = re.replace(file_name, "");
|
|
conversations.push(Self {
|
|
title: title.into_owned(),
|
|
path,
|
|
mtime: metadata.mtime.into(),
|
|
});
|
|
}
|
|
}
|
|
conversations.sort_unstable_by_key(|conversation| Reverse(conversation.mtime));
|
|
|
|
Ok(conversations)
|
|
}
|
|
}
|
|
|
|
pub fn init(cx: &mut AppContext) {
|
|
assistant_panel::init(cx);
|
|
}
|
|
|
|
#[cfg(test)]
|
|
#[ctor::ctor]
|
|
fn init_logger() {
|
|
if std::env::var("RUST_LOG").is_ok() {
|
|
env_logger::init();
|
|
}
|
|
}
|