oxios 1.13.0

Oxios Agent OS — Agent Operating System powered by oxi-sdk
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80

import { Gauge } from 'lucide-react'
import { useState } from 'react'
import { useTranslation } from 'react-i18next'
import { AgentBudgetCard } from '@/components/budget/agent-budget-card'
import { BudgetSummaryCard } from '@/components/budget/budget-summary'
import { SetBudgetDialog } from '@/components/budget/set-budget-dialog'
import { Card, CardContent } from '@/components/ui/card'
import { useBudgetDelete, useBudgetList, useBudgetReset, useBudgetSet } from '@/hooks/use-budget'
import type { AgentBudget } from '@/types/budget'

/** Agent budget management section — token/call budgets with sliding windows.
 *
 * Wires the previously orphaned budget components (AgentBudgetCard,
 * SetBudgetDialog, BudgetSummaryCard) into the cost page. These are
 * rate-limiting budgets (tokens + calls per window), separate from the
 * dollar-based spend limit.
 */
export function BudgetManagement() {
  const { t } = useTranslation()
  const { data, isLoading } = useBudgetList()
  const setMutation = useBudgetSet()
  const deleteMutation = useBudgetDelete()
  const resetMutation = useBudgetReset()

  const [dialogOpen, setDialogOpen] = useState(false)
  const [editingAgent, setEditingAgent] = useState<AgentBudget | null>(null)

  const agents = data?.agents ?? []
  const summary = data?.summary

  const handleEdit = (agent: AgentBudget) => {
    setEditingAgent(agent)
    setDialogOpen(true)
  }

  return (
    <div className="space-y-4">
      {summary && <BudgetSummaryCard summary={summary} />}

      <div className="flex items-center justify-between">
        <h3 className="flex items-center gap-2 text-sm font-medium text-muted-foreground">
          <Gauge className="h-4 w-4" />
          {t('budget.agentBudgets')}
        </h3>
      </div>

      {isLoading ? (
        <p className="text-sm text-muted-foreground py-4">{t('common.loading')}</p>
      ) : agents.length === 0 ? (
        <Card>
          <CardContent className="py-8 text-center">
            <p className="text-sm text-muted-foreground">{t('budget.noBudgetDataDescription')}</p>
          </CardContent>
        </Card>
      ) : (
        <div className="grid gap-4 sm:grid-cols-2 lg:grid-cols-3">
          {agents.map((agent) => (
            <AgentBudgetCard
              key={agent.agent_id}
              agent={agent}
              onEdit={handleEdit}
              onReset={(id) => resetMutation.mutate(id)}
              onRemove={(id) => deleteMutation.mutate(id)}
              isResetting={resetMutation.isPending}
              isRemoving={deleteMutation.isPending}
            />
          ))}
        </div>
      )}

      <SetBudgetDialog
        open={dialogOpen}
        onOpenChange={setDialogOpen}
        agent={editingAgent}
        onSubmit={(data) => setMutation.mutate(data)}
        isPending={setMutation.isPending}
      />
    </div>
  )
}