[feat]: Add memory system (#339)

block · Nov 26, 2024 · 5ba98c4 · 5ba98c4
1 parent 2875302
commit 5ba98c4
Show file tree

Hide file tree

Showing 6 changed files with 495 additions and 3 deletions.
diff --git a/crates/goose-cli/src/session.rs b/crates/goose-cli/src/session.rs
@@ -204,7 +204,6 @@ impl<'a> Session<'a> {
         self.agent.add_system(system);
         self.prompt
             .render(raw_message("Connected developer system."));
-
         let goosehints_system = Box::new(GooseHintsSystem::new());
         self.agent.add_system(goosehints_system);
         self.prompt

diff --git a/crates/goose-server/src/routes/reply.rs b/crates/goose-server/src/routes/reply.rs
@@ -1,3 +1,4 @@
+use crate::state::AppState;
 use axum::{
     extract::State,
     http::{self, HeaderMap, StatusCode},
@@ -25,8 +26,6 @@ use std::{
 use tokio::sync::mpsc;
 use tokio_stream::wrappers::ReceiverStream;
 
-use crate::state::AppState;
-
 // Types matching the incoming JSON structure
 #[derive(Debug, Deserialize)]
 struct ChatRequest {

diff --git a/crates/goose/Cargo.toml b/crates/goose/Cargo.toml
@@ -8,6 +8,7 @@ repository.workspace = true
 description.workspace = true
 
 [dependencies]
+dirs = "5.0.1"
 anyhow = "1.0"
 thiserror = "1.0"
 futures = "0.3"

diff --git a/crates/goose/src/lib.rs b/crates/goose/src/lib.rs
@@ -1,6 +1,7 @@
 pub mod agent;
 pub mod developer;
 pub mod errors;
+pub mod memory;
 pub mod models;
 pub mod prompt_template;
 pub mod providers;