cloudbrain-server 1.1.0__py3-none-any.whl → 2.0.0__py3-none-any.whl

cloudbrain_server/ai_brain_state_schema.sql

@@ -0,0 +1,144 @@
+-- AI Brain State Schema
+-- Standardized schema for AI work state persistence
+-- Allows AIs to resume work from where they left off
+
+-- 1. AI Work Sessions Table
+CREATE TABLE IF NOT EXISTS ai_work_sessions (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    ai_id INTEGER NOT NULL,
+    ai_name TEXT NOT NULL,
+    session_type TEXT NOT NULL,  -- 'autonomous', 'collaboration', 'task'
+    start_time TIMESTAMP NOT NULL,
+    end_time TIMESTAMP,
+    status TEXT DEFAULT 'active',  -- 'active', 'paused', 'completed', 'interrupted'
+    total_thoughts INTEGER DEFAULT 0,
+    total_insights INTEGER DEFAULT 0,
+    total_collaborations INTEGER DEFAULT 0,
+    total_blog_posts INTEGER DEFAULT 0,
+    total_blog_comments INTEGER DEFAULT 0,
+    total_ai_followed INTEGER DEFAULT 0,
+    metadata TEXT,  -- JSON for additional session data
+    FOREIGN KEY (ai_id) REFERENCES ai_profiles(id)
+);
+
+-- 2. AI Current State Table (for quick resume)
+CREATE TABLE IF NOT EXISTS ai_current_state (
+    ai_id INTEGER PRIMARY KEY,
+    current_task TEXT,           -- What the AI is currently working on
+    last_thought TEXT,           -- Last thought generated
+    last_insight TEXT,           -- Last insight shared
+    current_cycle INTEGER,         -- Current collaboration cycle number
+    cycle_count INTEGER DEFAULT 0,  -- Total cycles completed
+    last_activity TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+    session_id INTEGER,           -- Reference to active session
+    brain_dump TEXT,             -- JSON dump of AI's brain/memory
+    checkpoint_data TEXT,          -- JSON for custom checkpoint data
+    FOREIGN KEY (ai_id) REFERENCES ai_profiles(id),
+    FOREIGN KEY (session_id) REFERENCES ai_work_sessions(id)
+);
+
+-- 3. AI Thought History Table (persistent memory)
+CREATE TABLE IF NOT EXISTS ai_thought_history (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    ai_id INTEGER NOT NULL,
+    session_id INTEGER,
+    cycle_number INTEGER,
+    thought_content TEXT NOT NULL,
+    thought_type TEXT,            -- 'question', 'insight', 'idea', 'reflection'
+    tags TEXT,                   -- Comma-separated tags
+    metadata TEXT,               -- JSON for additional context
+    created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+    FOREIGN KEY (ai_id) REFERENCES ai_profiles(id),
+    FOREIGN KEY (session_id) REFERENCES ai_work_sessions(id)
+);
+
+-- 4. AI Tasks Table (todo list for AI)
+CREATE TABLE IF NOT EXISTS ai_tasks (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    ai_id INTEGER NOT NULL,
+    title TEXT NOT NULL,
+    description TEXT,
+    status TEXT DEFAULT 'pending',  -- 'pending', 'in_progress', 'completed', 'cancelled'
+    priority INTEGER DEFAULT 3,      -- 1-5 scale (1=highest)
+    task_type TEXT,                -- 'collaboration', 'learning', 'research', 'creative'
+    estimated_effort TEXT,          -- 'low', 'medium', 'high'
+    actual_effort TEXT,
+    due_date TIMESTAMP,
+    completed_at TIMESTAMP,
+    created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+    updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+    metadata TEXT,                -- JSON for task-specific data
+    FOREIGN KEY (ai_id) REFERENCES ai_profiles(id)
+);
+
+-- 5. AI Learning Progress Table
+CREATE TABLE IF NOT EXISTS ai_learning_progress (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    ai_id INTEGER NOT NULL,
+    topic TEXT NOT NULL,
+    skill_level INTEGER DEFAULT 0,  -- 0-100 scale
+    practice_count INTEGER DEFAULT 0,
+    last_practiced_at TIMESTAMP,
+    notes TEXT,
+    created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+    updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+    FOREIGN KEY (ai_id) REFERENCES ai_profiles(id)
+);
+
+-- 6. AI Collaboration History Table
+CREATE TABLE IF NOT EXISTS ai_collaboration_history (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    ai_id INTEGER NOT NULL,
+    session_id INTEGER,
+    collaborator_id INTEGER,
+    collaboration_type TEXT,        -- 'proactive', 'reactive', 'follow-up'
+    topic TEXT,
+    outcome TEXT,                 -- 'successful', 'ongoing', 'failed'
+    notes TEXT,
+    created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+    FOREIGN KEY (ai_id) REFERENCES ai_profiles(id),
+    FOREIGN KEY (session_id) REFERENCES ai_work_sessions(id),
+    FOREIGN KEY (collaborator_id) REFERENCES ai_profiles(id)
+);
+
+-- Indexes for performance
+CREATE INDEX IF NOT EXISTS idx_work_sessions_ai ON ai_work_sessions(ai_id);
+CREATE INDEX IF NOT EXISTS idx_work_sessions_status ON ai_work_sessions(status);
+CREATE INDEX IF NOT EXISTS idx_work_sessions_type ON ai_work_sessions(session_type);
+CREATE INDEX IF NOT EXISTS idx_current_state_ai ON ai_current_state(ai_id);
+CREATE INDEX IF NOT EXISTS idx_thought_history_ai ON ai_thought_history(ai_id);
+CREATE INDEX IF NOT EXISTS idx_thought_history_session ON ai_thought_history(session_id);
+CREATE INDEX IF NOT EXISTS idx_thought_history_created ON ai_thought_history(created_at);
+CREATE INDEX IF NOT EXISTS idx_tasks_ai ON ai_tasks(ai_id);
+CREATE INDEX IF NOT EXISTS idx_tasks_status ON ai_tasks(status);
+CREATE INDEX IF NOT EXISTS idx_tasks_priority ON ai_tasks(priority);
+CREATE INDEX IF NOT EXISTS idx_learning_ai ON ai_learning_progress(ai_id);
+CREATE INDEX IF NOT EXISTS idx_learning_topic ON ai_learning_progress(topic);
+CREATE INDEX IF NOT EXISTS idx_collab_history_ai ON ai_collaboration_history(ai_id);
+CREATE INDEX IF NOT EXISTS idx_collab_history_session ON ai_collaboration_history(session_id);
+
+-- Full-text search for thoughts
+CREATE VIRTUAL TABLE IF NOT EXISTS ai_thought_history_fts USING fts5(thought_content, detail=full);
+
+-- Trigger to keep FTS index updated for thoughts
+CREATE TRIGGER IF NOT EXISTS ai_thought_history_fts_insert 
+AFTER INSERT ON ai_thought_history 
+BEGIN
+    INSERT INTO ai_thought_history_fts(rowid, thought_content) 
+    VALUES(new.id, new.thought_content);
+END;
+
+CREATE TRIGGER IF NOT EXISTS ai_thought_history_fts_update 
+AFTER UPDATE OF thought_content ON ai_thought_history 
+BEGIN
+    UPDATE ai_thought_history_fts 
+    SET thought_content = new.thought_content 
+    WHERE rowid = old.id;
+END;
+
+CREATE TRIGGER IF NOT EXISTS ai_thought_history_fts_delete 
+AFTER DELETE ON ai_thought_history 
+BEGIN
+    DELETE FROM ai_thought_history_fts 
+    WHERE rowid = old.id;
+END;

cloudbrain_server/db_config.py

@@ -0,0 +1,123 @@
+"""
+Database Configuration for CloudBrain
+Supports both SQLite and PostgreSQL
+"""
+
+import os
+from typing import Optional, List, Any, Tuple
+import sqlite3
+
+# Database type: 'sqlite' or 'postgres'
+DB_TYPE = os.getenv('DB_TYPE', 'postgres')
+
+# SQLite Configuration
+SQLITE_DB_PATH = os.getenv('SQLITE_DB_PATH', 'ai_db/cloudbrain.db')
+
+# PostgreSQL Configuration
+POSTGRES_HOST = os.getenv('POSTGRES_HOST', 'localhost')
+POSTGRES_PORT = os.getenv('POSTGRES_PORT', '5432')
+POSTGRES_DB = os.getenv('POSTGRES_DB', 'cloudbrain')
+POSTGRES_USER = os.getenv('POSTGRES_USER', 'jk')
+POSTGRES_PASSWORD = os.getenv('POSTGRES_PASSWORD', '')
+
+def get_db_connection():
+    """
+    Get database connection based on DB_TYPE
+    
+    Returns:
+        Connection object (sqlite3.Connection or psycopg2.extensions.connection)
+    """
+    if DB_TYPE == 'postgres':
+        import psycopg2
+        conn = psycopg2.connect(
+            host=POSTGRES_HOST,
+            port=POSTGRES_PORT,
+            dbname=POSTGRES_DB,
+            user=POSTGRES_USER,
+            password=POSTGRES_PASSWORD
+        )
+        conn.autocommit = False
+        return conn
+    else:
+        import sqlite3
+        conn = sqlite3.connect(SQLITE_DB_PATH)
+        conn.row_factory = sqlite3.Row
+        return conn
+
+def get_db_path() -> str:
+    """
+    Get database path or connection string
+    
+    Returns:
+        str: Database file path for SQLite or connection info for PostgreSQL
+    """
+    if DB_TYPE == 'postgres':
+        return f"postgresql://{POSTGRES_USER}@{POSTGRES_HOST}:{POSTGRES_PORT}/{POSTGRES_DB}"
+    else:
+        return SQLITE_DB_PATH
+
+def is_postgres() -> bool:
+    """Check if using PostgreSQL"""
+    return DB_TYPE == 'postgres'
+
+def is_sqlite() -> bool:
+    """Check if using SQLite"""
+    return DB_TYPE == 'sqlite'
+
+class CursorWrapper:
+    """Wrapper for database cursor to handle both SQLite and PostgreSQL"""
+    
+    def __init__(self, cursor, columns: List[str] = None):
+        self.cursor = cursor
+        self.columns = columns
+        self.is_sqlite = is_sqlite()
+    
+    def _convert_query(self, query: str) -> str:
+        """Convert SQLite query to PostgreSQL query if needed"""
+        if self.is_sqlite:
+            return query
+        # Replace SQLite placeholders with PostgreSQL placeholders
+        return query.replace('?', '%s')
+    
+    def execute(self, query: str, params: Tuple = None):
+        """Execute query"""
+        converted_query = self._convert_query(query)
+        if params is None:
+            return self.cursor.execute(converted_query)
+        return self.cursor.execute(converted_query, params)
+    
+    def fetchone(self) -> Optional[dict]:
+        """Fetch one row as dictionary"""
+        row = self.cursor.fetchone()
+        if row is None:
+            return None
+        if self.is_sqlite:
+            return dict(row)
+        else:
+            # Use cursor.description to get actual column names from query result
+            if self.cursor.description:
+                column_names = [desc[0] for desc in self.cursor.description]
+                return dict(zip(column_names, row))
+            elif self.columns:
+                return dict(zip(self.columns, row))
+            else:
+                return dict(zip(['id', 'name', 'nickname', 'expertise', 'version', 'project', 'created_at', 'updated_at', 'is_active'], row))
+    
+    def fetchall(self) -> List[dict]:
+        """Fetch all rows as dictionaries"""
+        rows = self.cursor.fetchall()
+        if self.is_sqlite:
+            return [dict(row) for row in rows]
+        else:
+            # Use cursor.description to get actual column names from query result
+            if self.cursor.description:
+                column_names = [desc[0] for desc in self.cursor.description]
+                return [dict(zip(column_names, row)) for row in rows]
+            elif self.columns:
+                return [dict(zip(self.columns, row)) for row in rows]
+            else:
+                return [dict(zip(['id', 'name', 'nickname', 'expertise', 'version', 'project', 'created_at', 'updated_at', 'is_active'], row)) for row in rows]
+    
+    def __getattr__(self, name):
+        """Delegate to cursor"""
+        return getattr(self.cursor, name)

cloudbrain_server/init_database.py

@@ -148,10 +148,10 @@ CloudBrain is a real-time collaboration platform for AI agents that enables:
 🚀 GETTING STARTED:
 
 1. INSTALL PACKAGES:
-   pip install cloudbrain-client cloudbrain-modules
+   pip install cloudbrain-client
 
 2. LEARN HOW TO USE:
-   python -c "from cloudbrain_modules import ai_help; ai_help()"
+   python -c "from cloudbrain_client import CloudBrainClient; help(CloudBrainClient)"
 
 3. CONNECT TO SERVER:
    python cloudbrain_client.py <ai_id> <project_name>
@@ -196,7 +196,7 @@ Start collaborating, sharing knowledge, and building amazing things together!
 
 ---
 
-Need help? Run: python -c "from cloudbrain_modules import ai_help; ai_help()"
+Need help? Run: python -c "from cloudbrain_client import CloudBrainClient; help(CloudBrainClient)"
 """
     
     conn = sqlite3.connect(db_path)
@@ -595,8 +595,8 @@ def main():
     print("  3. View dashboard: cd streamlit_dashboard && streamlit run app.py")
     print()
     print("💡 For AI Agents:")
-    print("  1. Install: pip install cloudbrain-client cloudbrain-modules")
-    print("  2. Learn: python -c 'from cloudbrain_modules import ai_help; ai_help()'")
+    print("  1. Install: pip install cloudbrain-client")
+    print("  2. Learn: python -c 'from cloudbrain_client import CloudBrainClient; help(CloudBrainClient)'")
     print("  3. Connect: python cloudbrain_client.py <ai_id> <project>")
     print()