3ccb67bddd
Adds pkg/ratelimit for Gemini API rate limiting with sliding window
(RPM/TPM/RPD), persistent state, and token counting. Replaces the
bash agent-runner.sh with a native Go implementation under
`core ai dispatch {run,watch,status}` for local queue processing.
Rate limiting:
- Per-model quotas (RPM, TPM, RPD) with 1-minute sliding window
- WaitForCapacity blocks until capacity available or context cancelled
- Persistent state in ~/.core/ratelimits.yaml
- Default quotas for Gemini 3 Pro/Flash, 2.5 Pro, 2.0 Flash/Lite
- CountTokens helper calls Google tokenizer API
- CLI: core ai ratelimits {show,reset,count,config,check}
Dispatch runner:
- core ai dispatch run — process single ticket from queue
- core ai dispatch watch — daemon mode with configurable interval
- core ai dispatch status — show queue/active/done counts
- Supports claude/codex/gemini runners with rate-limited Gemini
- File-based locking with stale PID detection
- Completion handler updates issue labels on success/failure
Closes #42
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
94 lines
2.3 KiB
Go
94 lines
2.3 KiB
Go
// Package ai provides AI agent task management and Claude Code integration.
|
|
//
|
|
// Commands:
|
|
// - tasks: List tasks from the agentic service
|
|
// - task: View, claim, or auto-select tasks
|
|
// - task:update: Update task status and progress
|
|
// - task:complete: Mark tasks as completed or failed
|
|
// - task:commit: Create commits with task references
|
|
// - task:pr: Create pull requests linked to tasks
|
|
// - claude: Claude Code CLI integration (planned)
|
|
// - rag: RAG tools (ingest, query, collections)
|
|
// - metrics: View AI/security event metrics
|
|
package ai
|
|
|
|
import (
|
|
ragcmd "github.com/host-uk/core/internal/cmd/rag"
|
|
"github.com/host-uk/core/pkg/cli"
|
|
"github.com/host-uk/core/pkg/i18n"
|
|
)
|
|
|
|
func init() {
|
|
cli.RegisterCommands(AddAICommands)
|
|
}
|
|
|
|
var aiCmd = &cli.Command{
|
|
Use: "ai",
|
|
Short: i18n.T("cmd.ai.short"),
|
|
Long: i18n.T("cmd.ai.long"),
|
|
}
|
|
|
|
var claudeCmd = &cli.Command{
|
|
Use: "claude",
|
|
Short: i18n.T("cmd.ai.claude.short"),
|
|
Long: i18n.T("cmd.ai.claude.long"),
|
|
}
|
|
|
|
var claudeRunCmd = &cli.Command{
|
|
Use: "run",
|
|
Short: i18n.T("cmd.ai.claude.run.short"),
|
|
RunE: func(cmd *cli.Command, args []string) error {
|
|
return runClaudeCode()
|
|
},
|
|
}
|
|
|
|
var claudeConfigCmd = &cli.Command{
|
|
Use: "config",
|
|
Short: i18n.T("cmd.ai.claude.config.short"),
|
|
RunE: func(cmd *cli.Command, args []string) error {
|
|
return showClaudeConfig()
|
|
},
|
|
}
|
|
|
|
func initCommands() {
|
|
// Add Claude subcommands
|
|
claudeCmd.AddCommand(claudeRunCmd)
|
|
claudeCmd.AddCommand(claudeConfigCmd)
|
|
|
|
// Add Claude command to ai
|
|
aiCmd.AddCommand(claudeCmd)
|
|
|
|
// Add agentic task commands
|
|
AddAgenticCommands(aiCmd)
|
|
|
|
// Add RAG subcommands (core ai rag ...)
|
|
ragcmd.AddRAGSubcommands(aiCmd)
|
|
|
|
// Add metrics subcommand (core ai metrics)
|
|
addMetricsCommand(aiCmd)
|
|
|
|
// Add agent management commands (core ai agent ...)
|
|
AddAgentCommands(aiCmd)
|
|
|
|
// Add rate limit management commands (core ai ratelimits ...)
|
|
AddRateLimitCommands(aiCmd)
|
|
|
|
// Add dispatch commands (core ai dispatch run/watch/status)
|
|
AddDispatchCommands(aiCmd)
|
|
}
|
|
|
|
// AddAICommands registers the 'ai' command and all subcommands.
|
|
func AddAICommands(root *cli.Command) {
|
|
initCommands()
|
|
root.AddCommand(aiCmd)
|
|
}
|
|
|
|
func runClaudeCode() error {
|
|
// Placeholder - will integrate with claude CLI
|
|
return nil
|
|
}
|
|
|
|
func showClaudeConfig() error {
|
|
// Placeholder - will show claude configuration
|
|
return nil
|
|
}
|