improve agent

Author: Tricked-dev

src/main.rs

@@ -60,9 +60,9 @@ async fn main() -> color_eyre::Result<()> {
     dotenv::dotenv().ok();
 
     tracing_subscriber::fmt()
-        .with_env_filter(
-            tracing_subscriber::EnvFilter::new("debug,h2::codec::framed_read=off")
-        )
+        .with_env_filter(tracing_subscriber::EnvFilter::new(
+            "debug,h2::codec::framed_read=off,twilight_gateway::shard=off,twilight_http_ratelimiting::in_memory=info,rustls::client=info,hyper::client::connect=info",
+        ))
         .init();
 
     let mut cfg = Config::parse();

src/memory_creator.rs

@@ -7,12 +7,8 @@ use r2d2_sqlite::SqliteConnectionManager;
 use serde::{Deserialize, Serialize};
 use serde_rusqlite::from_row;
 use std::{collections::HashMap, sync::Arc};
-use wb_sqlite::InsertSync;
 
-use crate::{
-    config::Config,
-    database::{Memory, User},
-};
+use crate::{config::Config, database::User};
 
 /// JSON response structure for memory creation
 #[derive(Debug, Serialize, Deserialize)]
@@ -58,45 +54,21 @@ Respond ONLY with valid JSON in this exact format:
 IMPORTANT GUIDELINES:
 - Only create memories if there's meaningful information from THIS conversation
 - Each user should have AT MOST ONE memory entry per unique "key" category
-- The "key" should be a broad category like "preferences", "hobbies", "work", "personality", "relationships", "recent_activity"
+- The "key" should be a broad category like "preferences", "hobbies", "work", "personality", "relationships", "recent_activity" but can be anything like outside_hobbies could work too
 - The "content" should combine ALL related facts for that category into ONE comprehensive entry
 - Use exact usernames as they appear in the conversation
 - If there's nothing meaningful to remember, return an empty memories array
 
-EXAMPLE - CORRECT (combining multiple facts under one key):
-{{
-  "memories": [
-    {{
-      "username": "tricked.",
-      "key": "preferences",
-      "content": "Likes cats, dislikes insects"
-    }}
-  ]
-}}
-
-EXAMPLE - WRONG (duplicate keys for same user):
-{{
-  "memories": [
-    {{"username": "tricked.", "key": "preferences", "content": "Likes cats"}},
-    {{"username": "tricked.", "key": "preferences", "content": "Dislikes insects"}}
-  ]
-}}
-
 Remember: Output ONLY valid JSON, nothing else. Combine related information under the same key."#,
         context = context,
         participants = participants
     )
 }
 
 /// Resolve usernames to user IDs using the database
-fn resolve_username_to_id(
-    database: &r2d2::Pool<SqliteConnectionManager>,
-    username: &str,
-) -> Option<u64> {
+fn resolve_username_to_id(database: &r2d2::Pool<SqliteConnectionManager>, username: &str) -> Option<u64> {
     let db = database.get().ok()?;
-    let mut stmt = db
-        .prepare("SELECT * FROM user WHERE name = ? COLLATE NOCASE")
-        .ok()?;
+    let mut stmt = db.prepare("SELECT * FROM user WHERE name = ? COLLATE NOCASE").ok()?;
 
     stmt.query_one([username], |row| {
         from_row::<User>(row).map_err(|_| rusqlite::Error::QueryReturnedNoRows)
@@ -108,12 +80,7 @@ fn resolve_username_to_id(
 /// Insert a new memory in the database
 /// If a memory with the same user_id and key exists, delete it first then insert the new one
 /// This ensures the new memory has the latest timestamp and will be in the top 5 most recent
-fn insert_memory(
-    database: &r2d2::Pool<SqliteConnectionManager>,
-    user_id: u64,
-    key: &str,
-    content: &str,
-) -> Result<()> {
+fn insert_memory(database: &r2d2::Pool<SqliteConnectionManager>, user_id: u64, key: &str, content: &str) -> Result<()> {
     let db = database.get()?;
 
     // Delete any existing memory with the same user_id and key
@@ -122,23 +89,17 @@ fn insert_memory(
         rusqlite::params![user_id.to_string(), key],
     )?;
 
-    // Insert the new memory (will get a new ID and timestamp)
-    let memory = Memory {
-        id: 0, // Will be auto-generated
-        user_id: user_id.to_string(),
-        content: content.to_string(),
-        key: key.to_string(),
-    };
-    memory.insert_sync(&db)?;
+    // Insert the new memory (id will be auto-generated by AUTOINCREMENT)
+    db.execute(
+        "INSERT INTO memory (user_id, content, key) VALUES (?, ?, ?)",
+        rusqlite::params![user_id.to_string(), content, key],
+    )?;
 
     Ok(())
 }
 
 /// Process memory creation response and store in database
-fn process_memory_response(
-    database: &r2d2::Pool<SqliteConnectionManager>,
-    response_text: &str,
-) -> Result<usize> {
+fn process_memory_response(database: &r2d2::Pool<SqliteConnectionManager>, response_text: &str) -> Result<usize> {
     log::info!("Raw memory response: {}", response_text);
 
     // Try to extract JSON from the response (in case the model adds extra text)