@liangjie559567/ultrapower 5.5.12 → 5.5.13

package/templates/axiom/scripts/Check-Memory.ps1

@@ -1,68 +1,68 @@
-
-param (
-    [Parameter(Mandatory=$true)]
-    [string]$SessionId,
-    
-    [int]$Threshold = 2
-)
-
-$ErrorActionPreference = "SilentlyContinue"
-
-# 1. Define Paths
-$AgentRoot = Resolve-Path "$PSScriptRoot\..\.."
-$MemoryDir = Join-Path $AgentRoot ".agent\memory"
-$TargetFile = Join-Path $MemoryDir "active_context.md"
-$StatusFile = Join-Path $MemoryDir "watchdog_status.lock"
-$TempFile = Join-Path $MemoryDir "watchdog_status.tmp"
-
-# 2. Check Context Load (Latest PB File Size)
-$Status = "NORMAL"
-$SizeMB = 0
-
-# Path to Axiom Conversations Directory
-# Dynamic detection for Windows ($env:USERPROFILE) and macOS/Linux ($env:HOME)
-$HomeDir = if ($env:USERPROFILE) { $env:USERPROFILE } else { $env:HOME }
-$ConversationsDir = Join-Path $HomeDir ".gemini\axiom\conversations"
-
-if (Test-Path $ConversationsDir) {
-    # Get the most recently modified .pb file
-    $LatestPb = Get-ChildItem -Path $ConversationsDir -Filter "*.pb" | Sort-Object LastWriteTime -Descending | Select-Object -First 1
-    
-    if ($LatestPb) {
-        $SizeBytes = $LatestPb.Length
-        $SizeMB = [math]::Round($SizeBytes / 1MB, 4)
-        
-        Write-Host " [Watchdog] Monitoring: $($LatestPb.Name) ($SizeMB MB)" -ForegroundColor DarkGray
-        
-        # Threshold handling
-        if ($SizeMB -ge $Threshold) {
-            $Status = "CRITICAL"
-        } elseif ($SizeMB -ge ($Threshold * 0.8)) {
-            $Status = "WARNING"
-        }
-    } else {
-         Write-Host " [Watchdog] No .pb files found to monitor." -ForegroundColor Yellow
-    }
-} else {
-    Write-Host " [Watchdog] Conversations directory not found: $ConversationsDir" -ForegroundColor Red
-}
-
-# 3. Create JSON Payload
-$Payload = @{
-    status = $Status
-    size_mb = $SizeMB
-    limit_mb = $Threshold
-    timestamp = [int64]((Get-Date).ToUniversalTime() - (Get-Date "1970-01-01")).TotalSeconds
-    session_id = $SessionId
-} | ConvertTo-Json -Compress
-
-# 4. Atomic Write
-try {
-    $Payload | Set-Content -Path $TempFile -Encoding UTF8 -Force
-    Move-Item -Path $TempFile -Destination $StatusFile -Force
-} catch {
-    # Silent fail, maybe log if verbose
-    if ($PSBoundParameters['Verbose']) {
-        Write-Error $_
-    }
-}
+
+param (
+    [Parameter(Mandatory=$true)]
+    [string]$SessionId,
+    
+    [int]$Threshold = 2
+)
+
+$ErrorActionPreference = "SilentlyContinue"
+
+# 1. Define Paths
+$AgentRoot = Resolve-Path "$PSScriptRoot\..\.."
+$MemoryDir = Join-Path $AgentRoot ".agent\memory"
+$TargetFile = Join-Path $MemoryDir "active_context.md"
+$StatusFile = Join-Path $MemoryDir "watchdog_status.lock"
+$TempFile = Join-Path $MemoryDir "watchdog_status.tmp"
+
+# 2. Check Context Load (Latest PB File Size)
+$Status = "NORMAL"
+$SizeMB = 0
+
+# Path to Axiom Conversations Directory
+# Dynamic detection for Windows ($env:USERPROFILE) and macOS/Linux ($env:HOME)
+$HomeDir = if ($env:USERPROFILE) { $env:USERPROFILE } else { $env:HOME }
+$ConversationsDir = Join-Path $HomeDir ".gemini\axiom\conversations"
+
+if (Test-Path $ConversationsDir) {
+    # Get the most recently modified .pb file
+    $LatestPb = Get-ChildItem -Path $ConversationsDir -Filter "*.pb" | Sort-Object LastWriteTime -Descending | Select-Object -First 1
+    
+    if ($LatestPb) {
+        $SizeBytes = $LatestPb.Length
+        $SizeMB = [math]::Round($SizeBytes / 1MB, 4)
+        
+        Write-Host " [Watchdog] Monitoring: $($LatestPb.Name) ($SizeMB MB)" -ForegroundColor DarkGray
+        
+        # Threshold handling
+        if ($SizeMB -ge $Threshold) {
+            $Status = "CRITICAL"
+        } elseif ($SizeMB -ge ($Threshold * 0.8)) {
+            $Status = "WARNING"
+        }
+    } else {
+         Write-Host " [Watchdog] No .pb files found to monitor." -ForegroundColor Yellow
+    }
+} else {
+    Write-Host " [Watchdog] Conversations directory not found: $ConversationsDir" -ForegroundColor Red
+}
+
+# 3. Create JSON Payload
+$Payload = @{
+    status = $Status
+    size_mb = $SizeMB
+    limit_mb = $Threshold
+    timestamp = [int64]((Get-Date).ToUniversalTime() - (Get-Date "1970-01-01")).TotalSeconds
+    session_id = $SessionId
+} | ConvertTo-Json -Compress
+
+# 4. Atomic Write
+try {
+    $Payload | Set-Content -Path $TempFile -Encoding UTF8 -Force
+    Move-Item -Path $TempFile -Destination $StatusFile -Force
+} catch {
+    # Silent fail, maybe log if verbose
+    if ($PSBoundParameters['Verbose']) {
+        Write-Error $_
+    }
+}

package/templates/axiom/scripts/Poll-Memory.ps1

@@ -1,36 +1,36 @@
-param([int]$Interval = 2)
-
-$HomeDir = if ($env:USERPROFILE) { $env:USERPROFILE } else { $env:HOME }
-if (-not $HomeDir) {
-    Write-Host "Error: Could not determine HOME directory." -ForegroundColor Red
-    exit 1
-}
-
-$TargetDir = Join-Path $HomeDir ".gemini\axiom\conversations"
-Write-Host "Polling $TargetDir every $Interval seconds..." -ForegroundColor Cyan
-
-$LastTime = $null
-$LastSize = 0
-
-while ($true) {
-    if (Test-Path $TargetDir) {
-        $Latest = Get-ChildItem -Path $TargetDir -Filter "*.pb" | Sort-Object LastWriteTime -Descending | Select-Object -First 1
-        
-        if ($Latest) {
-            $CurrentTime = $Latest.LastWriteTime
-            $CurrentSize = $Latest.Length
-            
-            # If time changed or size changed
-            if ($LastTime -ne $CurrentTime -or $LastSize -ne $CurrentSize) {
-                $SizeMB = [math]::Round($CurrentSize / 1MB, 4)
-                Write-Host " [POLL] Change Detected: $($Latest.Name)" -ForegroundColor Yellow
-                Write-Host "    Time: $CurrentTime"
-                Write-Host "    Size: $SizeMB MB"
-                
-                $LastTime = $CurrentTime
-                $LastSize = $CurrentSize
-            }
-        }
-    }
-    Start-Sleep -Seconds $Interval
-}
+param([int]$Interval = 2)
+
+$HomeDir = if ($env:USERPROFILE) { $env:USERPROFILE } else { $env:HOME }
+if (-not $HomeDir) {
+    Write-Host "Error: Could not determine HOME directory." -ForegroundColor Red
+    exit 1
+}
+
+$TargetDir = Join-Path $HomeDir ".gemini\axiom\conversations"
+Write-Host "Polling $TargetDir every $Interval seconds..." -ForegroundColor Cyan
+
+$LastTime = $null
+$LastSize = 0
+
+while ($true) {
+    if (Test-Path $TargetDir) {
+        $Latest = Get-ChildItem -Path $TargetDir -Filter "*.pb" | Sort-Object LastWriteTime -Descending | Select-Object -First 1
+        
+        if ($Latest) {
+            $CurrentTime = $Latest.LastWriteTime
+            $CurrentSize = $Latest.Length
+            
+            # If time changed or size changed
+            if ($LastTime -ne $CurrentTime -or $LastSize -ne $CurrentSize) {
+                $SizeMB = [math]::Round($CurrentSize / 1MB, 4)
+                Write-Host " [POLL] Change Detected: $($Latest.Name)" -ForegroundColor Yellow
+                Write-Host "    Time: $CurrentTime"
+                Write-Host "    Size: $SizeMB MB"
+                
+                $LastTime = $CurrentTime
+                $LastSize = $CurrentSize
+            }
+        }
+    }
+    Start-Sleep -Seconds $Interval
+}

package/templates/axiom/scripts/Watch-Memory.ps1

@@ -1,149 +1,149 @@
-#!/usr/bin/env pwsh
-# .agent/scripts/Watch-Memory.ps1
-# Real-time Memory Watchdog: Monitors .pb file changes and updates status lock.
-
-param(
-    [int]$ThresholdMB = 2,
-    [bool]$ShowToasts = $true
-)
-
-# 1. Setup Paths
-$ScriptDir = $PSScriptRoot
-$AgentRoot = Resolve-Path "$ScriptDir\..\.."
-$MemoryDir = Join-Path $AgentRoot ".agent\memory"
-$StatusFile = Join-Path $MemoryDir "watchdog_status.lock"
-$TempFile = Join-Path $MemoryDir "watchdog_status.tmp"
-
-# Dynamic detection for Axiom Paths
-$HomeDir = [Environment]::GetFolderPath("UserProfile")
-if ([string]::IsNullOrEmpty($HomeDir)) {
-    $HomeDir = $HOME
-}
-
-if ([string]::IsNullOrEmpty($HomeDir)) {
-    Write-Host "Error: Could not determine HOME directory." -ForegroundColor Red
-    exit 1
-}
-
-$ConversationsDir = Join-Path $HomeDir ".gemini\axiom\conversations"
-
-if (-not (Test-Path $ConversationsDir)) {
-    Write-Host "Error: Conversations directory not found: $ConversationsDir" -ForegroundColor Red
-    exit 1
-}
-
-Write-Host " [Watchdog] Starting Real-time Monitor..." -ForegroundColor Cyan
-Write-Host "   Target: $ConversationsDir" -ForegroundColor DarkGray
-Write-Host "   Threshold: $ThresholdMB MB" -ForegroundColor DarkGray
-Write-Host "   Status File: $StatusFile" -ForegroundColor DarkGray
-
-# Global Cached State
-$Global:LastCheck = @{
-    FileName = ""
-    SizeMB   = 0.0
-    Status   = "NORMAL"
-    Time     = 0 # timestamp
-}
-
-# 2. Define Action Block
-$Action = {
-    $FilePath = $Event.SourceEventArgs.FullPath
-    $ChangeType = $Event.SourceEventArgs.ChangeType
-    $FileName = $Event.SourceEventArgs.Name
-    
-    # We only care about .pb files
-    if ($FileName -like "*.pb") {
-        try {
-            # Debounce: Wait a tiny bit for write to complete
-            Start-Sleep -Milliseconds 200
-            
-            $Item = Get-Item $FilePath -ErrorAction SilentlyContinue
-            if ($null -eq $Item) { return }
-            
-            $SizeMB = [math]::Round($Item.Length / 1MB, 4)
-            $Status = "NORMAL"
-            
-            if ($SizeMB -ge $ThresholdMB) {
-                $Status = "CRITICAL"
-                $Color = "Red"
-            }
-            elseif ($SizeMB -ge ($ThresholdMB * 0.8)) {
-                $Status = "WARNING"
-                $Color = "Yellow"
-            }
-            else {
-                $Color = "Green"
-            }
-
-            # --- OPTIMIZATION START ---
-            $CurrentTime = [int64]((Get-Date).ToUniversalTime() - (Get-Date "1970-01-01")).TotalSeconds
-            
-            # Check if redundant update
-            $IsSameFile = ($FileName -eq $Global:LastCheck.FileName)
-            $IsSameStatus = ($Status -eq $Global:LastCheck.Status)
-            $SizeDiff = [math]::Abs($SizeMB - $Global:LastCheck.SizeMB)
-            $TimeSinceLast = $CurrentTime - $Global:LastCheck.Time
-
-            # Rule 1: If exactly same status and size negligible change (< 0.01 MB), skip
-            if ($IsSameFile -and $IsSameStatus -and ($SizeDiff -lt 0.01)) {
-                return
-            }
-            
-            # Update Cache
-            $Global:LastCheck.FileName = $FileName
-            $Global:LastCheck.SizeMB = $SizeMB
-            $Global:LastCheck.Status = $Status
-            $Global:LastCheck.Time = $CurrentTime
-            # --- OPTIMIZATION END ---
-            
-            # Print Log
-            $TimeStr = Get-Date -Format "HH:mm:ss"
-            Write-Host (" [{0}] {1}: {2} -> {3} MB ({4})" -f $TimeStr, $ChangeType, $FileName, $SizeMB, $Status) -ForegroundColor $Color
-
-            # Update Lock File Logic
-            if ($Status -ne "NORMAL") {
-                $Payload = @{
-                    status     = $Status
-                    size_mb    = $SizeMB
-                    limit_mb   = $ThresholdMB
-                    timestamp  = $CurrentTime
-                    session_id = $FileName
-                } | ConvertTo-Json -Compress
-                
-                $Payload | Set-Content -Path $TempFile -Encoding UTF8 -Force
-                Move-Item -Path $TempFile -Destination $StatusFile -Force
-            }
-            else {
-                # Clear lock file if back to normal (But only do it ONCE when status CHANGES to normal)
-                if ($IsSameStatus -eq $false -and (Test-Path $StatusFile)) {
-                    $Payload = @{ status = "NORMAL"; timestamp = $CurrentTime; session_id = $FileName } | ConvertTo-Json -Compress
-                    $Payload | Set-Content -Path $TempFile -Encoding UTF8 -Force
-                    Move-Item -Path $TempFile -Destination $StatusFile -Force
-                    Write-Host " [Watchdog] cleared status lock." -ForegroundColor Gray
-                }
-            }
-        }
-        catch {
-            Write-Host "Error processing event: $_" -ForegroundColor Red
-        }
-    }
-}
-
-# 3. Setup FileSystemWatcher
-$Watcher = New-Object System.IO.FileSystemWatcher
-$Watcher.Path = $ConversationsDir
-$Watcher.Filter = "*.pb"
-$Watcher.IncludeSubdirectories = $false
-$Watcher.EnableRaisingEvents = $true
-
-# Register Events
-Register-ObjectEvent $Watcher "Changed" -Action $Action | Out-Null
-Register-ObjectEvent $Watcher "Created" -Action $Action | Out-Null
-Register-ObjectEvent $Watcher "Renamed" -Action $Action | Out-Null
-
-Write-Host " [Watchdog] Monitoring active. Press Ctrl+C to stop." -ForegroundColor Green
-
-# Keep script running
-while ($true) {
-    Start-Sleep -Seconds 1
-}
+#!/usr/bin/env pwsh
+# .agent/scripts/Watch-Memory.ps1
+# Real-time Memory Watchdog: Monitors .pb file changes and updates status lock.
+
+param(
+    [int]$ThresholdMB = 2,
+    [bool]$ShowToasts = $true
+)
+
+# 1. Setup Paths
+$ScriptDir = $PSScriptRoot
+$AgentRoot = Resolve-Path "$ScriptDir\..\.."
+$MemoryDir = Join-Path $AgentRoot ".agent\memory"
+$StatusFile = Join-Path $MemoryDir "watchdog_status.lock"
+$TempFile = Join-Path $MemoryDir "watchdog_status.tmp"
+
+# Dynamic detection for Axiom Paths
+$HomeDir = [Environment]::GetFolderPath("UserProfile")
+if ([string]::IsNullOrEmpty($HomeDir)) {
+    $HomeDir = $HOME
+}
+
+if ([string]::IsNullOrEmpty($HomeDir)) {
+    Write-Host "Error: Could not determine HOME directory." -ForegroundColor Red
+    exit 1
+}
+
+$ConversationsDir = Join-Path $HomeDir ".gemini\axiom\conversations"
+
+if (-not (Test-Path $ConversationsDir)) {
+    Write-Host "Error: Conversations directory not found: $ConversationsDir" -ForegroundColor Red
+    exit 1
+}
+
+Write-Host " [Watchdog] Starting Real-time Monitor..." -ForegroundColor Cyan
+Write-Host "   Target: $ConversationsDir" -ForegroundColor DarkGray
+Write-Host "   Threshold: $ThresholdMB MB" -ForegroundColor DarkGray
+Write-Host "   Status File: $StatusFile" -ForegroundColor DarkGray
+
+# Global Cached State
+$Global:LastCheck = @{
+    FileName = ""
+    SizeMB   = 0.0
+    Status   = "NORMAL"
+    Time     = 0 # timestamp
+}
+
+# 2. Define Action Block
+$Action = {
+    $FilePath = $Event.SourceEventArgs.FullPath
+    $ChangeType = $Event.SourceEventArgs.ChangeType
+    $FileName = $Event.SourceEventArgs.Name
+    
+    # We only care about .pb files
+    if ($FileName -like "*.pb") {
+        try {
+            # Debounce: Wait a tiny bit for write to complete
+            Start-Sleep -Milliseconds 200
+            
+            $Item = Get-Item $FilePath -ErrorAction SilentlyContinue
+            if ($null -eq $Item) { return }
+            
+            $SizeMB = [math]::Round($Item.Length / 1MB, 4)
+            $Status = "NORMAL"
+            
+            if ($SizeMB -ge $ThresholdMB) {
+                $Status = "CRITICAL"
+                $Color = "Red"
+            }
+            elseif ($SizeMB -ge ($ThresholdMB * 0.8)) {
+                $Status = "WARNING"
+                $Color = "Yellow"
+            }
+            else {
+                $Color = "Green"
+            }
+
+            # --- OPTIMIZATION START ---
+            $CurrentTime = [int64]((Get-Date).ToUniversalTime() - (Get-Date "1970-01-01")).TotalSeconds
+            
+            # Check if redundant update
+            $IsSameFile = ($FileName -eq $Global:LastCheck.FileName)
+            $IsSameStatus = ($Status -eq $Global:LastCheck.Status)
+            $SizeDiff = [math]::Abs($SizeMB - $Global:LastCheck.SizeMB)
+            $TimeSinceLast = $CurrentTime - $Global:LastCheck.Time
+
+            # Rule 1: If exactly same status and size negligible change (< 0.01 MB), skip
+            if ($IsSameFile -and $IsSameStatus -and ($SizeDiff -lt 0.01)) {
+                return
+            }
+            
+            # Update Cache
+            $Global:LastCheck.FileName = $FileName
+            $Global:LastCheck.SizeMB = $SizeMB
+            $Global:LastCheck.Status = $Status
+            $Global:LastCheck.Time = $CurrentTime
+            # --- OPTIMIZATION END ---
+            
+            # Print Log
+            $TimeStr = Get-Date -Format "HH:mm:ss"
+            Write-Host (" [{0}] {1}: {2} -> {3} MB ({4})" -f $TimeStr, $ChangeType, $FileName, $SizeMB, $Status) -ForegroundColor $Color
+
+            # Update Lock File Logic
+            if ($Status -ne "NORMAL") {
+                $Payload = @{
+                    status     = $Status
+                    size_mb    = $SizeMB
+                    limit_mb   = $ThresholdMB
+                    timestamp  = $CurrentTime
+                    session_id = $FileName
+                } | ConvertTo-Json -Compress
+                
+                $Payload | Set-Content -Path $TempFile -Encoding UTF8 -Force
+                Move-Item -Path $TempFile -Destination $StatusFile -Force
+            }
+            else {
+                # Clear lock file if back to normal (But only do it ONCE when status CHANGES to normal)
+                if ($IsSameStatus -eq $false -and (Test-Path $StatusFile)) {
+                    $Payload = @{ status = "NORMAL"; timestamp = $CurrentTime; session_id = $FileName } | ConvertTo-Json -Compress
+                    $Payload | Set-Content -Path $TempFile -Encoding UTF8 -Force
+                    Move-Item -Path $TempFile -Destination $StatusFile -Force
+                    Write-Host " [Watchdog] cleared status lock." -ForegroundColor Gray
+                }
+            }
+        }
+        catch {
+            Write-Host "Error processing event: $_" -ForegroundColor Red
+        }
+    }
+}
+
+# 3. Setup FileSystemWatcher
+$Watcher = New-Object System.IO.FileSystemWatcher
+$Watcher.Path = $ConversationsDir
+$Watcher.Filter = "*.pb"
+$Watcher.IncludeSubdirectories = $false
+$Watcher.EnableRaisingEvents = $true
+
+# Register Events
+Register-ObjectEvent $Watcher "Changed" -Action $Action | Out-Null
+Register-ObjectEvent $Watcher "Created" -Action $Action | Out-Null
+Register-ObjectEvent $Watcher "Renamed" -Action $Action | Out-Null
+
+Write-Host " [Watchdog] Monitoring active. Press Ctrl+C to stop." -ForegroundColor Green
+
+# Keep script running
+while ($true) {
+    Start-Sleep -Seconds 1
+}