session-context-usage.tsx 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. import { createMemo, Show } from "solid-js"
  2. import { Tooltip } from "@opencode-ai/ui/tooltip"
  3. import { ProgressCircle } from "@opencode-ai/ui/progress-circle"
  4. import { useSync } from "@/context/sync"
  5. import { useParams } from "@solidjs/router"
  6. import { AssistantMessage } from "@opencode-ai/sdk/v2"
  7. export function SessionContextUsage() {
  8. const sync = useSync()
  9. const params = useParams()
  10. const messages = createMemo(() => (params.id ? (sync.data.message[params.id] ?? []) : []))
  11. const cost = createMemo(() => {
  12. const total = messages().reduce((sum, x) => sum + (x.role === "assistant" ? x.cost : 0), 0)
  13. return new Intl.NumberFormat("en-US", {
  14. style: "currency",
  15. currency: "USD",
  16. }).format(total)
  17. })
  18. const context = createMemo(() => {
  19. const last = messages().findLast((x) => x.role === "assistant" && x.tokens.output > 0) as AssistantMessage
  20. if (!last) return
  21. const total =
  22. last.tokens.input + last.tokens.output + last.tokens.reasoning + last.tokens.cache.read + last.tokens.cache.write
  23. const model = sync.data.provider.all.find((x) => x.id === last.providerID)?.models[last.modelID]
  24. return {
  25. tokens: total.toLocaleString(),
  26. percentage: model?.limit.context ? Math.round((total / model.limit.context) * 100) : null,
  27. }
  28. })
  29. return (
  30. <Show when={context?.()}>
  31. {(ctx) => (
  32. <Tooltip
  33. value={
  34. <div class="flex flex-col gap-1">
  35. <div class="flex gap-3">
  36. <span class="opacity-70 text-right flex-1">Tokens</span>
  37. <span class="text-left flex-1">{ctx().tokens}</span>
  38. </div>
  39. <div class="flex gap-3">
  40. <span class="opacity-70 text-right flex-1">Usage</span>
  41. <span class="text-left flex-1">{ctx().percentage ?? 0}%</span>
  42. </div>
  43. <div class="flex gap-3">
  44. <span class="opacity-70 text-right flex-1">Cost</span>
  45. <span class="text-left flex-1">{cost()}</span>
  46. </div>
  47. </div>
  48. }
  49. placement="top"
  50. >
  51. <div class="flex items-center gap-1.5">
  52. <ProgressCircle size={16} strokeWidth={2} percentage={ctx().percentage ?? 0} />
  53. {/* <span class="text-12-medium text-text-weak">{`${ctx().percentage ?? 0}%`}</span> */}
  54. </div>
  55. </Tooltip>
  56. )}
  57. </Show>
  58. )
  59. }