reigncode-app 1.3.2
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/AGENTS.md +30 -0
- package/Dockerfile +21 -0
- package/README.md +51 -0
- package/bunfig.toml +3 -0
- package/create-effect-simplification-spec.md +515 -0
- package/e2e/AGENTS.md +226 -0
- package/e2e/actions.ts +1018 -0
- package/e2e/app/home.spec.ts +24 -0
- package/e2e/app/navigation.spec.ts +10 -0
- package/e2e/app/palette.spec.ts +20 -0
- package/e2e/app/server-default.spec.ts +58 -0
- package/e2e/app/session.spec.ts +16 -0
- package/e2e/app/titlebar-history.spec.ts +120 -0
- package/e2e/commands/input-focus.spec.ts +15 -0
- package/e2e/commands/panels.spec.ts +33 -0
- package/e2e/commands/tab-close.spec.ts +32 -0
- package/e2e/files/file-open.spec.ts +31 -0
- package/e2e/files/file-tree.spec.ts +56 -0
- package/e2e/files/file-viewer.spec.ts +156 -0
- package/e2e/fixtures.ts +154 -0
- package/e2e/models/model-picker.spec.ts +48 -0
- package/e2e/models/models-visibility.spec.ts +61 -0
- package/e2e/projects/project-edit.spec.ts +43 -0
- package/e2e/projects/projects-close.spec.ts +54 -0
- package/e2e/projects/projects-switch.spec.ts +116 -0
- package/e2e/projects/workspace-new-session.spec.ts +94 -0
- package/e2e/projects/workspaces.spec.ts +375 -0
- package/e2e/prompt/context.spec.ts +95 -0
- package/e2e/prompt/prompt-async.spec.ts +76 -0
- package/e2e/prompt/prompt-drop-file-uri.spec.ts +22 -0
- package/e2e/prompt/prompt-drop-file.spec.ts +30 -0
- package/e2e/prompt/prompt-history.spec.ts +184 -0
- package/e2e/prompt/prompt-mention.spec.ts +26 -0
- package/e2e/prompt/prompt-multiline.spec.ts +24 -0
- package/e2e/prompt/prompt-shell.spec.ts +62 -0
- package/e2e/prompt/prompt-slash-open.spec.ts +22 -0
- package/e2e/prompt/prompt-slash-share.spec.ts +64 -0
- package/e2e/prompt/prompt-slash-terminal.spec.ts +18 -0
- package/e2e/prompt/prompt.spec.ts +55 -0
- package/e2e/selectors.ts +75 -0
- package/e2e/session/session-child-navigation.spec.ts +37 -0
- package/e2e/session/session-composer-dock.spec.ts +530 -0
- package/e2e/session/session-model-persistence.spec.ts +359 -0
- package/e2e/session/session-review.spec.ts +426 -0
- package/e2e/session/session-undo-redo.spec.ts +233 -0
- package/e2e/session/session.spec.ts +174 -0
- package/e2e/settings/settings-keybinds.spec.ts +389 -0
- package/e2e/settings/settings-models.spec.ts +122 -0
- package/e2e/settings/settings-providers.spec.ts +136 -0
- package/e2e/settings/settings.spec.ts +519 -0
- package/e2e/sidebar/sidebar-popover-actions.spec.ts +118 -0
- package/e2e/sidebar/sidebar-session-links.spec.ts +30 -0
- package/e2e/sidebar/sidebar.spec.ts +40 -0
- package/e2e/status/status-popover.spec.ts +94 -0
- package/e2e/terminal/terminal-init.spec.ts +28 -0
- package/e2e/terminal/terminal-reconnect.spec.ts +46 -0
- package/e2e/terminal/terminal-tabs.spec.ts +168 -0
- package/e2e/terminal/terminal.spec.ts +18 -0
- package/e2e/thinking-level.spec.ts +25 -0
- package/e2e/tsconfig.json +9 -0
- package/e2e/utils.ts +63 -0
- package/happydom.ts +75 -0
- package/index.html +23 -0
- package/package.json +77 -0
- package/playwright.config.ts +45 -0
- package/public/_headers +17 -0
- package/public/oc-theme-preload.js +35 -0
- package/script/e2e-local.ts +180 -0
- package/src/addons/serialize.test.ts +319 -0
- package/src/addons/serialize.ts +634 -0
- package/src/app.tsx +308 -0
- package/src/components/debug-bar.tsx +443 -0
- package/src/components/dialog-connect-provider.tsx +617 -0
- package/src/components/dialog-custom-provider-form.ts +158 -0
- package/src/components/dialog-custom-provider.test.ts +80 -0
- package/src/components/dialog-custom-provider.tsx +329 -0
- package/src/components/dialog-edit-project.tsx +255 -0
- package/src/components/dialog-fork.tsx +108 -0
- package/src/components/dialog-manage-models.tsx +101 -0
- package/src/components/dialog-release-notes.tsx +144 -0
- package/src/components/dialog-select-directory.tsx +392 -0
- package/src/components/dialog-select-file.tsx +466 -0
- package/src/components/dialog-select-mcp.tsx +107 -0
- package/src/components/dialog-select-model-unpaid.tsx +137 -0
- package/src/components/dialog-select-model.tsx +220 -0
- package/src/components/dialog-select-provider.tsx +86 -0
- package/src/components/dialog-select-server.tsx +649 -0
- package/src/components/dialog-settings.tsx +73 -0
- package/src/components/file-tree.test.ts +78 -0
- package/src/components/file-tree.tsx +507 -0
- package/src/components/link.tsx +26 -0
- package/src/components/model-tooltip.tsx +91 -0
- package/src/components/prompt-input/attachments.test.ts +44 -0
- package/src/components/prompt-input/attachments.ts +201 -0
- package/src/components/prompt-input/build-request-parts.test.ts +312 -0
- package/src/components/prompt-input/build-request-parts.ts +175 -0
- package/src/components/prompt-input/context-items.tsx +88 -0
- package/src/components/prompt-input/drag-overlay.tsx +25 -0
- package/src/components/prompt-input/editor-dom.test.ts +99 -0
- package/src/components/prompt-input/editor-dom.ts +148 -0
- package/src/components/prompt-input/files.ts +66 -0
- package/src/components/prompt-input/history.test.ts +153 -0
- package/src/components/prompt-input/history.ts +256 -0
- package/src/components/prompt-input/image-attachments.tsx +58 -0
- package/src/components/prompt-input/paste.ts +24 -0
- package/src/components/prompt-input/placeholder.test.ts +48 -0
- package/src/components/prompt-input/placeholder.ts +15 -0
- package/src/components/prompt-input/slash-popover.tsx +141 -0
- package/src/components/prompt-input/submit.test.ts +346 -0
- package/src/components/prompt-input/submit.ts +579 -0
- package/src/components/prompt-input.tsx +1595 -0
- package/src/components/server/server-row.tsx +130 -0
- package/src/components/session/index.ts +5 -0
- package/src/components/session/session-context-breakdown.test.ts +61 -0
- package/src/components/session/session-context-breakdown.ts +132 -0
- package/src/components/session/session-context-format.ts +20 -0
- package/src/components/session/session-context-metrics.test.ts +101 -0
- package/src/components/session/session-context-metrics.ts +82 -0
- package/src/components/session/session-context-tab.tsx +339 -0
- package/src/components/session/session-header.tsx +486 -0
- package/src/components/session/session-new-view.tsx +91 -0
- package/src/components/session/session-sortable-tab.tsx +70 -0
- package/src/components/session/session-sortable-terminal-tab.tsx +193 -0
- package/src/components/session-context-usage.tsx +122 -0
- package/src/components/settings-general.tsx +585 -0
- package/src/components/settings-keybinds.tsx +453 -0
- package/src/components/settings-list.tsx +5 -0
- package/src/components/settings-models.tsx +137 -0
- package/src/components/settings-providers.tsx +251 -0
- package/src/components/status-popover.tsx +419 -0
- package/src/components/terminal.tsx +653 -0
- package/src/components/titlebar-history.test.ts +63 -0
- package/src/components/titlebar-history.ts +57 -0
- package/src/components/titlebar.tsx +312 -0
- package/src/constants/file-picker.ts +89 -0
- package/src/context/command-keybind.test.ts +69 -0
- package/src/context/command.test.ts +25 -0
- package/src/context/command.tsx +437 -0
- package/src/context/comments.test.ts +186 -0
- package/src/context/comments.tsx +243 -0
- package/src/context/file/content-cache.ts +88 -0
- package/src/context/file/path.test.ts +360 -0
- package/src/context/file/path.ts +151 -0
- package/src/context/file/tree-store.ts +170 -0
- package/src/context/file/types.ts +41 -0
- package/src/context/file/view-cache.ts +146 -0
- package/src/context/file/watcher.test.ts +149 -0
- package/src/context/file/watcher.ts +53 -0
- package/src/context/file-content-eviction-accounting.test.ts +65 -0
- package/src/context/file.tsx +280 -0
- package/src/context/global-sdk.tsx +232 -0
- package/src/context/global-sync/bootstrap.ts +206 -0
- package/src/context/global-sync/child-store.test.ts +38 -0
- package/src/context/global-sync/child-store.ts +281 -0
- package/src/context/global-sync/event-reducer.test.ts +552 -0
- package/src/context/global-sync/event-reducer.ts +359 -0
- package/src/context/global-sync/eviction.ts +28 -0
- package/src/context/global-sync/queue.ts +83 -0
- package/src/context/global-sync/session-cache.test.ts +102 -0
- package/src/context/global-sync/session-cache.ts +62 -0
- package/src/context/global-sync/session-load.ts +25 -0
- package/src/context/global-sync/session-prefetch.test.ts +96 -0
- package/src/context/global-sync/session-prefetch.ts +100 -0
- package/src/context/global-sync/session-trim.test.ts +59 -0
- package/src/context/global-sync/session-trim.ts +56 -0
- package/src/context/global-sync/types.ts +133 -0
- package/src/context/global-sync/utils.ts +25 -0
- package/src/context/global-sync.test.ts +122 -0
- package/src/context/global-sync.tsx +408 -0
- package/src/context/highlights.tsx +233 -0
- package/src/context/language.tsx +248 -0
- package/src/context/layout-scroll.test.ts +64 -0
- package/src/context/layout-scroll.ts +126 -0
- package/src/context/layout.test.ts +69 -0
- package/src/context/layout.tsx +937 -0
- package/src/context/local.tsx +422 -0
- package/src/context/model-variant.test.ts +86 -0
- package/src/context/model-variant.ts +52 -0
- package/src/context/models.tsx +163 -0
- package/src/context/notification.tsx +373 -0
- package/src/context/permission-auto-respond.test.ts +102 -0
- package/src/context/permission-auto-respond.ts +51 -0
- package/src/context/permission.tsx +277 -0
- package/src/context/platform.tsx +99 -0
- package/src/context/prompt.tsx +297 -0
- package/src/context/sdk.tsx +49 -0
- package/src/context/server.tsx +295 -0
- package/src/context/settings.tsx +241 -0
- package/src/context/sync-optimistic.test.ts +123 -0
- package/src/context/sync.tsx +618 -0
- package/src/context/terminal-title.ts +51 -0
- package/src/context/terminal.test.ts +82 -0
- package/src/context/terminal.tsx +437 -0
- package/src/entry.tsx +144 -0
- package/src/env.d.ts +18 -0
- package/src/hooks/use-providers.ts +44 -0
- package/src/i18n/ar.ts +855 -0
- package/src/i18n/br.ts +867 -0
- package/src/i18n/bs.ts +943 -0
- package/src/i18n/da.ts +937 -0
- package/src/i18n/de.ts +879 -0
- package/src/i18n/en.ts +948 -0
- package/src/i18n/es.ts +950 -0
- package/src/i18n/fr.ts +878 -0
- package/src/i18n/ja.ts +861 -0
- package/src/i18n/ko.ts +860 -0
- package/src/i18n/no.ts +944 -0
- package/src/i18n/parity.test.ts +32 -0
- package/src/i18n/pl.ts +865 -0
- package/src/i18n/ru.ts +946 -0
- package/src/i18n/th.ts +933 -0
- package/src/i18n/tr.ts +952 -0
- package/src/i18n/zh.ts +930 -0
- package/src/i18n/zht.ts +925 -0
- package/src/index.css +29 -0
- package/src/index.ts +6 -0
- package/src/pages/directory-layout.tsx +88 -0
- package/src/pages/error.tsx +327 -0
- package/src/pages/home.tsx +131 -0
- package/src/pages/layout/deep-links.ts +50 -0
- package/src/pages/layout/helpers.test.ts +211 -0
- package/src/pages/layout/helpers.ts +98 -0
- package/src/pages/layout/inline-editor.tsx +126 -0
- package/src/pages/layout/sidebar-items.tsx +437 -0
- package/src/pages/layout/sidebar-project.tsx +384 -0
- package/src/pages/layout/sidebar-shell.tsx +125 -0
- package/src/pages/layout/sidebar-workspace.tsx +504 -0
- package/src/pages/layout.tsx +2509 -0
- package/src/pages/session/composer/index.ts +2 -0
- package/src/pages/session/composer/session-composer-region.tsx +255 -0
- package/src/pages/session/composer/session-composer-state.test.ts +128 -0
- package/src/pages/session/composer/session-composer-state.ts +249 -0
- package/src/pages/session/composer/session-followup-dock.tsx +109 -0
- package/src/pages/session/composer/session-permission-dock.tsx +74 -0
- package/src/pages/session/composer/session-question-dock.tsx +449 -0
- package/src/pages/session/composer/session-request-tree.ts +52 -0
- package/src/pages/session/composer/session-revert-dock.tsx +99 -0
- package/src/pages/session/composer/session-todo-dock.tsx +330 -0
- package/src/pages/session/file-tab-scroll.test.ts +40 -0
- package/src/pages/session/file-tab-scroll.ts +67 -0
- package/src/pages/session/file-tabs.tsx +456 -0
- package/src/pages/session/handoff.ts +36 -0
- package/src/pages/session/helpers.test.ts +181 -0
- package/src/pages/session/helpers.ts +198 -0
- package/src/pages/session/message-gesture.test.ts +62 -0
- package/src/pages/session/message-gesture.ts +21 -0
- package/src/pages/session/message-id-from-hash.ts +6 -0
- package/src/pages/session/message-timeline.tsx +1013 -0
- package/src/pages/session/review-tab.tsx +170 -0
- package/src/pages/session/session-layout.ts +20 -0
- package/src/pages/session/session-model-helpers.test.ts +51 -0
- package/src/pages/session/session-model-helpers.ts +16 -0
- package/src/pages/session/session-side-panel.tsx +453 -0
- package/src/pages/session/terminal-label.ts +16 -0
- package/src/pages/session/terminal-panel.test.ts +25 -0
- package/src/pages/session/terminal-panel.tsx +326 -0
- package/src/pages/session/use-session-commands.tsx +495 -0
- package/src/pages/session/use-session-hash-scroll.test.ts +16 -0
- package/src/pages/session/use-session-hash-scroll.ts +197 -0
- package/src/pages/session.tsx +1841 -0
- package/src/sst-env.d.ts +12 -0
- package/src/testing/model-selection.ts +80 -0
- package/src/testing/prompt.ts +56 -0
- package/src/testing/session-composer.ts +84 -0
- package/src/testing/terminal.ts +118 -0
- package/src/theme-preload.test.ts +46 -0
- package/src/utils/agent.ts +23 -0
- package/src/utils/aim.ts +138 -0
- package/src/utils/base64.ts +10 -0
- package/src/utils/comment-note.ts +88 -0
- package/src/utils/id.ts +99 -0
- package/src/utils/notification-click.test.ts +27 -0
- package/src/utils/notification-click.ts +13 -0
- package/src/utils/persist.test.ts +115 -0
- package/src/utils/persist.ts +476 -0
- package/src/utils/prompt.test.ts +44 -0
- package/src/utils/prompt.ts +203 -0
- package/src/utils/runtime-adapters.test.ts +62 -0
- package/src/utils/runtime-adapters.ts +39 -0
- package/src/utils/same.ts +6 -0
- package/src/utils/scoped-cache.test.ts +69 -0
- package/src/utils/scoped-cache.ts +104 -0
- package/src/utils/server-errors.test.ts +131 -0
- package/src/utils/server-errors.ts +80 -0
- package/src/utils/server-health.test.ts +123 -0
- package/src/utils/server-health.ts +91 -0
- package/src/utils/server.ts +22 -0
- package/src/utils/solid-dnd.tsx +49 -0
- package/src/utils/sound.ts +117 -0
- package/src/utils/terminal-writer.test.ts +64 -0
- package/src/utils/terminal-writer.ts +65 -0
- package/src/utils/time.ts +22 -0
- package/src/utils/uuid.test.ts +78 -0
- package/src/utils/uuid.ts +12 -0
- package/src/utils/worktree.test.ts +46 -0
- package/src/utils/worktree.ts +73 -0
- package/sst-env.d.ts +10 -0
- package/tsconfig.json +26 -0
- package/vite.config.ts +15 -0
- package/vite.js +26 -0
|
@@ -0,0 +1,130 @@
|
|
|
1
|
+
import { Tooltip } from "@reign-labs/ui/tooltip"
|
|
2
|
+
import {
|
|
3
|
+
children,
|
|
4
|
+
createEffect,
|
|
5
|
+
createMemo,
|
|
6
|
+
createSignal,
|
|
7
|
+
type JSXElement,
|
|
8
|
+
onCleanup,
|
|
9
|
+
onMount,
|
|
10
|
+
type ParentProps,
|
|
11
|
+
Show,
|
|
12
|
+
} from "solid-js"
|
|
13
|
+
import { useLanguage } from "@/context/language"
|
|
14
|
+
import { type ServerConnection, serverName } from "@/context/server"
|
|
15
|
+
import type { ServerHealth } from "@/utils/server-health"
|
|
16
|
+
|
|
17
|
+
interface ServerRowProps extends ParentProps {
|
|
18
|
+
conn: ServerConnection.Any
|
|
19
|
+
status?: ServerHealth
|
|
20
|
+
class?: string
|
|
21
|
+
nameClass?: string
|
|
22
|
+
versionClass?: string
|
|
23
|
+
dimmed?: boolean
|
|
24
|
+
badge?: JSXElement
|
|
25
|
+
showCredentials?: boolean
|
|
26
|
+
}
|
|
27
|
+
|
|
28
|
+
export function ServerRow(props: ServerRowProps) {
|
|
29
|
+
const language = useLanguage()
|
|
30
|
+
const [truncated, setTruncated] = createSignal(false)
|
|
31
|
+
let nameRef: HTMLSpanElement | undefined
|
|
32
|
+
let versionRef: HTMLSpanElement | undefined
|
|
33
|
+
const name = createMemo(() => serverName(props.conn))
|
|
34
|
+
|
|
35
|
+
const check = () => {
|
|
36
|
+
const nameTruncated = nameRef ? nameRef.scrollWidth > nameRef.clientWidth : false
|
|
37
|
+
const versionTruncated = versionRef ? versionRef.scrollWidth > versionRef.clientWidth : false
|
|
38
|
+
setTruncated(nameTruncated || versionTruncated)
|
|
39
|
+
}
|
|
40
|
+
|
|
41
|
+
createEffect(() => {
|
|
42
|
+
name()
|
|
43
|
+
props.conn.http.url
|
|
44
|
+
props.status?.version
|
|
45
|
+
queueMicrotask(check)
|
|
46
|
+
})
|
|
47
|
+
|
|
48
|
+
onMount(() => {
|
|
49
|
+
check()
|
|
50
|
+
if (typeof ResizeObserver !== "function") return
|
|
51
|
+
const observer = new ResizeObserver(check)
|
|
52
|
+
if (nameRef) observer.observe(nameRef)
|
|
53
|
+
if (versionRef) observer.observe(versionRef)
|
|
54
|
+
onCleanup(() => observer.disconnect())
|
|
55
|
+
})
|
|
56
|
+
|
|
57
|
+
const tooltipValue = () => (
|
|
58
|
+
<span class="flex items-center gap-2">
|
|
59
|
+
<span>{serverName(props.conn, true)}</span>
|
|
60
|
+
<Show when={props.status?.version}>
|
|
61
|
+
<span class="text-text-invert-weak">v{props.status?.version}</span>
|
|
62
|
+
</Show>
|
|
63
|
+
</span>
|
|
64
|
+
)
|
|
65
|
+
|
|
66
|
+
const badge = children(() => props.badge)
|
|
67
|
+
|
|
68
|
+
return (
|
|
69
|
+
<Tooltip
|
|
70
|
+
class="flex-1 min-w-0"
|
|
71
|
+
value={tooltipValue()}
|
|
72
|
+
contentStyle={{ "max-width": "none", "white-space": "nowrap" }}
|
|
73
|
+
placement="top-start"
|
|
74
|
+
inactive={!truncated() && !props.conn.displayName}
|
|
75
|
+
>
|
|
76
|
+
<div class={props.class} classList={{ "opacity-50": props.dimmed }}>
|
|
77
|
+
<div class="flex flex-col items-start min-w-0 w-full">
|
|
78
|
+
<div class="flex flex-row items-center gap-2 min-w-0 w-full">
|
|
79
|
+
<span ref={nameRef} class={`${props.nameClass ?? "truncate"} min-w-0`}>
|
|
80
|
+
{name()}
|
|
81
|
+
</span>
|
|
82
|
+
<Show
|
|
83
|
+
when={badge()}
|
|
84
|
+
fallback={
|
|
85
|
+
<Show when={props.status?.version}>
|
|
86
|
+
<span
|
|
87
|
+
ref={versionRef}
|
|
88
|
+
class={`${props.versionClass ?? "text-text-weak text-14-regular truncate"} min-w-0`}
|
|
89
|
+
>
|
|
90
|
+
v{props.status?.version}
|
|
91
|
+
</span>
|
|
92
|
+
</Show>
|
|
93
|
+
}
|
|
94
|
+
>
|
|
95
|
+
{(badge) => badge()}
|
|
96
|
+
</Show>
|
|
97
|
+
</div>
|
|
98
|
+
<Show when={props.showCredentials && props.conn.type === "http" && props.conn}>
|
|
99
|
+
{(conn) => (
|
|
100
|
+
<div class="flex flex-row gap-3">
|
|
101
|
+
<span>
|
|
102
|
+
{conn().http.username ? (
|
|
103
|
+
<span class="text-text-weak">{conn().http.username}</span>
|
|
104
|
+
) : (
|
|
105
|
+
<span class="text-text-weaker">{language.t("server.row.noUsername")}</span>
|
|
106
|
+
)}
|
|
107
|
+
</span>
|
|
108
|
+
{conn().http.password && <span class="text-text-weak">••••••••</span>}
|
|
109
|
+
</div>
|
|
110
|
+
)}
|
|
111
|
+
</Show>
|
|
112
|
+
</div>
|
|
113
|
+
{props.children}
|
|
114
|
+
</div>
|
|
115
|
+
</Tooltip>
|
|
116
|
+
)
|
|
117
|
+
}
|
|
118
|
+
|
|
119
|
+
export function ServerHealthIndicator(props: { health?: ServerHealth }) {
|
|
120
|
+
return (
|
|
121
|
+
<div
|
|
122
|
+
classList={{
|
|
123
|
+
"size-1.5 rounded-full shrink-0": true,
|
|
124
|
+
"bg-icon-success-base": props.health?.healthy === true,
|
|
125
|
+
"bg-icon-critical-base": props.health?.healthy === false,
|
|
126
|
+
"bg-border-weak-base": props.health === undefined,
|
|
127
|
+
}}
|
|
128
|
+
/>
|
|
129
|
+
)
|
|
130
|
+
}
|
|
@@ -0,0 +1,5 @@
|
|
|
1
|
+
export { SessionHeader } from "./session-header"
|
|
2
|
+
export { SessionContextTab } from "./session-context-tab"
|
|
3
|
+
export { SortableTab, FileVisual } from "./session-sortable-tab"
|
|
4
|
+
export { SortableTerminalTab } from "./session-sortable-terminal-tab"
|
|
5
|
+
export { NewSessionView } from "./session-new-view"
|
|
@@ -0,0 +1,61 @@
|
|
|
1
|
+
import { describe, expect, test } from "bun:test"
|
|
2
|
+
import type { Message, Part } from "@reign-labs/sdk/v2/client"
|
|
3
|
+
import { estimateSessionContextBreakdown } from "./session-context-breakdown"
|
|
4
|
+
|
|
5
|
+
const user = (id: string) => {
|
|
6
|
+
return {
|
|
7
|
+
id,
|
|
8
|
+
role: "user",
|
|
9
|
+
time: { created: 1 },
|
|
10
|
+
} as unknown as Message
|
|
11
|
+
}
|
|
12
|
+
|
|
13
|
+
const assistant = (id: string) => {
|
|
14
|
+
return {
|
|
15
|
+
id,
|
|
16
|
+
role: "assistant",
|
|
17
|
+
time: { created: 1 },
|
|
18
|
+
} as unknown as Message
|
|
19
|
+
}
|
|
20
|
+
|
|
21
|
+
describe("estimateSessionContextBreakdown", () => {
|
|
22
|
+
test("estimates tokens and keeps remaining tokens as other", () => {
|
|
23
|
+
const messages = [user("u1"), assistant("a1")]
|
|
24
|
+
const parts = {
|
|
25
|
+
u1: [{ type: "text", text: "hello world" }] as unknown as Part[],
|
|
26
|
+
a1: [{ type: "text", text: "assistant response" }] as unknown as Part[],
|
|
27
|
+
}
|
|
28
|
+
|
|
29
|
+
const output = estimateSessionContextBreakdown({
|
|
30
|
+
messages,
|
|
31
|
+
parts,
|
|
32
|
+
input: 20,
|
|
33
|
+
systemPrompt: "system prompt",
|
|
34
|
+
})
|
|
35
|
+
|
|
36
|
+
const map = Object.fromEntries(output.map((segment) => [segment.key, segment.tokens]))
|
|
37
|
+
expect(map.system).toBe(4)
|
|
38
|
+
expect(map.user).toBe(3)
|
|
39
|
+
expect(map.assistant).toBe(5)
|
|
40
|
+
expect(map.other).toBe(8)
|
|
41
|
+
})
|
|
42
|
+
|
|
43
|
+
test("scales segments when estimates exceed input", () => {
|
|
44
|
+
const messages = [user("u1"), assistant("a1")]
|
|
45
|
+
const parts = {
|
|
46
|
+
u1: [{ type: "text", text: "x".repeat(400) }] as unknown as Part[],
|
|
47
|
+
a1: [{ type: "text", text: "y".repeat(400) }] as unknown as Part[],
|
|
48
|
+
}
|
|
49
|
+
|
|
50
|
+
const output = estimateSessionContextBreakdown({
|
|
51
|
+
messages,
|
|
52
|
+
parts,
|
|
53
|
+
input: 10,
|
|
54
|
+
systemPrompt: "z".repeat(200),
|
|
55
|
+
})
|
|
56
|
+
|
|
57
|
+
const total = output.reduce((sum, segment) => sum + segment.tokens, 0)
|
|
58
|
+
expect(total).toBeLessThanOrEqual(10)
|
|
59
|
+
expect(output.every((segment) => segment.width <= 100)).toBeTrue()
|
|
60
|
+
})
|
|
61
|
+
})
|
|
@@ -0,0 +1,132 @@
|
|
|
1
|
+
import type { Message, Part } from "@reign-labs/sdk/v2/client"
|
|
2
|
+
|
|
3
|
+
export type SessionContextBreakdownKey = "system" | "user" | "assistant" | "tool" | "other"
|
|
4
|
+
|
|
5
|
+
export type SessionContextBreakdownSegment = {
|
|
6
|
+
key: SessionContextBreakdownKey
|
|
7
|
+
tokens: number
|
|
8
|
+
width: number
|
|
9
|
+
percent: number
|
|
10
|
+
}
|
|
11
|
+
|
|
12
|
+
const estimateTokens = (chars: number) => Math.ceil(chars / 4)
|
|
13
|
+
const toPercent = (tokens: number, input: number) => (tokens / input) * 100
|
|
14
|
+
const toPercentLabel = (tokens: number, input: number) => Math.round(toPercent(tokens, input) * 10) / 10
|
|
15
|
+
|
|
16
|
+
const charsFromUserPart = (part: Part) => {
|
|
17
|
+
if (part.type === "text") return part.text.length
|
|
18
|
+
if (part.type === "file") return part.source?.text.value.length ?? 0
|
|
19
|
+
if (part.type === "agent") return part.source?.value.length ?? 0
|
|
20
|
+
return 0
|
|
21
|
+
}
|
|
22
|
+
|
|
23
|
+
const charsFromAssistantPart = (part: Part) => {
|
|
24
|
+
if (part.type === "text") return { assistant: part.text.length, tool: 0 }
|
|
25
|
+
if (part.type === "reasoning") return { assistant: part.text.length, tool: 0 }
|
|
26
|
+
if (part.type !== "tool") return { assistant: 0, tool: 0 }
|
|
27
|
+
|
|
28
|
+
const input = Object.keys(part.state.input).length * 16
|
|
29
|
+
if (part.state.status === "pending") return { assistant: 0, tool: input + part.state.raw.length }
|
|
30
|
+
if (part.state.status === "completed") return { assistant: 0, tool: input + part.state.output.length }
|
|
31
|
+
if (part.state.status === "error") return { assistant: 0, tool: input + part.state.error.length }
|
|
32
|
+
return { assistant: 0, tool: input }
|
|
33
|
+
}
|
|
34
|
+
|
|
35
|
+
const build = (
|
|
36
|
+
tokens: { system: number; user: number; assistant: number; tool: number; other: number },
|
|
37
|
+
input: number,
|
|
38
|
+
) => {
|
|
39
|
+
return [
|
|
40
|
+
{
|
|
41
|
+
key: "system",
|
|
42
|
+
tokens: tokens.system,
|
|
43
|
+
},
|
|
44
|
+
{
|
|
45
|
+
key: "user",
|
|
46
|
+
tokens: tokens.user,
|
|
47
|
+
},
|
|
48
|
+
{
|
|
49
|
+
key: "assistant",
|
|
50
|
+
tokens: tokens.assistant,
|
|
51
|
+
},
|
|
52
|
+
{
|
|
53
|
+
key: "tool",
|
|
54
|
+
tokens: tokens.tool,
|
|
55
|
+
},
|
|
56
|
+
{
|
|
57
|
+
key: "other",
|
|
58
|
+
tokens: tokens.other,
|
|
59
|
+
},
|
|
60
|
+
]
|
|
61
|
+
.filter((x) => x.tokens > 0)
|
|
62
|
+
.map((x) => ({
|
|
63
|
+
key: x.key,
|
|
64
|
+
tokens: x.tokens,
|
|
65
|
+
width: toPercent(x.tokens, input),
|
|
66
|
+
percent: toPercentLabel(x.tokens, input),
|
|
67
|
+
})) as SessionContextBreakdownSegment[]
|
|
68
|
+
}
|
|
69
|
+
|
|
70
|
+
export function estimateSessionContextBreakdown(args: {
|
|
71
|
+
messages: Message[]
|
|
72
|
+
parts: Record<string, Part[] | undefined>
|
|
73
|
+
input: number
|
|
74
|
+
systemPrompt?: string
|
|
75
|
+
}) {
|
|
76
|
+
if (!args.input) return []
|
|
77
|
+
|
|
78
|
+
const counts = args.messages.reduce(
|
|
79
|
+
(acc, msg) => {
|
|
80
|
+
const parts = args.parts[msg.id] ?? []
|
|
81
|
+
if (msg.role === "user") {
|
|
82
|
+
const user = parts.reduce((sum, part) => sum + charsFromUserPart(part), 0)
|
|
83
|
+
return { ...acc, user: acc.user + user }
|
|
84
|
+
}
|
|
85
|
+
|
|
86
|
+
if (msg.role !== "assistant") return acc
|
|
87
|
+
const assistant = parts.reduce(
|
|
88
|
+
(sum, part) => {
|
|
89
|
+
const next = charsFromAssistantPart(part)
|
|
90
|
+
return {
|
|
91
|
+
assistant: sum.assistant + next.assistant,
|
|
92
|
+
tool: sum.tool + next.tool,
|
|
93
|
+
}
|
|
94
|
+
},
|
|
95
|
+
{ assistant: 0, tool: 0 },
|
|
96
|
+
)
|
|
97
|
+
return {
|
|
98
|
+
...acc,
|
|
99
|
+
assistant: acc.assistant + assistant.assistant,
|
|
100
|
+
tool: acc.tool + assistant.tool,
|
|
101
|
+
}
|
|
102
|
+
},
|
|
103
|
+
{
|
|
104
|
+
system: args.systemPrompt?.length ?? 0,
|
|
105
|
+
user: 0,
|
|
106
|
+
assistant: 0,
|
|
107
|
+
tool: 0,
|
|
108
|
+
},
|
|
109
|
+
)
|
|
110
|
+
|
|
111
|
+
const tokens = {
|
|
112
|
+
system: estimateTokens(counts.system),
|
|
113
|
+
user: estimateTokens(counts.user),
|
|
114
|
+
assistant: estimateTokens(counts.assistant),
|
|
115
|
+
tool: estimateTokens(counts.tool),
|
|
116
|
+
}
|
|
117
|
+
const estimated = tokens.system + tokens.user + tokens.assistant + tokens.tool
|
|
118
|
+
|
|
119
|
+
if (estimated <= args.input) {
|
|
120
|
+
return build({ ...tokens, other: args.input - estimated }, args.input)
|
|
121
|
+
}
|
|
122
|
+
|
|
123
|
+
const scale = args.input / estimated
|
|
124
|
+
const scaled = {
|
|
125
|
+
system: Math.floor(tokens.system * scale),
|
|
126
|
+
user: Math.floor(tokens.user * scale),
|
|
127
|
+
assistant: Math.floor(tokens.assistant * scale),
|
|
128
|
+
tool: Math.floor(tokens.tool * scale),
|
|
129
|
+
}
|
|
130
|
+
const total = scaled.system + scaled.user + scaled.assistant + scaled.tool
|
|
131
|
+
return build({ ...scaled, other: Math.max(0, args.input - total) }, args.input)
|
|
132
|
+
}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import { DateTime } from "luxon"
|
|
2
|
+
|
|
3
|
+
export function createSessionContextFormatter(locale: string) {
|
|
4
|
+
return {
|
|
5
|
+
number(value: number | null | undefined) {
|
|
6
|
+
if (value === undefined) return "—"
|
|
7
|
+
if (value === null) return "—"
|
|
8
|
+
return value.toLocaleString(locale)
|
|
9
|
+
},
|
|
10
|
+
percent(value: number | null | undefined) {
|
|
11
|
+
if (value === undefined) return "—"
|
|
12
|
+
if (value === null) return "—"
|
|
13
|
+
return value.toLocaleString(locale) + "%"
|
|
14
|
+
},
|
|
15
|
+
time(value: number | undefined) {
|
|
16
|
+
if (!value) return "—"
|
|
17
|
+
return DateTime.fromMillis(value).setLocale(locale).toLocaleString(DateTime.DATETIME_MED)
|
|
18
|
+
},
|
|
19
|
+
}
|
|
20
|
+
}
|
|
@@ -0,0 +1,101 @@
|
|
|
1
|
+
import { describe, expect, test } from "bun:test"
|
|
2
|
+
import type { Message } from "@reign-labs/sdk/v2/client"
|
|
3
|
+
import { getSessionContextMetrics } from "./session-context-metrics"
|
|
4
|
+
|
|
5
|
+
const assistant = (
|
|
6
|
+
id: string,
|
|
7
|
+
tokens: { input: number; output: number; reasoning: number; read: number; write: number },
|
|
8
|
+
cost: number,
|
|
9
|
+
providerID = "openai",
|
|
10
|
+
modelID = "gpt-4.1",
|
|
11
|
+
) => {
|
|
12
|
+
return {
|
|
13
|
+
id,
|
|
14
|
+
role: "assistant",
|
|
15
|
+
providerID,
|
|
16
|
+
modelID,
|
|
17
|
+
cost,
|
|
18
|
+
tokens: {
|
|
19
|
+
input: tokens.input,
|
|
20
|
+
output: tokens.output,
|
|
21
|
+
reasoning: tokens.reasoning,
|
|
22
|
+
cache: {
|
|
23
|
+
read: tokens.read,
|
|
24
|
+
write: tokens.write,
|
|
25
|
+
},
|
|
26
|
+
},
|
|
27
|
+
time: { created: 1 },
|
|
28
|
+
} as unknown as Message
|
|
29
|
+
}
|
|
30
|
+
|
|
31
|
+
const user = (id: string) => {
|
|
32
|
+
return {
|
|
33
|
+
id,
|
|
34
|
+
role: "user",
|
|
35
|
+
cost: 0,
|
|
36
|
+
time: { created: 1 },
|
|
37
|
+
} as unknown as Message
|
|
38
|
+
}
|
|
39
|
+
|
|
40
|
+
describe("getSessionContextMetrics", () => {
|
|
41
|
+
test("computes totals and usage from latest assistant with tokens", () => {
|
|
42
|
+
const messages = [
|
|
43
|
+
user("u1"),
|
|
44
|
+
assistant("a1", { input: 0, output: 0, reasoning: 0, read: 0, write: 0 }, 0.5),
|
|
45
|
+
assistant("a2", { input: 300, output: 100, reasoning: 50, read: 25, write: 25 }, 1.25),
|
|
46
|
+
]
|
|
47
|
+
const providers = [
|
|
48
|
+
{
|
|
49
|
+
id: "openai",
|
|
50
|
+
name: "OpenAI",
|
|
51
|
+
models: {
|
|
52
|
+
"gpt-4.1": {
|
|
53
|
+
name: "GPT-4.1",
|
|
54
|
+
limit: { context: 1000 },
|
|
55
|
+
},
|
|
56
|
+
},
|
|
57
|
+
},
|
|
58
|
+
]
|
|
59
|
+
|
|
60
|
+
const metrics = getSessionContextMetrics(messages, providers)
|
|
61
|
+
|
|
62
|
+
expect(metrics.totalCost).toBe(1.75)
|
|
63
|
+
expect(metrics.context?.message.id).toBe("a2")
|
|
64
|
+
expect(metrics.context?.total).toBe(500)
|
|
65
|
+
expect(metrics.context?.usage).toBe(50)
|
|
66
|
+
expect(metrics.context?.providerLabel).toBe("OpenAI")
|
|
67
|
+
expect(metrics.context?.modelLabel).toBe("GPT-4.1")
|
|
68
|
+
})
|
|
69
|
+
|
|
70
|
+
test("preserves fallback labels and null usage when model metadata is missing", () => {
|
|
71
|
+
const messages = [assistant("a1", { input: 40, output: 10, reasoning: 0, read: 0, write: 0 }, 0.1, "p-1", "m-1")]
|
|
72
|
+
const providers = [{ id: "p-1", models: {} }]
|
|
73
|
+
|
|
74
|
+
const metrics = getSessionContextMetrics(messages, providers)
|
|
75
|
+
|
|
76
|
+
expect(metrics.context?.providerLabel).toBe("p-1")
|
|
77
|
+
expect(metrics.context?.modelLabel).toBe("m-1")
|
|
78
|
+
expect(metrics.context?.limit).toBeUndefined()
|
|
79
|
+
expect(metrics.context?.usage).toBeNull()
|
|
80
|
+
})
|
|
81
|
+
|
|
82
|
+
test("recomputes when message array is mutated in place", () => {
|
|
83
|
+
const messages = [assistant("a1", { input: 10, output: 10, reasoning: 10, read: 10, write: 10 }, 0.25)]
|
|
84
|
+
const providers = [{ id: "openai", models: {} }]
|
|
85
|
+
|
|
86
|
+
const one = getSessionContextMetrics(messages, providers)
|
|
87
|
+
messages.push(assistant("a2", { input: 100, output: 20, reasoning: 0, read: 0, write: 0 }, 0.75))
|
|
88
|
+
const two = getSessionContextMetrics(messages, providers)
|
|
89
|
+
|
|
90
|
+
expect(one.context?.message.id).toBe("a1")
|
|
91
|
+
expect(two.context?.message.id).toBe("a2")
|
|
92
|
+
expect(two.totalCost).toBe(1)
|
|
93
|
+
})
|
|
94
|
+
|
|
95
|
+
test("returns empty metrics when inputs are undefined", () => {
|
|
96
|
+
const metrics = getSessionContextMetrics(undefined, undefined)
|
|
97
|
+
|
|
98
|
+
expect(metrics.totalCost).toBe(0)
|
|
99
|
+
expect(metrics.context).toBeUndefined()
|
|
100
|
+
})
|
|
101
|
+
})
|
|
@@ -0,0 +1,82 @@
|
|
|
1
|
+
import type { AssistantMessage, Message } from "@reign-labs/sdk/v2/client"
|
|
2
|
+
|
|
3
|
+
type Provider = {
|
|
4
|
+
id: string
|
|
5
|
+
name?: string
|
|
6
|
+
models: Record<string, Model | undefined>
|
|
7
|
+
}
|
|
8
|
+
|
|
9
|
+
type Model = {
|
|
10
|
+
name?: string
|
|
11
|
+
limit: {
|
|
12
|
+
context: number
|
|
13
|
+
}
|
|
14
|
+
}
|
|
15
|
+
|
|
16
|
+
type Context = {
|
|
17
|
+
message: AssistantMessage
|
|
18
|
+
provider?: Provider
|
|
19
|
+
model?: Model
|
|
20
|
+
providerLabel: string
|
|
21
|
+
modelLabel: string
|
|
22
|
+
limit: number | undefined
|
|
23
|
+
input: number
|
|
24
|
+
output: number
|
|
25
|
+
reasoning: number
|
|
26
|
+
cacheRead: number
|
|
27
|
+
cacheWrite: number
|
|
28
|
+
total: number
|
|
29
|
+
usage: number | null
|
|
30
|
+
}
|
|
31
|
+
|
|
32
|
+
type Metrics = {
|
|
33
|
+
totalCost: number
|
|
34
|
+
context: Context | undefined
|
|
35
|
+
}
|
|
36
|
+
|
|
37
|
+
const tokenTotal = (msg: AssistantMessage) => {
|
|
38
|
+
return msg.tokens.input + msg.tokens.output + msg.tokens.reasoning + msg.tokens.cache.read + msg.tokens.cache.write
|
|
39
|
+
}
|
|
40
|
+
|
|
41
|
+
const lastAssistantWithTokens = (messages: Message[]) => {
|
|
42
|
+
for (let i = messages.length - 1; i >= 0; i--) {
|
|
43
|
+
const msg = messages[i]
|
|
44
|
+
if (msg.role !== "assistant") continue
|
|
45
|
+
if (tokenTotal(msg) <= 0) continue
|
|
46
|
+
return msg
|
|
47
|
+
}
|
|
48
|
+
}
|
|
49
|
+
|
|
50
|
+
const build = (messages: Message[] = [], providers: Provider[] = []): Metrics => {
|
|
51
|
+
const totalCost = messages.reduce((sum, msg) => sum + (msg.role === "assistant" ? msg.cost : 0), 0)
|
|
52
|
+
const message = lastAssistantWithTokens(messages)
|
|
53
|
+
if (!message) return { totalCost, context: undefined }
|
|
54
|
+
|
|
55
|
+
const provider = providers.find((item) => item.id === message.providerID)
|
|
56
|
+
const model = provider?.models[message.modelID]
|
|
57
|
+
const limit = model?.limit.context
|
|
58
|
+
const total = tokenTotal(message)
|
|
59
|
+
|
|
60
|
+
return {
|
|
61
|
+
totalCost,
|
|
62
|
+
context: {
|
|
63
|
+
message,
|
|
64
|
+
provider,
|
|
65
|
+
model,
|
|
66
|
+
providerLabel: provider?.name ?? message.providerID,
|
|
67
|
+
modelLabel: model?.name ?? message.modelID,
|
|
68
|
+
limit,
|
|
69
|
+
input: message.tokens.input,
|
|
70
|
+
output: message.tokens.output,
|
|
71
|
+
reasoning: message.tokens.reasoning,
|
|
72
|
+
cacheRead: message.tokens.cache.read,
|
|
73
|
+
cacheWrite: message.tokens.cache.write,
|
|
74
|
+
total,
|
|
75
|
+
usage: limit ? Math.round((total / limit) * 100) : null,
|
|
76
|
+
},
|
|
77
|
+
}
|
|
78
|
+
}
|
|
79
|
+
|
|
80
|
+
export function getSessionContextMetrics(messages: Message[] = [], providers: Provider[] = []) {
|
|
81
|
+
return build(messages, providers)
|
|
82
|
+
}
|