Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
06586a6dc8 | ||
|
|
55fc5ee97b | ||
|
|
d9aea8e9c8 | ||
|
|
0048324d8a | ||
|
|
f7ef218507 |
@@ -1,5 +1,12 @@
|
||||
name: Release image
|
||||
|
||||
# Builds and pushes the multi-arch image to GHCR, the registry the installer
|
||||
# defaults to.
|
||||
#
|
||||
# The Gitea registry is on a VPN, which a GitHub runner can never reach, so CI
|
||||
# does not touch it at all: `scripts/mirror-gitea.sh` copies a release into it
|
||||
# from the VPN side, in parallel and after the fact. Keeping the two apart is
|
||||
# what makes a registry outage unable to fail a release.
|
||||
on:
|
||||
release:
|
||||
types: [published]
|
||||
@@ -30,13 +37,6 @@ jobs:
|
||||
username: ${{ github.actor }}
|
||||
password: ${{ secrets.GITHUB_TOKEN }}
|
||||
|
||||
- name: Log in to Gitea
|
||||
uses: docker/login-action@v3
|
||||
with:
|
||||
registry: git.tben.me
|
||||
username: ${{ secrets.GITEA_USERNAME }}
|
||||
password: ${{ secrets.GITEA_TOKEN }}
|
||||
|
||||
# The version the app reports on Settings > General when the build
|
||||
# isn't a published release.
|
||||
- name: Read the source version
|
||||
@@ -45,6 +45,11 @@ jobs:
|
||||
set -euo pipefail
|
||||
echo "version=$(sed -n -E 's/^[[:space:]]*"version": "([^"]+)",?$/\1/p' package.json | head -n1)" >> "$GITHUB_OUTPUT"
|
||||
|
||||
# The same four tags for every kind of build: a release gets its own
|
||||
# version plus the moving `latest` and `major.minor`, so that
|
||||
# scripts/mirror-gitea.sh can copy whichever of them still belong to
|
||||
# it, and a dispatch build gets the commit tag only, since it is not a
|
||||
# release and must not move `latest`.
|
||||
- name: Work out image tags
|
||||
id: tags
|
||||
env:
|
||||
@@ -57,32 +62,24 @@ jobs:
|
||||
if [ "$EVENT" = "release" ] && [ -n "$RELEASE_TAG" ]; then
|
||||
version="$RELEASE_TAG"
|
||||
minor="$(printf '%s' "$version" | sed -E 's/^v?([0-9]+)\.([0-9]+).*/\1.\2/')"
|
||||
ghcr="$(printf '%s\n' \
|
||||
tags="$(printf '%s\n' \
|
||||
"ghcr.io/moduletrace/cobble:latest" \
|
||||
"ghcr.io/moduletrace/cobble:${version}" \
|
||||
"ghcr.io/moduletrace/cobble:${minor}" \
|
||||
"ghcr.io/moduletrace/cobble:${sha}")"
|
||||
gitea="$(printf '%s\n' \
|
||||
"git.tben.me/moduletrace/cobble:latest" \
|
||||
"git.tben.me/moduletrace/cobble:${version}" \
|
||||
"git.tben.me/moduletrace/cobble:${minor}" \
|
||||
"git.tben.me/moduletrace/cobble:${sha}")"
|
||||
else
|
||||
version="$SOURCE_VERSION"
|
||||
ghcr="ghcr.io/moduletrace/cobble:${sha}"
|
||||
gitea="git.tben.me/moduletrace/cobble:${sha}"
|
||||
tags="ghcr.io/moduletrace/cobble:${sha}"
|
||||
fi
|
||||
{
|
||||
echo "version=$version"
|
||||
echo "ghcr<<EOF"
|
||||
echo "$ghcr"
|
||||
echo "EOF"
|
||||
echo "gitea<<EOF"
|
||||
echo "$gitea"
|
||||
echo "tags<<EOF"
|
||||
echo "$tags"
|
||||
echo "EOF"
|
||||
} >> "$GITHUB_OUTPUT"
|
||||
|
||||
# The registry the installer defaults to, so this one must succeed.
|
||||
# The only push CI makes. The installer defaults to this registry, so
|
||||
# this step failing has to fail the release.
|
||||
- name: Build and push to GHCR
|
||||
uses: docker/build-push-action@v6
|
||||
with:
|
||||
@@ -92,62 +89,8 @@ jobs:
|
||||
push: true
|
||||
build-args: |
|
||||
COBBLE_VERSION=${{ steps.tags.outputs.version }}
|
||||
tags: ${{ steps.tags.outputs.ghcr }}
|
||||
tags: ${{ steps.tags.outputs.tags }}
|
||||
provenance: false
|
||||
sbom: false
|
||||
cache-from: type=gha
|
||||
cache-to: type=gha,mode=max
|
||||
|
||||
# The mirror's registry, from the image just built: no second build,
|
||||
# and Cloudflare rate limiting git.tben.me (429/1015 on blob uploads)
|
||||
# can never fail a release the installer depends on. It must still be
|
||||
# loud: `continue-on-error` alone leaves the check green when the
|
||||
# mirror does nothing at all, which is how the 413 below went unseen
|
||||
# for four releases.
|
||||
- name: Mirror to Gitea
|
||||
id: mirror
|
||||
continue-on-error: true
|
||||
env:
|
||||
GITEA_TAGS: ${{ steps.tags.outputs.gitea }}
|
||||
GHCR_IMAGES: ${{ steps.tags.outputs.ghcr }}
|
||||
run: |
|
||||
set -uo pipefail
|
||||
first_ghcr="$(printf '%s\n' "$GHCR_IMAGES" | head -n1)"
|
||||
args=()
|
||||
while IFS= read -r tag; do
|
||||
[ -n "$tag" ] || continue
|
||||
args+=(--tag "${tag}")
|
||||
done <<< "$GITEA_TAGS"
|
||||
|
||||
output="$(mktemp)"
|
||||
status=0
|
||||
docker buildx imagetools create "${args[@]}" "${first_ghcr}" 2>&1 | tee "$output" || status=$?
|
||||
|
||||
if [ "$status" -eq 0 ]; then
|
||||
{
|
||||
echo "### Gitea mirror"
|
||||
echo
|
||||
echo "Mirrored \`${first_ghcr}\` to:"
|
||||
echo
|
||||
printf -- '- \`%s\`\n' $GITEA_TAGS
|
||||
} >> "$GITHUB_STEP_SUMMARY"
|
||||
exit 0
|
||||
fi
|
||||
|
||||
# The usual cause: git.tben.me sits behind Cloudflare, whose
|
||||
# request-body cap is 100 MB, and this image has a 243 MiB
|
||||
# `bun install` layer that no route through it can carry.
|
||||
reason="$(grep -oE '[0-9]{3} [A-Za-z ]{0,40}' "$output" | tail -n1)"
|
||||
{
|
||||
echo "### Gitea mirror failed"
|
||||
echo
|
||||
echo "\`${first_ghcr}\` did **not** reach the Gitea registry. The release itself is fine: the installer defaults to GHCR."
|
||||
echo
|
||||
echo '```'
|
||||
tail -n 12 "$output"
|
||||
echo '```'
|
||||
echo
|
||||
echo "See the [registry push limits](https://developers.cloudflare.com/workers/platform/limits/): git.tben.me is behind Cloudflare, which refuses any request body over 100 MB, and this image's \`bun install\` layer alone is about 243 MiB."
|
||||
} >> "$GITHUB_STEP_SUMMARY"
|
||||
echo "::warning title=Gitea mirror failed::${reason:-docker buildx imagetools create failed}; the image did not reach git.tben.me. See the job summary."
|
||||
exit "$status"
|
||||
|
||||
@@ -134,7 +134,7 @@ The app serves plain HTTP only; TLS is the reverse proxy's or tunnel's job, and
|
||||
|
||||
The container runs with `--network cobble`, `-p BIND:PORT`, `.env` at `/app/.env` read only, the host data directory at `/data`, the docker socket at `/var/run/docker.sock`, `MAIN_SERVER=1` and `PORT`, `--init`, `--cpus 1` and `--memory 2048m` (a light orchestration process: the heavy work happens in each computer's own container; overridable with `--cpus`/`--memory`), a `unless-stopped` restart policy and bounded json-file logs. `/data` must stay a bind mount: `resolveHostPath` turns paths under the app's own mounts into host paths for computer containers, and a named volume would break them. `DATA_DIR=/data` and `DB_DIR=/data/db` come from the Dockerfile; `entrypoint.sh` keeps bun-sqlite's schema record on the same volume.
|
||||
|
||||
Publishing is `scripts/publish-github.sh` and `scripts/publish-gitea.sh`: the shared flow (argument parsing, the release gate, tag preparation, the push, the release notes) lives in `scripts/lib/publish-common.sh` and `scripts/lib/publish-release.sh`, and only the release object differs (`gh release create` for GitHub, which is what fires the image build, and `curl` against the Gitea API with a `GITEA_TOKEN` or the token `tea login` stored). Both are marked-block scripts tested by `scripts/publish-tools.test.ts`, which lifts the blocks out with the `install.test.ts` approach. Images are built by GitHub Actions on a published release (`.github/workflows/release.yml`), for `linux/amd64` and `linux/arm64`, and pushed to both `ghcr.io/moduletrace/cobble` and `git.tben.me/moduletrace/cobble` with `latest`, version, `major.minor` and commit-sha tags. The version they are built with is passed as the `COBBLE_VERSION` build arg (the published tag, else `package.json`'s version read from `<version>`), which the image turns into `ENV COBBLE_VERSION` and `runningVersion` (`src/utils/running-version.ts`) reports on Settings > General; bump `package.json`'s `version` with a release, since a local build and a workflow-dispatch build fall back to it. `.github/workflows/ci.yml` runs `bun run typecheck` and `bun run test` on every push and pull request, and checks that `install.sh` parses. The Dockerfile pins Bun 1.4.3 to match CI. The repository is mirrored on Gitea, and the installer defaults to the GHCR image with `--image git.tben.me/moduletrace/cobble` as the Gitea alternative.
|
||||
Publishing is `scripts/publish-github.sh` and `scripts/publish-gitea.sh`: the shared flow (argument parsing, the release gate, tag preparation, the push, the release notes) lives in `scripts/lib/publish-common.sh` and `scripts/lib/publish-release.sh`, and only the release object differs (`gh release create` for GitHub, which is what fires the image build, and `curl` against the Gitea API with a `GITEA_TOKEN` or the token `tea login` stored). Both are marked-block scripts tested by `scripts/publish-tools.test.ts`, which lifts the blocks out with the `install.test.ts` approach. Images are built by GitHub Actions on a published release (`.github/workflows/release.yml`), for `linux/amd64` and `linux/arm64`, and pushed to `ghcr.io/moduletrace/cobble` with `latest`, version, `major.minor` and commit-sha tags. CI never touches the Gitea registry: it is on a VPN, which a runner cannot reach, so `scripts/mirror-gitea.sh` (`bun run mirror:gitea`) copies a release into `git.tben.me/moduletrace/cobble` afterwards, from the Gitea host over ssh, running skopeo there in a container and reading every destination tag's digest back against GHCR's; it only moves `latest` and `major.minor` while that release still owns them in GHCR, so mirroring an older one never rolls them back. The two run in parallel and neither can fail the other, which is the point: the installer defaults to GHCR, so a registry outage cannot fail a release. The version they are built with is passed as the `COBBLE_VERSION` build arg (the published tag, else `package.json`'s version read from `<version>`), which the image turns into `ENV COBBLE_VERSION` and `runningVersion` (`src/utils/running-version.ts`) reports on Settings > General; bump `package.json`'s `version` with a release, since a local build and a workflow-dispatch build fall back to it. `.github/workflows/ci.yml` runs `bun run typecheck` and `bun run test` on every push and pull request, and checks that `install.sh` parses. The Dockerfile pins Bun 1.4.3 to match CI. The repository is mirrored on Gitea, and the installer defaults to the GHCR image with `--image git.tben.me/moduletrace/cobble` as the Gitea alternative.
|
||||
|
||||
### Envs
|
||||
|
||||
@@ -196,7 +196,7 @@ Every folder on the data volume, browsable read-only: a bot's own, a workspace's
|
||||
- **Automatic picks** (`rank-auto-models`): only for the `automatic` harnesses in `Harnesses` (text, vision, image); transcription and speech have none and are not available until a model is picked (the model select reads "Not available", no fallback checkbox). Free models that can do the job (on the setting's provider when it has one; a provider without free models needs a picked model) (`model-serves-harness`: text needs tools), by vendor `AutoModelVendorPreference` (DeepSeek, Qwen, NVIDIA, Liquid), then other vendors that look competent (`AppData.HarnessAutoMin*`), then routers; within a vendor by intelligence index, then full-size before small, newest first.
|
||||
- **Calling:** `resolveHarnessCandidates` (pinned, then automatic) -> `withCandidates` tries them in order with process-wide cooldowns (`resolve/harness-cooldowns.ts`): rate limits and outages rest the model (until `Retry-After` when given), 401/402 rest the whole provider, 403/404 rest the model for an hour. Errors are `HarnessError` (`code` provider|unavailable|aborted, `retryable`), classified in `to-harness-error.ts`. Entry points: `text/generate-text.ts` (one turn), `text/run-text-loop.ts` (tool loop, sticks to one model, wraps up after `HarnessMaxToolSteps`, trims the oldest tool output before each turn to fit the model's context, `fit-transcript-to-context.ts`), `media/read-image.ts`, `media/read-media.ts` (documents locally via `fileToText`, images and audio through their harnesses), `audio/transcribe-audio.ts`, `audio/speak-text.ts` (returns the provider's audio as a stream, falling back to the next model only until one answers), `image/generate-image.ts`, `embeddings/embed-text.ts` (batches, then `fit-vector-dimensions.ts` pads or cuts each vector to `AppData.EmbeddingDimensions` and normalizes it, since a provider's width varies). A text model that can't take an image or audio part gets it read by the vision/transcription harness first (`prepare-messages-for-model.ts`).
|
||||
- **Per-bot models:** in a workspace, its `workspace_harness_settings` row wins (`source: "workspace"`). Otherwise a worker bot inherits the defaults per harness unless it has a `bot_harness_settings` row (same fields as `harness_settings`, unique per bot and harness). `grabHarnessSetting({ harness, bot_id })` returns the bot's row (`source: "bot"`) or the default; the coordinator always gets the default. Every entry point takes `bot_id` and passes it on (text loop -> media reading/transcription), so a bot's run uses its own models throughout. Saved by `saveBotHarnessSetting` (`inherit` deletes the row); the page data is `grabBotHarnessOverview`. A bot's chat header has a ghost model icon (`(partials)/header-icon-button.tsx`; the text model and provider show under the bot's name) that opens a modal with every harness (`(partials)/bot-models-modal.tsx`); the modal and `/admin/bots/[bot_id]/models` both use `(partials)/bot-harness-form.tsx`. On the coordinator the modal edits the defaults themselves (`api/admin/harness/save-settings`, the same settings as Settings, under Models, without the providers): super admins only, the icon is disabled for everyone else. Every one of these wraps `settings/models/(partials)/harness-settings-form.tsx`: per harness a provider select first (`harness-provider-select.tsx`: any provider, one provider, or on a bot "Same as the default"), then the model select narrowed to it. Removing a provider (`deleteHarnessProvider`) puts default and bot settings that picked it or its models back on any provider (`save` prop; "Same as the default" shows when the overview item has `default`).
|
||||
- **Tools:** `HarnessTool` (built per run with its bot bound): own tools validate with a zod `schema`; tools described elsewhere pass a JSON schema as `parameters` and validate themselves. `run-tool-call.ts` reports bad calls and failures back to the model as `Error: ...`, and caps every tool call at `AppData.HarnessToolTimeoutMs` (aborted, its result handed back as an error the model can work around). Bot tools in `src/harness/tools/storage/`: `list_files`, `read_file`, `write_file`, `search_files`, `move_file`, `delete_file`, `ingest_file`, `copy_file`, `generate_image`, `make_pdf` (`grabBotFileTools`, every run, see Bot files and folders). Bot tools in `src/harness/tools/bot/`: `update_memory`, `update_instructions` (`grabBotTools`, every run), `start_task` and, for the coordinator with workers, `delegate_task` (`grabBotChatTools`, chat replies only; the job's `requested_by_bot_id` is the coordinator, and `relayJobToRequester` posts the worker's result or failure in its conversation, pushed as `relay` by `pushBotNotice`), `ask_user_to_take_over` (see Take-over), plus for the coordinator `create_workspace` and `update_workspace` (a new workspace with its shared context, or changing one; both also in its tasks, so a task that gathers material can write it into the workspace), `create_bot` (add a worker with its name, role and instructions, and join it to a workspace straight away) and, in its own chat, `add_bot_to_workspace` (put an existing worker in one of the user's workspaces) and `remove_bot_from_workspace` (take it out, keeping its conversation there), `assign_computer` (grant or revoke a worker's direct access, `add/remove-bot-to/from-computer`; access through a workspace stays and the result says so), `update_bot` (change a worker's name, role or core instructions, `src/functions/backend/db/bots/update-bot.ts`), `find_bot` and `find_workspace` (look the crew, or the live workspaces, up by id, name, role/description or instructions/context), `delete_bot` (a worker with its conversations, tasks, schedules and workspace places; the coordinator can't go) and `delete_workspace` (its conversations, tasks, schedules and links; its bots and computers stay). `search_computers` (`src/harness/tools/computer/search-computers-tool.ts`, chat replies and tasks) runs one read-only command on all of a bot's computers at once over the sandbox REST API (`run-computer-command.ts`), each output under its `[computer:ID]`, secrets redacted, capped by `AppData.ComputerSearch*`.
|
||||
- **Tools:** `HarnessTool` (built per run with its bot bound): own tools validate with a zod `schema`; tools described elsewhere pass a JSON schema as `parameters` and validate themselves. `run-tool-call.ts` reports bad calls and failures back to the model as `Error: ...`, and caps every tool call at `AppData.HarnessToolTimeoutMs` (aborted, its result handed back as an error the model can work around). Bot tools in `src/harness/tools/storage/`: `list_files`, `read_file`, `write_file`, `search_files`, `move_file`, `delete_file`, `ingest_file`, `copy_file`, `generate_image`, `make_pdf` (`grabBotFileTools`, every run, see Bot files and folders). Bot tools in `src/harness/tools/bot/`: `update_memory`, `update_instructions` (`grabBotTools`, every run), `start_task` and, for the coordinator with workers, `delegate_task` (`grabBotChatTools`, chat replies only; a task handed over in the coordinator's own chat runs in the worker's own chat unless it names a `workspace_id`, which puts it in that workspace: the worker's workspace instructions and memory, the workspace's context, files and computers apply, and it needs membership, added first via `addBotToWorkspace` when missing; handed over from inside a workspace it always runs there and a different `workspace_id` is refused; the job's `requested_by_bot_id` is the coordinator, and `relayJobToRequester` posts the worker's result or failure in its conversation, pushed as `relay` by `pushBotNotice`), `ask_user_to_take_over` (see Take-over), plus for the coordinator `create_workspace` and `update_workspace` (a new workspace with its shared context, or changing one; both also in its tasks, so a task that gathers material can write it into the workspace), `create_bot` (add a worker with its name, role and instructions, and join it to a workspace straight away) and, in its own chat, `add_bot_to_workspace` (put an existing worker in one of the user's workspaces) and `remove_bot_from_workspace` (take it out, keeping its conversation there), `assign_computer` (grant or revoke a worker's direct access, `add/remove-bot-to/from-computer`; access through a workspace stays and the result says so), `update_bot` (change a worker's name, role or core instructions, `src/functions/backend/db/bots/update-bot.ts`), `find_bot` and `find_workspace` (look the crew, or the live workspaces, up by id, name, role/description or instructions/context), `delete_bot` (a worker with its conversations, tasks, schedules and workspace places; the coordinator can't go) and `delete_workspace` (its conversations, tasks, schedules and links; its bots and computers stay). `search_computers` (`src/harness/tools/computer/search-computers-tool.ts`, chat replies and tasks) runs one read-only command on all of a bot's computers at once over the sandbox REST API (`run-computer-command.ts`), each output under its `[computer:ID]`, secrets redacted, capped by `AppData.ComputerSearch*`.
|
||||
- **MCP:** `src/harness/mcp/`: `connectMcpServer` (Streamable HTTP, SSE fallback), `grabMcpTools` (each server tool becomes a `HarnessTool` with its JSON schema), `mcpResultToText` (text as is, images read by the vision harness, `isError` thrown). Computers are reached this way: `openComputerTools` (`src/harness/tools/computer/`) connects to the computer's AIO sandbox at `grabComputerMcpUrl` (the `MAIN_COMPUTER_URL` app setting for the main computer, for a dev server outside docker, e.g. `http://localhost:8170`, else `http://<hostname>:<port>/mcp`, or its IP address when the hostname can't be reached), leaves out `ComputerMcpExcludedTools`, and sets the computer's envs (see Envs).
|
||||
- **MCPs and custom tools (extensions):** installs add their own MCP servers (`mcp_servers`, Settings, under MCPs) and tools (`custom_tools`, Settings, under Tools). A bot gets the ones given to it (`bot_mcp_servers`, `bot_custom_tools`) and, in a workspace, that workspace's (`workspace_mcp_servers`, `workspace_custom_tools`); `botExtensionsSql` (`src/utils/bot-extensions-sql.ts`, tested) resolves a run's, `grabBotExtensions` reads them. `all_bots` ("every bot by default") only decides what bots are given: turning it on gives it to every bot (`attachToAllBots`, the coordinator included) and every new bot gets it (`attachDefaultExtensions`, from `createBot` and `ensureCoordinatorBot`); turning it off takes it from no one, and a bot that drops one keeps it dropped (`src/scripts/attach-every-bot-extensions.ts`, safe to rerun, moved installs from the old "reaches every bot" meaning). Super admins define them. Super admins and admins (the super admin alone for the coordinator, `assertBotExtensionManager`) give and take them per bot on the bot's settings page and under its chat's composer (`composer-tools.tsx`: one "N tools" button (`composer-tools-modal.tsx`) counting everything it can use there, which opens a modal listing them (a workspace's not removable) with the add picker, so ten of them don't fill the composer; beside it `composer-computer-button.tsx` opens the bot's current computer in the viewer), and per workspace on the workspace's settings page; the settings pages share `settings/(partials)/extensions-group.tsx`, all through `useExtensions` (`src/hooks/`), `grabBotExtensionsOverview` / `grabWorkspaceExtensionsOverview` (`bot_extensions` / `workspace_extensions` page props, both on `grabExtensionsOverview`) and `api/admin/bots/` and `api/admin/workspaces/` `add-extension`, `remove-extension` (`setBotExtension`, `setWorkspaceExtension`); the MCPs and Tools pages' Access modal sets bots and workspaces in bulk (`setExtensionAccess`, an admin never touches the coordinator's or `all_bots`). Helpers in `src/functions/backend/db/extensions/`, API in `api/admin/mcps/` and `api/admin/tools/`, zod in `src/schemas/extension-schema.ts`.
|
||||
- **Skills** are the third extension kind (`skills`, Settings, under Skills; `bot_skills` / `workspace_skills`, same `all_bots` rules, `ExtensionTables.kind = "skill"`). A skill is a runbook: `name`, a unique `slug` a bot loads it by, a one-line `description` (when it applies) and the markdown `body`, capped by `AppData.SkillBodyMaxChars`. Unlike an MCP server or a tool it has no `status`/`last_error`: nothing connects or runs, and it never reports a failure (`ExtensionRow` and `ComposerToolRow` key the state line off `kind == "skill"`). A run sees only the slug and description; `openBotSkills` (`src/harness/tools/skills/`, fed by the `skills` that `grabBotExtensions` already reads) adds one `load_skill` tool that returns the body, cut at `AppData.SkillBodyShownChars` with a line saying so, and throws the valid slugs for an unknown one. Slugs are checked by `checkSkillSlug` (`src/utils/check-skill-slug.ts`, tested); `load_skill` is in `ReservedToolNames` so a custom tool can't shadow it. The settings page is `api/admin/skills/` (`save-skill`, `delete-skill` super admin only, `set-skill-access`), `grabSkillsOverview`; `deleteSkill` deletes the two link tables itself, since foreign keys are off outside the server process. `openBotExtensions` (`src/harness/tools/extensions/`) turns them into harness tools for chat replies and tasks (built-ins win a name clash; `close()` in the runners' `finally`; `note` tells the bot about a server it can't use). Templates (`fillTemplate`, `src/utils/fill-template.ts`): `{{args.x}}` and `{{env.KEY}}`, envs from `grabRunEnvs` (the bot's own beat the workspace's, which beat general; app settings never, no need to be set on computers); secret values are redacted from every result.
|
||||
@@ -206,15 +206,28 @@ Every folder on the data volume, browsable read-only: a bot's own, a workspace's
|
||||
- **Replies and tasks:** `run-bot-reply.ts` answers from the last `BotReplyHistoryMessages` messages (`botMessagesToHarness`) with `ChatReplyNote`, the bot's computers (`compose-computers-note.ts`: what they are and that they're reached through `start_task`) and the chat tools, at most `BotReplyMaxToolSteps` steps, and finishes silently if the bot already had the last word (`enqueueBotReply` reuses a pending reply, so a burst of messages gets one). `run-bot-task.ts` runs the brief (`compose-job-brief.ts`) with `TaskRunNote`, the bot tools, the file tools and the job's computer tools, closing the MCP connection after. Both add `BotFilesNote` (where a bot's files live, that they persist, and to search before saying it doesn't know) and collect `attachments` for files the bot wrote with `attach`, which `completeBotJob` links to the result message. Retryable harness errors go back to the queue's backoff, the rest are `BotJobFatalError`.
|
||||
- **API:** `api/admin/harness/` (`save-provider` also syncs, `sync-provider`, `delete-provider`, `save-settings`), super admin only, each returning the fresh `grabHarnessOverview()` page data.
|
||||
|
||||
### Token usage
|
||||
|
||||
Every model call is written down in `token_usages` as it happens, so a message can show what it cost and `/admin/usage` can total the install's spending by day, by model and by provider.
|
||||
|
||||
- **Whose call it was:** `src/harness/usage/harness-usage-scope.ts` is an `AsyncLocalStorage` scope (`user_id`, `bot_id`, `workspace_id`, `session_id`, `job_id`, `message_id`) that a run enters once, in `run-bot-job.ts` (`runWithHarnessUsageScope`), so everything underneath - the text loop's turns, a session summary, a transcription, an image, an embedding - is attributed without threading the ids through every harness function. The store is merged, not replaced: a nested run keeps the outer ids and overrides only what it knows better, and a `null` is a real value that clears a field (a bot's own chat has no workspace). `withUsageScope` (`with-usage-scope.ts`) wraps one call the app makes on a user's behalf outside any run (the chat's transcribe and speak routes), and `runWithFallbackScope` (`run-with-fallback-scope.ts`) attributes work whose run may be missing - a file the ingest queue embeds has no run, so it is charged to its owner (`grabUsageOwnerUserId`, the workspace's creator else the bot's) - while `mode: "own"` forces the given scope (the session namer, whose job is already complete and must not pay for it). Without an owner nothing is written: a row no one can be shown is not worth a row.
|
||||
- **Recording:** `record-harness-usage.ts` writes one row and never throws (a model that answered must not lose its answer to a failed insert; it logs with `logError` instead). `build-usage-call.ts` turns the candidate that really answered into the row's provider and model (`provider_id`/`provider_name` and the catalog `model_id` plus the provider's `model_slug`), and providers that report no usage still get a row, so a call is never invisible. Every entry point calls it in `withCandidates`'s `on_success`: `generate-text.ts`, `read-image.ts`, `transcribe-audio.ts`, `speak-text.ts` (billed per character, so the text's length stands in for what was spoken), `generate-image.ts`, `embed-text.ts`.
|
||||
- **Messages:** a reply or task job's calls carry the job's id as they happen, since its message doesn't exist yet; `complete-bot-job.ts` then runs `UPDATE token_usages SET message_id = ? WHERE job_id = ? AND message_id IS NULL` once the message is inserted, so the thread can show what the answer cost. Calls that produce no message (naming a session, embedding a file, transcribing a dictation) keep a null `message_id` and are still counted in the totals.
|
||||
- **Message usage:** `grab-message-usage.ts` sums a batch of messages' calls (one `GROUP BY message_id`) and `grab-bot-messages.ts` attaches it as `BotMessageWithAttachments.usage` (`TokenUsageSummary`: prompt, completion, total, cost, calls). `formatTokens` (`src/utils/format-tokens.ts`) writes a count as a person reads it (842 / 1.2K / 842K / 1.2M). `MessageUsageChip` (`bots/[bot_id]/(partials)/message-usage-chip.tsx`) sits in the actions row under a bot message beside its timestamp and play button, showing the total with the breakdown in its `title`, and renders nothing when a message has no calls.
|
||||
- **Usage page:** `/admin/usage`, a `RailPages` entry like Routines and Activity rather than a settings sub page, since it is the install's own record rather than a setting (super admins and admins only, `canManageCrew`; the dict entry's `can_view` hides it for anyone else, and `grab-usage-server-props.ts` redirects them to the bots). Helpers in `src/functions/backend/db/usage/`: `grab-usage-overview.ts` runs `grab-usage-totals.ts` (one total and the same split per harness), `grab-usage-days.ts` -> `bucket-usage-days.ts` (one bucket per day, empty days at zero), `grab-usage-models.ts` (per provider, model slug and harness, busiest first, joined to `harness_models` for a display name), plus the previous period of the same length and a rolling week. The window comes from `grab-usage-window.ts` (`UsageDefaultDays` 30, `UsageMaxDays` 365, clamped). API `api/admin/usage` (GET only) for another period, zod in `src/schemas/usage-schema.ts`.
|
||||
- **Days are local, not UTC:** `grabUsageDays` groups in SQL with `date(created_at / 1000, 'unixepoch', 'localtime')`, matching the local midnights `grabUsageWindow` builds and the `YYYY-MM-DD` `localDate` writes. Grouping by the bare UTC day put a late evening's calls in the previous bucket and silently dropped them whenever no bucket carried that name (the chart's total disagreed with the totals card); `bucket-usage-days.test.ts` pins this. Never name a local day with `toISOString().slice(0, 10)`.
|
||||
- **The page:** `index.server.ts` -> `(functions)/grab-usage-server-props.ts` (crew managers only, the period in the address), `index.tsx` in `SettingsShell` with `wide`, `(sections)/usage-section.tsx` (state in the address through `history.pushState`, `use-usage-overview.ts` loading another period), `(partials)/usage-totals.tsx` (tokens, model calls, cost, previous period with a change label), `usage-day-chart.tsx` + `usage-day-bar.tsx` (bars from static `h-*` classes in `(data)/usage-bar-heights.ts`, since inline styles are not allowed), `usage-harness-breakdown.tsx` + `usage-harness-row.tsx`, and `usage-models-table.tsx` + `usage-model-row.tsx` (calls, in, out, tokens, **price per thousand tokens** worked out from cost and tokens, which is what makes one model's cost comparable with another's).
|
||||
|
||||
## Admin Route Hierarchy
|
||||
|
||||
Live routes under `/admin` (see `AdminSections`): `/admin/bots`, `/admin/workspaces` (the index opens the first workspace), `/admin/routines`, `/admin/logs`, `/admin/help` and `/admin/settings`. `/admin` redirects to `/admin/bots`; a workspace's dashboard is `/admin/workspaces/<id>`. The routes in detail:
|
||||
|
||||
- `/admin/bots` (live) - the Cobble chat shell, plus `[bot_id]/`
|
||||
- `/admin/routines` (live) - every schedule the user's bots have (see Schedules), opened from "Routines" in the rail's user popover
|
||||
- `/admin/usage` (live) - what the crew's model calls spent, per day, per kind of call and per model, for crew managers (see Token usage), opened from "Usage" in the rail's user popover
|
||||
- `/admin/logs` (live) - Activity, the audit log: super admins only (others are redirected; the popover's "Activity" item shows to super admins only). Rendered in `SettingsShell` with `wide`
|
||||
- `/admin/help` (live) - familiar problems and their fixes, one `SettingsGroup` per topic (`(data)/help-topics.ts`, markdown rendered with the chat's `MessageMarkdownComponents`, the topic id as the group's anchor). First topic: signing in to Google and other strict sites with Cookie-Editor over the Desktop view (`HelpSignInTopicId` in `src/dict/help-dict.ts`; `ask_user_to_take_over` points bots to it)
|
||||
- Routines, Files, Activity and Help are the `RailPages` (`src/dict/rail-pages-dict.ts`), opened from the rail's user popover (which lists them, then Settings and Sign out). On them the rail shows its own aside (`(partials)/rail-pages-list.tsx`): a back link to the bots, then those pages, like settings
|
||||
- Routines, Files, Usage, Activity and Help are the `RailPages` (`src/dict/rail-pages-dict.ts`), opened from the rail's user popover (which lists them, then Settings and Sign out). On them the rail shows its own aside (`(partials)/rail-pages-list.tsx`): a back link to the bots, then those pages, like settings. A page carries `super_admin_only` or a `can_view` rule (Usage is `canManageCrew`), and `canSeeRailItem` (`src/utils/rbac/can-see-rail-item.ts`, tested) decides both the popover and the aside, so a page is never offered in one and hidden in the other
|
||||
- `/admin/settings/backups` - super admin only (listed in `SettingsPages`), two groups. Database snapshots: create, list, restore, download and delete in `db/.backups` (`bun-sqlite backup`); "Restore from file" beside "Create backup" opens `upload-restore-modal.tsx` (upload a SQLite database file, which is saved to `.backups` then restored with `bun-sqlite import <file>`, since `restore` is interactive). Whole-data archives: `write-data-backup-archive.ts` tars all of `DATA_DIR` (database, media, computer homes, bot and workspace files, MCP installs) into `BACKUP_ARCHIVE_DIR` as `cobble-data-<epoch-ms>.tar.gz` (via `tar -czf` into a `.part` name, renamed on success so a cut-off run is never listed, with a `--transform` putting every member under one root folder named after the archive), keeping the newest `AppData.DataBackupsKept` and never touching `db/.backups`; its own "Restore from file" uploads a tar.gz (`upload-restore-data.ts`, gzip header checked first). `restore-data-backup.ts` extracts an archive to a temp dir, requires exactly one root folder holding a `db` folder (`findArchiveRoot`, since an upload is saved under a fresh timestamp and its inner name no longer matches the file name), takes a safety archive before touching anything (`write-data-backup-archive.ts`, not the locked `create-data-backup.ts`), empties `DATA_DIR` (the snapshots included) and copies the archive's in, then `restart-app.ts` restarts the process through the docker socket after the response lands (`process.exit`, the container's restart policy brings it back), because the open SQLite handle points at the replaced file. Row actions are icon-only (`RowIconButton`, `src/components/general/row-icon-button.tsx`): restore, download, delete. `BACKUP_ARCHIVE_DIR` is a `ProcessOnlyEnvKeys` start-up path (the installer mounts `$COBBLE_HOME/backups-archives` there; outside docker it falls back to `~/.cobble/backups`). Every backup and restore, of either kind, runs one at a time via `with-backup-run-lock.ts`. Helpers in `src/functions/backend/backups/`, API routes in `api/admin/backups/`
|
||||
- `/admin/settings/database` - super admin only (listed in `SettingsPages`), a read-only browser of the install's SQLite database, no writes anywhere in it. `grabDatabaseTables` reads the catalog once (every table and its columns, SQLite's own `sqlite_*` tables and sqlite-vec's `bot_file_vectors_*` shadow tables left out) and adds each table's `COUNT(*)` and its size from `dbstat` (index pages folded back onto their table). `grabTableRecords` then returns one page (`AppData.DatabaseBrowserPageSize`, 50) of the open table's records newest key first, every value already cut in SQL at `AppData.DatabaseBrowserCellMaxChars` (400) so a table of long texts never ships a megabyte of JSON, and a `truncated` flag per cell; the search matches over the concatenation of every column's text (`buildSearchClause`: every word, or the whole phrase), which is what also works on a virtual table, where `MATCH` and `LIKE` over a vector blob do not. A record's key is its single primary key column, else its rowid, else nothing (a virtual table with no primary key reads as plain cells); clicking a cell's row opens `grabTableRecord` for that key, which returns the record whole up to `AppData.DatabaseBrowserValueMaxChars` (20000), the answer to a cell cut at three lines. `clampPage` puts a page past the end on the last one. The page's state (table, search, mode, page) lives in the address through `databaseHref`, so a table or a search can be linked to; the open table's first page comes with the page props and anything else loads through the API. Helpers in `src/functions/backend/database/` (`build-table-catalog-sql`, `build-table-sizes-sql`, `build-records-sql`, `build-search-clause`, `resolve-table-key`, `build-record-row`, `quote-sql-identifier`), API `api/admin/database/records` and `record` (GET only, `grabSuperAdminUser`), zod in `src/schemas/database-schema.ts`, page under `src/pages/admin/settings/database/`
|
||||
- `/admin/settings` (live) - redirects to `SettingsDefaultPath` (`/admin/settings/general`). Sub pages: `general/` (theme: System/Light/Dark per device via `useThemePreference`, stored in `localStorage[AppData.ThemeStorageKey]`, the same key `ThemeInitScript` and twui's `ColorSchemeSelector` read; applies instantly, no save bar; for super admins the prompt appendix (`prompt-section.tsx`, `system-prompt-append-form.tsx`) saving `api/admin/settings/system-prompt-append` into the `bot_system_prompt_append` app setting; and, for `canManageCrew`, the bot file quota: `storage-section.tsx`, `storage-quota-form.tsx` saving `api/admin/settings/storage-quota` into `app_settings`; and, for everyone, the running version: `version-section.tsx`, `version-line.tsx` showing `pageProps.running_version` from `runningVersion` (`src/utils/running-version.ts`: `COBBLE_VERSION`, the release tag the image was built with, else `package.json`'s version, in development too, with `(development)` after it and no such suffix in production), `profile/` (current user profile), `notifications/` (this device's Web Push switch, a test send, and every device the user subscribed), `change-password/` (set/change own password) `models/` (super admin only, `super_admin_only` in the dict hides it from the rail: model providers and which model each harness uses) and `environment/` (super admin only: app settings and the envs bots and computers use, per scope), each wrapped in `SettingsShell` (`(partials)/settings-shell.tsx`: 64px pane header like the conversation header, then one centered column). Content is built from `SettingsGroup` blocks (label + description left, controls right, hairline-separated, no cards) and ends with the sticky `SettingsSaveBar` (Save enabled only when dirty, Discard, one-line status). Saves don't reload the page. Keep save notices in component state: `useFormInit` clears `status` whenever the form changes. `bots/` (the crew: chat, edit with `BotForm`, models, delete a worker via `api/admin/bots/delete-bot`), `workspaces/` (open, create through the Add flow, delete), `computers/` (state and resources, live; super admins create, edit with `AddComputerForm`, retry, delete), `mcps/`, `tools/` and `skills/` (see MCPs, custom tools and skills: super admins add, edit, refresh or try, delete; admins only set Access, `extension-access-modal.tsx`; add, edit, access and try open in modals (`AddModalShell`). A skill has no status: nothing connects or runs, so `skill-row.tsx` puts its slug, who gets it (`extension-access-label.ts`) and its description where a server's or tool's status line goes), each MCP server or tool row with a status line under it (`extension-status-line.tsx`: connected or not, last run worked or failed, when, and why), those five for `canManageCrew` only (`grab-crew-settings-server-props.ts`, `can_view` in the dict) and `users/` (for `canListUsers`: super admins and admins; the dict's `can_view` gates it in the rail; add with a password (`canCreateUser`), edit (`canUpdateUser`), both with the one `UserForm` (`admin/settings/users/(partials)/user-form/`, `useFormInit`, photo included, saved in one request), revoke (adds `revoked`, `canRevokeUser`), delete (`canRevokeUser`'s rules, never yourself), pick their user types from `UserTypes` (`user-types-select.tsx`: only the ones `filterAssignableUserTypes` allows, never `UneditableUserTypes`; at least one, since login needs a type; `saveUser` checks each grant and removal with `userTypeAssignment`; the super admin's types are never shown or changed there, and they can't be revoked; the password is entered twice, `SaveUserRequestSchema` checks they match), delete with everything theirs; `api/admin/users/save-user`, `delete-user`). The lists share `SettingsList`, `SettingsListRow`, `SettingsRowAction` and `use-settings-list`. Sub pages are declared once in `src/dict/settings-pages-dict.ts`; on every `/admin/settings/*` page the rail swaps the crew for a back link plus that list. Add a sub page by adding its folder and a dict entry
|
||||
|
||||
@@ -245,15 +245,16 @@ queue and Web Push.
|
||||
|
||||
## Images
|
||||
|
||||
Every release publishes a multi-arch image to both registries:
|
||||
Every release publishes a multi-arch image to
|
||||
`ghcr.io/moduletrace/cobble`, which is what the installer pulls. Tags are
|
||||
`latest`, the release version, `major.minor` and the commit sha.
|
||||
|
||||
- `ghcr.io/moduletrace/cobble`
|
||||
- `git.tben.me/moduletrace/cobble`
|
||||
|
||||
Tags are `latest`, the release version, `major.minor` and the commit sha.
|
||||
Install from Gitea with
|
||||
`--image git.tben.me/moduletrace/cobble`, or build from source with
|
||||
`docker build -t cobble:local .` and `--image cobble --version local --no-pull`.
|
||||
The same image is copied into `git.tben.me/moduletrace/cobble` by
|
||||
`bun run mirror:gitea`, which runs after the build over the VPN (the Gitea
|
||||
registry is not on the public internet, so GitHub Actions cannot reach it).
|
||||
Install from Gitea with `--image git.tben.me/moduletrace/cobble`, or build
|
||||
from source with `docker build -t cobble:local .` and
|
||||
`--image cobble --version local --no-pull`.
|
||||
|
||||
A release is published from a clean `main` with the two scripts, which tag the
|
||||
`package.json` version, push and create the release objects:
|
||||
@@ -264,7 +265,9 @@ scripts/publish-gitea.sh # or: bun run publish:gitea
|
||||
```
|
||||
|
||||
Both take `--version VERSION`, `--dry-run` and `--yes`. Only the GitHub release
|
||||
builds the image; see [RELEASING.md](RELEASING.md) for the full checklist.
|
||||
builds the image, and the Gitea registry is filled in afterwards, once that
|
||||
build has finished and from the VPN, by `bun run mirror:gitea`; see
|
||||
[RELEASING.md](RELEASING.md) for the full checklist.
|
||||
|
||||
## Development
|
||||
|
||||
|
||||
+14
-6
@@ -62,16 +62,24 @@ them in that order, since only the GitHub release builds images.
|
||||
gh release create v0.1.0 --title v0.1.0 --notes-file notes.md --latest
|
||||
|
||||
7. [ ] The release workflow builds `linux/amd64` and `linux/arm64` and
|
||||
pushes to both registries:
|
||||
- `ghcr.io/moduletrace/cobble`
|
||||
- `git.tben.me/moduletrace/cobble`
|
||||
pushes to `ghcr.io/moduletrace/cobble`, the registry the installer
|
||||
defaults to. It does not touch Gitea: that registry is on the VPN,
|
||||
and a GitHub runner cannot reach it.
|
||||
8. [ ] Check the manifests:
|
||||
|
||||
docker buildx imagetools inspect ghcr.io/moduletrace/cobble:latest
|
||||
docker buildx imagetools inspect git.tben.me/moduletrace/cobble:latest
|
||||
|
||||
9. [ ] Confirm the GHCR package is public and pull it anonymously.
|
||||
10. [ ] Publish the release to Gitea as well:
|
||||
10. [ ] Once the build has finished, copy the image into the Gitea registry
|
||||
from the VPN:
|
||||
|
||||
GITEA_TOKEN=... bun run mirror:gitea # or: scripts/mirror-gitea.sh
|
||||
|
||||
It copies the version and commit tags, plus `latest` and
|
||||
`major.minor` while this release still owns them, and reads every
|
||||
destination digest back. A failure here does not affect the release:
|
||||
the installer keeps working from GHCR.
|
||||
11. [ ] Publish the release to Gitea as well:
|
||||
|
||||
scripts/publish-gitea.sh # or: bun run publish:gitea
|
||||
|
||||
@@ -79,7 +87,7 @@ them in that order, since only the GitHub release builds images.
|
||||
git push gitea main
|
||||
git push gitea refs/tags/v0.1.0
|
||||
|
||||
11. [ ] Update every install with the one-liner:
|
||||
12. [ ] Update every install with the one-liner:
|
||||
|
||||
curl -fsSL https://url.moduletrace.com/cobble | sudo bash
|
||||
|
||||
|
||||
+143
@@ -2866,6 +2866,149 @@ const schema: BUN_SQLITE_DatabaseSchemaType = {
|
||||
},
|
||||
],
|
||||
},
|
||||
{
|
||||
tableName: "token_usages",
|
||||
tableDescription: `One model call that reported its usage: the tokens it spent, on which model and provider, for whom and for what. A bot message's total is the sum of the calls it took; the usage page sums them by model, provider and period`,
|
||||
fields: [
|
||||
{
|
||||
fieldName: "user_id",
|
||||
fieldDescription: `Whose work it was: the user whose conversation the call ran for`,
|
||||
dataType: "INTEGER",
|
||||
foreignKey: {
|
||||
cascadeDelete: true,
|
||||
destinationTableName: "users",
|
||||
destinationTableColumnName: "id",
|
||||
destinationTableColumnType: "INTEGER",
|
||||
foreignKeyName: "token_usage_user_fk",
|
||||
},
|
||||
},
|
||||
{
|
||||
fieldName: "bot_id",
|
||||
fieldDescription: `The bot it was for, when it was for one`,
|
||||
dataType: "INTEGER",
|
||||
foreignKey: {
|
||||
cascadeDelete: true,
|
||||
destinationTableName: "bots",
|
||||
destinationTableColumnName: "id",
|
||||
destinationTableColumnType: "INTEGER",
|
||||
foreignKeyName: "token_usage_bot_fk",
|
||||
},
|
||||
},
|
||||
{
|
||||
fieldName: "workspace_id",
|
||||
fieldDescription: `The workspace the call was in; null for a bot's own chat`,
|
||||
dataType: "INTEGER",
|
||||
foreignKey: {
|
||||
cascadeDelete: true,
|
||||
destinationTableName: "workspaces",
|
||||
destinationTableColumnName: "id",
|
||||
destinationTableColumnType: "INTEGER",
|
||||
foreignKeyName: "token_usage_workspace_fk",
|
||||
},
|
||||
},
|
||||
{
|
||||
fieldName: "session_id",
|
||||
fieldDescription: `The conversation it was in`,
|
||||
dataType: "INTEGER",
|
||||
foreignKey: {
|
||||
cascadeDelete: true,
|
||||
destinationTableName: "bot_sessions",
|
||||
destinationTableColumnName: "id",
|
||||
destinationTableColumnType: "INTEGER",
|
||||
foreignKeyName: "token_usage_session_fk",
|
||||
},
|
||||
},
|
||||
{
|
||||
fieldName: "message_id",
|
||||
fieldDescription: `The message it produced, so the thread can show what an answer cost. Null for calls that produce no message (naming a session, embedding a file)`,
|
||||
dataType: "INTEGER",
|
||||
foreignKey: {
|
||||
cascadeDelete: true,
|
||||
destinationTableName: "bot_messages",
|
||||
destinationTableColumnName: "id",
|
||||
destinationTableColumnType: "INTEGER",
|
||||
foreignKeyName: "token_usage_message_fk",
|
||||
},
|
||||
},
|
||||
{
|
||||
fieldName: "job_id",
|
||||
fieldDescription: `The queued job it ran for, when it ran for one`,
|
||||
dataType: "INTEGER",
|
||||
foreignKey: {
|
||||
cascadeDelete: true,
|
||||
destinationTableName: "bots_queue",
|
||||
destinationTableColumnName: "id",
|
||||
destinationTableColumnType: "INTEGER",
|
||||
foreignKeyName: "token_usage_job_fk",
|
||||
},
|
||||
},
|
||||
{
|
||||
fieldName: "harness",
|
||||
fieldDescription: `What the model was used for: a chat turn, reading media, transcribing, speaking, making an image, embedding`,
|
||||
dataType: "TEXT",
|
||||
options: Harnesses.map((h) => h.value),
|
||||
},
|
||||
{
|
||||
fieldName: "provider_id",
|
||||
fieldDescription: `The provider that served it. Not a foreign key: it stays readable after the provider is deleted`,
|
||||
dataType: "INTEGER",
|
||||
},
|
||||
{
|
||||
fieldName: "provider_name",
|
||||
fieldDescription: `The provider's name as it was, so a deleted provider still reads`,
|
||||
dataType: "TEXT",
|
||||
},
|
||||
{
|
||||
fieldName: "model_id",
|
||||
fieldDescription: `The catalog row of the model. Not a foreign key: a model the provider stops listing still counts`,
|
||||
dataType: "INTEGER",
|
||||
},
|
||||
{
|
||||
fieldName: "model_slug",
|
||||
fieldDescription: `The provider's model id, as it was called`,
|
||||
dataType: "TEXT",
|
||||
},
|
||||
{
|
||||
fieldName: "prompt_tokens",
|
||||
fieldDescription: `Tokens sent: the prompt, the system instructions and the tool results`,
|
||||
dataType: "INTEGER",
|
||||
defaultValue: 0,
|
||||
},
|
||||
{
|
||||
fieldName: "completion_tokens",
|
||||
fieldDescription: `Tokens written back: the answer, its tool calls and its reasoning`,
|
||||
dataType: "INTEGER",
|
||||
defaultValue: 0,
|
||||
},
|
||||
{
|
||||
fieldName: "cost",
|
||||
fieldDescription: `The provider's price in USD, when it reports one`,
|
||||
dataType: "REAL",
|
||||
},
|
||||
],
|
||||
indexes: [
|
||||
{
|
||||
indexName: "idx_token_usages_created_at",
|
||||
indexTableFields: ["created_at"],
|
||||
},
|
||||
{
|
||||
indexName: "idx_token_usages_message_id",
|
||||
indexTableFields: ["message_id"],
|
||||
},
|
||||
{
|
||||
indexName: "idx_token_usages_provider_model",
|
||||
indexTableFields: ["provider_id", "model_slug"],
|
||||
},
|
||||
{
|
||||
indexName: "idx_token_usages_user_created_at",
|
||||
indexTableFields: ["user_id", "created_at"],
|
||||
},
|
||||
{
|
||||
indexName: "idx_token_usages_bot_created_at",
|
||||
indexTableFields: ["bot_id", "created_at"],
|
||||
},
|
||||
],
|
||||
},
|
||||
{
|
||||
tableName: "bot_file_vectors",
|
||||
tableDescription: `sqlite-vec vectors of a file's chunks, at \`AppData.EmbeddingDimensions\` wide whatever the model returns. A vec0 table is only recreated with \`bunx bun-sqlite schema --vector\`, so the entrypoint never passes it`,
|
||||
|
||||
+73
-1
@@ -48,6 +48,7 @@ export const BunSQLiteTables = [
|
||||
"envs",
|
||||
"bot_files",
|
||||
"bot_file_chunks",
|
||||
"token_usages",
|
||||
"bot_file_vectors",
|
||||
"file_ingest_queue",
|
||||
"app_settings",
|
||||
@@ -1717,6 +1718,77 @@ export type BUN_SQLITE_COBBLE_BOT_FILE_CHUNKS = {
|
||||
token_estimate?: number | "";
|
||||
}
|
||||
|
||||
export type BUN_SQLITE_COBBLE_TOKEN_USAGES = {
|
||||
/**
|
||||
* The unique identifier of the record.
|
||||
*/
|
||||
id?: number | "";
|
||||
/**
|
||||
* The time when the record was created. (Unix Timestamp)
|
||||
*/
|
||||
created_at?: number | "";
|
||||
/**
|
||||
* The time when the record was updated. (Unix Timestamp)
|
||||
*/
|
||||
updated_at?: number | "";
|
||||
/**
|
||||
* Whose work it was: the user whose conversation the call ran for
|
||||
*/
|
||||
user_id?: number | "";
|
||||
/**
|
||||
* The bot it was for, when it was for one
|
||||
*/
|
||||
bot_id?: number | "";
|
||||
/**
|
||||
* The workspace the call was in; null for a bot's own chat
|
||||
*/
|
||||
workspace_id?: number | "";
|
||||
/**
|
||||
* The conversation it was in
|
||||
*/
|
||||
session_id?: number | "";
|
||||
/**
|
||||
* The message it produced, so the thread can show what an answer cost. Null for calls that produce no message (naming a session, embedding a file)
|
||||
*/
|
||||
message_id?: number | "";
|
||||
/**
|
||||
* The queued job it ran for, when it ran for one
|
||||
*/
|
||||
job_id?: number | "";
|
||||
/**
|
||||
* What the model was used for: a chat turn, reading media, transcribing, speaking, making an image, embedding
|
||||
*/
|
||||
harness?: "text" | "vision" | "transcription" | "speech" | "image" | "embedding" | "";
|
||||
/**
|
||||
* The provider that served it. Not a foreign key: it stays readable after the provider is deleted
|
||||
*/
|
||||
provider_id?: number | "";
|
||||
/**
|
||||
* The provider's name as it was, so a deleted provider still reads
|
||||
*/
|
||||
provider_name?: string;
|
||||
/**
|
||||
* The catalog row of the model. Not a foreign key: a model the provider stops listing still counts
|
||||
*/
|
||||
model_id?: number | "";
|
||||
/**
|
||||
* The provider's model id, as it was called
|
||||
*/
|
||||
model_slug?: string;
|
||||
/**
|
||||
* Tokens sent: the prompt, the system instructions and the tool results
|
||||
*/
|
||||
prompt_tokens?: number | "";
|
||||
/**
|
||||
* Tokens written back: the answer, its tool calls and its reasoning
|
||||
*/
|
||||
completion_tokens?: number | "";
|
||||
/**
|
||||
* The provider's price in USD, when it reports one
|
||||
*/
|
||||
cost?: number | "";
|
||||
}
|
||||
|
||||
export type BUN_SQLITE_COBBLE_BOT_FILE_VECTORS = {
|
||||
/**
|
||||
* The unique identifier of the record.
|
||||
@@ -1800,4 +1872,4 @@ export type BUN_SQLITE_COBBLE_APP_SETTINGS = {
|
||||
value?: string;
|
||||
}
|
||||
|
||||
export type BUN_SQLITE_COBBLE_ALL_TYPEDEFS = BUN_SQLITE_COBBLE_USERS & BUN_SQLITE_COBBLE_USER_TYPES & BUN_SQLITE_COBBLE_SSO_LOGIN_CODES & BUN_SQLITE_COBBLE_USER_PASSWORDS & BUN_SQLITE_COBBLE_MEDIA & BUN_SQLITE_COBBLE_MEDIA_PARADIGMS & BUN_SQLITE_COBBLE_BOTS & BUN_SQLITE_COBBLE_BOT_SESSIONS & BUN_SQLITE_COBBLE_BOT_MESSAGES & BUN_SQLITE_COBBLE_BOT_MESSAGE_REACTIONS & BUN_SQLITE_COBBLE_BOT_TASKS & BUN_SQLITE_COBBLE_BOT_TASK_STEPS & BUN_SQLITE_COBBLE_TASKS & BUN_SQLITE_COBBLE_TASK_MEDIA & BUN_SQLITE_COBBLE_TASK_NOTES & BUN_SQLITE_COBBLE_BOTS_QUEUE & BUN_SQLITE_COBBLE_BOTS_QUEUE_UPDATES & BUN_SQLITE_COBBLE_CRON_JOBS & BUN_SQLITE_COBBLE_WORKSPACES & BUN_SQLITE_COBBLE_WORKSPACE_BOTS & BUN_SQLITE_COBBLE_WORKSPACE_USERS & BUN_SQLITE_COBBLE_BOT_USERS & BUN_SQLITE_COBBLE_BOT_USER_MEMORIES & BUN_SQLITE_COBBLE_BOT_USER_FACES & BUN_SQLITE_COBBLE_BOT_MESSAGE_MEDIA & BUN_SQLITE_COBBLE_CONTAINERS & BUN_SQLITE_COBBLE_COMPUTERS & BUN_SQLITE_COBBLE_BOT_COMPUTERS & BUN_SQLITE_COBBLE_WORKSPACE_COMPUTERS & BUN_SQLITE_COBBLE_HARNESS_PROVIDERS & BUN_SQLITE_COBBLE_HARNESS_MODELS & BUN_SQLITE_COBBLE_HARNESS_SETTINGS & BUN_SQLITE_COBBLE_BOT_HARNESS_SETTINGS & BUN_SQLITE_COBBLE_WORKSPACE_HARNESS_SETTINGS & BUN_SQLITE_COBBLE_MCP_SERVERS & BUN_SQLITE_COBBLE_CUSTOM_TOOLS & BUN_SQLITE_COBBLE_SKILLS & BUN_SQLITE_COBBLE_BOT_MCP_SERVERS & BUN_SQLITE_COBBLE_WORKSPACE_MCP_SERVERS & BUN_SQLITE_COBBLE_BOT_CUSTOM_TOOLS & BUN_SQLITE_COBBLE_WORKSPACE_CUSTOM_TOOLS & BUN_SQLITE_COBBLE_BOT_SKILLS & BUN_SQLITE_COBBLE_WORKSPACE_SKILLS & BUN_SQLITE_COBBLE_LOGS & BUN_SQLITE_COBBLE_USER_PRESENCE & BUN_SQLITE_COBBLE_PUSH_SUBSCRIPTIONS & BUN_SQLITE_COBBLE_ENVS & BUN_SQLITE_COBBLE_BOT_FILES & BUN_SQLITE_COBBLE_BOT_FILE_CHUNKS & BUN_SQLITE_COBBLE_BOT_FILE_VECTORS & BUN_SQLITE_COBBLE_FILE_INGEST_QUEUE & BUN_SQLITE_COBBLE_APP_SETTINGS
|
||||
export type BUN_SQLITE_COBBLE_ALL_TYPEDEFS = BUN_SQLITE_COBBLE_USERS & BUN_SQLITE_COBBLE_USER_TYPES & BUN_SQLITE_COBBLE_SSO_LOGIN_CODES & BUN_SQLITE_COBBLE_USER_PASSWORDS & BUN_SQLITE_COBBLE_MEDIA & BUN_SQLITE_COBBLE_MEDIA_PARADIGMS & BUN_SQLITE_COBBLE_BOTS & BUN_SQLITE_COBBLE_BOT_SESSIONS & BUN_SQLITE_COBBLE_BOT_MESSAGES & BUN_SQLITE_COBBLE_BOT_MESSAGE_REACTIONS & BUN_SQLITE_COBBLE_BOT_TASKS & BUN_SQLITE_COBBLE_BOT_TASK_STEPS & BUN_SQLITE_COBBLE_TASKS & BUN_SQLITE_COBBLE_TASK_MEDIA & BUN_SQLITE_COBBLE_TASK_NOTES & BUN_SQLITE_COBBLE_BOTS_QUEUE & BUN_SQLITE_COBBLE_BOTS_QUEUE_UPDATES & BUN_SQLITE_COBBLE_CRON_JOBS & BUN_SQLITE_COBBLE_WORKSPACES & BUN_SQLITE_COBBLE_WORKSPACE_BOTS & BUN_SQLITE_COBBLE_WORKSPACE_USERS & BUN_SQLITE_COBBLE_BOT_USERS & BUN_SQLITE_COBBLE_BOT_USER_MEMORIES & BUN_SQLITE_COBBLE_BOT_USER_FACES & BUN_SQLITE_COBBLE_BOT_MESSAGE_MEDIA & BUN_SQLITE_COBBLE_CONTAINERS & BUN_SQLITE_COBBLE_COMPUTERS & BUN_SQLITE_COBBLE_BOT_COMPUTERS & BUN_SQLITE_COBBLE_WORKSPACE_COMPUTERS & BUN_SQLITE_COBBLE_HARNESS_PROVIDERS & BUN_SQLITE_COBBLE_HARNESS_MODELS & BUN_SQLITE_COBBLE_HARNESS_SETTINGS & BUN_SQLITE_COBBLE_BOT_HARNESS_SETTINGS & BUN_SQLITE_COBBLE_WORKSPACE_HARNESS_SETTINGS & BUN_SQLITE_COBBLE_MCP_SERVERS & BUN_SQLITE_COBBLE_CUSTOM_TOOLS & BUN_SQLITE_COBBLE_SKILLS & BUN_SQLITE_COBBLE_BOT_MCP_SERVERS & BUN_SQLITE_COBBLE_WORKSPACE_MCP_SERVERS & BUN_SQLITE_COBBLE_BOT_CUSTOM_TOOLS & BUN_SQLITE_COBBLE_WORKSPACE_CUSTOM_TOOLS & BUN_SQLITE_COBBLE_BOT_SKILLS & BUN_SQLITE_COBBLE_WORKSPACE_SKILLS & BUN_SQLITE_COBBLE_LOGS & BUN_SQLITE_COBBLE_USER_PRESENCE & BUN_SQLITE_COBBLE_PUSH_SUBSCRIPTIONS & BUN_SQLITE_COBBLE_ENVS & BUN_SQLITE_COBBLE_BOT_FILES & BUN_SQLITE_COBBLE_BOT_FILE_CHUNKS & BUN_SQLITE_COBBLE_TOKEN_USAGES & BUN_SQLITE_COBBLE_BOT_FILE_VECTORS & BUN_SQLITE_COBBLE_FILE_INGEST_QUEUE & BUN_SQLITE_COBBLE_APP_SETTINGS
|
||||
+3
-2
@@ -1,6 +1,6 @@
|
||||
{
|
||||
"name": "cobble",
|
||||
"version": "0.1.5",
|
||||
"version": "0.1.6",
|
||||
"module": "index.ts",
|
||||
"type": "module",
|
||||
"private": true,
|
||||
@@ -19,7 +19,8 @@
|
||||
"test": "bun test",
|
||||
"check:bundle": "bun src/scripts/check-bundle-size.ts",
|
||||
"publish:github": "bash scripts/publish-github.sh",
|
||||
"publish:gitea": "bash scripts/publish-gitea.sh"
|
||||
"publish:gitea": "bash scripts/publish-gitea.sh",
|
||||
"mirror:gitea": "bash scripts/mirror-gitea.sh"
|
||||
},
|
||||
"devDependencies": {
|
||||
"@next/mdx": "^16.3.0",
|
||||
|
||||
Executable
+318
@@ -0,0 +1,318 @@
|
||||
#!/usr/bin/env bash
|
||||
# Copies a published Cobble image from GHCR into the Gitea registry.
|
||||
#
|
||||
# scripts/mirror-gitea.sh
|
||||
#
|
||||
# The Gitea registry is on the VPN and GitHub Actions runners are on the
|
||||
# public internet, so CI can never reach it. The release workflow therefore
|
||||
# builds and pushes to GHCR only, and this script mirrors the result
|
||||
# separately: a registry that is unreachable, slow or full can never fail a
|
||||
# release the installer depends on.
|
||||
#
|
||||
# It copies from the Gitea host itself, because that host reaches GHCR over
|
||||
# the public internet and its own registry over the docker network it shares
|
||||
# with it. Nothing has to be installed on that host: skopeo runs there in a
|
||||
# container.
|
||||
#
|
||||
# `skopeo copy --all` keeps the manifest list and its signatures, so every
|
||||
# architecture comes across and the digest is unchanged. Each destination tag
|
||||
# is then read back and compared, so a partial copy is reported as a failure
|
||||
# instead of being assumed to have worked.
|
||||
#
|
||||
# Run it after scripts/publish-github.sh: that release is what builds the
|
||||
# image, so there is nothing to copy until it has finished.
|
||||
set -Eeuo pipefail
|
||||
|
||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||
# shellcheck source=lib/publish-common.sh
|
||||
. "$SCRIPT_DIR/lib/publish-common.sh"
|
||||
# shellcheck source=lib/publish-release.sh
|
||||
. "$SCRIPT_DIR/lib/publish-release.sh"
|
||||
|
||||
SOURCE_REPO="${MIRROR_SOURCE_REPO:-moduletrace/cobble}"
|
||||
SOURCE_REGISTRY_HOST="${MIRROR_SOURCE_REGISTRY_HOST:-ghcr.io}"
|
||||
HOST="${MIRROR_HOST:-10.7.0.4}"
|
||||
HOST_USER="${MIRROR_HOST_USER:-root}"
|
||||
REGISTRY_HOST="${MIRROR_REGISTRY_HOST:-gitea:3000}"
|
||||
REGISTRY_REPO="${MIRROR_REGISTRY_REPO:-moduletrace/cobble}"
|
||||
# The name a person types, which is not the one the copy uses: the host
|
||||
# reaches its registry by docker network name, while an install reaches it
|
||||
# through the proxy in front of it.
|
||||
REGISTRY_PUBLIC_HOST="${MIRROR_REGISTRY_PUBLIC_HOST:-git.tben.me}"
|
||||
NETWORK="${MIRROR_NETWORK:-cloudflare}"
|
||||
SKOPEO_IMAGE="${MIRROR_SKOPEO_IMAGE:-quay.io/skopeo/stable:latest}"
|
||||
MIRROR_USER="${MIRROR_USER:-cobble}"
|
||||
CONNECT_TIMEOUT="${MIRROR_CONNECT_TIMEOUT:-10}"
|
||||
|
||||
usage() {
|
||||
cat <<EOF
|
||||
Copies a published Cobble image from GHCR into the Gitea registry.
|
||||
|
||||
Usage: scripts/mirror-gitea.sh [options]
|
||||
|
||||
Options:
|
||||
--version VERSION Version to mirror, with or without the leading v.
|
||||
Defaults to package.json's version.
|
||||
--yes, -y Do not prompt.
|
||||
--dry-run Print what would run, copy nothing.
|
||||
-h, --help Show this help.
|
||||
|
||||
The version is copied as vX.Y.Z and sha-<commit>. It is copied as latest and
|
||||
major.minor too, but only while it is still the newest release in GHCR, so
|
||||
mirroring an older release never rolls those two tags backwards. Every
|
||||
destination is read back and its digest compared with GHCR's.
|
||||
|
||||
Run it after scripts/publish-github.sh, and after the image build for that
|
||||
release has finished.
|
||||
|
||||
Environment:
|
||||
GITEA_TOKEN Token for the registry (required).
|
||||
MIRROR_HOST The Gitea host to copy from over the VPN.
|
||||
Default: 10.7.0.4
|
||||
MIRROR_HOST_USER Its ssh user. Default: root
|
||||
MIRROR_REGISTRY_HOST The registry as that host sees it, by its docker
|
||||
network name. Default: gitea:3000
|
||||
MIRROR_REGISTRY_PUBLIC_HOST
|
||||
The registry's public name, for the pull command
|
||||
it prints. Default: git.tben.me
|
||||
MIRROR_SOURCE_REGISTRY_HOST
|
||||
Where the release workflow pushed.
|
||||
Default: ghcr.io
|
||||
MIRROR_NETWORK The docker network the registry is on, so the
|
||||
copy container can resolve it.
|
||||
Default: cloudflare
|
||||
MIRROR_USER The registry user for basic auth. Default: cobble
|
||||
MIRROR_SKOPEO_IMAGE The skopeo image the host runs.
|
||||
Default: quay.io/skopeo/stable:latest
|
||||
DRY_RUN=1, ASSUME_YES=1 Same as --dry-run and --yes.
|
||||
EOF
|
||||
}
|
||||
|
||||
parse_publish_args "$@"
|
||||
if [ "$HELP" = "1" ]; then
|
||||
usage
|
||||
exit 0
|
||||
fi
|
||||
reject_unknown_args "$0"
|
||||
|
||||
# MARK:mirror-tags
|
||||
# The moving tag a release owns: major.minor, read from the version the same
|
||||
# way the release workflow reads it, so the two never disagree.
|
||||
mirror_minor_tag() {
|
||||
printf '%s' "${1#v}" | sed -E 's/^([0-9]+)\.([0-9]+).*/\1.\2/'
|
||||
}
|
||||
|
||||
# The commit tag for a release: the first 12 characters of the commit the
|
||||
# tag points at, which is `sha-${GITHUB_SHA::12}` in the workflow.
|
||||
mirror_sha_tag() {
|
||||
local sha
|
||||
sha="$(git rev-parse -q --verify "refs/tags/$1^{commit}" 2>/dev/null || true)"
|
||||
if [ -z "$sha" ]; then
|
||||
return 1
|
||||
fi
|
||||
printf 'sha-%s' "${sha:0:12}"
|
||||
}
|
||||
|
||||
# The tags this release owns outright, which no newer release claims.
|
||||
mirror_fixed_tags() {
|
||||
local sha_tag
|
||||
printf '%s\n' "$1"
|
||||
if sha_tag="$(mirror_sha_tag "$1")"; then
|
||||
printf '%s\n' "$sha_tag"
|
||||
fi
|
||||
}
|
||||
|
||||
# The tags a newer release takes over.
|
||||
mirror_moving_tags() {
|
||||
printf 'latest\n%s\n' "$(mirror_minor_tag "$1")"
|
||||
}
|
||||
# END:mirror-tags
|
||||
|
||||
# MARK:mirror-guard
|
||||
# Whether this release may still write the moving tags. It may while GHCR's
|
||||
# `latest` is its own image: once a newer release has moved `latest`, copying
|
||||
# an older version into the Gitea registry must not drag it back.
|
||||
mirror_owns_moving_tags() {
|
||||
if [ -z "$2" ]; then
|
||||
# Nothing to compare against, so this release is taken as the newest.
|
||||
return 0
|
||||
fi
|
||||
[ "$1" = "$2" ]
|
||||
}
|
||||
# END:mirror-guard
|
||||
|
||||
# MARK:mirror-digest
|
||||
MANIFEST_ACCEPT="application/vnd.oci.image.index.v1+json, application/vnd.docker.distribution.manifest.list.v2+json, application/vnd.oci.image.manifest.v1+json, application/vnd.docker.distribution.manifest.v2+json"
|
||||
|
||||
# Reads a tag's digest from GHCR over the public internet. The token is
|
||||
# anonymous: the image is public, and a registry answers a bare manifest
|
||||
# request with 401 until it is given one.
|
||||
github_registry_digest() {
|
||||
local tag="$1"
|
||||
local token
|
||||
token="$(
|
||||
curl -sSf "https://$SOURCE_REGISTRY_HOST/token?scope=repository:$SOURCE_REPO:pull" 2>/dev/null |
|
||||
sed -n 's/.*"token":"\([^"]*\)".*/\1/p'
|
||||
)" || return 1
|
||||
if [ -z "$token" ]; then
|
||||
return 1
|
||||
fi
|
||||
curl -sSf -o /dev/null -D- \
|
||||
-H "Authorization: Bearer $token" \
|
||||
-H "Accept: $MANIFEST_ACCEPT" \
|
||||
"https://$SOURCE_REGISTRY_HOST/v2/$SOURCE_REPO/manifests/$tag" 2>/dev/null |
|
||||
tr -d '\r' | sed -n 's/^docker-content-digest: //Ip' | head -n1
|
||||
}
|
||||
# END:mirror-digest
|
||||
|
||||
# MARK:mirror-remote
|
||||
# Runs a bash script on the Gitea host, which is the one place that reaches
|
||||
# both registries. The script arrives on stdin, so nothing is installed there
|
||||
# and no quoting has to survive a round trip through ssh.
|
||||
mirror_on_host() {
|
||||
ssh -o BatchMode=yes -o ConnectTimeout="$CONNECT_TIMEOUT" "$HOST_USER@$HOST" \
|
||||
"MIRROR_NETWORK='$NETWORK' MIRROR_SKOPEO_IMAGE='$SKOPEO_IMAGE' bash -s"
|
||||
}
|
||||
|
||||
# Reads a destination tag's digest back from the registry. Empty means the
|
||||
# tag is not there, which a caller reports differently from a mismatch.
|
||||
mirror_digest_on_host() {
|
||||
mirror_on_host <<REMOTE
|
||||
set -uo pipefail
|
||||
docker run --rm --network "\$MIRROR_NETWORK" "\$MIRROR_SKOPEO_IMAGE" \\
|
||||
inspect --tls-verify=false --creds '$MIRROR_USER:$GITEA_TOKEN' \\
|
||||
--format '{{.Digest}}' 'docker://$REGISTRY_HOST/$REGISTRY_REPO:$1' 2>/dev/null | tail -n1
|
||||
REMOTE
|
||||
}
|
||||
|
||||
# Copies one tag, leaving skopeo's output in MIRROR_COPY_LOG so a failure can
|
||||
# be shown without streaming its progress into the report.
|
||||
mirror_copy_on_host() {
|
||||
MIRROR_COPY_LOG="$(mktemp -t cobble-mirror.XXXXXX)"
|
||||
mirror_on_host <<REMOTE >"$MIRROR_COPY_LOG" 2>&1
|
||||
set -uo pipefail
|
||||
docker run --rm --network "\$MIRROR_NETWORK" "\$MIRROR_SKOPEO_IMAGE" \\
|
||||
copy --all --retry-times 5 \\
|
||||
--src-tls-verify=true --dest-tls-verify=false \\
|
||||
--dest-creds '$MIRROR_USER:$GITEA_TOKEN' \\
|
||||
'docker://$SOURCE_REGISTRY_HOST/$SOURCE_REPO:$1' \\
|
||||
'docker://$REGISTRY_HOST/$REGISTRY_REPO:$2'
|
||||
REMOTE
|
||||
}
|
||||
# END:mirror-remote
|
||||
|
||||
# MARK:mirror-main
|
||||
require_repo_root
|
||||
|
||||
step "Resolving the version"
|
||||
if [ -z "$VERSION_RAW" ]; then
|
||||
VERSION_RAW="$(source_version)"
|
||||
if [ -z "$VERSION_RAW" ]; then
|
||||
fail "No version in package.json and none passed with --version."
|
||||
fi
|
||||
detail "Read from package.json: $VERSION_RAW"
|
||||
fi
|
||||
validate_version "$VERSION_RAW"
|
||||
TAG="$(normalize_version "$VERSION_RAW")"
|
||||
ok "Mirroring $TAG."
|
||||
|
||||
require_cmd git
|
||||
require_cmd ssh "Install an ssh client."
|
||||
require_cmd curl "Install curl."
|
||||
|
||||
if [ -z "${GITEA_TOKEN:-}" ]; then
|
||||
fail "No Gitea token. Set GITEA_TOKEN to a token whose user can push packages to the $REGISTRY_REPO owner."
|
||||
fi
|
||||
|
||||
step "Checking the released image in $SOURCE_REGISTRY_HOST"
|
||||
SOURCE_DIGEST="$(github_registry_digest "$TAG")" ||
|
||||
fail "No $TAG in $SOURCE_REGISTRY_HOST/$SOURCE_REPO. Publish the GitHub release first: scripts/publish-github.sh builds the image."
|
||||
if [ -z "$SOURCE_DIGEST" ]; then
|
||||
fail "No $TAG in $SOURCE_REGISTRY_HOST/$SOURCE_REPO. Publish the GitHub release first: scripts/publish-github.sh builds the image."
|
||||
fi
|
||||
ok "$TAG is $SOURCE_DIGEST."
|
||||
|
||||
LATEST_DIGEST="$(github_registry_digest latest || true)"
|
||||
|
||||
TAGS_TO_COPY="$(mirror_fixed_tags "$TAG")"
|
||||
MOVING_SKIPPED=""
|
||||
if mirror_owns_moving_tags "$SOURCE_DIGEST" "$LATEST_DIGEST"; then
|
||||
TAGS_TO_COPY="$(
|
||||
printf '%s\n%s\n' "$TAGS_TO_COPY" "$(mirror_moving_tags "$TAG")" |
|
||||
awk 'NF && !seen[$0]++'
|
||||
)"
|
||||
else
|
||||
MOVING_SKIPPED="$(mirror_moving_tags "$TAG" | tr '\n' ' ')"
|
||||
fi
|
||||
|
||||
step "Mirroring $TAG into the Gitea registry"
|
||||
printf ' %shost%s %s\n' "$C_DIM" "$C_RESET" "$HOST_USER@$HOST"
|
||||
printf ' %sregistry%s %s/%s\n' "$C_DIM" "$C_RESET" "$REGISTRY_HOST" "$REGISTRY_REPO"
|
||||
printf ' %ssource%s %s/%s\n' "$C_DIM" "$C_RESET" "$SOURCE_REGISTRY_HOST" "$SOURCE_REPO"
|
||||
printf ' %stags%s %s\n' "$C_DIM" "$C_RESET" "$(printf '%s' "$TAGS_TO_COPY" | tr '\n' ' ')"
|
||||
if [ -n "$MOVING_SKIPPED" ]; then
|
||||
warn "$SOURCE_REGISTRY_HOST/$SOURCE_REPO:latest belongs to a newer release, so ${MOVING_SKIPPED% } is left where it is."
|
||||
fi
|
||||
|
||||
step "Checking the Gitea host"
|
||||
if is_dry_run; then
|
||||
detail "[dry run] ssh -o BatchMode=yes $HOST_USER@$HOST 'bash -s'"
|
||||
remote_hostname="$HOST"
|
||||
else
|
||||
remote_hostname="$(
|
||||
mirror_on_host <<'REMOTE'
|
||||
hostname
|
||||
REMOTE
|
||||
)" || fail "Could not reach $HOST_USER@$HOST over ssh. The Gitea host is on the VPN: connect to it first."
|
||||
ok "Reached $remote_hostname."
|
||||
fi
|
||||
|
||||
confirm "Mirror $TAG into $REGISTRY_HOST/$REGISTRY_REPO?"
|
||||
|
||||
failed=0
|
||||
for dest_tag in $TAGS_TO_COPY; do
|
||||
step "Copying $TAG to $dest_tag"
|
||||
|
||||
if is_dry_run; then
|
||||
detail "[dry run] skopeo copy --all $SOURCE_REGISTRY_HOST/$SOURCE_REPO:$TAG -> $REGISTRY_HOST/$REGISTRY_REPO:$dest_tag"
|
||||
continue
|
||||
fi
|
||||
|
||||
if ! mirror_copy_on_host "$TAG" "$dest_tag"; then
|
||||
sed -n '$!{/^$/d};p' "${MIRROR_COPY_LOG:-/dev/null}" 2>/dev/null | tail -n 8 >&2
|
||||
warn "The copy to $dest_tag failed."
|
||||
failed=1
|
||||
continue
|
||||
fi
|
||||
|
||||
# The digest is what makes the two registries the same image: a copy that
|
||||
# stored a different manifest list would read back differently here.
|
||||
copied_digest="$(mirror_digest_on_host "$dest_tag")"
|
||||
if [ "$copied_digest" != "$SOURCE_DIGEST" ]; then
|
||||
warn "$dest_tag reads back as ${copied_digest:-nothing}, not $SOURCE_DIGEST."
|
||||
failed=1
|
||||
continue
|
||||
fi
|
||||
ok "$dest_tag is $SOURCE_DIGEST."
|
||||
done
|
||||
|
||||
if [ -n "${MIRROR_COPY_LOG:-}" ] && [ -f "$MIRROR_COPY_LOG" ]; then
|
||||
rm -f "$MIRROR_COPY_LOG"
|
||||
fi
|
||||
|
||||
step "Mirrored $TAG"
|
||||
printf ' %sregistry%s %s/%s\n' "$C_DIM" "$C_RESET" "$REGISTRY_HOST" "$REGISTRY_REPO"
|
||||
printf ' %stags%s %s\n' "$C_DIM" "$C_RESET" "$(printf '%s' "$TAGS_TO_COPY" | tr '\n' ' ')"
|
||||
printf ' %sdigest%s %s\n' "$C_DIM" "$C_RESET" "$SOURCE_DIGEST"
|
||||
if [ "$DRY_RUN" = "1" ]; then
|
||||
printf '\n%sDry run: copied nothing.%s\n' "$C_YELLOW" "$C_RESET"
|
||||
fi
|
||||
|
||||
if [ "$failed" = "1" ]; then
|
||||
fail "The Gitea registry does not match $SOURCE_REGISTRY_HOST/$SOURCE_REPO. The release itself is unaffected: the installer defaults to GHCR."
|
||||
fi
|
||||
|
||||
printf '\nAn install that can reach the Gitea registry can pull it with:\n'
|
||||
printf ' curl -fsSL https://url.moduletrace.com/cobble | sudo bash -s -- --image %s/%s:%s\n' \
|
||||
"$REGISTRY_PUBLIC_HOST" "$REGISTRY_REPO" "$TAG"
|
||||
# END:mirror-main
|
||||
@@ -8,9 +8,10 @@
|
||||
# creates the Gitea release over the API.
|
||||
#
|
||||
# Gitea has no release workflow, so nothing is built here: the image is built
|
||||
# once by .github/workflows/release.yml and mirrored to
|
||||
# git.tben.me/moduletrace/cobble. This script only publishes the source and
|
||||
# the release object.
|
||||
# once by .github/workflows/release.yml and copied into the Gitea registry by
|
||||
# scripts/mirror-gitea.sh, run from the VPN side, since a GitHub runner can
|
||||
# never reach it. This script only publishes the source and the release
|
||||
# object.
|
||||
#
|
||||
# The token comes from GITEA_TOKEN, or from the Gitea section of ~/.config/tea
|
||||
# if the tea CLI is installed. It needs the `write:repository` scope on the
|
||||
@@ -221,6 +222,5 @@ else
|
||||
fi
|
||||
|
||||
print_release_summary "$REMOTE" "$BRANCH" "https://${REMOTE_HOST}/${REPO_SLUG}/releases/tag/$TAG"
|
||||
printf '\nThe image is built by the GitHub release; mirror it into the Gitea registry with:\n'
|
||||
printf ' docker buildx imagetools create --tag git.tben.me/moduletrace/cobble:latest \\\n'
|
||||
printf ' ghcr.io/moduletrace/cobble:latest\n'
|
||||
printf '\nThe image is built by the GitHub release, which cannot reach this registry.\nOnce that build has finished, copy it in over the VPN with:\n'
|
||||
printf ' bun run mirror:gitea\n'
|
||||
|
||||
@@ -7,11 +7,12 @@
|
||||
# installer syntax), tags HEAD with the package.json version, pushes main and
|
||||
# the tag to the `origin` remote, then creates the GitHub release. Publishing
|
||||
# that release is what fires .github/workflows/release.yml, which builds the
|
||||
# multi-arch image and pushes it to GHCR (and mirrors it to Gitea).
|
||||
# multi-arch image and pushes it to GHCR, the registry the installer defaults
|
||||
# to. It never pushes to Gitea: that registry is on the VPN, out of a
|
||||
# runner's reach, so scripts/mirror-gitea.sh copies the image separately.
|
||||
#
|
||||
# Run scripts/publish-gitea.sh afterwards for the source mirror's own release
|
||||
# object. There is no existing publish script in the repo: this and its Gitea
|
||||
# twin are it.
|
||||
# object, and scripts/mirror-gitea.sh once the image build has finished.
|
||||
set -Eeuo pipefail
|
||||
|
||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||
|
||||
@@ -4,6 +4,7 @@ import path from "path";
|
||||
const ROOT = path.join(import.meta.dir, "..");
|
||||
const COMMON = path.join(ROOT, "scripts/lib/publish-common.sh");
|
||||
const RELEASE = path.join(ROOT, "scripts/lib/publish-release.sh");
|
||||
const MIRROR = path.join(ROOT, "scripts/mirror-gitea.sh");
|
||||
const SCRIPTS = [
|
||||
path.join(ROOT, "scripts/publish-github.sh"),
|
||||
path.join(ROOT, "scripts/publish-gitea.sh"),
|
||||
@@ -99,13 +100,21 @@ async function pathWithoutJq(): Promise<string> {
|
||||
}
|
||||
|
||||
describe("publish scripts syntax", () => {
|
||||
for (const script of [...SCRIPTS, COMMON, RELEASE]) {
|
||||
for (const script of [...SCRIPTS, MIRROR, COMMON, RELEASE]) {
|
||||
test(`${path.relative(ROOT, script)} parses`, async () => {
|
||||
const proc = Bun.spawn({ cmd: ["bash", "-n", script] });
|
||||
expect(await proc.exited).toBe(0);
|
||||
});
|
||||
}
|
||||
|
||||
test("the mirror script sources the shared libraries", async () => {
|
||||
const source = await Bun.file(MIRROR).text();
|
||||
|
||||
expect(source).toContain('SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"');
|
||||
expect(source).toContain('. "$SCRIPT_DIR/lib/publish-common.sh"');
|
||||
expect(source).toContain('. "$SCRIPT_DIR/lib/publish-release.sh"');
|
||||
});
|
||||
|
||||
test("both publish scripts source the shared library", async () => {
|
||||
for (const script of SCRIPTS) {
|
||||
const source = await Bun.file(script).text();
|
||||
@@ -141,6 +150,156 @@ describe("publish scripts remote wiring", () => {
|
||||
});
|
||||
});
|
||||
|
||||
describe("mirror script wiring", () => {
|
||||
test("CI never pushes to the Gitea registry", async () => {
|
||||
const workflow = await Bun.file(
|
||||
path.join(ROOT, ".github/workflows/release.yml"),
|
||||
).text();
|
||||
|
||||
expect(workflow).not.toContain("git.tben.me");
|
||||
expect(workflow).not.toContain("GITEA_TOKEN");
|
||||
expect(workflow).not.toContain("imagetools create");
|
||||
expect(workflow).toContain("Build and push to GHCR");
|
||||
});
|
||||
|
||||
test("the mirror script copies every tag the workflow publishes", async () => {
|
||||
const [workflow, mirror] = await Promise.all([
|
||||
Bun.file(path.join(ROOT, ".github/workflows/release.yml")).text(),
|
||||
Bun.file(MIRROR).text(),
|
||||
]);
|
||||
|
||||
// The workflow's own four tags, so a mirror tag can never be invented
|
||||
// here and go missing there.
|
||||
for (const tag of ["latest", "${version}", "${minor}", "${sha}"]) {
|
||||
expect(workflow).toContain(tag);
|
||||
}
|
||||
expect(mirror).toContain("mirror_minor_tag");
|
||||
expect(mirror).toContain("mirror_sha_tag");
|
||||
expect(mirror).toContain("mirror_fixed_tags");
|
||||
expect(mirror).toContain("mirror_moving_tags");
|
||||
});
|
||||
|
||||
test("the mirror script asks before copying and verifies each digest", async () => {
|
||||
const source = await Bun.file(MIRROR).text();
|
||||
|
||||
expect(source).toContain("confirm \"Mirror $TAG into");
|
||||
expect(source).toContain("mirror_digest_on_host");
|
||||
expect(source).toContain("skopeo");
|
||||
expect(source).toContain("--all");
|
||||
});
|
||||
|
||||
test("the pull command it prints names the public host, not the docker one", async () => {
|
||||
const source = await Bun.file(MIRROR).text();
|
||||
|
||||
// The copy reaches the registry by docker network name, which is not
|
||||
// resolvable for an install; printing that would hand out a broken
|
||||
// command.
|
||||
expect(source).toContain("MIRROR_REGISTRY_PUBLIC_HOST");
|
||||
expect(source).toContain('"$REGISTRY_PUBLIC_HOST" "$REGISTRY_REPO" "$TAG"');
|
||||
expect(source).not.toContain('"${REGISTRY_HOST%%:*}" "$REGISTRY_REPO" "$TAG"');
|
||||
});
|
||||
|
||||
test("without a token the mirror script refuses instead of guessing", async () => {
|
||||
const proc = Bun.spawn({
|
||||
cmd: ["bash", MIRROR, "--yes"],
|
||||
cwd: ROOT,
|
||||
env: { ...process.env, GITEA_TOKEN: "" },
|
||||
stdout: "pipe",
|
||||
stderr: "pipe",
|
||||
});
|
||||
const [stderr, code] = await Promise.all([
|
||||
new Response(proc.stderr).text(),
|
||||
proc.exited,
|
||||
]);
|
||||
|
||||
expect(code).toBe(1);
|
||||
expect(stderr).toContain("No Gitea token");
|
||||
});
|
||||
});
|
||||
|
||||
describe("mirror lib tag handling", () => {
|
||||
test("major.minor is read exactly as the release workflow reads it", async () => {
|
||||
const block = await blockCaller({ file: MIRROR, name: "mirror-tags" });
|
||||
|
||||
for (const [version, expected] of [
|
||||
["v0.1.5", "0.1"],
|
||||
["v1.2.3", "1.2"],
|
||||
["v10.20.30", "10.20"],
|
||||
["v0.1.5-rc.1", "0.1"],
|
||||
] as const) {
|
||||
const out = await runBash({
|
||||
script: [block, `mirror_minor_tag '${version}'`].join("\n"),
|
||||
});
|
||||
expect(out).toBe(expected);
|
||||
}
|
||||
});
|
||||
|
||||
test("fixed tags are the version and its commit, moving ones the rest", async () => {
|
||||
const block = await blockCaller({ file: MIRROR, name: "mirror-tags" });
|
||||
|
||||
const fixed = await runBash({
|
||||
script: [block, "mirror_fixed_tags v0.1.5"].join("\n"),
|
||||
});
|
||||
expect(fixed.split("\n")[0]).toBe("v0.1.5");
|
||||
|
||||
const moving = await runBash({
|
||||
script: [block, "mirror_moving_tags v0.1.5"].join("\n"),
|
||||
});
|
||||
expect(moving.split("\n")).toEqual(["latest", "0.1"]);
|
||||
});
|
||||
|
||||
test("a tag with no commit on disk still mirrors its version", async () => {
|
||||
const block = await blockCaller({ file: MIRROR, name: "mirror-tags" });
|
||||
|
||||
const out = await runBash({
|
||||
script: [block, "mirror_fixed_tags v99.99.99"].join("\n"),
|
||||
});
|
||||
|
||||
expect(out).toBe("v99.99.99");
|
||||
});
|
||||
});
|
||||
|
||||
describe("mirror lib moving tag guard", () => {
|
||||
test("a release may move latest while it owns it", async () => {
|
||||
const block = await blockCaller({ file: MIRROR, name: "mirror-guard" });
|
||||
|
||||
const out = await runBash({
|
||||
script: [
|
||||
block,
|
||||
'mirror_owns_moving_tags sha256:aa sha256:aa && printf owns',
|
||||
].join("\n"),
|
||||
});
|
||||
|
||||
expect(out).toBe("owns");
|
||||
});
|
||||
|
||||
test("a newer release's latest is never rolled back", async () => {
|
||||
const block = await blockCaller({ file: MIRROR, name: "mirror-guard" });
|
||||
|
||||
const out = await runBash({
|
||||
script: [
|
||||
block,
|
||||
'if mirror_owns_moving_tags sha256:aa sha256:bb; then printf owns; else printf refuses; fi',
|
||||
].join("\n"),
|
||||
});
|
||||
|
||||
expect(out).toBe("refuses");
|
||||
});
|
||||
|
||||
test("with no latest to compare, the release is taken as newest", async () => {
|
||||
const block = await blockCaller({ file: MIRROR, name: "mirror-guard" });
|
||||
|
||||
const out = await runBash({
|
||||
script: [
|
||||
block,
|
||||
'mirror_owns_moving_tags sha256:aa "" && printf owns',
|
||||
].join("\n"),
|
||||
});
|
||||
|
||||
expect(out).toBe("owns");
|
||||
});
|
||||
});
|
||||
|
||||
describe("publish lib version handling", () => {
|
||||
test("normalizes a version with and without the v", async () => {
|
||||
const block = await blockCaller({ name: "version" });
|
||||
|
||||
@@ -4,6 +4,7 @@ import { AppData } from "@/src/data/app-data";
|
||||
import { SessionNamePrompt } from "@/src/data/harness-prompts";
|
||||
import grabBotMessages from "@/src/functions/backend/db/bots/grab-bot-messages";
|
||||
import generateText from "@/src/harness/text/generate-text";
|
||||
import { runWithHarnessUsageScope } from "@/src/harness/usage/harness-usage-scope";
|
||||
import parseSessionName from "./parse-session-name";
|
||||
|
||||
type Params = {
|
||||
@@ -30,7 +31,7 @@ export default async function nameBotSession({
|
||||
}: Params): Promise<boolean> {
|
||||
const session = (
|
||||
await BunSQLite.sql<BUN_SQLITE_COBBLE_BOT_SESSIONS>({
|
||||
sql: `SELECT id, title, summary FROM bot_sessions WHERE id = ? LIMIT 1`,
|
||||
sql: `SELECT id, user_id, title, summary FROM bot_sessions WHERE id = ? LIMIT 1`,
|
||||
values: [session_id],
|
||||
})
|
||||
).payload?.[0];
|
||||
@@ -57,14 +58,30 @@ export default async function nameBotSession({
|
||||
)
|
||||
.join("\n\n");
|
||||
|
||||
const result = await generateText({
|
||||
messages: [
|
||||
{ role: "system", content: SessionNamePrompt },
|
||||
{ role: "user", content: transcript },
|
||||
],
|
||||
bot_id,
|
||||
workspace_id,
|
||||
signal,
|
||||
/**
|
||||
* The naming turn is a model call too, so it is recorded for this
|
||||
* conversation. It runs after the job is complete, on purpose, so it
|
||||
* has no job and no message of its own to charge it to.
|
||||
*/
|
||||
const result = await runWithHarnessUsageScope({
|
||||
scope: {
|
||||
user_id: session.user_id || undefined,
|
||||
bot_id,
|
||||
workspace_id: workspace_id || null,
|
||||
session_id: session.id,
|
||||
job_id: null,
|
||||
message_id: null,
|
||||
},
|
||||
task: () =>
|
||||
generateText({
|
||||
messages: [
|
||||
{ role: "system", content: SessionNamePrompt },
|
||||
{ role: "user", content: transcript },
|
||||
],
|
||||
bot_id,
|
||||
workspace_id,
|
||||
signal,
|
||||
}),
|
||||
});
|
||||
|
||||
const name = parseSessionName({ text: result.text });
|
||||
|
||||
@@ -1,6 +1,7 @@
|
||||
import grabBotById from "@/src/functions/backend/db/bots/grab-bot-by-id";
|
||||
import grabJobUserId from "@/src/functions/backend/db/bots-queue/grab-job-user-id";
|
||||
import HarnessError from "@/src/harness/harness-error";
|
||||
import { runWithHarnessUsageScope } from "@/src/harness/usage/harness-usage-scope";
|
||||
import BotJobFatalError from "./bot-job-fatal-error";
|
||||
import runBotReply from "./run-bot-reply";
|
||||
import runBotTask from "./run-bot-task";
|
||||
@@ -33,9 +34,25 @@ const runBotJob: BotJobRunner = async (context) => {
|
||||
const owned = { ...bot, id: bot.id };
|
||||
|
||||
try {
|
||||
return job.kind == "reply"
|
||||
? await runBotReply({ ...context, bot: owned, user_id })
|
||||
: await runBotTask({ ...context, bot: owned, user_id });
|
||||
/**
|
||||
* Everything the job's models do is recorded for this user, this
|
||||
* bot, this conversation and this job. A reply's text loop adds
|
||||
* the message it is producing to the one-turn scope it opens.
|
||||
*/
|
||||
return await runWithHarnessUsageScope({
|
||||
scope: {
|
||||
user_id,
|
||||
bot_id: bot.id,
|
||||
workspace_id: job.workspace_id || null,
|
||||
session_id: job.session_id || null,
|
||||
job_id: job.id || null,
|
||||
message_id: null,
|
||||
},
|
||||
task: () =>
|
||||
job.kind == "reply"
|
||||
? runBotReply({ ...context, bot: owned, user_id })
|
||||
: runBotTask({ ...context, bot: owned, user_id }),
|
||||
});
|
||||
} catch (error: unknown) {
|
||||
if (
|
||||
error instanceof HarnessError &&
|
||||
|
||||
@@ -328,6 +328,16 @@ export const AppData = {
|
||||
* How many older messages each scroll to the top of the thread loads.
|
||||
*/
|
||||
ConversationOlderMessages: 30,
|
||||
/**
|
||||
* Days of usage the usage page totals by default, and the most it will
|
||||
* total at once.
|
||||
*/
|
||||
UsageDefaultDays: 30,
|
||||
UsageMaxDays: 365,
|
||||
/**
|
||||
* Models listed per page on the usage page, the busiest first.
|
||||
*/
|
||||
UsageModelPageSize: 20,
|
||||
/**
|
||||
* How much of a bot's last message the rail's lists carry, as plain
|
||||
* text (`messagePreview`): a row shows one line of it.
|
||||
|
||||
@@ -27,6 +27,9 @@ Permission to draft is not permission to send, permission to analyze is not perm
|
||||
## 2. Delegate well
|
||||
|
||||
- Pick by role, instructions and tools, not by name.
|
||||
- Say where the work runs, and mean it. \`delegate_task\` takes an optional \`workspace_id\`, and that one argument decides the whole context the worker has: its own chat, or a workspace. A task in the worker's own chat has only that worker's own memory, files and directly assigned computers. A task in a workspace has the workspace's shared context, the worker's instructions and memory for that workspace, the project's files, and the workspace's computers as well as the worker's. The result also lands in a different conversation: the worker's own chat, or its conversation in that workspace beside the rest of that project. Choose by where the work belongs, not by which is easier to type.
|
||||
- Handed over from a conversation inside a workspace, the task always runs in that workspace and \`workspace_id\` is not accepted: say so, don't pretend you moved it.
|
||||
- Delegating into a workspace the worker isn't in adds it to that workspace as a side effect. Say that in your reply when it happens; don't add members silently for a one-off.
|
||||
- Give a full brief: objective, inputs, deliverable, acceptance criteria, constraints.
|
||||
- Check \`list_tasks\` first. Never give two bots the same job, and don't restart work an open task already covers.
|
||||
- Track what comes back, judge it against the request, and integrate it. Don't pass a worker's unchecked claim on as fact.
|
||||
|
||||
@@ -115,7 +115,7 @@ Every action counts here: saving memory or instructions, creating or changing a
|
||||
/**
|
||||
* Added to the coordinator's chat prompt: it hands work to the crew.
|
||||
*/
|
||||
export const CoordinatorReplyNote = `As the coordinator, hand work that fits a crew member's role to them with delegate_task. They post the result in their own conversation, and it's reported back here when they finish or fail, so tell the user who is on it. When a report comes back, pass on what matters to the user, follow up on anything left, and propose what comes next: the next piece of work, which bot would take it, or a question that settles the direction. Once the user agrees, you can also grow the crew with create_bot, put a worker in a workspace with add_bot_to_workspace, and create or change a workspace with create_workspace and update_workspace. When you name a crew member, write its ref instead of its name, e.g. ${botRef({ id: 3 })}: the user sees it as the bot's name, linked to its conversation.`;
|
||||
export const CoordinatorReplyNote = `As the coordinator, hand work that fits a crew member's role to them with delegate_task. They post the result in their own conversation, and it's reported back here when they finish or fail, so tell the user who is on it and where: a task with no workspace_id runs in that bot's own chat, on its own memory, files and directly assigned computers; one with workspace_id runs in the workspace, on the workspace's context, files and computers as well as the bot's, and lands in that project's conversation. Never swap one for the other without saying so. When a report comes back, pass on what matters to the user, follow up on anything left, and propose what comes next: the next piece of work, which bot would take it, or a question that settles the direction. Once the user agrees, you can also grow the crew with create_bot, put a worker in a workspace with add_bot_to_workspace, and create or change a workspace with create_workspace and update_workspace. When you name a crew member, write its ref instead of its name, e.g. ${botRef({ id: 3 })}: the user sees it as the bot's name, linked to its conversation.`;
|
||||
|
||||
/**
|
||||
* Added to the coordinator's chat prompt instead while it has no workers,
|
||||
|
||||
@@ -7,6 +7,7 @@ export const AdminSections = [
|
||||
"/admin/workspaces",
|
||||
"/admin/routines",
|
||||
"/admin/files",
|
||||
"/admin/usage",
|
||||
"/admin/logs",
|
||||
"/admin/help",
|
||||
"/admin/settings",
|
||||
|
||||
@@ -1,4 +1,6 @@
|
||||
import { Activity, CircleHelp, Clock, FolderOpen, type LucideIcon } from "lucide-react";
|
||||
import { Activity, BarChart3, CircleHelp, Clock, FolderOpen, type LucideIcon } from "lucide-react";
|
||||
import type { BUN_SQLITE_COBBLE_USER_TYPES } from "@/db/types/db";
|
||||
import canManageCrew from "@/src/utils/rbac/can-manage-crew";
|
||||
import { HelpPath } from "./help-dict";
|
||||
|
||||
export type RailPage = {
|
||||
@@ -9,11 +11,16 @@ export type RailPage = {
|
||||
* Shown and served to super admins only.
|
||||
*/
|
||||
super_admin_only?: boolean;
|
||||
/**
|
||||
* Who sees it, when that's an RBAC rule rather than super admins only.
|
||||
*/
|
||||
can_view?: (params: { user_types?: Pick<BUN_SQLITE_COBBLE_USER_TYPES, "user_type">[] | null }) => boolean;
|
||||
};
|
||||
|
||||
export const RoutinesPath = "/admin/routines";
|
||||
export const FilesPath = "/admin/files";
|
||||
export const ActivityPath = "/admin/logs";
|
||||
export const UsagePath = "/admin/usage";
|
||||
|
||||
/**
|
||||
* Pages opened from the rail's user popover that have their own rail: a
|
||||
@@ -22,6 +29,7 @@ export const ActivityPath = "/admin/logs";
|
||||
export const RailPages: RailPage[] = [
|
||||
{ title: "Routines", url: RoutinesPath, Icon: Clock },
|
||||
{ title: "Files", url: FilesPath, Icon: FolderOpen },
|
||||
{ title: "Usage", url: UsagePath, Icon: BarChart3, can_view: canManageCrew },
|
||||
{ title: "Activity", url: ActivityPath, Icon: Activity, super_admin_only: true },
|
||||
{ title: "Help", url: HelpPath, Icon: CircleHelp },
|
||||
];
|
||||
|
||||
@@ -7,6 +7,10 @@ import type {
|
||||
import { AppData } from "@/src/data/app-data";
|
||||
import embedText from "@/src/harness/embeddings/embed-text";
|
||||
import HarnessError from "@/src/harness/harness-error";
|
||||
import { grabHarnessUsageScope } from "@/src/harness/usage/harness-usage-scope";
|
||||
import type { HarnessUsageScope } from "@/src/harness/usage/types";
|
||||
import runWithFallbackScope from "@/src/harness/usage/run-with-fallback-scope";
|
||||
import grabUsageOwnerUserId from "@/src/functions/backend/db/usage/grab-usage-owner";
|
||||
import chunkText from "./chunk-text";
|
||||
import extractBotFileText from "./extract-bot-file-text";
|
||||
|
||||
@@ -72,13 +76,27 @@ export default async function ingestBotFile({
|
||||
return { status: "skipped", chunks: 0 };
|
||||
}
|
||||
|
||||
/**
|
||||
* A bot's own run already carries its user. The ingest queue has no run
|
||||
* behind it, so everything this file costs - reading an image or an
|
||||
* audio file, embedding its chunks - is entered under the owner's scope
|
||||
* instead: it spends real tokens and must show on the usage page.
|
||||
*/
|
||||
const owner_scope: HarnessUsageScope | null = grabHarnessUsageScope()?.user_id
|
||||
? null
|
||||
: await resolveOwnerScope({ file });
|
||||
|
||||
try {
|
||||
const text = await extractBotFileText({
|
||||
write_path: media?.media_write_path,
|
||||
text_content: media?.text_content,
|
||||
name: file.name,
|
||||
mime_type: file.mime_type,
|
||||
read_media,
|
||||
const text = await runWithFallbackScope({
|
||||
scope: owner_scope,
|
||||
task: () =>
|
||||
extractBotFileText({
|
||||
write_path: media?.media_write_path,
|
||||
text_content: media?.text_content,
|
||||
name: file.name,
|
||||
mime_type: file.mime_type,
|
||||
read_media,
|
||||
}),
|
||||
});
|
||||
|
||||
await clearChunks({ file_id: file.id });
|
||||
@@ -107,11 +125,11 @@ export default async function ingestBotFile({
|
||||
*/
|
||||
for (let start = 0; start < chunks.length; start += AppData.EmbeddingBatchSize) {
|
||||
const batch = chunks.slice(start, start + AppData.EmbeddingBatchSize);
|
||||
const { vectors } = await embedText({
|
||||
input: batch,
|
||||
bot_id: file.bot_id!,
|
||||
workspace_id: file.scope == "workspace" ? file.owner_id : null,
|
||||
const { vectors } = await runEmbed({
|
||||
scope: owner_scope,
|
||||
signal,
|
||||
file,
|
||||
batch,
|
||||
});
|
||||
|
||||
for (let i = 0; i < vectors.length; i++) {
|
||||
@@ -252,3 +270,54 @@ async function markIngested({
|
||||
values: [chunks > 0 ? 1 : 0, now, file_id],
|
||||
}).catch(() => {});
|
||||
}
|
||||
|
||||
/**
|
||||
* One batch of a file's chunks, embedded under whatever attribution owns
|
||||
* it: the run that asked for the ingest, else the file's owner.
|
||||
*/
|
||||
async function runEmbed({
|
||||
scope,
|
||||
signal,
|
||||
file,
|
||||
batch,
|
||||
}: {
|
||||
scope: HarnessUsageScope | null;
|
||||
signal?: AbortSignal;
|
||||
file: BUN_SQLITE_COBBLE_BOT_FILES;
|
||||
batch: string[];
|
||||
}) {
|
||||
return await runWithFallbackScope({
|
||||
scope,
|
||||
task: () =>
|
||||
embedText({
|
||||
input: batch,
|
||||
bot_id: file.bot_id!,
|
||||
workspace_id: file.scope == "workspace" ? file.owner_id : null,
|
||||
signal,
|
||||
}),
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* The attribution a file the queue ingests reports under: its workspace's
|
||||
* creator when it shares one, else its bot's.
|
||||
*/
|
||||
async function resolveOwnerScope({
|
||||
file,
|
||||
}: {
|
||||
file: BUN_SQLITE_COBBLE_BOT_FILES;
|
||||
}): Promise<HarnessUsageScope | null> {
|
||||
const workspace_id = file.scope == "workspace" ? file.owner_id : null;
|
||||
const user_id = await grabUsageOwnerUserId({ bot_id: file.bot_id, workspace_id });
|
||||
|
||||
if (!user_id) return null;
|
||||
|
||||
return {
|
||||
user_id,
|
||||
bot_id: file.scope == "bot" ? file.owner_id : file.bot_id,
|
||||
workspace_id,
|
||||
session_id: null,
|
||||
message_id: null,
|
||||
job_id: null,
|
||||
};
|
||||
}
|
||||
|
||||
@@ -89,6 +89,18 @@ export default async function completeBotJob({
|
||||
media_ids: attachments,
|
||||
});
|
||||
|
||||
if (job.id && message?.id) {
|
||||
/**
|
||||
* The job's model calls were written down as they happened, before
|
||||
* its message existed. Now that it does, they belong to it, so the
|
||||
* thread can show what the answer cost.
|
||||
*/
|
||||
await BunSQLite.sql({
|
||||
sql: `UPDATE token_usages SET message_id = ? WHERE job_id = ? AND message_id IS NULL`,
|
||||
values: [message.id, job.id],
|
||||
}).catch(() => {});
|
||||
}
|
||||
|
||||
const user_id = await grabJobUserId({ job });
|
||||
|
||||
const bot = !user_id
|
||||
|
||||
@@ -1,6 +1,7 @@
|
||||
import type { BUN_SQLITE_COBBLE_BOT_MESSAGES } from "@/db/types/db";
|
||||
import type { BotMessageWithAttachments } from "@/src/types";
|
||||
import BunSQLite from "@moduletrace/bun-sqlite";
|
||||
import grabMessageUsage from "../usage/grab-message-usage";
|
||||
import grabMessagesMedia from "./grab-messages-media";
|
||||
import grabMessagesReactions from "./grab-messages-reactions";
|
||||
import grabMessagesReplies from "./grab-messages-replies";
|
||||
@@ -53,12 +54,13 @@ export default async function grabBotMessages({
|
||||
const messages = res.payload || [];
|
||||
const message_ids = messages.map((message) => Number(message.id));
|
||||
|
||||
const [media, replies, reactions] = await Promise.all([
|
||||
const [media, replies, reactions, usage] = await Promise.all([
|
||||
grabMessagesMedia({ message_ids }),
|
||||
grabMessagesReplies({ message_ids }),
|
||||
user_id
|
||||
? grabMessagesReactions({ message_ids, user_id })
|
||||
: Promise.resolve(new Map()),
|
||||
grabMessageUsage({ message_ids }),
|
||||
]);
|
||||
|
||||
/**
|
||||
@@ -74,6 +76,8 @@ export default async function grabBotMessages({
|
||||
const files = media.get(id);
|
||||
const reply_to = replies.get(id);
|
||||
|
||||
const spent = usage.get(id);
|
||||
|
||||
return {
|
||||
...message,
|
||||
...(files
|
||||
@@ -88,6 +92,7 @@ export default async function grabBotMessages({
|
||||
: {}),
|
||||
...(reply_to ? { reply_to } : {}),
|
||||
...(user_id ? { reactions: reactions.get(id) || [] } : {}),
|
||||
...(spent ? { usage: spent } : {}),
|
||||
};
|
||||
});
|
||||
}
|
||||
|
||||
@@ -0,0 +1,101 @@
|
||||
import { describe, expect, test } from "bun:test";
|
||||
import bucketUsageDays, { localDate } from "./bucket-usage-days";
|
||||
import { DayMs, startOfDay } from "./grab-usage-window";
|
||||
|
||||
function row(local_day: string, tokens: number) {
|
||||
return {
|
||||
local_day,
|
||||
calls: 1,
|
||||
prompt_tokens: tokens,
|
||||
completion_tokens: 0,
|
||||
cost: null,
|
||||
};
|
||||
}
|
||||
|
||||
const now = new Date(2026, 9, 23, 15, 0, 0).getTime();
|
||||
const end = now;
|
||||
const start = startOfDay({ at: now }) - 2 * DayMs;
|
||||
|
||||
describe("localDate", () => {
|
||||
test("names the local calendar day, not the UTC one", () => {
|
||||
/**
|
||||
* Half an hour past local midnight is still the previous UTC day
|
||||
* for anyone east of Greenwich, and the bug the chart had: the row
|
||||
* was counted under one name and looked for under another.
|
||||
*/
|
||||
const at = new Date(2026, 9, 23, 0, 30, 0).getTime();
|
||||
|
||||
expect(localDate({ at })).toBe("2026-10-23");
|
||||
});
|
||||
|
||||
test("pads a single-digit month and day", () => {
|
||||
expect(localDate({ at: new Date(2026, 0, 5, 12, 0, 0).getTime() })).toBe("2026-01-05");
|
||||
});
|
||||
});
|
||||
|
||||
describe("bucketUsageDays", () => {
|
||||
test("one bucket per day of the window, oldest first", () => {
|
||||
const days = bucketUsageDays({ rows: [], start, end });
|
||||
|
||||
expect(days.length).toBe(3);
|
||||
expect(days.map((day) => day.label)).toEqual(
|
||||
[2, 1, 0].map((back) => localDate({ at: startOfDay({ at: now }) - back * DayMs })),
|
||||
);
|
||||
});
|
||||
|
||||
test("every row lands in its day, and yesterday's late call does not vanish", () => {
|
||||
const today = localDate({ at: startOfDay({ at: now }) });
|
||||
const yesterday = localDate({ at: startOfDay({ at: now }) - DayMs });
|
||||
|
||||
const days = bucketUsageDays({
|
||||
rows: [row(today, 1500), row(yesterday, 400)],
|
||||
start,
|
||||
end,
|
||||
});
|
||||
|
||||
const by_label = new Map(days.map((day) => [day.label, day.total_tokens]));
|
||||
|
||||
expect(by_label.get(today)).toBe(1500);
|
||||
expect(by_label.get(yesterday)).toBe(400);
|
||||
expect(days.reduce((total, day) => total + day.total_tokens, 0)).toBe(1900);
|
||||
});
|
||||
|
||||
test("a day with nothing in it is still a bucket, at zero", () => {
|
||||
const days = bucketUsageDays({
|
||||
rows: [row(localDate({ at: startOfDay({ at: now }) }), 100)],
|
||||
start,
|
||||
end,
|
||||
});
|
||||
|
||||
expect(days.filter((day) => day.total_tokens === 0).length).toBe(2);
|
||||
expect(days.every((day) => day.calls === 0 || day.calls === 1)).toBe(true);
|
||||
});
|
||||
|
||||
test("a bucket's midnight is that day's local midnight", () => {
|
||||
const days = bucketUsageDays({ rows: [], start, end });
|
||||
|
||||
for (const day of days) {
|
||||
expect(day.at).toBe(startOfDay({ at: day.at }));
|
||||
expect(localDate({ at: day.at })).toBe(day.label);
|
||||
}
|
||||
});
|
||||
|
||||
test("cost and calls are summed, not counted twice", () => {
|
||||
const label = localDate({ at: startOfDay({ at: now }) });
|
||||
|
||||
const days = bucketUsageDays({
|
||||
rows: [
|
||||
{ ...row(label, 100), calls: 3, cost: 0.25 },
|
||||
{ ...row(label, 50), calls: 2, cost: 0.5 },
|
||||
],
|
||||
start,
|
||||
end,
|
||||
});
|
||||
|
||||
const bucket = days.find((day) => day.label === label);
|
||||
|
||||
expect(bucket?.total_tokens).toBe(150);
|
||||
expect(bucket?.cost).toBe(0.75);
|
||||
expect(days.reduce((total, day) => total + day.calls, 0)).toBe(5);
|
||||
});
|
||||
});
|
||||
@@ -0,0 +1,115 @@
|
||||
import { DayMs, startOfDay } from "./grab-usage-window";
|
||||
import { toSummary } from "./grab-usage-totals";
|
||||
import type { TokenUsageSummary } from "@/src/harness/usage/types";
|
||||
|
||||
type Row = {
|
||||
/**
|
||||
* The day the call was made, `YYYY-MM-DD`, already in local time.
|
||||
*/
|
||||
local_day: string;
|
||||
calls: number | null;
|
||||
prompt_tokens: number | null;
|
||||
completion_tokens: number | null;
|
||||
cost: number | null;
|
||||
};
|
||||
|
||||
type Params = {
|
||||
rows: Row[];
|
||||
start: number;
|
||||
end: number;
|
||||
};
|
||||
|
||||
export type UsageDay = {
|
||||
/**
|
||||
* The day's local midnight (epoch ms).
|
||||
*/
|
||||
at: number;
|
||||
/**
|
||||
* `YYYY-MM-DD`, in local time.
|
||||
*/
|
||||
label: string;
|
||||
} & TokenUsageSummary;
|
||||
|
||||
/**
|
||||
* The rows SQL grouped by local day, laid onto one bucket per day of the
|
||||
* window, oldest first. Days with nothing in them are still there, at
|
||||
* zero, so the bars keep their spacing, and every row lands in a bucket:
|
||||
* a day outside the window would be a bug in the query's own bounds.
|
||||
*/
|
||||
export default function bucketUsageDays({ rows, start, end }: Params): UsageDay[] {
|
||||
const by_day = new Map<string, TokenUsageSummary>();
|
||||
|
||||
for (const row of rows) {
|
||||
/**
|
||||
* Accumulated, not replaced: the query groups by day, so a repeated
|
||||
* day means someone asked for two sets of rows, and the later one
|
||||
* must add to the earlier rather than hide it.
|
||||
*/
|
||||
const label = String(row.local_day);
|
||||
const summary = toSummary({ row });
|
||||
|
||||
by_day.set(label, mergeSummaries({ into: by_day.get(label), summary }));
|
||||
}
|
||||
|
||||
const days: UsageDay[] = [];
|
||||
const last = startOfDay({ at: end });
|
||||
|
||||
for (let at = startOfDay({ at: start }); at <= last; at += DayMs) {
|
||||
const label = localDate({ at });
|
||||
|
||||
days.push({
|
||||
at,
|
||||
label,
|
||||
...(by_day.get(label) || emptySummary()),
|
||||
});
|
||||
}
|
||||
|
||||
return days;
|
||||
}
|
||||
|
||||
/**
|
||||
* The day a local midnight falls on, in local time.
|
||||
*
|
||||
* Never `toISOString`: an epoch millisecond read in UTC is a different
|
||||
* calendar day for anyone east or west of Greenwich, and a call counted
|
||||
* under one day would then look for a bucket named after another and
|
||||
* disappear from the chart (which is exactly what it did before).
|
||||
*/
|
||||
export function localDate({ at }: { at: number }) {
|
||||
const date = new Date(at);
|
||||
|
||||
return [
|
||||
date.getFullYear(),
|
||||
String(date.getMonth() + 1).padStart(2, "0"),
|
||||
String(date.getDate()).padStart(2, "0"),
|
||||
].join("-");
|
||||
}
|
||||
|
||||
function emptySummary(): TokenUsageSummary {
|
||||
return { prompt_tokens: 0, completion_tokens: 0, total_tokens: 0, calls: 0 };
|
||||
}
|
||||
|
||||
/**
|
||||
* One summary added onto another. A day that reported no cost still costs
|
||||
* nothing; only a real cost is carried, so an unpriced day doesn't turn a
|
||||
* priced one into zero.
|
||||
*/
|
||||
function mergeSummaries({
|
||||
into,
|
||||
summary,
|
||||
}: {
|
||||
into?: TokenUsageSummary;
|
||||
summary: TokenUsageSummary;
|
||||
}): TokenUsageSummary {
|
||||
const prompt_tokens = (into?.prompt_tokens || 0) + summary.prompt_tokens;
|
||||
const completion_tokens = (into?.completion_tokens || 0) + summary.completion_tokens;
|
||||
const cost = (into?.cost || 0) + (summary.cost || 0);
|
||||
|
||||
return {
|
||||
prompt_tokens,
|
||||
completion_tokens,
|
||||
total_tokens: prompt_tokens + completion_tokens,
|
||||
...(cost > 0 ? { cost } : {}),
|
||||
calls: (into?.calls || 0) + summary.calls,
|
||||
};
|
||||
}
|
||||
@@ -0,0 +1,56 @@
|
||||
import BunSQLite from "@moduletrace/bun-sqlite";
|
||||
import type { TokenUsageSummary } from "@/src/harness/usage/types";
|
||||
|
||||
type Params = {
|
||||
/**
|
||||
* The messages whose calls are summed. A message with no calls is
|
||||
* absent from the answer.
|
||||
*/
|
||||
message_ids: number[];
|
||||
};
|
||||
|
||||
/**
|
||||
* What each of these messages cost, summed over every model call it took
|
||||
* (chat turns, media readings, image generations). This is what a
|
||||
* message's token line shows.
|
||||
*/
|
||||
export default async function grabMessageUsage({ message_ids }: Params) {
|
||||
const totals = new Map<number, TokenUsageSummary>();
|
||||
const ids = [...new Set(message_ids.map(Number).filter((id) => id > 0))];
|
||||
|
||||
if (!ids.length) return totals;
|
||||
|
||||
const res = await BunSQLite.sql<{
|
||||
message_id: number;
|
||||
prompt_tokens: number | null;
|
||||
completion_tokens: number | null;
|
||||
cost: number | null;
|
||||
calls: number;
|
||||
}>({
|
||||
sql: `SELECT message_id,
|
||||
SUM(prompt_tokens) AS prompt_tokens,
|
||||
SUM(completion_tokens) AS completion_tokens,
|
||||
SUM(cost) AS cost,
|
||||
COUNT(*) AS calls
|
||||
FROM token_usages
|
||||
WHERE message_id IN (${ids.map(() => "?").join(", ")})
|
||||
GROUP BY message_id`,
|
||||
values: ids,
|
||||
});
|
||||
|
||||
for (const row of res.payload || []) {
|
||||
const prompt_tokens = Number(row.prompt_tokens) || 0;
|
||||
const completion_tokens = Number(row.completion_tokens) || 0;
|
||||
const cost = Number(row.cost);
|
||||
|
||||
totals.set(Number(row.message_id), {
|
||||
prompt_tokens,
|
||||
completion_tokens,
|
||||
total_tokens: prompt_tokens + completion_tokens,
|
||||
...(Number.isFinite(cost) && cost > 0 ? { cost } : {}),
|
||||
calls: Number(row.calls) || 0,
|
||||
});
|
||||
}
|
||||
|
||||
return totals;
|
||||
}
|
||||
@@ -0,0 +1,42 @@
|
||||
import BunSQLite from "@moduletrace/bun-sqlite";
|
||||
import bucketUsageDays, { type UsageDay } from "./bucket-usage-days";
|
||||
|
||||
export type { UsageDay };
|
||||
|
||||
type Params = {
|
||||
start: number;
|
||||
end: number;
|
||||
};
|
||||
|
||||
/**
|
||||
* One bucket per day in the window, oldest first, each the tokens spent
|
||||
* that day. Days with nothing in them are still there, at zero, so the
|
||||
* bars keep their spacing.
|
||||
*
|
||||
* SQLite has no timezone support, so the day is taken with its `localtime`
|
||||
* modifier, which reads the same server timezone the page's own day
|
||||
* boundaries come from. Grouping by the bare UTC day instead put a late
|
||||
* evening's calls in the previous day's bucket, and dropped them whenever
|
||||
* no bucket carried that name.
|
||||
*/
|
||||
export default async function grabUsageDays({ start, end }: Params): Promise<UsageDay[]> {
|
||||
const res = await BunSQLite.sql<{
|
||||
local_day: string;
|
||||
calls: number | null;
|
||||
prompt_tokens: number | null;
|
||||
completion_tokens: number | null;
|
||||
cost: number | null;
|
||||
}>({
|
||||
sql: `SELECT date(created_at / 1000, 'unixepoch', 'localtime') AS local_day,
|
||||
COUNT(*) AS calls,
|
||||
SUM(prompt_tokens) AS prompt_tokens,
|
||||
SUM(completion_tokens) AS completion_tokens,
|
||||
SUM(cost) AS cost
|
||||
FROM token_usages
|
||||
WHERE created_at >= ? AND created_at < ?
|
||||
GROUP BY local_day`,
|
||||
values: [start, end],
|
||||
});
|
||||
|
||||
return bucketUsageDays({ rows: res.payload || [], start, end });
|
||||
}
|
||||
@@ -0,0 +1,75 @@
|
||||
import BunSQLite from "@moduletrace/bun-sqlite";
|
||||
import { toSummary } from "./grab-usage-totals";
|
||||
import type { TokenUsageSummary } from "@/src/harness/usage/types";
|
||||
|
||||
type Params = {
|
||||
start: number;
|
||||
end: number;
|
||||
/**
|
||||
* How many the page shows, busiest first.
|
||||
*/
|
||||
limit: number;
|
||||
};
|
||||
|
||||
export type UsageModelRow = {
|
||||
/**
|
||||
* The provider's own id for the model, as the call recorded it
|
||||
* (`deepseek/deepseek-chat`). This is what a person compares, not the
|
||||
* catalog row's numeric id.
|
||||
*/
|
||||
model_slug: string;
|
||||
/**
|
||||
* The catalog's display name, when the model is still listed. A model
|
||||
* a provider stopped listing keeps its slug and loses its name.
|
||||
*/
|
||||
model_name: string | null;
|
||||
provider_name: string;
|
||||
provider_id: number | null;
|
||||
harness: string;
|
||||
} & TokenUsageSummary;
|
||||
|
||||
/**
|
||||
* Every model the period ran, busiest first (by tokens), with what it
|
||||
* spent. This is how one works out what a model costs: its tokens against
|
||||
* its calls, for the window asked for.
|
||||
*/
|
||||
export default async function grabUsageModels({ start, end, limit }: Params): Promise<UsageModelRow[]> {
|
||||
const res = await BunSQLite.sql<{
|
||||
model_slug: string | null;
|
||||
model_name: string | null;
|
||||
provider_name: string | null;
|
||||
provider_id: number | null;
|
||||
harness: string | null;
|
||||
calls: number | null;
|
||||
prompt_tokens: number | null;
|
||||
completion_tokens: number | null;
|
||||
cost: number | null;
|
||||
}>({
|
||||
sql: `SELECT u.model_slug,
|
||||
m.name AS model_name,
|
||||
u.provider_name,
|
||||
u.provider_id,
|
||||
u.harness,
|
||||
COUNT(*) AS calls,
|
||||
SUM(u.prompt_tokens) AS prompt_tokens,
|
||||
SUM(u.completion_tokens) AS completion_tokens,
|
||||
SUM(u.cost) AS cost
|
||||
FROM token_usages u
|
||||
LEFT JOIN harness_models m
|
||||
ON m.provider_id = u.provider_id AND m.slug = u.model_slug
|
||||
WHERE u.created_at >= ? AND u.created_at < ?
|
||||
GROUP BY u.provider_name, u.model_slug, u.harness
|
||||
ORDER BY SUM(u.prompt_tokens) + SUM(u.completion_tokens) DESC
|
||||
LIMIT ?`,
|
||||
values: [start, end, limit],
|
||||
});
|
||||
|
||||
return (res.payload || []).map((row) => ({
|
||||
model_slug: row.model_slug || "unknown",
|
||||
model_name: row.model_name || null,
|
||||
provider_name: row.provider_name || "unknown",
|
||||
provider_id: row.provider_id == null ? null : Number(row.provider_id),
|
||||
harness: String(row.harness || "text"),
|
||||
...toSummary({ row }),
|
||||
}));
|
||||
}
|
||||
@@ -0,0 +1,53 @@
|
||||
import { AppData } from "@/src/data/app-data";
|
||||
import type { UsageOverview } from "@/src/types";
|
||||
import grabUsageDays from "./grab-usage-days";
|
||||
import grabUsageModels from "./grab-usage-models";
|
||||
import grabUsageTotals from "./grab-usage-totals";
|
||||
import grabUsageWindow, { DayMs } from "./grab-usage-window";
|
||||
|
||||
type Params = {
|
||||
/**
|
||||
* How many days back to total, today included.
|
||||
*/
|
||||
days?: number | string | null;
|
||||
now?: number;
|
||||
};
|
||||
|
||||
/**
|
||||
* Everything the usage page shows for one window: the totals, the same day
|
||||
* by day, per harness and per model, plus the day before the window and a
|
||||
* rolling week, so the page can say whether work is going up or down.
|
||||
*/
|
||||
export default async function grabUsageOverview({
|
||||
days,
|
||||
now = Date.now(),
|
||||
}: Params): Promise<UsageOverview> {
|
||||
const window = grabUsageWindow({ days, now });
|
||||
const previous_start = window.start - window.days * DayMs;
|
||||
|
||||
const [totals, days_by_day, models, previous, week] = await Promise.all([
|
||||
grabUsageTotals({ start: window.start, end: window.end }),
|
||||
grabUsageDays({ start: window.start, end: window.end }),
|
||||
grabUsageModels({
|
||||
start: window.start,
|
||||
end: window.end,
|
||||
limit: AppData.UsageModelPageSize,
|
||||
}),
|
||||
grabUsageTotals({ start: previous_start, end: window.start }),
|
||||
grabUsageTotals({ start: window.end - 7 * DayMs, end: window.end }),
|
||||
]);
|
||||
|
||||
return {
|
||||
window: {
|
||||
days: window.days,
|
||||
start: window.start,
|
||||
end: window.end,
|
||||
},
|
||||
totals: totals.totals,
|
||||
by_harness: totals.by_harness,
|
||||
days: days_by_day,
|
||||
models,
|
||||
previous: previous.totals,
|
||||
week: week.totals,
|
||||
};
|
||||
}
|
||||
@@ -0,0 +1,47 @@
|
||||
import BunSQLite from "@moduletrace/bun-sqlite";
|
||||
|
||||
type Params = {
|
||||
/**
|
||||
* The bot whose file it is, or the bot the search ran for.
|
||||
*/
|
||||
bot_id?: number | string | null;
|
||||
/**
|
||||
* The workspace whose file it is, when it shares one.
|
||||
*/
|
||||
workspace_id?: number | string | null;
|
||||
};
|
||||
|
||||
/**
|
||||
* Whose usage a call made outside anyone's run belongs to: a file ingested
|
||||
* by the queue, or a search that ran with no conversation behind it. The
|
||||
* workspace's creator when there is a workspace, else the bot's, so an
|
||||
* embedding never goes uncounted.
|
||||
*/
|
||||
export default async function grabUsageOwnerUserId({
|
||||
bot_id,
|
||||
workspace_id,
|
||||
}: Params): Promise<number | null> {
|
||||
const workspace = Number(workspace_id);
|
||||
|
||||
if (Number.isFinite(workspace) && workspace > 0) {
|
||||
const row = await BunSQLite.sql<{ user_id: number }>({
|
||||
sql: `SELECT user_id FROM workspaces WHERE id = ? LIMIT 1`,
|
||||
values: [workspace],
|
||||
}).then((res) => res.payload?.[0]);
|
||||
|
||||
if (row?.user_id) return Number(row.user_id) || null;
|
||||
}
|
||||
|
||||
const bot = Number(bot_id);
|
||||
|
||||
if (Number.isFinite(bot) && bot > 0) {
|
||||
const row = await BunSQLite.sql<{ user_id: number }>({
|
||||
sql: `SELECT user_id FROM bots WHERE id = ? LIMIT 1`,
|
||||
values: [bot],
|
||||
}).then((res) => res.payload?.[0]);
|
||||
|
||||
if (row?.user_id) return Number(row.user_id) || null;
|
||||
}
|
||||
|
||||
return null;
|
||||
}
|
||||
@@ -0,0 +1,97 @@
|
||||
import BunSQLite from "@moduletrace/bun-sqlite";
|
||||
import type { TokenUsageSummary } from "@/src/harness/usage/types";
|
||||
|
||||
type Params = {
|
||||
/**
|
||||
* The window's start and end (epoch ms), from `grabUsageWindow`.
|
||||
*/
|
||||
start: number;
|
||||
end: number;
|
||||
};
|
||||
|
||||
type UsageRow = {
|
||||
calls: number | null;
|
||||
prompt_tokens: number | null;
|
||||
completion_tokens: number | null;
|
||||
cost: number | null;
|
||||
};
|
||||
|
||||
/**
|
||||
* One total for a whole period, and the same split by the harnesses the
|
||||
* calls were for.
|
||||
*/
|
||||
export type UsageTotals = {
|
||||
totals: TokenUsageSummary;
|
||||
by_harness: (TokenUsageSummary & { harness: string })[];
|
||||
};
|
||||
|
||||
/**
|
||||
* What every model call in the period cost, in total and per harness
|
||||
* (chat, media reading, transcribing, speaking, images, embeddings).
|
||||
*/
|
||||
export default async function grabUsageTotals({ start, end }: Params): Promise<UsageTotals> {
|
||||
const res = await BunSQLite.sql<UsageRow & { harness: string | null }>({
|
||||
sql: `SELECT harness,
|
||||
COUNT(*) AS calls,
|
||||
SUM(prompt_tokens) AS prompt_tokens,
|
||||
SUM(completion_tokens) AS completion_tokens,
|
||||
SUM(cost) AS cost
|
||||
FROM token_usages
|
||||
WHERE created_at >= ? AND created_at < ?
|
||||
GROUP BY harness`,
|
||||
values: [start, end],
|
||||
});
|
||||
|
||||
const by_harness = (res.payload || []).map((row) => ({
|
||||
harness: String(row.harness || "text"),
|
||||
...toSummary({ row }),
|
||||
}));
|
||||
|
||||
return {
|
||||
totals: sumSummary({ summaries: by_harness }),
|
||||
by_harness: by_harness.sort((a, b) => b.total_tokens - a.total_tokens),
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* Adds summaries into one. A call that reported no tokens still counts as
|
||||
* a call.
|
||||
*/
|
||||
export function sumSummary({ summaries }: { summaries: TokenUsageSummary[] }): TokenUsageSummary {
|
||||
let prompt_tokens = 0;
|
||||
let completion_tokens = 0;
|
||||
let cost = 0;
|
||||
let calls = 0;
|
||||
|
||||
for (const summary of summaries) {
|
||||
prompt_tokens += summary.prompt_tokens;
|
||||
completion_tokens += summary.completion_tokens;
|
||||
cost += summary.cost || 0;
|
||||
calls += summary.calls;
|
||||
}
|
||||
|
||||
return {
|
||||
prompt_tokens,
|
||||
completion_tokens,
|
||||
total_tokens: prompt_tokens + completion_tokens,
|
||||
...(cost > 0 ? { cost } : {}),
|
||||
calls,
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* One grouped row (any shape with the four sums) as a summary.
|
||||
*/
|
||||
export function toSummary({ row }: { row: UsageRow }): TokenUsageSummary {
|
||||
const prompt_tokens = Number(row.prompt_tokens) || 0;
|
||||
const completion_tokens = Number(row.completion_tokens) || 0;
|
||||
const cost = Number(row.cost);
|
||||
|
||||
return {
|
||||
prompt_tokens,
|
||||
completion_tokens,
|
||||
total_tokens: prompt_tokens + completion_tokens,
|
||||
...(Number.isFinite(cost) && cost > 0 ? { cost } : {}),
|
||||
calls: Number(row.calls) || 0,
|
||||
};
|
||||
}
|
||||
@@ -0,0 +1,57 @@
|
||||
import { AppData } from "@/src/data/app-data";
|
||||
|
||||
type Params = {
|
||||
/**
|
||||
* How many days back the window starts, today included. Clamped to
|
||||
* what the page offers.
|
||||
*/
|
||||
days?: number | string | null;
|
||||
/**
|
||||
* The moment "now" is, so a whole page (its totals and its days) is
|
||||
* read at one instant.
|
||||
*/
|
||||
now?: number;
|
||||
};
|
||||
|
||||
/**
|
||||
* The period the usage page totals: a start, an end and the day buckets
|
||||
* between them, oldest first, each a local midnight.
|
||||
*/
|
||||
export default function grabUsageWindow({ days, now = Date.now() }: Params) {
|
||||
const asked = Number(days);
|
||||
const count =
|
||||
Number.isFinite(asked) && asked > 0
|
||||
? Math.min(Math.round(asked), AppData.UsageMaxDays)
|
||||
: AppData.UsageDefaultDays;
|
||||
|
||||
const end = now;
|
||||
/**
|
||||
* Whole days back from the start of today, so every bucket is a day's
|
||||
* work rather than a sliding window's slice.
|
||||
*/
|
||||
const today = startOfDay({ at: now });
|
||||
const start = today - (count - 1) * DayMs;
|
||||
|
||||
return {
|
||||
days: count,
|
||||
start,
|
||||
end,
|
||||
/**
|
||||
* One bucket per day in the window, oldest first.
|
||||
*/
|
||||
buckets: Array.from({ length: count }, (_, index) => start + index * DayMs),
|
||||
};
|
||||
}
|
||||
|
||||
export const DayMs = 24 * 60 * 60 * 1000;
|
||||
|
||||
/**
|
||||
* Local midnight of the day the moment falls in.
|
||||
*/
|
||||
export function startOfDay({ at }: { at: number }) {
|
||||
const date = new Date(at);
|
||||
|
||||
date.setHours(0, 0, 0, 0);
|
||||
|
||||
return date.getTime();
|
||||
}
|
||||
@@ -2,7 +2,9 @@ import grabHarnessSetting from "@/src/functions/backend/db/harness/grab-harness-
|
||||
import grabHarnessAdapter from "../providers/grab-harness-adapter";
|
||||
import resolveHarnessCandidates from "../resolve/resolve-harness-candidates";
|
||||
import withCandidates from "../resolve/with-candidates";
|
||||
import type { HarnessCandidate } from "../types";
|
||||
import type { HarnessAudioStream, HarnessCandidate } from "../types";
|
||||
import buildUsageCall from "../usage/build-usage-call";
|
||||
import recordHarnessUsage from "../usage/record-harness-usage";
|
||||
|
||||
type Params = {
|
||||
text: string;
|
||||
@@ -39,9 +41,22 @@ export default async function speakText({
|
||||
grabHarnessSetting({ harness: "speech", bot_id, workspace_id }),
|
||||
]);
|
||||
|
||||
const { result } = await withCandidates({
|
||||
const { result } = await withCandidates<HarnessAudioStream>({
|
||||
harness: "speech",
|
||||
candidates,
|
||||
/**
|
||||
* Speech is billed per character and streamed, so there is no
|
||||
* completion count to read: the call is recorded as it is, with
|
||||
* the text's length standing in for what was spoken.
|
||||
*/
|
||||
on_success: async ({ candidate }) =>
|
||||
await recordHarnessUsage({
|
||||
call: buildUsageCall({
|
||||
harness: "speech",
|
||||
candidate,
|
||||
usage: { prompt_tokens: text.length, completion_tokens: 0 },
|
||||
}),
|
||||
}),
|
||||
run: async (candidate) =>
|
||||
await grabHarnessAdapter({ kind: candidate.provider.kind }).speak({
|
||||
candidate,
|
||||
|
||||
@@ -4,7 +4,9 @@ import HarnessError from "../harness-error";
|
||||
import grabHarnessAdapter from "../providers/grab-harness-adapter";
|
||||
import resolveHarnessCandidates from "../resolve/resolve-harness-candidates";
|
||||
import withCandidates from "../resolve/with-candidates";
|
||||
import type { HarnessAudioInput } from "../types";
|
||||
import type { HarnessAudioInput, HarnessUsage } from "../types";
|
||||
import buildUsageCall from "../usage/build-usage-call";
|
||||
import recordHarnessUsage from "../usage/record-harness-usage";
|
||||
|
||||
type Params = {
|
||||
audio: HarnessAudioInput;
|
||||
@@ -46,15 +48,23 @@ export default async function transcribeAudio({
|
||||
|
||||
const final_signal = signal || new AbortController().signal;
|
||||
|
||||
const { result } = await withCandidates({
|
||||
const { result } = await withCandidates<{ text: string; usage?: HarnessUsage }>({
|
||||
harness: "transcription",
|
||||
candidates,
|
||||
on_success: async ({ candidate, result: transcript }): Promise<void> =>
|
||||
await recordHarnessUsage({
|
||||
call: buildUsageCall({
|
||||
harness: "transcription",
|
||||
candidate,
|
||||
usage: transcript.usage,
|
||||
}),
|
||||
}),
|
||||
run: async (candidate) => {
|
||||
const adapter = grabHarnessAdapter({
|
||||
kind: candidate.provider.kind,
|
||||
});
|
||||
|
||||
const { text } = await adapter.transcribe({
|
||||
const { text, usage } = await adapter.transcribe({
|
||||
candidate,
|
||||
audio,
|
||||
language,
|
||||
@@ -69,9 +79,9 @@ export default async function transcribeAudio({
|
||||
});
|
||||
}
|
||||
|
||||
return text.trim();
|
||||
return { text: text.trim(), usage };
|
||||
},
|
||||
});
|
||||
|
||||
return result;
|
||||
return result.text;
|
||||
}
|
||||
|
||||
@@ -1,6 +1,9 @@
|
||||
import grabHarnessAdapter from "../providers/grab-harness-adapter";
|
||||
import resolveHarnessCandidates from "../resolve/resolve-harness-candidates";
|
||||
import withCandidates from "../resolve/with-candidates";
|
||||
import type { HarnessEmbedResult } from "../types";
|
||||
import buildUsageCall from "../usage/build-usage-call";
|
||||
import recordHarnessUsage from "../usage/record-harness-usage";
|
||||
import fitVectorDimensions from "./fit-vector-dimensions";
|
||||
|
||||
type Params = {
|
||||
@@ -52,9 +55,17 @@ export default async function embedText({
|
||||
workspace_id,
|
||||
});
|
||||
|
||||
const { result } = await withCandidates({
|
||||
const { result } = await withCandidates<HarnessEmbedResult>({
|
||||
harness: "embedding",
|
||||
candidates,
|
||||
on_success: async ({ candidate, result: embedded }): Promise<void> =>
|
||||
await recordHarnessUsage({
|
||||
call: buildUsageCall({
|
||||
harness: "embedding",
|
||||
candidate,
|
||||
usage: embedded.usage,
|
||||
}),
|
||||
}),
|
||||
run: async (candidate) =>
|
||||
await grabHarnessAdapter({ kind: candidate.provider.kind }).embed({
|
||||
candidate,
|
||||
|
||||
@@ -1,6 +1,9 @@
|
||||
import grabHarnessAdapter from "../providers/grab-harness-adapter";
|
||||
import resolveHarnessCandidates from "../resolve/resolve-harness-candidates";
|
||||
import withCandidates from "../resolve/with-candidates";
|
||||
import type { HarnessImageOutput, HarnessUsage } from "../types";
|
||||
import buildUsageCall from "../usage/build-usage-call";
|
||||
import recordHarnessUsage from "../usage/record-harness-usage";
|
||||
|
||||
type Params = {
|
||||
prompt: string;
|
||||
@@ -35,9 +38,17 @@ export default async function generateImage({
|
||||
}: Params) {
|
||||
const candidates = await resolveHarnessCandidates({ harness: "image", bot_id, workspace_id });
|
||||
|
||||
const { result } = await withCandidates({
|
||||
const { result } = await withCandidates<{ images: HarnessImageOutput[]; usage?: HarnessUsage }>({
|
||||
harness: "image",
|
||||
candidates,
|
||||
on_success: async ({ candidate, result: made }): Promise<void> =>
|
||||
await recordHarnessUsage({
|
||||
call: buildUsageCall({
|
||||
harness: "image",
|
||||
candidate,
|
||||
usage: made.usage,
|
||||
}),
|
||||
}),
|
||||
run: async (candidate) =>
|
||||
await grabHarnessAdapter({
|
||||
kind: candidate.provider.kind,
|
||||
|
||||
@@ -3,7 +3,10 @@ import HarnessError from "../harness-error";
|
||||
import grabHarnessAdapter from "../providers/grab-harness-adapter";
|
||||
import resolveHarnessCandidates from "../resolve/resolve-harness-candidates";
|
||||
import withCandidates from "../resolve/with-candidates";
|
||||
import type { HarnessUsage } from "../types";
|
||||
import grabChatParams from "../text/grab-chat-params";
|
||||
import buildUsageCall from "../usage/build-usage-call";
|
||||
import recordHarnessUsage from "../usage/record-harness-usage";
|
||||
|
||||
type Params = {
|
||||
/**
|
||||
@@ -42,9 +45,17 @@ export default async function readImage({ url, focus, signal, bot_id, workspace_
|
||||
workspace_id,
|
||||
});
|
||||
|
||||
const { result } = await withCandidates({
|
||||
const { result } = await withCandidates<{ text: string; usage?: HarnessUsage }>({
|
||||
harness: "vision",
|
||||
candidates,
|
||||
on_success: async ({ candidate, result: answer }): Promise<void> =>
|
||||
await recordHarnessUsage({
|
||||
call: buildUsageCall({
|
||||
harness: "vision",
|
||||
candidate,
|
||||
usage: answer.usage,
|
||||
}),
|
||||
}),
|
||||
run: async (candidate) => {
|
||||
const answer = await grabHarnessAdapter({
|
||||
kind: candidate.provider.kind,
|
||||
@@ -83,9 +94,9 @@ export default async function readImage({ url, focus, signal, bot_id, workspace_
|
||||
});
|
||||
}
|
||||
|
||||
return answer.text.trim();
|
||||
return { text: answer.text.trim(), usage: answer.usage };
|
||||
},
|
||||
});
|
||||
|
||||
return result;
|
||||
return result.text;
|
||||
}
|
||||
|
||||
@@ -13,6 +13,11 @@ const embed: HarnessAdapter["embed"] = async ({ candidate, input, signal }) => {
|
||||
const { provider, model } = candidate;
|
||||
const preset = grabProviderPreset({ kind: provider.kind });
|
||||
const vectors: number[][] = [];
|
||||
/**
|
||||
* What the provider said the batch cost. Most report it once per
|
||||
* request; the rows are opened before the loop so each batch counts.
|
||||
*/
|
||||
let prompt_tokens = 0;
|
||||
|
||||
/**
|
||||
* Every OpenAI-compatible API puts embeddings at `/embeddings`.
|
||||
@@ -47,6 +52,8 @@ const embed: HarnessAdapter["embed"] = async ({ candidate, input, signal }) => {
|
||||
});
|
||||
}
|
||||
|
||||
prompt_tokens += Number(json?.usage?.prompt_tokens) || 0;
|
||||
|
||||
for (const item of items) {
|
||||
const vector = Array.isArray(item) ? item : item?.embedding;
|
||||
|
||||
@@ -62,7 +69,12 @@ const embed: HarnessAdapter["embed"] = async ({ candidate, input, signal }) => {
|
||||
}
|
||||
}
|
||||
|
||||
return { vectors };
|
||||
return {
|
||||
vectors,
|
||||
...(prompt_tokens
|
||||
? { usage: { prompt_tokens, completion_tokens: 0 } }
|
||||
: {}),
|
||||
};
|
||||
};
|
||||
|
||||
export default embed;
|
||||
|
||||
@@ -81,7 +81,20 @@ const generateImage: HarnessAdapter["generateImage"] = async ({
|
||||
});
|
||||
}
|
||||
|
||||
return { images };
|
||||
/**
|
||||
* Image models bill per image, not per token: only a provider that
|
||||
* volunteers a usage figure is recorded with one.
|
||||
*/
|
||||
const usage =
|
||||
json?.usage && typeof json.usage == "object"
|
||||
? {
|
||||
prompt_tokens: Number(json.usage.prompt_tokens) || 0,
|
||||
completion_tokens: Number(json.usage.completion_tokens) || 0,
|
||||
...(typeof json.usage.cost == "number" ? { cost: json.usage.cost } : {}),
|
||||
}
|
||||
: undefined;
|
||||
|
||||
return { images, ...(usage ? { usage } : {}) };
|
||||
};
|
||||
|
||||
export default generateImage;
|
||||
|
||||
@@ -50,7 +50,22 @@ const transcribe: HarnessAdapter["transcribe"] = async ({
|
||||
});
|
||||
}
|
||||
|
||||
return { text: String(json?.text || "").trim() };
|
||||
/**
|
||||
* Providers that bill by audio seconds sometimes report a token count;
|
||||
* the OpenAI shape puts it under `usage`. Left out when absent, so the
|
||||
* row records that the call happened without a token count.
|
||||
*/
|
||||
const usage =
|
||||
json?.usage && typeof json.usage == "object"
|
||||
? {
|
||||
prompt_tokens: Number(json.usage.prompt_tokens ?? json.usage.input_tokens) || 0,
|
||||
completion_tokens:
|
||||
Number(json.usage.completion_tokens ?? json.usage.output_tokens) || 0,
|
||||
...(typeof json.usage.cost == "number" ? { cost: json.usage.cost } : {}),
|
||||
}
|
||||
: undefined;
|
||||
|
||||
return { text: String(json?.text || "").trim(), ...(usage ? { usage } : {}) };
|
||||
};
|
||||
|
||||
export default transcribe;
|
||||
|
||||
@@ -14,6 +14,15 @@ type Params<T> = {
|
||||
* uses, so it doesn't switch models mid-task without a reason.
|
||||
*/
|
||||
prefer_key?: string;
|
||||
/**
|
||||
* Called with what the candidate that worked returned, before the call
|
||||
* comes back to the caller: where a harness writes its usage down.
|
||||
* Never fails the call (the caller passes a recorder that can't throw).
|
||||
*/
|
||||
on_success?: (params: {
|
||||
candidate: HarnessCandidate;
|
||||
result: T;
|
||||
}) => Promise<void> | void;
|
||||
cooldowns?: HarnessCooldowns;
|
||||
now?: () => number;
|
||||
};
|
||||
@@ -42,6 +51,7 @@ export default async function withCandidates<T>({
|
||||
candidates,
|
||||
run,
|
||||
prefer_key,
|
||||
on_success,
|
||||
cooldowns = DefaultCooldowns,
|
||||
now = Date.now,
|
||||
}: Params<T>): Promise<{ result: T; candidate: HarnessCandidate }> {
|
||||
@@ -93,7 +103,11 @@ export default async function withCandidates<T>({
|
||||
if (!isReady(candidate)) continue;
|
||||
|
||||
try {
|
||||
return { result: await run(candidate), candidate };
|
||||
const result = await run(candidate);
|
||||
|
||||
await on_success?.({ candidate, result });
|
||||
|
||||
return { result, candidate };
|
||||
} catch (error: unknown) {
|
||||
if (!(error instanceof HarnessError)) throw error;
|
||||
if (error.code == "aborted") throw error;
|
||||
|
||||
@@ -8,6 +8,8 @@ import type {
|
||||
HarnessMessage,
|
||||
HarnessToolDefinition,
|
||||
} from "../types";
|
||||
import buildUsageCall from "../usage/build-usage-call";
|
||||
import recordHarnessUsage from "../usage/record-harness-usage";
|
||||
import grabChatParams from "./grab-chat-params";
|
||||
import prepareMessagesForModel from "./prepare-messages-for-model";
|
||||
import stripThinkTags from "./strip-think-tags";
|
||||
@@ -47,6 +49,14 @@ export type GenerateTextResult = HarnessChatResult & {
|
||||
candidate: HarnessCandidate;
|
||||
};
|
||||
|
||||
/**
|
||||
* One turn as `withCandidates` carries it: the answer the provider gave,
|
||||
* with the reasoning tags stripped and an empty answer turned into null.
|
||||
*/
|
||||
type ChatAnswer = Omit<HarnessChatResult, "text"> & {
|
||||
text: string | null;
|
||||
};
|
||||
|
||||
/**
|
||||
* One model turn on the text harness, falling back across its models.
|
||||
* Media the model can't take is read by the helper harnesses first.
|
||||
@@ -71,10 +81,18 @@ export default async function generateText({
|
||||
? undefined
|
||||
: await grabChatParams({ harness: "text", bot_id, workspace_id });
|
||||
|
||||
const { result, candidate } = await withCandidates({
|
||||
const { result, candidate } = await withCandidates<ChatAnswer>({
|
||||
harness: "text",
|
||||
candidates: resolved,
|
||||
prefer_key,
|
||||
on_success: async ({ candidate: answered, result: answer }): Promise<void> =>
|
||||
await recordHarnessUsage({
|
||||
call: buildUsageCall({
|
||||
harness: "text",
|
||||
candidate: answered,
|
||||
usage: answer.usage,
|
||||
}),
|
||||
}),
|
||||
run: async (candidate) => {
|
||||
const answer = await grabHarnessAdapter({
|
||||
kind: candidate.provider.kind,
|
||||
|
||||
@@ -0,0 +1,72 @@
|
||||
import { describe, expect, test } from "bun:test";
|
||||
import type { BUN_SQLITE_COBBLE_BOTS, BUN_SQLITE_COBBLE_WORKSPACES } from "@/db/types/db";
|
||||
import delegateTaskTool from "./delegate-task-tool";
|
||||
|
||||
const worker = { id: 5, name: "Researcher", role: "research", kind: "worker" } as BUN_SQLITE_COBBLE_BOTS;
|
||||
const workspace = { id: 3, name: "Launch", archived: 0 } as BUN_SQLITE_COBBLE_WORKSPACES;
|
||||
const signal = new AbortController().signal;
|
||||
|
||||
function coordinatorTool({
|
||||
workspace_id,
|
||||
workers = [worker],
|
||||
workspaces = [workspace],
|
||||
}: { workspace_id?: number; workers?: BUN_SQLITE_COBBLE_BOTS[]; workspaces?: BUN_SQLITE_COBBLE_WORKSPACES[] } = {}) {
|
||||
return delegateTaskTool({
|
||||
user_id: 1,
|
||||
coordinator_id: 1,
|
||||
session_id: 10,
|
||||
workspace_id: workspace_id ?? null,
|
||||
coordinator_name: "Coordinator",
|
||||
workers,
|
||||
workspaces,
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* The tool's own confines, kept to what needs no database: the argument it
|
||||
* accepts and the refusal that keeps a workspace hand-over in its
|
||||
* workspace. Every run that gets past those reaches the DB.
|
||||
*/
|
||||
describe("delegateTaskTool", () => {
|
||||
test("takes an optional workspace the task runs in", () => {
|
||||
const tool = coordinatorTool();
|
||||
|
||||
expect(tool.name).toBe("delegate_task");
|
||||
expect(tool.schema?.safeParse({ bot_id: 5, title: "Dig", instructions: "Read the docs." }).success).toBe(true);
|
||||
expect(
|
||||
tool.schema?.safeParse({ bot_id: 5, title: "Dig", instructions: "Read the docs.", workspace_id: 3 }).success,
|
||||
).toBe(true);
|
||||
});
|
||||
|
||||
test("names the workspaces it can hand work into", () => {
|
||||
expect(coordinatorTool().description).toContain("3: Launch");
|
||||
});
|
||||
|
||||
test("in a workspace, says the task runs there and offers no other", () => {
|
||||
const tool = coordinatorTool({ workspace_id: 3 });
|
||||
|
||||
expect(tool.description).toContain("runs there");
|
||||
expect(tool.schema?.safeParse({ bot_id: 5, title: "Dig", instructions: "x", workspace_id: 9 }).success).toBe(
|
||||
true,
|
||||
);
|
||||
});
|
||||
|
||||
test("refuses to move a hand-over out of the workspace it came from", async () => {
|
||||
await expect(
|
||||
coordinatorTool({ workspace_id: 3 }).run({
|
||||
args: { bot_id: 5, title: "Dig", instructions: "x", workspace_id: 9 },
|
||||
signal,
|
||||
}),
|
||||
).rejects.toThrow(/runs in that workspace/);
|
||||
});
|
||||
|
||||
test("accepts the workspace it came from", async () => {
|
||||
/** Reaches the DB and fails there for a test id, never on the guard. */
|
||||
await expect(
|
||||
coordinatorTool({ workspace_id: 3 }).run({
|
||||
args: { bot_id: 5, title: "Dig", instructions: "x", workspace_id: 3 },
|
||||
signal,
|
||||
}),
|
||||
).rejects.not.toThrow(/runs in that workspace/);
|
||||
});
|
||||
});
|
||||
@@ -1,8 +1,12 @@
|
||||
import { z } from "zod";
|
||||
import type { BUN_SQLITE_COBBLE_BOTS } from "@/db/types/db";
|
||||
import type { BUN_SQLITE_COBBLE_BOTS, BUN_SQLITE_COBBLE_WORKSPACES } from "@/db/types/db";
|
||||
import pushBotCreated from "@/src/bots-queue/push-bot-created";
|
||||
import pushBotNotice from "@/src/bots-queue/push-bot-notice";
|
||||
import enqueueBotJob from "@/src/functions/backend/db/bots-queue/enqueue-bot-job";
|
||||
import grabOpenBotTask from "@/src/functions/backend/db/bots-queue/grab-open-bot-task";
|
||||
import addBotToWorkspace from "@/src/functions/backend/db/workspaces/add-bot-to-workspace";
|
||||
import grabUserWorkspaces from "@/src/functions/backend/db/workspaces/grab-user-workspaces";
|
||||
import grabWorkspaceBots from "@/src/functions/backend/db/workspaces/grab-workspace-bots";
|
||||
import insertBotMessage from "@/src/functions/backend/db/bots/insert-bot-message";
|
||||
import grabCrewWorkers from "@/src/functions/backend/db/bots/grab-crew-workers";
|
||||
import grabBotComputers from "@/src/functions/backend/db/computers/grab-bot-computers";
|
||||
@@ -32,19 +36,32 @@ type Params = {
|
||||
* The worker bots it can hand work to (in a workspace, its members).
|
||||
*/
|
||||
workers: BUN_SQLITE_COBBLE_BOTS[];
|
||||
/**
|
||||
* The user's live workspaces, for the picker. In a workspace there's
|
||||
* nothing to pick: the task runs there.
|
||||
*/
|
||||
workspaces?: BUN_SQLITE_COBBLE_WORKSPACES[];
|
||||
};
|
||||
|
||||
type Args = {
|
||||
bot_id: number;
|
||||
title: string;
|
||||
instructions: string;
|
||||
/**
|
||||
* The workspace the worker runs the task in, from the coordinator's own
|
||||
* chat. Left out, the worker runs it in its own chat, on its own
|
||||
* memory and its direct computers only.
|
||||
*/
|
||||
workspace_id?: number;
|
||||
};
|
||||
|
||||
/**
|
||||
* Lets the coordinator hand a task to a worker bot. The worker runs it on
|
||||
* the computer it's on (if it has one) and posts the result in its own
|
||||
* conversation, where a note says who asked; the result (or failure) is
|
||||
* also reported back in the coordinator's conversation.
|
||||
* Lets the coordinator hand a task to a worker bot: in its own chat, or in
|
||||
* one of the user's workspaces, which decides the work's context entirely.
|
||||
* The worker runs it on the computer it's on there (if it has one) and
|
||||
* posts the result in that conversation, where a note says who asked; the
|
||||
* result (or failure) is also reported back in the coordinator's
|
||||
* conversation.
|
||||
*/
|
||||
export default function delegateTaskTool({
|
||||
user_id,
|
||||
@@ -53,33 +70,79 @@ export default function delegateTaskTool({
|
||||
workspace_id,
|
||||
coordinator_name,
|
||||
workers,
|
||||
workspaces = [],
|
||||
}: Params): HarnessTool<Args> {
|
||||
const crew = workers
|
||||
.map((bot) => `${bot.id}: ${bot.name}${bot.role ? ` (${bot.role})` : ""}`)
|
||||
.join("; ");
|
||||
const places = workspaces
|
||||
.map((w) => `${w.id}: ${w.name}`)
|
||||
.join("; ");
|
||||
|
||||
return {
|
||||
name: "delegate_task",
|
||||
description: `Hand a task to a crew member. They run it in the background and post the result in their own conversation; it's also reported back here when they finish or fail. Write the instructions as a full brief: what to do, where, and how they'll know it's done. Crew: ${crew}.`,
|
||||
description: `Hand a task to a crew member. They run it in the background and post the result in their own conversation; it's also reported back here when they finish or fail. Write the instructions as a full brief: what to do, where, and how they'll know it's done. Crew: ${crew}.${
|
||||
workspace_id
|
||||
? ` You are working in a workspace, so the task runs there: in this workspace's context.`
|
||||
: ` Leave workspace_id out to run it in the crew member's own chat: only its own memory, files and directly assigned computers apply there, and the user finds the result by talking to it on its own. Add workspace_id to run it in one of the user's workspaces instead: the workspace's shared context, the member's instructions and memory there, its files and its computers (the workspace's own and the member's) all apply, and the result lands in its conversation in that workspace, beside the project it belongs to. Pick a workspace when the work belongs to that project; the member is added to it when it isn't in it yet. Workspaces: ${places || "none"}.`
|
||||
}`,
|
||||
schema: z.object({
|
||||
bot_id: z.number().int(),
|
||||
title: z.string().min(1).max(80),
|
||||
instructions: z.string().min(1).max(8000),
|
||||
workspace_id: z
|
||||
.number()
|
||||
.int()
|
||||
.optional()
|
||||
.describe(
|
||||
workspace_id
|
||||
? "Not available here: a task handed over in a workspace runs in that workspace."
|
||||
: "Run the task in this workspace: its shared context, memory, files and computers apply. Leave it out to run it in the crew member's own chat.",
|
||||
),
|
||||
}),
|
||||
run: async ({ args }) => {
|
||||
const fresh = await grabCrewWorkers({ workspace_id });
|
||||
/**
|
||||
* In a workspace the task can't leave it: the run is already
|
||||
* there, and that is the whole point of the hand-over.
|
||||
*/
|
||||
if (workspace_id && args.workspace_id && args.workspace_id != Number(workspace_id)) {
|
||||
throw new Error(
|
||||
`A task handed over from a workspace runs in that workspace. Leave workspace_id out, or hand it over from your own chat.`,
|
||||
);
|
||||
}
|
||||
|
||||
const target_workspace_id = workspace_id
|
||||
? Number(workspace_id)
|
||||
: args.workspace_id
|
||||
? Number(args.workspace_id)
|
||||
: null;
|
||||
|
||||
const fresh = await grabCrewWorkers({ workspace_id: target_workspace_id });
|
||||
const worker = fresh.find((bot) => bot.id == args.bot_id);
|
||||
|
||||
if (!worker?.id) {
|
||||
throw new Error(
|
||||
`No crew member has id ${args.bot_id}. Crew: ${fresh
|
||||
.map((bot) => `${bot.id}: ${bot.name}`)
|
||||
.join("; ") || "none"}.`,
|
||||
`No crew member has id ${args.bot_id}${
|
||||
target_workspace_id ? ` in workspace ${target_workspace_id}` : ""
|
||||
}. Crew: ${
|
||||
fresh.map((bot) => `${bot.id}: ${bot.name}`).join("; ") || "none"
|
||||
}.`,
|
||||
);
|
||||
}
|
||||
|
||||
const computers = await grabBotComputers({ bot_id: worker.id, workspace_id });
|
||||
const worker_session = await ensureBotSession({ user_id, bot_id: worker.id, workspace_id });
|
||||
const joined = target_workspace_id
|
||||
? await ensureWorkerInWorkspace({ user_id, worker, target_workspace_id })
|
||||
: "";
|
||||
|
||||
const computers = await grabBotComputers({
|
||||
bot_id: worker.id,
|
||||
workspace_id: target_workspace_id,
|
||||
});
|
||||
const worker_session = await ensureBotSession({
|
||||
user_id,
|
||||
bot_id: worker.id,
|
||||
workspace_id: target_workspace_id,
|
||||
});
|
||||
const open = await grabOpenBotTask({
|
||||
bot_id: worker.id,
|
||||
session_id: Number(worker_session.id),
|
||||
@@ -108,12 +171,58 @@ export default function delegateTaskTool({
|
||||
message: await insertBotMessage({
|
||||
bot_id: worker.id,
|
||||
session_id: Number(worker_session.id),
|
||||
workspace_id: workspace_id ? Number(workspace_id) : null,
|
||||
workspace_id: target_workspace_id,
|
||||
text: `${coordinator_name} asked me to: **${args.title}**`,
|
||||
}),
|
||||
});
|
||||
|
||||
return `Task #${job_id} is with ${worker.name}. Their result will be in their conversation and reported back here.`;
|
||||
return `Task #${job_id} is with ${worker.name}${
|
||||
target_workspace_id ? ` in workspace ${target_workspace_id}` : " in their own chat"
|
||||
}.${joined} Their result will be in that conversation and reported back here.`;
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
type EnsureParams = {
|
||||
user_id: number | string;
|
||||
worker: BUN_SQLITE_COBBLE_BOTS;
|
||||
target_workspace_id: number;
|
||||
};
|
||||
|
||||
/**
|
||||
* A worker can only run a task in a workspace it's in, and a workspace's
|
||||
* members were already listed by `grabCrewWorkers`. Anyone else is added
|
||||
* first, since delegation is exactly the moment the broker decides who
|
||||
* does the work where.
|
||||
*/
|
||||
async function ensureWorkerInWorkspace({
|
||||
user_id,
|
||||
worker,
|
||||
target_workspace_id,
|
||||
}: EnsureParams) {
|
||||
if (!worker.id) return "";
|
||||
|
||||
const workspaces = (await grabUserWorkspaces({ user_id })).filter((w) => !w.archived);
|
||||
const workspace = workspaces.find((w) => w.id == target_workspace_id);
|
||||
|
||||
if (!workspace) {
|
||||
throw new Error(
|
||||
`No workspace has id ${target_workspace_id}. Workspaces: ${
|
||||
workspaces.map((w) => `${w.id}: ${w.name}`).join("; ") || "none"
|
||||
}. Hand the task over without workspace_id to run it in the crew member's own chat.`,
|
||||
);
|
||||
}
|
||||
|
||||
const members = await grabWorkspaceBots({ user_id, workspace_id: target_workspace_id });
|
||||
|
||||
if (members.some((bot) => bot.id == worker.id)) return "";
|
||||
|
||||
await addBotToWorkspace({
|
||||
user_id,
|
||||
workspace_id: target_workspace_id,
|
||||
bot_id: Number(worker.id),
|
||||
});
|
||||
await pushBotCreated({ bot: worker });
|
||||
|
||||
return ` ${worker.name} wasn't in "${workspace.name}", so it was added to it first.`;
|
||||
}
|
||||
|
||||
@@ -61,7 +61,9 @@ type Params = {
|
||||
* for the coordinator (the super admin's alone) create_workspace and
|
||||
* update_workspace (a workspace and its shared context), create_bot to
|
||||
* grow the crew, delegate_task to every worker (the workspace's in a
|
||||
* workspace) plus, in its own chat, find_bot and find_workspace,
|
||||
* workspace, where the task runs; from its own chat, in the worker's own
|
||||
* chat or in a workspace it names) plus, in its own chat, find_bot and
|
||||
* find_workspace,
|
||||
* add_bot_to_workspace and remove_bot_from_workspace, assign_computer,
|
||||
* update_bot, delete_bot and delete_workspace (which workspace, which
|
||||
* computers, and which name, role and core instructions each crew member
|
||||
@@ -150,6 +152,7 @@ export default async function grabBotChatTools({
|
||||
workspace_id,
|
||||
coordinator_name: bot.name || "The coordinator",
|
||||
workers,
|
||||
workspaces,
|
||||
}),
|
||||
);
|
||||
|
||||
|
||||
+25
-2
@@ -138,6 +138,18 @@ export type HarnessUsage = {
|
||||
cost?: number;
|
||||
};
|
||||
|
||||
/**
|
||||
* What a harness does with a model, for its usage rows and the totals
|
||||
* that read them. Mirrors the `harness` column of `token_usages`.
|
||||
*/
|
||||
export type HarnessUsageKind =
|
||||
| "text"
|
||||
| "vision"
|
||||
| "transcription"
|
||||
| "speech"
|
||||
| "image"
|
||||
| "embedding";
|
||||
|
||||
export type HarnessChatParams = {
|
||||
candidate: HarnessCandidate;
|
||||
messages: HarnessMessage[];
|
||||
@@ -238,11 +250,22 @@ export type HarnessEmbedResult = {
|
||||
*/
|
||||
export type HarnessAdapter = {
|
||||
chat: (params: HarnessChatParams) => Promise<HarnessChatResult>;
|
||||
transcribe: (params: HarnessTranscribeParams) => Promise<{ text: string }>;
|
||||
/**
|
||||
* `usage` when the provider says how many tokens it heard, which
|
||||
* OpenAI-style speech-to-text endpoints do only sometimes.
|
||||
*/
|
||||
transcribe: (params: HarnessTranscribeParams) => Promise<{
|
||||
text: string;
|
||||
usage?: HarnessUsage;
|
||||
}>;
|
||||
speak: (params: HarnessSpeakParams) => Promise<HarnessAudioStream>;
|
||||
/**
|
||||
* Image models report what they made rather than how many tokens it
|
||||
* took, so `usage` is only whatever the provider volunteers.
|
||||
*/
|
||||
generateImage: (
|
||||
params: HarnessImageParams,
|
||||
) => Promise<{ images: HarnessImageOutput[] }>;
|
||||
) => Promise<{ images: HarnessImageOutput[]; usage?: HarnessUsage }>;
|
||||
embed: (params: HarnessEmbedParams) => Promise<HarnessEmbedResult>;
|
||||
listModels: (params: {
|
||||
provider: HarnessProviderConfig;
|
||||
|
||||
@@ -0,0 +1,102 @@
|
||||
import { describe, expect, test } from "bun:test";
|
||||
import type { HarnessCandidate } from "../types";
|
||||
import buildUsageCall from "./build-usage-call";
|
||||
|
||||
function model(overrides: { row_id?: number } = {}) {
|
||||
return {
|
||||
slug: "deepseek/deepseek-chat",
|
||||
name: "DeepSeek Chat",
|
||||
input_modalities: ["text"],
|
||||
output_modalities: ["text"],
|
||||
supports_tools: true,
|
||||
...overrides,
|
||||
} as HarnessCandidate["model"];
|
||||
}
|
||||
|
||||
function candidate(overrides: Partial<HarnessCandidate> = {}): HarnessCandidate {
|
||||
return {
|
||||
provider: {
|
||||
id: 2,
|
||||
name: "OpenRouter",
|
||||
kind: "openrouter",
|
||||
} as HarnessCandidate["provider"],
|
||||
key: "openrouter:deepseek/deepseek-chat",
|
||||
source: "auto",
|
||||
model: model({ row_id: 41 }),
|
||||
...overrides,
|
||||
};
|
||||
}
|
||||
|
||||
describe("buildUsageCall", () => {
|
||||
test("names the candidate that really answered and what it spent", () => {
|
||||
const call = buildUsageCall({
|
||||
harness: "text",
|
||||
candidate: candidate(),
|
||||
usage: { prompt_tokens: 1200, completion_tokens: 340, cost: 0.0042 },
|
||||
});
|
||||
|
||||
expect(call).toEqual({
|
||||
harness: "text",
|
||||
provider_id: 2,
|
||||
provider_name: "OpenRouter",
|
||||
model_id: 41,
|
||||
model_slug: "deepseek/deepseek-chat",
|
||||
prompt_tokens: 1200,
|
||||
completion_tokens: 340,
|
||||
cost: 0.0042,
|
||||
});
|
||||
});
|
||||
|
||||
test("a provider with no name reads as its kind", () => {
|
||||
const call = buildUsageCall({
|
||||
harness: "image",
|
||||
candidate: candidate({
|
||||
provider: { id: 4, name: "", kind: "openai" } as HarnessCandidate["provider"],
|
||||
}),
|
||||
usage: { prompt_tokens: 0, completion_tokens: 0 },
|
||||
});
|
||||
|
||||
expect(call.provider_name).toBe("openai");
|
||||
});
|
||||
|
||||
test("a provider that reports no usage still gets a row", () => {
|
||||
const call = buildUsageCall({ harness: "transcription", candidate: candidate() });
|
||||
|
||||
expect(call.prompt_tokens).toBe(0);
|
||||
expect(call.completion_tokens).toBe(0);
|
||||
expect(call.cost).toBeUndefined();
|
||||
});
|
||||
|
||||
test("a candidate with no catalog row has no model id to point at", () => {
|
||||
const call = buildUsageCall({
|
||||
harness: "text",
|
||||
candidate: candidate({
|
||||
model: model({ slug: "made-up-model" } as any),
|
||||
}),
|
||||
usage: { prompt_tokens: 10, completion_tokens: 5 },
|
||||
});
|
||||
|
||||
expect(call.model_id).toBeUndefined();
|
||||
expect(call.model_slug).toBe("made-up-model");
|
||||
});
|
||||
|
||||
test("a cost the provider did not report is left out, not zeroed", () => {
|
||||
const call = buildUsageCall({
|
||||
harness: "text",
|
||||
candidate: candidate(),
|
||||
usage: { prompt_tokens: 10, completion_tokens: 5, cost: Number.NaN },
|
||||
});
|
||||
|
||||
expect(call.cost).toBeUndefined();
|
||||
});
|
||||
|
||||
test("a free provider's zero cost is kept as zero", () => {
|
||||
const call = buildUsageCall({
|
||||
harness: "text",
|
||||
candidate: candidate(),
|
||||
usage: { prompt_tokens: 10, completion_tokens: 5, cost: 0 },
|
||||
});
|
||||
|
||||
expect(call.cost).toBe(0);
|
||||
});
|
||||
});
|
||||
@@ -0,0 +1,37 @@
|
||||
import type { HarnessCandidate, HarnessUsage } from "../types";
|
||||
import type { HarnessUsageCall } from "./types";
|
||||
|
||||
type Params = {
|
||||
harness: HarnessUsageCall["harness"];
|
||||
candidate: HarnessCandidate;
|
||||
usage?: HarnessUsage;
|
||||
};
|
||||
|
||||
/**
|
||||
* The candidate that really answered a call, and what it reported, as the
|
||||
* row `recordHarnessUsage` writes. Providers that report no usage at all
|
||||
* still get a row (no tokens), so a call is never invisible.
|
||||
*/
|
||||
export default function buildUsageCall({ harness, candidate, usage }: Params): HarnessUsageCall {
|
||||
return {
|
||||
harness,
|
||||
provider_id: candidate.provider.id,
|
||||
provider_name: candidate.provider.name || candidate.provider.kind,
|
||||
model_id: modelIdOf(candidate),
|
||||
model_slug: candidate.model.slug,
|
||||
prompt_tokens: Number(usage?.prompt_tokens) || 0,
|
||||
completion_tokens: Number(usage?.completion_tokens) || 0,
|
||||
cost: typeof usage?.cost == "number" && Number.isFinite(usage.cost) ? usage.cost : undefined,
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* The `harness_models` row id, when the candidate carries the catalog row
|
||||
* it was resolved from (`withCandidates` doesn't drop it; a candidate made
|
||||
* by hand may not have one).
|
||||
*/
|
||||
function modelIdOf(candidate: HarnessCandidate) {
|
||||
const row_id = (candidate.model as { row_id?: number }).row_id;
|
||||
|
||||
return typeof row_id == "number" ? row_id : undefined;
|
||||
}
|
||||
@@ -0,0 +1,106 @@
|
||||
import { describe, expect, test } from "bun:test";
|
||||
import { grabHarnessUsageScope, runWithHarnessUsageScope } from "./harness-usage-scope";
|
||||
|
||||
describe("harness usage scope", () => {
|
||||
test("nothing is attributed outside a run", () => {
|
||||
expect(grabHarnessUsageScope()).toBeUndefined();
|
||||
});
|
||||
|
||||
test("a run's scope is visible to everything it calls", async () => {
|
||||
const scope = await runWithHarnessUsageScope({
|
||||
scope: { user_id: 7, bot_id: 3, session_id: 11 },
|
||||
task: async () => grabHarnessUsageScope(),
|
||||
});
|
||||
|
||||
expect(scope).toEqual({ user_id: 7, bot_id: 3, session_id: 11 });
|
||||
});
|
||||
|
||||
test("a nested run keeps what the outer one knew", async () => {
|
||||
const scope = await runWithHarnessUsageScope({
|
||||
scope: { user_id: 7, bot_id: 3, session_id: 11, job_id: 5 },
|
||||
task: () =>
|
||||
runWithHarnessUsageScope({
|
||||
scope: { message_id: 42 },
|
||||
task: async () => grabHarnessUsageScope(),
|
||||
}),
|
||||
});
|
||||
|
||||
expect(scope).toEqual({
|
||||
user_id: 7,
|
||||
bot_id: 3,
|
||||
session_id: 11,
|
||||
job_id: 5,
|
||||
message_id: 42,
|
||||
});
|
||||
});
|
||||
|
||||
test("a nested run overrides only what it sets", async () => {
|
||||
const scope = await runWithHarnessUsageScope({
|
||||
scope: { user_id: 7, bot_id: 3, workspace_id: 9 },
|
||||
task: () =>
|
||||
runWithHarnessUsageScope({
|
||||
scope: { bot_id: 4 },
|
||||
task: async () => grabHarnessUsageScope(),
|
||||
}),
|
||||
});
|
||||
|
||||
expect(scope).toEqual({ user_id: 7, bot_id: 4, workspace_id: 9 });
|
||||
});
|
||||
|
||||
test("an undefined field leaves the outer value alone", async () => {
|
||||
const scope = await runWithHarnessUsageScope({
|
||||
scope: { user_id: 7, bot_id: 3, job_id: 5 },
|
||||
task: () =>
|
||||
runWithHarnessUsageScope({
|
||||
scope: { job_id: undefined, message_id: 42 },
|
||||
task: async () => grabHarnessUsageScope(),
|
||||
}),
|
||||
});
|
||||
|
||||
expect(scope).toEqual({ user_id: 7, bot_id: 3, job_id: 5, message_id: 42 });
|
||||
});
|
||||
|
||||
test("a null field clears the outer value, since it means 'none'", async () => {
|
||||
const scope = await runWithHarnessUsageScope({
|
||||
scope: { user_id: 7, bot_id: 3, workspace_id: 9, job_id: 5 },
|
||||
task: () =>
|
||||
runWithHarnessUsageScope({
|
||||
scope: { workspace_id: null, job_id: null },
|
||||
task: async () => grabHarnessUsageScope(),
|
||||
}),
|
||||
});
|
||||
|
||||
expect(scope).toEqual({ user_id: 7, bot_id: 3, workspace_id: null, job_id: null });
|
||||
});
|
||||
|
||||
test("the scope is gone once the run returns", async () => {
|
||||
await runWithHarnessUsageScope({
|
||||
scope: { user_id: 7 },
|
||||
task: async () => undefined,
|
||||
});
|
||||
|
||||
expect(grabHarnessUsageScope()).toBeUndefined();
|
||||
});
|
||||
|
||||
test("two concurrent runs never see each other's scope", async () => {
|
||||
const [first, second] = await Promise.all([
|
||||
runWithHarnessUsageScope({
|
||||
scope: { user_id: 1 },
|
||||
task: async () => {
|
||||
await Bun.sleep(5);
|
||||
return grabHarnessUsageScope();
|
||||
},
|
||||
}),
|
||||
runWithHarnessUsageScope({
|
||||
scope: { user_id: 2 },
|
||||
task: async () => {
|
||||
await Bun.sleep(1);
|
||||
return grabHarnessUsageScope();
|
||||
},
|
||||
}),
|
||||
]);
|
||||
|
||||
expect(first).toEqual({ user_id: 1 });
|
||||
expect(second).toEqual({ user_id: 2 });
|
||||
});
|
||||
});
|
||||
@@ -0,0 +1,63 @@
|
||||
import { AsyncLocalStorage } from "node:async_hooks";
|
||||
import type { HarnessUsageScope } from "./types";
|
||||
|
||||
/**
|
||||
* The attribution a run's model calls carry.
|
||||
*
|
||||
* A bot's run is entered once (`runWithHarnessUsageScope`) and everything
|
||||
* it calls underneath - the text loop's turns, a session summary, a
|
||||
* transcription, an image, an embedding - is written down for that user,
|
||||
* bot, session and job without threading the ids through every harness
|
||||
* function. One of those calls is itself entered by a `run` that has to
|
||||
* swap the scope it reports under (`withHarnessUsageScope`): the session
|
||||
* namer runs outside the job's scope, on purpose, since the job it named
|
||||
* is already complete.
|
||||
*
|
||||
* The store is merged, not replaced: a call that only knows the message id
|
||||
* (the text loop attaching a turn's usage) keeps the run's user, bot and
|
||||
* job.
|
||||
*/
|
||||
const storage = new AsyncLocalStorage<HarnessUsageScope>();
|
||||
|
||||
/**
|
||||
* Runs `task` with `scope` as its attribution. Ambient values are merged
|
||||
* under it, so a nested call keeps what the outer one set and overrides
|
||||
* only what it knows better (null clears a field).
|
||||
*/
|
||||
export async function runWithHarnessUsageScope<T>(
|
||||
{ scope, task }: { scope: HarnessUsageScope; task: () => Promise<T> },
|
||||
): Promise<T> {
|
||||
return await storage.run(mergeScopes({ base: storage.getStore(), overlay: scope }), task);
|
||||
}
|
||||
|
||||
/**
|
||||
* The attribution a call right now carries, if any.
|
||||
*/
|
||||
export function grabHarnessUsageScope(): HarnessUsageScope | undefined {
|
||||
return storage.getStore();
|
||||
}
|
||||
|
||||
function mergeScopes({
|
||||
base,
|
||||
overlay,
|
||||
}: {
|
||||
base?: HarnessUsageScope;
|
||||
overlay: HarnessUsageScope;
|
||||
}): HarnessUsageScope {
|
||||
const merged: HarnessUsageScope = { ...base };
|
||||
|
||||
for (const [key, value] of Object.entries(overlay) as [
|
||||
keyof HarnessUsageScope,
|
||||
HarnessUsageScope[keyof HarnessUsageScope],
|
||||
][]) {
|
||||
if (value === undefined) continue;
|
||||
|
||||
/**
|
||||
* A null is a real value here: it says "none", e.g. a bot's own
|
||||
* chat has no workspace, and must clear what the outer scope had.
|
||||
*/
|
||||
(merged as Record<string, unknown>)[key] = value;
|
||||
}
|
||||
|
||||
return merged;
|
||||
}
|
||||
@@ -0,0 +1,92 @@
|
||||
import BunSQLite from "@moduletrace/bun-sqlite";
|
||||
import type {
|
||||
BUN_SQLITE_COBBLE_TOKEN_USAGES,
|
||||
BunSQLiteTables,
|
||||
} from "@/db/types/db";
|
||||
import { logError } from "@/src/functions/backend/logger";
|
||||
import { grabHarnessUsageScope } from "./harness-usage-scope";
|
||||
import type { HarnessUsageCall, HarnessUsageScope } from "./types";
|
||||
|
||||
type Params = {
|
||||
/**
|
||||
* The scope the call ran under. Left out, the ambient one
|
||||
* (`grabHarnessUsageScope`) is used, which is what every entry point
|
||||
* does: the ids came in through `runWithHarnessUsageScope`.
|
||||
*/
|
||||
scope?: HarnessUsageScope;
|
||||
/**
|
||||
* Which harness and which model answered, with what the provider
|
||||
* reported it cost.
|
||||
*/
|
||||
call: HarnessUsageCall;
|
||||
};
|
||||
|
||||
/**
|
||||
* Writes one model call into `token_usages`: the tokens it spent, on which
|
||||
* model and provider, for whom, and which message or job it was for. This
|
||||
* is what a message's token line, a bot's and a provider's totals and the
|
||||
* usage page all read.
|
||||
*
|
||||
* Never throws: a model that answered must not have its answer lost
|
||||
* because the bookkeeping failed. A call without an owner (no scope, no
|
||||
* user) is not written at all, since nothing could show it.
|
||||
*/
|
||||
export default async function recordHarnessUsage({ scope, call }: Params) {
|
||||
const resolved = scope || grabHarnessUsageScope();
|
||||
const user_id = Number(resolved?.user_id) || null;
|
||||
|
||||
if (!user_id) return;
|
||||
|
||||
const row: BUN_SQLITE_COBBLE_TOKEN_USAGES = {
|
||||
user_id,
|
||||
bot_id: numberOrNull(resolved?.bot_id),
|
||||
workspace_id: numberOrNull(resolved?.workspace_id),
|
||||
session_id: numberOrNull(resolved?.session_id),
|
||||
message_id: numberOrNull(resolved?.message_id),
|
||||
job_id: numberOrNull(resolved?.job_id),
|
||||
harness: call.harness,
|
||||
provider_id: call.provider_id,
|
||||
provider_name: call.provider_name,
|
||||
model_id: call.model_id,
|
||||
model_slug: call.model_slug,
|
||||
prompt_tokens: Math.max(0, Math.round(call.prompt_tokens || 0)),
|
||||
completion_tokens: Math.max(0, Math.round(call.completion_tokens || 0)),
|
||||
cost: typeof call.cost == "number" && Number.isFinite(call.cost) ? call.cost : undefined,
|
||||
};
|
||||
|
||||
try {
|
||||
const res = await BunSQLite.insert<
|
||||
BUN_SQLITE_COBBLE_TOKEN_USAGES,
|
||||
(typeof BunSQLiteTables)[number]
|
||||
>({
|
||||
table: "token_usages",
|
||||
data: [row],
|
||||
});
|
||||
|
||||
if (!res.success) {
|
||||
throw new Error(res.msg || `The usage wasn't saved.`);
|
||||
}
|
||||
} catch (error: unknown) {
|
||||
await logError({
|
||||
title: "Model usage not recorded",
|
||||
error,
|
||||
metadata: {
|
||||
harness: call.harness,
|
||||
model_slug: call.model_slug || null,
|
||||
provider_name: call.provider_name || null,
|
||||
user_id,
|
||||
},
|
||||
}).catch(() => {});
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* SQLite stores NULL for what isn't given; `Number(undefined)` is NaN,
|
||||
* which bun-sqlite would refuse. An absent id is `undefined`, not `null`:
|
||||
* bun-sqlite's row types spell "no value" that way.
|
||||
*/
|
||||
function numberOrNull(value: unknown): number | undefined {
|
||||
const number = Number(value);
|
||||
|
||||
return Number.isFinite(number) && number > 0 ? number : undefined;
|
||||
}
|
||||
@@ -0,0 +1,81 @@
|
||||
import { describe, expect, test } from "bun:test";
|
||||
import { grabHarnessUsageScope, runWithHarnessUsageScope } from "./harness-usage-scope";
|
||||
import runWithFallbackScope from "./run-with-fallback-scope";
|
||||
|
||||
const owner = { user_id: 9, bot_id: 3, session_id: null, message_id: null, job_id: null };
|
||||
|
||||
describe("runWithFallbackScope", () => {
|
||||
test("the enclosing run keeps the work it already owns", async () => {
|
||||
const seen = await runWithHarnessUsageScope({
|
||||
scope: { user_id: 7, bot_id: 3, job_id: 5 },
|
||||
task: () =>
|
||||
runWithFallbackScope({
|
||||
scope: owner,
|
||||
task: async () => grabHarnessUsageScope(),
|
||||
}),
|
||||
});
|
||||
|
||||
expect(seen).toEqual({ user_id: 7, bot_id: 3, job_id: 5 });
|
||||
});
|
||||
|
||||
test("work with no run is attributed to its owner", async () => {
|
||||
const seen = await runWithFallbackScope({
|
||||
scope: owner,
|
||||
task: async () => grabHarnessUsageScope(),
|
||||
});
|
||||
|
||||
expect(seen).toEqual(owner);
|
||||
});
|
||||
|
||||
test("nobody to charge leaves the work unattributed rather than failing", async () => {
|
||||
const seen = await runWithFallbackScope({
|
||||
scope: null,
|
||||
task: async () => grabHarnessUsageScope(),
|
||||
});
|
||||
|
||||
expect(seen).toBeUndefined();
|
||||
});
|
||||
|
||||
test("a scope with no user is not an owner", async () => {
|
||||
const seen = await runWithFallbackScope({
|
||||
scope: { ...owner, user_id: undefined },
|
||||
task: async () => grabHarnessUsageScope(),
|
||||
});
|
||||
|
||||
expect(seen).toBeUndefined();
|
||||
});
|
||||
|
||||
test("work that must be charged on its own overrides the enclosing run", async () => {
|
||||
const seen = await runWithHarnessUsageScope({
|
||||
scope: { user_id: 7, bot_id: 3, job_id: 5 },
|
||||
task: () =>
|
||||
runWithFallbackScope({
|
||||
scope: owner,
|
||||
mode: "own",
|
||||
task: async () => grabHarnessUsageScope(),
|
||||
}),
|
||||
});
|
||||
|
||||
expect(seen).toEqual(owner);
|
||||
});
|
||||
|
||||
test("the work's result is passed through", async () => {
|
||||
const result = await runWithFallbackScope({
|
||||
scope: owner,
|
||||
task: async () => "the answer",
|
||||
});
|
||||
|
||||
expect(result).toBe("the answer");
|
||||
});
|
||||
|
||||
test("a failure inside the work is not swallowed", async () => {
|
||||
expect(
|
||||
runWithFallbackScope({
|
||||
scope: owner,
|
||||
task: async () => {
|
||||
throw new Error("the provider refused");
|
||||
},
|
||||
}),
|
||||
).rejects.toThrow("the provider refused");
|
||||
});
|
||||
});
|
||||
@@ -0,0 +1,41 @@
|
||||
import { grabHarnessUsageScope, runWithHarnessUsageScope } from "./harness-usage-scope";
|
||||
import type { HarnessUsageScope } from "./types";
|
||||
|
||||
type Params<T> = {
|
||||
/**
|
||||
* What to attribute the work to when nothing already owns it: a file
|
||||
* ingested by the queue has no run behind it, but its owner must still
|
||||
* see what the embedding cost. Null when there is nobody to charge.
|
||||
*/
|
||||
scope: HarnessUsageScope | null;
|
||||
task: () => Promise<T>;
|
||||
/**
|
||||
* How the work is attributed: `enclosing` keeps whatever run it is in,
|
||||
* `own` always reports under `scope` (the session namer, whose job is
|
||||
* already complete and must not be charged to it).
|
||||
*/
|
||||
mode?: "enclosing" | "own";
|
||||
};
|
||||
|
||||
/**
|
||||
* Runs work under an attribution the run may not have. Nothing is entered
|
||||
* when a run is already in charge, so an ingest a bot asked for is charged
|
||||
* to that bot's user and not to the file's owner.
|
||||
*/
|
||||
export default async function runWithFallbackScope<T>({
|
||||
scope,
|
||||
task,
|
||||
mode = "enclosing",
|
||||
}: Params<T>): Promise<T> {
|
||||
if (mode == "enclosing" && grabHarnessUsageScope()?.user_id) {
|
||||
return await task();
|
||||
}
|
||||
|
||||
const user_id = Number(scope?.user_id);
|
||||
|
||||
if (!Number.isFinite(user_id) || user_id <= 0 || user_id === null) {
|
||||
return await task();
|
||||
}
|
||||
|
||||
return await runWithHarnessUsageScope({ scope: { ...scope, user_id }, task });
|
||||
}
|
||||
@@ -0,0 +1,60 @@
|
||||
/**
|
||||
* Who and what a model call was for. Every harness call that reports its
|
||||
* usage is written down with this (`/api/auth/crud` no: `token_usages`),
|
||||
* so a message, a session, a bot or a provider can be totaled later.
|
||||
*
|
||||
* It is deliberately shaped slightly differently from the table's
|
||||
* columns: `user_id` is required here (a row without an owner can't be
|
||||
* shown or totaled for anyone), the rest are optional.
|
||||
*/
|
||||
export type HarnessUsageScope = {
|
||||
/**
|
||||
* Whose work it was: the user whose conversation the call ran for.
|
||||
*/
|
||||
user_id?: number | string;
|
||||
bot_id?: number | string | null;
|
||||
workspace_id?: number | string | null;
|
||||
session_id?: number | string | null;
|
||||
/**
|
||||
* The queued job it ran for, when it ran for one.
|
||||
*/
|
||||
job_id?: number | string | null;
|
||||
/**
|
||||
* The message it produced, for the ones that produce one.
|
||||
*/
|
||||
message_id?: number | string | null;
|
||||
};
|
||||
|
||||
/**
|
||||
* What the caller knows about the call beyond its scope: which harness
|
||||
* asked, and which candidate answered. Both are filled in by the call's
|
||||
* own entry point, never by the code that runs it.
|
||||
*/
|
||||
export type HarnessUsageCall = {
|
||||
harness: "text" | "vision" | "transcription" | "speech" | "image" | "embedding";
|
||||
provider_id?: number;
|
||||
provider_name?: string;
|
||||
model_id?: number;
|
||||
model_slug?: string;
|
||||
prompt_tokens: number;
|
||||
completion_tokens: number;
|
||||
cost?: number;
|
||||
};
|
||||
|
||||
/**
|
||||
* A message's usage, as its thread shows it and as `token_usages` stores
|
||||
* it: what one answer cost, summed over its model calls.
|
||||
*/
|
||||
export type TokenUsageSummary = {
|
||||
prompt_tokens: number;
|
||||
completion_tokens: number;
|
||||
total_tokens: number;
|
||||
/**
|
||||
* In USD, when the provider reported it.
|
||||
*/
|
||||
cost?: number;
|
||||
/**
|
||||
* How many model calls it took.
|
||||
*/
|
||||
calls: number;
|
||||
};
|
||||
@@ -0,0 +1,19 @@
|
||||
import { runWithHarnessUsageScope } from "./harness-usage-scope";
|
||||
import type { HarnessUsageScope } from "./types";
|
||||
|
||||
/**
|
||||
* Runs one call the app makes on a user's behalf inside the scope its
|
||||
* usage is written down with (transcribing a dictation, speaking a
|
||||
* message, previewing a voice). The bot's run is already inside a scope of
|
||||
* its own, so this only adds what the run didn't know: no job, no message,
|
||||
* and the session the call belongs to.
|
||||
*/
|
||||
export default async function withUsageScope<T>({
|
||||
scope,
|
||||
task,
|
||||
}: {
|
||||
scope: HarnessUsageScope;
|
||||
task: () => Promise<T>;
|
||||
}): Promise<T> {
|
||||
return await runWithHarnessUsageScope({ scope, task });
|
||||
}
|
||||
@@ -1,4 +1,5 @@
|
||||
import { LogOut, SlidersHorizontal, type LucideIcon } from "lucide-react";
|
||||
import type { BUN_SQLITE_COBBLE_USER_TYPES } from "@/db/types/db";
|
||||
import { SettingsDefaultPath } from "@/src/dict/settings-pages-dict";
|
||||
import { RailPages } from "@/src/dict/rail-pages-dict";
|
||||
|
||||
@@ -18,12 +19,16 @@ export type RailUserMenuItem = {
|
||||
* Shown to super admins only.
|
||||
*/
|
||||
super_admin_only?: boolean;
|
||||
/**
|
||||
* Who sees it, when that's an RBAC rule rather than super admins only.
|
||||
*/
|
||||
can_view?: (params: { user_types?: Pick<BUN_SQLITE_COBBLE_USER_TYPES, "user_type">[] | null }) => boolean;
|
||||
};
|
||||
|
||||
/**
|
||||
* The popover items: the rail pages (routines, activity, help), settings,
|
||||
* sign out. Nothing else lives in the rail: these destinations are only
|
||||
* reachable from here.
|
||||
* The popover items: the rail pages (routines, files, usage, activity,
|
||||
* help), settings, sign out. Nothing else lives in the rail: these
|
||||
* destinations are only reachable from here.
|
||||
*/
|
||||
export const RailUserMenuItems: RailUserMenuItem[] = [
|
||||
...RailPages,
|
||||
|
||||
@@ -2,6 +2,7 @@ import { useContext } from "react";
|
||||
import Divider from "@moduletrace/twui/layout/Divider";
|
||||
import { RailPages } from "@/src/dict/rail-pages-dict";
|
||||
import { AppContext } from "@/src/pages/__root";
|
||||
import canSeeRailItem from "@/src/utils/rbac/can-see-rail-item";
|
||||
import isPassThroughUser from "@/src/utils/rbac/is-pass-through-user";
|
||||
import RailBackLink from "./rail-back-link";
|
||||
import RailSettingsLink from "./rail-settings-link";
|
||||
@@ -11,8 +12,9 @@ type Props = {
|
||||
};
|
||||
|
||||
/**
|
||||
* The rail on routines, activity and help: back to the bots, then those
|
||||
* pages (super admin ones for super admins only).
|
||||
* The rail on routines, usage, activity and help: back to the bots, then
|
||||
* those pages (super admin ones for super admins only, crew managers for
|
||||
* usage).
|
||||
*/
|
||||
export default function RailPagesList({ pathname }: Props) {
|
||||
const { pageProps } = useContext(AppContext);
|
||||
@@ -25,8 +27,12 @@ export default function RailPagesList({ pathname }: Props) {
|
||||
<>
|
||||
<RailBackLink />
|
||||
{/* <Divider className="shrink-0 mx-1 @rail:mx-3 my-2 w-auto border-hairline dark:border-hairline" /> */}
|
||||
{RailPages.filter(
|
||||
(page) => is_super_admin || !page.super_admin_only,
|
||||
{RailPages.filter((page) =>
|
||||
canSeeRailItem({
|
||||
item: page,
|
||||
is_super_admin,
|
||||
user_types: pageProps?.user_types,
|
||||
}),
|
||||
).map((page) => (
|
||||
<RailSettingsLink
|
||||
key={page.url}
|
||||
|
||||
@@ -2,6 +2,7 @@ import { useContext } from "react";
|
||||
import Divider from "@moduletrace/twui/layout/Divider";
|
||||
import { SettingsPages } from "@/src/dict/settings-pages-dict";
|
||||
import { AppContext } from "@/src/pages/__root";
|
||||
import canSeeRailItem from "@/src/utils/rbac/can-see-rail-item";
|
||||
import isPassThroughUser from "@/src/utils/rbac/is-pass-through-user";
|
||||
import RailBackLink from "./rail-back-link";
|
||||
import RailSettingsLink from "./rail-settings-link";
|
||||
@@ -27,9 +28,11 @@ export default function RailSettingsList({ pathname }: Props) {
|
||||
<RailBackLink />
|
||||
{/* <Divider className="shrink-0 mx-1 @rail:mx-3 my-2 w-auto border-hairline dark:border-hairline" /> */}
|
||||
{SettingsPages.filter((page) =>
|
||||
page.can_view
|
||||
? page.can_view({ user_types: pageProps?.user_types })
|
||||
: is_super_admin || !page.super_admin_only,
|
||||
canSeeRailItem({
|
||||
item: page,
|
||||
is_super_admin,
|
||||
user_types: pageProps?.user_types,
|
||||
}),
|
||||
).map((page) => (
|
||||
<RailSettingsLink
|
||||
key={page.url}
|
||||
|
||||
@@ -9,18 +9,23 @@ import RailUserMenuEntry from "./rail-user-menu-entry";
|
||||
import RailUserMenuAction from "./rail-user-menu-action";
|
||||
import InstallAppModal from "./install-app-modal";
|
||||
import { AppContext } from "@/src/pages/__root";
|
||||
import canSeeRailItem from "@/src/utils/rbac/can-see-rail-item";
|
||||
import isPassThroughUser from "@/src/utils/rbac/is-pass-through-user";
|
||||
import useInstallPrompt from "@/src/hooks/use-install-prompt";
|
||||
|
||||
/**
|
||||
* 220px popover: the only element in the shell with elevation. Super
|
||||
* admin items show to super admins only. Install app shows while the
|
||||
* browser can install it; Safari gets the Share menu steps instead.
|
||||
* 220px popover: the only element in the shell with elevation. Items
|
||||
* follow the same rule as the rail's lists: super admin ones show to
|
||||
* super admins only, RBAC-gated ones by their own rule. Install app shows
|
||||
* while the browser can install it; Safari gets the Share menu steps
|
||||
* instead.
|
||||
*/
|
||||
export default function RailUserMenu() {
|
||||
const { pageProps } = useContext(AppContext);
|
||||
const is_super_admin = isPassThroughUser({ user_types: pageProps?.user_types });
|
||||
const shown = RailUserMenuItems.filter((item) => is_super_admin || !item.super_admin_only);
|
||||
const shown = RailUserMenuItems.filter((item) =>
|
||||
canSeeRailItem({ item, is_super_admin, user_types: pageProps?.user_types }),
|
||||
);
|
||||
const items = shown.filter((item) => !item.destructive);
|
||||
const destructive = shown.filter((item) => item.destructive);
|
||||
|
||||
|
||||
@@ -0,0 +1,162 @@
|
||||
import { useLayoutEffect, useRef, useState, type RefObject } from "react";
|
||||
import type { BotMessageWithAttachments } from "@/src/types";
|
||||
import {
|
||||
ThreadJumpWindowMs,
|
||||
scrollThreadToBottom,
|
||||
shouldFollowThread,
|
||||
threadOffset,
|
||||
} from "../(utils)/thread-scroll";
|
||||
|
||||
type Params = {
|
||||
/**
|
||||
* The thread's messages, oldest first, pending ones included.
|
||||
*/
|
||||
messages: BotMessageWithAttachments[];
|
||||
};
|
||||
|
||||
/**
|
||||
* Keeps the thread's bottom in view: a chat sitting at the bottom follows
|
||||
* every new message, a chat scrolled up stays where it is (except for a
|
||||
* message the user just sent, which brings the thread to it) and offers a
|
||||
* button to get back down. Whether it sits at the bottom is read from where
|
||||
* the content ends, not from `scrollTop`, whose sign a reversed column
|
||||
* reports differently across engines. Content that grows while the thread
|
||||
* is at the bottom (images landing, a wider window) is followed too.
|
||||
*/
|
||||
export default function useThreadScroll({ messages }: Params) {
|
||||
const column_ref = useRef<HTMLDivElement>(null);
|
||||
const content_ref = useRef<HTMLDivElement>(null);
|
||||
const [jump_visible, setJumpVisible] = useState(false);
|
||||
/** Whether the bottom is being held, so nothing fights the user's scroll. */
|
||||
const following = useRef(true);
|
||||
/**
|
||||
* A pressed jump is still animating, so the scrolling it causes is its
|
||||
* own. It ends when the thread arrives, when a scroll moves away from
|
||||
* the bottom (the user took over), or after the window, whichever first.
|
||||
*/
|
||||
const jump = useRef({ active: false, closest: 0, until: 0 });
|
||||
const previous = useRef<{
|
||||
last_id?: number | string | null;
|
||||
count: number;
|
||||
}>({ count: 0 });
|
||||
|
||||
const measure = () => {
|
||||
const column = column_ref.current;
|
||||
const content = content_ref.current;
|
||||
|
||||
if (!column || !content) return null;
|
||||
|
||||
return threadOffset({
|
||||
content_bottom: content.getBoundingClientRect().bottom,
|
||||
column_bottom: column.getBoundingClientRect().bottom,
|
||||
});
|
||||
};
|
||||
|
||||
/** The user's own scrolling: it decides whether the thread follows. */
|
||||
const onScroll = () => {
|
||||
const offset = measure();
|
||||
|
||||
if (!offset) return;
|
||||
|
||||
if (jump.current.active) {
|
||||
const animating =
|
||||
Date.now() < jump.current.until &&
|
||||
offset.distance <= jump.current.closest;
|
||||
|
||||
jump.current.closest = offset.distance;
|
||||
jump.current.active = animating;
|
||||
|
||||
if (animating) return;
|
||||
}
|
||||
|
||||
following.current = offset.near_bottom;
|
||||
setJumpVisible(!offset.near_bottom);
|
||||
};
|
||||
|
||||
const jumpToBottom = () => {
|
||||
const column = column_ref.current;
|
||||
|
||||
if (!column) return;
|
||||
|
||||
const reduced = Boolean(
|
||||
globalThis.matchMedia?.("(prefers-reduced-motion: reduce)").matches,
|
||||
);
|
||||
|
||||
jump.current = {
|
||||
active: !reduced,
|
||||
closest: Number.POSITIVE_INFINITY,
|
||||
until: Date.now() + ThreadJumpWindowMs,
|
||||
};
|
||||
following.current = true;
|
||||
setJumpVisible(false);
|
||||
|
||||
scrollThreadToBottom({
|
||||
element: column,
|
||||
behavior: reduced ? "auto" : "smooth",
|
||||
});
|
||||
};
|
||||
|
||||
useLayoutEffect(() => {
|
||||
const column = column_ref.current;
|
||||
const newest = messages[messages.length - 1];
|
||||
const before = previous.current;
|
||||
|
||||
/**
|
||||
* The list was replaced (a new session, the first render) when the
|
||||
* message that was last is gone.
|
||||
*/
|
||||
const kept_last_message =
|
||||
before.count > 0 &&
|
||||
messages.some((message) => message.id == before.last_id);
|
||||
|
||||
previous.current = { last_id: newest?.id, count: messages.length };
|
||||
|
||||
if (!column) return;
|
||||
|
||||
const follow = shouldFollowThread({
|
||||
near_bottom: following.current,
|
||||
newest_author: newest?.author,
|
||||
kept_last_message,
|
||||
});
|
||||
|
||||
if (!follow) {
|
||||
setJumpVisible(true);
|
||||
return;
|
||||
}
|
||||
|
||||
following.current = true;
|
||||
jump.current.active = false;
|
||||
setJumpVisible(false);
|
||||
scrollThreadToBottom({ element: column });
|
||||
}, [messages]);
|
||||
|
||||
useLayoutEffect(() => {
|
||||
const column = column_ref.current;
|
||||
const content = content_ref.current;
|
||||
|
||||
if (!column || !content) return;
|
||||
|
||||
/**
|
||||
* A message that grows after it lands (an image, a widened window)
|
||||
* keeps the bottom in view while the thread is there, and leaves a
|
||||
* chat scrolled up alone.
|
||||
*/
|
||||
const observer = new ResizeObserver(() => {
|
||||
if (!following.current) return;
|
||||
|
||||
scrollThreadToBottom({ element: column });
|
||||
});
|
||||
|
||||
observer.observe(content);
|
||||
|
||||
return () => observer.disconnect();
|
||||
}, []);
|
||||
|
||||
return {
|
||||
column_ref,
|
||||
content_ref: content_ref as RefObject<HTMLDivElement>,
|
||||
jump_visible,
|
||||
jump_to_bottom: jumpToBottom,
|
||||
on_scroll: onScroll,
|
||||
};
|
||||
}
|
||||
@@ -12,6 +12,7 @@ import GroupTimestamp from "./group-timestamp";
|
||||
import MessageActions from "./message-actions";
|
||||
import MessageReactions from "./message-reactions";
|
||||
import MessageReplyQuote from "./message-reply-quote";
|
||||
import MessageUsageChip from "./message-usage-chip";
|
||||
import NeedsYouChip from "./needs-you-chip";
|
||||
import SpeakMessageButton from "./speak-message-button";
|
||||
|
||||
@@ -74,6 +75,7 @@ function BotMessageLine({ message, bot_name, at }: Props) {
|
||||
<Row className="gap-3! items-center">
|
||||
{message.needs_you ? <NeedsYouChip /> : null}
|
||||
{at ? <GroupTimestamp at={at} /> : null}
|
||||
<MessageUsageChip usage={message.usage} />
|
||||
{can_speak ? (
|
||||
<SpeakMessageButton message_id={Number(message.id)} />
|
||||
) : null}
|
||||
|
||||
@@ -1,15 +1,17 @@
|
||||
import { useMemo, useRef, type RefObject } from "react";
|
||||
import { useMemo, type RefObject } from "react";
|
||||
import Stack from "@moduletrace/twui/layout/Stack";
|
||||
import type {
|
||||
BotMessageWithAttachments,
|
||||
BotWithLastMessage,
|
||||
} from "@/src/types";
|
||||
import useMessageSelection from "../(hooks)/use-message-selection";
|
||||
import useThreadScroll from "../(hooks)/use-thread-scroll";
|
||||
import groupMessages from "../(utils)/group-messages";
|
||||
import BotThinkingRow from "./bot-thinking-row";
|
||||
import MessageGroupBlock from "./message-group-block";
|
||||
import OlderMessagesLoader from "./older-messages-loader";
|
||||
import QuoteSelectionButton from "./quote-selection-button";
|
||||
import ScrollToBottomButton from "./scroll-to-bottom-button";
|
||||
|
||||
type Props = {
|
||||
bot: BotWithLastMessage;
|
||||
@@ -51,7 +53,9 @@ type Props = {
|
||||
* While the bot works on a reply or a task, a status row sits under the
|
||||
* last one. Text selected in a message can be quoted in the composer.
|
||||
* Older messages load as it scrolls to the top; the reversed column keeps
|
||||
* the view where it was as they land above it.
|
||||
* the view where it was as they land above it. The thread follows new
|
||||
* messages while it sits at its bottom, stays put while it's scrolled up
|
||||
* and then offers a button to get back down.
|
||||
*/
|
||||
export default function MessageThread({
|
||||
bot,
|
||||
@@ -64,42 +68,52 @@ export default function MessageThread({
|
||||
older,
|
||||
}: Props) {
|
||||
const groups = useMemo(() => groupMessages({ messages }), [messages]);
|
||||
const ref = useRef<HTMLDivElement>(null);
|
||||
const { selection, clear } = useMessageSelection({ ref });
|
||||
const scroll = useThreadScroll({ messages });
|
||||
const { selection, clear } = useMessageSelection({ ref: scroll.column_ref });
|
||||
|
||||
return (
|
||||
<Stack
|
||||
componentRef={ref as RefObject<HTMLDivElement>}
|
||||
className="grow min-h-0 gap-0! items-stretch overflow-y-auto flex-col-reverse">
|
||||
<Stack className="relative grow min-h-0 gap-0! items-stretch">
|
||||
<Stack
|
||||
role="log"
|
||||
aria-live="polite"
|
||||
aria-label={`Conversation with ${bot.name}`}
|
||||
className="w-full max-w-container mx-auto gap-4! items-stretch px-3 md:px-6 pt-6 pb-6"
|
||||
componentRef={scroll.column_ref as RefObject<HTMLDivElement>}
|
||||
className="grow min-h-0 gap-0! items-stretch overflow-y-auto flex-col-reverse"
|
||||
onScroll={scroll.on_scroll}
|
||||
>
|
||||
{older?.has_more ? (
|
||||
<OlderMessagesLoader
|
||||
loading={older.loading}
|
||||
error={older.error}
|
||||
onLoad={older.loadOlder}
|
||||
/>
|
||||
) : null}
|
||||
{groups.map((group) => (
|
||||
<MessageGroupBlock
|
||||
key={group.key}
|
||||
<Stack
|
||||
componentRef={scroll.content_ref as RefObject<HTMLDivElement>}
|
||||
role="log"
|
||||
aria-live="polite"
|
||||
aria-label={`Conversation with ${bot.name}`}
|
||||
className="w-full max-w-container mx-auto gap-4! items-stretch px-3 md:px-6 pt-6 pb-6"
|
||||
>
|
||||
{older?.has_more ? (
|
||||
<OlderMessagesLoader
|
||||
loading={older.loading}
|
||||
error={older.error}
|
||||
onLoad={older.loadOlder}
|
||||
/>
|
||||
) : null}
|
||||
{groups.map((group) => (
|
||||
<MessageGroupBlock
|
||||
key={group.key}
|
||||
bot={bot}
|
||||
group={group}
|
||||
/>
|
||||
))}
|
||||
<BotThinkingRow
|
||||
bot={bot}
|
||||
group={group}
|
||||
progress={progress}
|
||||
reply_progress={reply_progress}
|
||||
awaiting_reply={
|
||||
messages[messages.length - 1]?.author == "user"
|
||||
}
|
||||
receiving={receiving}
|
||||
has_task={has_task}
|
||||
/>
|
||||
))}
|
||||
<BotThinkingRow
|
||||
bot={bot}
|
||||
progress={progress}
|
||||
reply_progress={reply_progress}
|
||||
awaiting_reply={messages[messages.length - 1]?.author == "user"}
|
||||
receiving={receiving}
|
||||
has_task={has_task}
|
||||
/>
|
||||
</Stack>
|
||||
</Stack>
|
||||
{scroll.jump_visible ? (
|
||||
<ScrollToBottomButton onClick={scroll.jump_to_bottom} />
|
||||
) : null}
|
||||
{selection && onQuote ? (
|
||||
<QuoteSelectionButton
|
||||
selection={selection}
|
||||
|
||||
@@ -0,0 +1,35 @@
|
||||
import Span from "@moduletrace/twui/layout/Span";
|
||||
import type { TokenUsageSummary } from "@/src/harness/usage/types";
|
||||
import formatTokens from "@/src/utils/format-tokens";
|
||||
|
||||
type Props = {
|
||||
usage?: TokenUsageSummary | null;
|
||||
};
|
||||
|
||||
/**
|
||||
* What one answer cost, in the row under it beside its timestamp: the
|
||||
* tokens every model call behind it spent (a chat turn, reading an image,
|
||||
* making one), and the price when the provider reported it. The hover
|
||||
* title breaks it down.
|
||||
*/
|
||||
export default function MessageUsageChip({ usage }: Props) {
|
||||
if (!usage || usage.total_tokens <= 0) return null;
|
||||
|
||||
const { prompt_tokens, completion_tokens, total_tokens, cost, calls } = usage;
|
||||
const title = [
|
||||
`${formatTokens({ tokens: prompt_tokens })} in, ${formatTokens({ tokens: completion_tokens })} out`,
|
||||
`${calls} model ${calls == 1 ? "call" : "calls"}`,
|
||||
typeof cost == "number" && cost > 0 ? `$${cost.toFixed(4)}` : "",
|
||||
]
|
||||
.filter(Boolean)
|
||||
.join(" · ");
|
||||
|
||||
return (
|
||||
<Span
|
||||
title={title}
|
||||
className="text-[11px] leading-4 text-ink-secondary tabular-nums whitespace-nowrap cursor-help"
|
||||
>
|
||||
{formatTokens({ tokens: total_tokens })} tokens
|
||||
</Span>
|
||||
);
|
||||
}
|
||||
@@ -0,0 +1,28 @@
|
||||
import Button from "@moduletrace/twui/layout/Button";
|
||||
import { ArrowDown } from "lucide-react";
|
||||
|
||||
type Props = {
|
||||
onClick: () => void;
|
||||
};
|
||||
|
||||
/**
|
||||
* "Jump to latest", floating over the bottom of the thread while it sits
|
||||
* above it: a bot's reply that landed out of sight, or a long scroll back
|
||||
* through the conversation.
|
||||
*/
|
||||
export default function ScrollToBottomButton({ onClick }: Props) {
|
||||
return (
|
||||
<Button
|
||||
type="button"
|
||||
color="secondary"
|
||||
title="Jump to the latest message"
|
||||
aria-label="Jump to the latest message"
|
||||
variant="outlined"
|
||||
onClick={onClick}
|
||||
className="absolute bottom-3 left-1/2 z-20 shrink-0 -translate-x-1/2 gap-1.5! px-2.5! py-1.5! rounded-default-sm border border-solid border-hairline bg-surface shadow-popover text-xs font-bold text-ink hover:bg-sunken focus-visible:ring-2 focus-visible:ring-ink"
|
||||
>
|
||||
<ArrowDown size={13} aria-hidden />
|
||||
Jump to latest
|
||||
</Button>
|
||||
);
|
||||
}
|
||||
@@ -0,0 +1,98 @@
|
||||
import { describe, expect, test } from "bun:test";
|
||||
import {
|
||||
ThreadBottomSlack,
|
||||
shouldFollowThread,
|
||||
threadOffset,
|
||||
} from "./thread-scroll";
|
||||
|
||||
describe("threadOffset", () => {
|
||||
test("content whose bottom edge is the column's is at the bottom", () => {
|
||||
expect(
|
||||
threadOffset({ content_bottom: 800, column_bottom: 800 }),
|
||||
).toEqual({ distance: 0, near_bottom: true });
|
||||
});
|
||||
|
||||
test("content ending just short of the column's bottom is at the bottom", () => {
|
||||
expect(
|
||||
threadOffset({ content_bottom: 780, column_bottom: 800 }),
|
||||
).toEqual({ distance: 0, near_bottom: true });
|
||||
});
|
||||
|
||||
test("a small offset past the bottom still counts as being there", () => {
|
||||
const offset = threadOffset({
|
||||
content_bottom: 800 + ThreadBottomSlack - 1,
|
||||
column_bottom: 800,
|
||||
});
|
||||
|
||||
expect(offset.near_bottom).toBe(true);
|
||||
expect(offset.distance).toBe(ThreadBottomSlack - 1);
|
||||
});
|
||||
|
||||
test("one pixel past the slack is not the bottom anymore", () => {
|
||||
const offset = threadOffset({
|
||||
content_bottom: 800 + ThreadBottomSlack + 1,
|
||||
column_bottom: 800,
|
||||
});
|
||||
|
||||
expect(offset.near_bottom).toBe(false);
|
||||
expect(offset.distance).toBe(ThreadBottomSlack + 1);
|
||||
});
|
||||
|
||||
test("scrolled up by a long way is far from the bottom", () => {
|
||||
expect(
|
||||
threadOffset({ content_bottom: 4800, column_bottom: 800 }),
|
||||
).toEqual({ distance: 4000, near_bottom: false });
|
||||
});
|
||||
|
||||
test("the slack is configurable", () => {
|
||||
expect(
|
||||
threadOffset({
|
||||
content_bottom: 1000,
|
||||
column_bottom: 800,
|
||||
slack: 200,
|
||||
}).near_bottom,
|
||||
).toBe(true);
|
||||
});
|
||||
});
|
||||
|
||||
describe("shouldFollowThread", () => {
|
||||
test("a chat at the bottom follows a bot's reply", () => {
|
||||
expect(
|
||||
shouldFollowThread({
|
||||
near_bottom: true,
|
||||
newest_author: "bot",
|
||||
kept_last_message: true,
|
||||
}),
|
||||
).toBe(true);
|
||||
});
|
||||
|
||||
test("a chat scrolled up stays put when a bot's reply lands", () => {
|
||||
expect(
|
||||
shouldFollowThread({
|
||||
near_bottom: false,
|
||||
newest_author: "bot",
|
||||
kept_last_message: true,
|
||||
}),
|
||||
).toBe(false);
|
||||
});
|
||||
|
||||
test("a message the user sent always brings the thread to it", () => {
|
||||
expect(
|
||||
shouldFollowThread({
|
||||
near_bottom: false,
|
||||
newest_author: "user",
|
||||
kept_last_message: true,
|
||||
}),
|
||||
).toBe(true);
|
||||
});
|
||||
|
||||
test("a list that replaced the old one opens at its bottom", () => {
|
||||
expect(
|
||||
shouldFollowThread({
|
||||
near_bottom: false,
|
||||
newest_author: "bot",
|
||||
kept_last_message: false,
|
||||
}),
|
||||
).toBe(true);
|
||||
});
|
||||
});
|
||||
@@ -0,0 +1,75 @@
|
||||
/**
|
||||
* Pixels from the bottom that still count as being at it, so a mouse wheel
|
||||
* tick or a rounding difference doesn't hide the latest message.
|
||||
*/
|
||||
export const ThreadBottomSlack = 48;
|
||||
|
||||
/**
|
||||
* How long a smooth jump to the bottom runs, in milliseconds. The scrolling
|
||||
* it causes is ignored for that long (so the button doesn't flicker back
|
||||
* mid-animation), and the window ends on its own however it lands.
|
||||
*/
|
||||
export const ThreadJumpWindowMs = 800;
|
||||
|
||||
/**
|
||||
* How far the thread's content sits below the bottom of its column, in
|
||||
* pixels, and whether that still counts as being at the bottom. It reads
|
||||
* the two rects rather than `scrollTop`, whose sign a reversed column
|
||||
* (`flex-col-reverse`) reports differently across engines: the content's
|
||||
* bottom edge is the column's when the thread is at its bottom, and past it
|
||||
* by however far the thread has been scrolled up.
|
||||
*/
|
||||
export function threadOffset({
|
||||
content_bottom,
|
||||
column_bottom,
|
||||
slack = ThreadBottomSlack,
|
||||
}: {
|
||||
content_bottom: number;
|
||||
column_bottom: number;
|
||||
slack?: number;
|
||||
}) {
|
||||
const distance = content_bottom - column_bottom;
|
||||
const at_bottom = distance <= slack;
|
||||
|
||||
return {
|
||||
distance: at_bottom && distance < 0 ? 0 : distance,
|
||||
near_bottom: at_bottom,
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* Whether the thread should show a change of its messages, which keeps the
|
||||
* bottom in view while the user is there: it follows while at the bottom, a
|
||||
* message the user sent always counts, and a list that replaced the old one
|
||||
* (a new session, a first render) opens at its bottom. A chat scrolled up
|
||||
* stays where it is when a bot answers.
|
||||
*/
|
||||
export function shouldFollowThread({
|
||||
near_bottom,
|
||||
newest_author,
|
||||
kept_last_message,
|
||||
}: {
|
||||
near_bottom: boolean;
|
||||
newest_author?: string | null;
|
||||
kept_last_message: boolean;
|
||||
}) {
|
||||
if (!kept_last_message) return true;
|
||||
if (newest_author == "user") return true;
|
||||
|
||||
return near_bottom;
|
||||
}
|
||||
|
||||
/**
|
||||
* Puts the thread at its bottom, animated or at once. A reversed column
|
||||
* clamps a past-the-end `scrollTop` back onto its bottom, so the same call
|
||||
* serves either direction.
|
||||
*/
|
||||
export function scrollThreadToBottom({
|
||||
element,
|
||||
behavior = "auto",
|
||||
}: {
|
||||
element: HTMLElement;
|
||||
behavior?: ScrollBehavior;
|
||||
}) {
|
||||
element.scrollTo({ top: element.scrollHeight, behavior });
|
||||
}
|
||||
@@ -0,0 +1,37 @@
|
||||
/**
|
||||
* The heights a day's bar can take, shortest first. A day's tokens are
|
||||
* placed on this scale rather than given a height in pixels, so every bar
|
||||
* is a static utility class the stylesheet already carries: no inline
|
||||
* style, and the theme still decides the colour.
|
||||
*/
|
||||
export const UsageBarHeights = [
|
||||
"h-0.5",
|
||||
"h-1",
|
||||
"h-1.5",
|
||||
"h-2",
|
||||
"h-2.5",
|
||||
"h-3",
|
||||
"h-4",
|
||||
"h-5",
|
||||
"h-6",
|
||||
"h-8",
|
||||
"h-10",
|
||||
"h-12",
|
||||
"h-14",
|
||||
"h-16",
|
||||
"h-20",
|
||||
"h-24",
|
||||
];
|
||||
|
||||
/**
|
||||
* Where a value sits on that scale, against the window's busiest day. A
|
||||
* day with work on it is never shorter than one step, so it is visible.
|
||||
*/
|
||||
export function usageBarHeight({ value, max }: { value: number; max: number }) {
|
||||
if (value <= 0) return UsageBarHeights[0]!;
|
||||
|
||||
const ratio = max > 0 ? value / max : 0;
|
||||
const step = Math.max(1, Math.round(ratio * (UsageBarHeights.length - 1)));
|
||||
|
||||
return UsageBarHeights[Math.min(step, UsageBarHeights.length - 1)]!;
|
||||
}
|
||||
@@ -0,0 +1,12 @@
|
||||
/**
|
||||
* What each harness is called on the usage page, in words rather than the
|
||||
* harness's own key.
|
||||
*/
|
||||
export const UsageHarnessTitles: Record<string, string> = {
|
||||
text: "Chat and tasks",
|
||||
vision: "Reading media",
|
||||
transcription: "Transcription",
|
||||
speech: "Speech",
|
||||
image: "Image generation",
|
||||
embedding: "Embeddings",
|
||||
};
|
||||
@@ -0,0 +1,27 @@
|
||||
export type UsageRange = {
|
||||
/**
|
||||
* The days back the range totals, today included.
|
||||
*/
|
||||
days: number;
|
||||
title: string;
|
||||
};
|
||||
|
||||
/**
|
||||
* The periods the usage page offers, shortest first. A week and a month
|
||||
* are what most questions need; a quarter is for spotting a trend.
|
||||
*/
|
||||
export const UsageRanges: UsageRange[] = [
|
||||
{ days: 7, title: "7 days" },
|
||||
{ days: 30, title: "30 days" },
|
||||
{ days: 90, title: "90 days" },
|
||||
];
|
||||
|
||||
/**
|
||||
* The range asked for in the address, or the page's default when it isn't
|
||||
* one of the offered ones.
|
||||
*/
|
||||
export function parseUsageRange({ value }: { value?: string | null }): UsageRange | null {
|
||||
const days = Number(value);
|
||||
|
||||
return UsageRanges.find((range) => range.days == days) || null;
|
||||
}
|
||||
@@ -0,0 +1,35 @@
|
||||
import type { PagePropsType } from "@/src/types";
|
||||
import type { BunextPageServerFn } from "@moduletrace/bunext/types";
|
||||
import { AdminHomePath } from "@/src/dict/admin-sections-dict";
|
||||
import userAuth from "@/src/functions/backend/auth/user-auth";
|
||||
import grabUsageOverview from "@/src/functions/backend/db/usage/grab-usage-overview";
|
||||
import canManageCrew from "@/src/utils/rbac/can-manage-crew";
|
||||
import { parseUsageRange } from "../(data)/usage-ranges";
|
||||
|
||||
type Params = Parameters<BunextPageServerFn<PagePropsType>>[0];
|
||||
|
||||
/**
|
||||
* The usage page's data for the period in the address (the default when
|
||||
* it carries none). Crew managers only, since it is the whole install's
|
||||
* work rather than one user's conversation; everyone else lands on the
|
||||
* bots.
|
||||
*/
|
||||
export default async function grabUsageServerProps(params: Params) {
|
||||
const { user, user_types } = await userAuth({ req: params.req });
|
||||
|
||||
if (!user?.id) {
|
||||
return { redirect: { destination: "/auth/login" } };
|
||||
}
|
||||
|
||||
if (!canManageCrew({ user_types })) {
|
||||
return { redirect: { destination: AdminHomePath } };
|
||||
}
|
||||
|
||||
const range = parseUsageRange({ value: params.query?.range });
|
||||
|
||||
return {
|
||||
props: {
|
||||
usage: await grabUsageOverview({ days: range?.days }).catch(() => null),
|
||||
},
|
||||
};
|
||||
}
|
||||
@@ -0,0 +1,13 @@
|
||||
import type { TokenUsageSummary } from "@/src/harness/usage/types";
|
||||
|
||||
type Params = {
|
||||
rows: TokenUsageSummary[];
|
||||
};
|
||||
|
||||
/**
|
||||
* The tokens every row of a breakdown spent together, which is the
|
||||
* denominator of its shares.
|
||||
*/
|
||||
export default function sumTokens({ rows }: Params) {
|
||||
return rows.reduce((total, row) => total + row.total_tokens, 0);
|
||||
}
|
||||
@@ -0,0 +1,15 @@
|
||||
import { UsagePath } from "@/src/dict/rail-pages-dict";
|
||||
|
||||
type Params = {
|
||||
days?: number;
|
||||
};
|
||||
|
||||
/**
|
||||
* The usage page's address for the period it is showing, so a range can be
|
||||
* linked to and the browser's own back button walks back through them.
|
||||
*/
|
||||
export default function usageHref({ days }: Params): string {
|
||||
if (!days) return UsagePath;
|
||||
|
||||
return `${UsagePath}?range=${days}`;
|
||||
}
|
||||
@@ -0,0 +1,63 @@
|
||||
import fetchApi from "@moduletrace/twui/utils/fetch/fetchApi";
|
||||
import type { APIResponseObject } from "@moduletrace/bunext/types";
|
||||
import { useEffect, useState } from "react";
|
||||
import type { UsageOverview } from "@/src/types";
|
||||
|
||||
type Params = {
|
||||
days: number;
|
||||
/**
|
||||
* The totals the page already carries, when it covers this period.
|
||||
*/
|
||||
page_usage?: UsageOverview | null;
|
||||
};
|
||||
|
||||
/**
|
||||
* The usage totals for one period. The page's own copy is used when it
|
||||
* covers that period; another one loads from the API, so switching the
|
||||
* range doesn't reload the page.
|
||||
*/
|
||||
export default function useUsageOverview({ days, page_usage }: Params) {
|
||||
const from_page = Boolean(page_usage && page_usage.window.days == days);
|
||||
const [usage, set_usage] = useState<UsageOverview | null>(from_page ? page_usage || null : null);
|
||||
const [loading, set_loading] = useState(!from_page);
|
||||
const [error, set_error] = useState<string>();
|
||||
|
||||
useEffect(() => {
|
||||
if (from_page) {
|
||||
set_usage(page_usage || null);
|
||||
set_loading(false);
|
||||
set_error(undefined);
|
||||
return;
|
||||
}
|
||||
|
||||
const controller = new AbortController();
|
||||
|
||||
const load = async () => {
|
||||
set_loading(true);
|
||||
set_error(undefined);
|
||||
|
||||
const res = await fetchApi<{}, APIResponseObject<UsageOverview>>(
|
||||
`/api/admin/usage?days=${days}`,
|
||||
{ method: "GET" },
|
||||
).catch(() => null);
|
||||
|
||||
if (controller.signal.aborted) return;
|
||||
|
||||
set_loading(false);
|
||||
|
||||
if (!res?.success || !res.singleRes) {
|
||||
set_usage(null);
|
||||
set_error(res?.msg || `Couldn't read the usage.`);
|
||||
return;
|
||||
}
|
||||
|
||||
set_usage(res.singleRes);
|
||||
};
|
||||
|
||||
load().catch(() => {});
|
||||
|
||||
return () => controller.abort();
|
||||
}, [days, from_page, page_usage]);
|
||||
|
||||
return { usage, loading, error };
|
||||
}
|
||||
@@ -0,0 +1,39 @@
|
||||
import Stack from "@moduletrace/twui/layout/Stack";
|
||||
import { twMerge } from "tailwind-merge";
|
||||
import dayjs from "dayjs";
|
||||
import type { UsageOverview } from "@/src/types";
|
||||
import formatTokens from "@/src/utils/format-tokens";
|
||||
import { usageBarHeight } from "../(data)/usage-bar-heights";
|
||||
|
||||
type Props = {
|
||||
day: UsageOverview["days"][number];
|
||||
max: number;
|
||||
};
|
||||
|
||||
/**
|
||||
* One day's bar in the chart: its tokens above nothing, the bar itself,
|
||||
* and the title spells the day out with what it spent. Hovering it shows
|
||||
* that title.
|
||||
*/
|
||||
export default function UsageDayBar({ day, max }: Props) {
|
||||
const title = `${dayjs(day.at).format("ddd, MMM D")}: ${formatTokens({ tokens: day.total_tokens })} tokens, ${daysCalls({ calls: day.calls })}`;
|
||||
|
||||
return (
|
||||
<Stack
|
||||
title={title}
|
||||
className="grow basis-[3px] min-w-[3px] max-w-10 h-24 justify-end items-stretch gap-0!"
|
||||
>
|
||||
<Stack
|
||||
className={twMerge(
|
||||
"w-full rounded-t-[2px]",
|
||||
day.total_tokens ? "bg-ink" : "bg-hairline",
|
||||
usageBarHeight({ value: day.total_tokens, max }),
|
||||
)}
|
||||
/>
|
||||
</Stack>
|
||||
);
|
||||
}
|
||||
|
||||
function daysCalls({ calls }: { calls: number }) {
|
||||
return `${calls} ${calls == 1 ? "call" : "calls"}`;
|
||||
}
|
||||
@@ -0,0 +1,51 @@
|
||||
import Row from "@moduletrace/twui/layout/Row";
|
||||
import Span from "@moduletrace/twui/layout/Span";
|
||||
import Stack from "@moduletrace/twui/layout/Stack";
|
||||
import dayjs from "dayjs";
|
||||
import type { UsageOverview } from "@/src/types";
|
||||
import formatTokens from "@/src/utils/format-tokens";
|
||||
import SettingsGroup from "@/src/pages/admin/settings/(partials)/settings-group";
|
||||
import UsageDayBar from "./usage-day-bar";
|
||||
|
||||
type Props = {
|
||||
days: UsageOverview["days"];
|
||||
};
|
||||
|
||||
/**
|
||||
* The period day by day, one bar a day, so a spike or a quiet stretch is
|
||||
* read at a glance. The busiest day's total labels the top.
|
||||
*/
|
||||
export default function UsageDayChart({ days }: Props) {
|
||||
const max = days.reduce((highest, day) => Math.max(highest, day.total_tokens), 0);
|
||||
const first = days[0];
|
||||
const last = days[days.length - 1];
|
||||
|
||||
return (
|
||||
<SettingsGroup
|
||||
title="Day by day"
|
||||
description={
|
||||
max
|
||||
? `Busiest day: ${formatTokens({ tokens: max })} tokens.`
|
||||
: "Nothing spent in this period yet."
|
||||
}
|
||||
>
|
||||
<Stack className="w-full gap-2!">
|
||||
<Row className="w-full gap-1! items-end">
|
||||
{days.map((day) => (
|
||||
<UsageDayBar day={day} max={max} key={day.at} />
|
||||
))}
|
||||
</Row>
|
||||
{first && last ? (
|
||||
<Row className="w-full gap-2! items-center justify-between">
|
||||
<Span className="text-[11px] leading-4 text-ink-secondary">
|
||||
{dayjs(first.at).format("MMM D")}
|
||||
</Span>
|
||||
<Span className="text-[11px] leading-4 text-ink-secondary">
|
||||
{dayjs(last.at).format("MMM D")}
|
||||
</Span>
|
||||
</Row>
|
||||
) : null}
|
||||
</Stack>
|
||||
</SettingsGroup>
|
||||
);
|
||||
}
|
||||
@@ -0,0 +1,55 @@
|
||||
import Center from "@moduletrace/twui/layout/Center";
|
||||
import Row from "@moduletrace/twui/layout/Row";
|
||||
import Span from "@moduletrace/twui/layout/Span";
|
||||
import Stack from "@moduletrace/twui/layout/Stack";
|
||||
import type { UsageOverview } from "@/src/types";
|
||||
import SettingsGroup from "@/src/pages/admin/settings/(partials)/settings-group";
|
||||
import sumTokens from "../(functions)/sum-tokens";
|
||||
import UsageHarnessRow from "./usage-harness-row";
|
||||
|
||||
type Props = {
|
||||
rows: UsageOverview["by_harness"];
|
||||
};
|
||||
|
||||
/**
|
||||
* What each kind of model call spent: a chat turn, reading an image,
|
||||
* transcribing, speaking, making a picture, embedding a file. Their totals
|
||||
* add up to the period's.
|
||||
*/
|
||||
export default function UsageHarnessBreakdown({ rows }: Props) {
|
||||
if (!rows.length) {
|
||||
return (
|
||||
<SettingsGroup title="By kind of call" description="No model calls in this period yet.">
|
||||
<Center className="p-8">
|
||||
<Span size="small">Nothing to break down.</Span>
|
||||
</Center>
|
||||
</SettingsGroup>
|
||||
);
|
||||
}
|
||||
|
||||
const total_tokens = sumTokens({ rows });
|
||||
|
||||
return (
|
||||
<SettingsGroup title="By kind of call" description="Where the period's tokens went.">
|
||||
<Stack className="w-full gap-0!">
|
||||
<Row className="w-full pb-2 gap-3! items-center border-0 border-b border-solid border-hairline">
|
||||
<Span className="text-[11px] uppercase tracking-wide text-ink-secondary grow">
|
||||
Kind
|
||||
</Span>
|
||||
<Span className="text-[11px] uppercase tracking-wide text-ink-secondary w-10 text-right">
|
||||
Share
|
||||
</Span>
|
||||
<Span className="text-[11px] uppercase tracking-wide text-ink-secondary w-20 text-right">
|
||||
Tokens
|
||||
</Span>
|
||||
<Span className="text-[11px] uppercase tracking-wide text-ink-secondary w-16 text-right">
|
||||
Cost
|
||||
</Span>
|
||||
</Row>
|
||||
{rows.map((row) => (
|
||||
<UsageHarnessRow row={row} total_tokens={total_tokens} key={row.harness} />
|
||||
))}
|
||||
</Stack>
|
||||
</SettingsGroup>
|
||||
);
|
||||
}
|
||||
@@ -0,0 +1,38 @@
|
||||
import Row from "@moduletrace/twui/layout/Row";
|
||||
import Span from "@moduletrace/twui/layout/Span";
|
||||
import type { UsageOverview } from "@/src/types";
|
||||
import formatTokens from "@/src/utils/format-tokens";
|
||||
import { UsageHarnessTitles } from "../(data)/usage-harness-titles";
|
||||
|
||||
type Props = {
|
||||
row: UsageOverview["by_harness"][number];
|
||||
/**
|
||||
* Every harness's tokens in the period, for the share.
|
||||
*/
|
||||
total_tokens: number;
|
||||
};
|
||||
|
||||
/**
|
||||
* One harness's line in the breakdown: what it is, its share of the
|
||||
* period's tokens, and what it spent.
|
||||
*/
|
||||
export default function UsageHarnessRow({ row, total_tokens }: Props) {
|
||||
const share = total_tokens ? Math.round((row.total_tokens / total_tokens) * 100) : 0;
|
||||
|
||||
return (
|
||||
<Row className="w-full py-2 gap-3! items-center border-0 border-b border-solid border-hairline last:border-b-0">
|
||||
<Span className="text-[13px] leading-5 text-ink grow min-w-0 truncate">
|
||||
{UsageHarnessTitles[row.harness] || row.harness}
|
||||
</Span>
|
||||
<Span className="text-[12px] leading-4 text-ink-secondary tabular-nums w-10 text-right">
|
||||
{share}%
|
||||
</Span>
|
||||
<Span className="text-[13px] leading-5 text-ink tabular-nums w-20 text-right">
|
||||
{formatTokens({ tokens: row.total_tokens })}
|
||||
</Span>
|
||||
<Span className="text-[12px] leading-4 text-ink-secondary tabular-nums w-16 text-right">
|
||||
{row.cost ? `$${row.cost.toFixed(2)}` : "-"}
|
||||
</Span>
|
||||
</Row>
|
||||
);
|
||||
}
|
||||
@@ -0,0 +1,23 @@
|
||||
import Span from "@moduletrace/twui/layout/Span";
|
||||
import { twMerge } from "tailwind-merge";
|
||||
|
||||
type Props = {
|
||||
title: string;
|
||||
className?: string;
|
||||
};
|
||||
|
||||
/**
|
||||
* A column heading of the models table, right-aligned over its figures.
|
||||
*/
|
||||
export default function UsageModelHeadCell({ title, className }: Props) {
|
||||
return (
|
||||
<Span
|
||||
className={twMerge(
|
||||
"text-[11px] uppercase tracking-wide text-ink-secondary text-right shrink-0",
|
||||
className,
|
||||
)}
|
||||
>
|
||||
{title}
|
||||
</Span>
|
||||
);
|
||||
}
|
||||
@@ -0,0 +1,52 @@
|
||||
import Row from "@moduletrace/twui/layout/Row";
|
||||
import Span from "@moduletrace/twui/layout/Span";
|
||||
import Stack from "@moduletrace/twui/layout/Stack";
|
||||
import type { UsageOverview } from "@/src/types";
|
||||
import formatTokens from "@/src/utils/format-tokens";
|
||||
import { UsageHarnessTitles } from "../(data)/usage-harness-titles";
|
||||
|
||||
type Props = {
|
||||
row: UsageOverview["models"][number];
|
||||
};
|
||||
|
||||
/**
|
||||
* One model's line: which provider and model it was, what it was used for,
|
||||
* how many calls and tokens it took, and the price of a thousand tokens
|
||||
* worked out from both, which is what makes models comparable.
|
||||
*/
|
||||
export default function UsageModelRow({ row }: Props) {
|
||||
const per_thousand = row.cost && row.total_tokens
|
||||
? `$${((row.cost / row.total_tokens) * 1000).toFixed(3)}`
|
||||
: "-";
|
||||
|
||||
return (
|
||||
<Row className="w-full py-2.5 gap-3! items-center border-0 border-b border-solid border-hairline last:border-b-0">
|
||||
<Stack className="grow min-w-0 gap-0!">
|
||||
<Span className="text-[13px] leading-5 text-ink truncate" title={row.model_slug}>
|
||||
{row.model_name || row.model_slug}
|
||||
</Span>
|
||||
<Span className="text-[11px] leading-4 text-ink-secondary truncate">
|
||||
{row.provider_name} - {UsageHarnessTitles[row.harness] || row.harness}
|
||||
</Span>
|
||||
</Stack>
|
||||
<Span className="text-[12px] leading-4 text-ink-secondary tabular-nums w-14 text-right">
|
||||
{formatTokens({ tokens: row.calls })}
|
||||
</Span>
|
||||
<Span className="text-[12px] leading-4 text-ink-secondary tabular-nums w-16 text-right">
|
||||
{formatTokens({ tokens: row.prompt_tokens })}
|
||||
</Span>
|
||||
<Span className="text-[12px] leading-4 text-ink-secondary tabular-nums w-16 text-right">
|
||||
{formatTokens({ tokens: row.completion_tokens })}
|
||||
</Span>
|
||||
<Span className="text-[13px] leading-5 text-ink tabular-nums w-16 text-right">
|
||||
{formatTokens({ tokens: row.total_tokens })}
|
||||
</Span>
|
||||
<Span className="text-[12px] leading-4 text-ink-secondary tabular-nums w-16 text-right">
|
||||
{per_thousand}
|
||||
</Span>
|
||||
<Span className="text-[12px] leading-4 text-ink-secondary tabular-nums w-14 text-right">
|
||||
{row.cost ? `$${row.cost.toFixed(2)}` : "-"}
|
||||
</Span>
|
||||
</Row>
|
||||
);
|
||||
}
|
||||
@@ -0,0 +1,55 @@
|
||||
import Center from "@moduletrace/twui/layout/Center";
|
||||
import Row from "@moduletrace/twui/layout/Row";
|
||||
import Span from "@moduletrace/twui/layout/Span";
|
||||
import Stack from "@moduletrace/twui/layout/Stack";
|
||||
import type { UsageOverview } from "@/src/types";
|
||||
import SettingsGroup from "@/src/pages/admin/settings/(partials)/settings-group";
|
||||
import UsageModelHeadCell from "./usage-model-head-cell";
|
||||
import UsageModelRow from "./usage-model-row";
|
||||
|
||||
type Props = {
|
||||
rows: UsageOverview["models"];
|
||||
days: number;
|
||||
};
|
||||
|
||||
/**
|
||||
* Every model the period ran, busiest first. A row's price per thousand
|
||||
* tokens is what makes one model's cost comparable with another's, and
|
||||
* how many calls each took says which one earned it.
|
||||
*/
|
||||
export default function UsageModelsTable({ rows, days }: Props) {
|
||||
if (!rows.length) {
|
||||
return (
|
||||
<SettingsGroup title="By model" description="No model calls in this period yet.">
|
||||
<Center className="p-8">
|
||||
<Span size="small">Nothing to list.</Span>
|
||||
</Center>
|
||||
</SettingsGroup>
|
||||
);
|
||||
}
|
||||
|
||||
return (
|
||||
<SettingsGroup
|
||||
title="By model"
|
||||
description={`The models this install ran in the last ${days} days, busiest first. The price per thousand tokens is read from what the provider reported.`}
|
||||
>
|
||||
<Stack className="w-full gap-0! overflow-x-auto">
|
||||
<Row className="w-full pb-2 gap-3! items-center border-0 border-b border-solid border-hairline min-w-[640px]">
|
||||
<UsageModelHeadCell className="grow text-left" title="Model" />
|
||||
<UsageModelHeadCell className="w-14" title="Calls" />
|
||||
<UsageModelHeadCell className="w-16" title="In" />
|
||||
<UsageModelHeadCell className="w-16" title="Out" />
|
||||
<UsageModelHeadCell className="w-16" title="Tokens" />
|
||||
<UsageModelHeadCell className="w-16" title="Per 1K" />
|
||||
<UsageModelHeadCell className="w-14" title="Cost" />
|
||||
</Row>
|
||||
{rows.map((row) => (
|
||||
<UsageModelRow
|
||||
row={row}
|
||||
key={`${row.provider_name}-${row.model_slug}-${row.harness}`}
|
||||
/>
|
||||
))}
|
||||
</Stack>
|
||||
</SettingsGroup>
|
||||
);
|
||||
}
|
||||
@@ -0,0 +1,35 @@
|
||||
import Button from "@moduletrace/twui/layout/Button";
|
||||
import Row from "@moduletrace/twui/layout/Row";
|
||||
import { UsageRanges } from "../(data)/usage-ranges";
|
||||
|
||||
type Props = {
|
||||
days: number;
|
||||
loading?: boolean;
|
||||
onChange: (days: number) => void;
|
||||
};
|
||||
|
||||
/**
|
||||
* The periods the page totals, one button each. The one showing is the
|
||||
* one that reads as pressed.
|
||||
*/
|
||||
export default function UsageRangePicker({ days, loading, onChange }: Props) {
|
||||
return (
|
||||
<Row className="gap-1! items-center">
|
||||
{UsageRanges.map((range) => (
|
||||
<Button
|
||||
key={range.days}
|
||||
title={range.title}
|
||||
type="button"
|
||||
size="small"
|
||||
variant={range.days == days ? "outlined" : "ghost"}
|
||||
color="text"
|
||||
disabled={loading}
|
||||
onClick={() => onChange(range.days)}
|
||||
className="px-2.5! py-1! text-[12px]! rounded-default-xs"
|
||||
>
|
||||
{range.title}
|
||||
</Button>
|
||||
))}
|
||||
</Row>
|
||||
);
|
||||
}
|
||||
@@ -0,0 +1,27 @@
|
||||
import Span from "@moduletrace/twui/layout/Span";
|
||||
import Stack from "@moduletrace/twui/layout/Stack";
|
||||
|
||||
type Props = {
|
||||
label: string;
|
||||
value: string;
|
||||
/**
|
||||
* A line under the figure, e.g. the comparison with an earlier period.
|
||||
*/
|
||||
hint?: string;
|
||||
};
|
||||
|
||||
/**
|
||||
* One figure of the page's totals: what it is above, the number itself
|
||||
* big, and a quiet line under it.
|
||||
*/
|
||||
export default function UsageStat({ label, value, hint }: Props) {
|
||||
return (
|
||||
<Stack className="grow basis-[160px] min-w-0 gap-1!">
|
||||
<Span className="text-[11px] uppercase tracking-wide text-ink-secondary">{label}</Span>
|
||||
<Span className="text-2xl leading-8 font-bold text-ink tabular-nums">{value}</Span>
|
||||
{hint ? (
|
||||
<Span className="text-[12px] leading-4 text-ink-secondary">{hint}</Span>
|
||||
) : null}
|
||||
</Stack>
|
||||
);
|
||||
}
|
||||
@@ -0,0 +1,58 @@
|
||||
import Row from "@moduletrace/twui/layout/Row";
|
||||
import type { UsageOverview } from "@/src/types";
|
||||
import formatTokens from "@/src/utils/format-tokens";
|
||||
import { UsageHarnessTitles } from "../(data)/usage-harness-titles";
|
||||
import UsageStat from "./usage-stat";
|
||||
|
||||
type Props = {
|
||||
usage: UsageOverview;
|
||||
};
|
||||
|
||||
/**
|
||||
* The page's headline figures for the period: what it spent, how many
|
||||
* calls that took, the price when providers report it, the busiest
|
||||
* harness, and how the total compares with the period before it.
|
||||
*/
|
||||
export default function UsageTotals({ usage }: Props) {
|
||||
const { totals, previous, by_harness } = usage;
|
||||
const busiest = by_harness[0];
|
||||
|
||||
return (
|
||||
<Row className="w-full py-8 gap-6! items-start flex-wrap border-0 border-b border-solid border-hairline">
|
||||
<UsageStat
|
||||
label="Tokens"
|
||||
value={formatTokens({ tokens: totals.total_tokens })}
|
||||
hint={`${formatTokens({ tokens: totals.prompt_tokens })} in, ${formatTokens({ tokens: totals.completion_tokens })} out`}
|
||||
/>
|
||||
<UsageStat
|
||||
label="Model calls"
|
||||
value={formatTokens({ tokens: totals.calls })}
|
||||
hint={busiest ? `${UsageHarnessTitles[busiest.harness] || busiest.harness} most of it` : undefined}
|
||||
/>
|
||||
<UsageStat
|
||||
label="Cost"
|
||||
value={totals.cost ? `$${totals.cost.toFixed(2)}` : "Not reported"}
|
||||
hint={totals.cost ? "As the providers reported it" : "Free models, or no price"}
|
||||
/>
|
||||
<UsageStat
|
||||
label="Previous period"
|
||||
value={changeLabel({ current: totals.total_tokens, previous: previous.total_tokens })}
|
||||
hint={`${formatTokens({ tokens: previous.total_tokens })} then`}
|
||||
/>
|
||||
</Row>
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* How this period compares with the one before it: a share up or down, or
|
||||
* that there is nothing to compare with yet.
|
||||
*/
|
||||
function changeLabel({ current, previous }: { current: number; previous: number }) {
|
||||
if (!previous) return current ? "New" : "Nothing yet";
|
||||
|
||||
const change = Math.round(((current - previous) / previous) * 100);
|
||||
|
||||
if (change == 0) return "Level";
|
||||
|
||||
return `${change > 0 ? "+" : ""}${change}%`;
|
||||
}
|
||||
@@ -0,0 +1,59 @@
|
||||
import Center from "@moduletrace/twui/layout/Center";
|
||||
import Row from "@moduletrace/twui/layout/Row";
|
||||
import Span from "@moduletrace/twui/layout/Span";
|
||||
import { useContext, useState } from "react";
|
||||
import { AppContext } from "@/src/pages/__root";
|
||||
import { UsageRanges } from "../(data)/usage-ranges";
|
||||
import usageHref from "../(functions)/usage-href";
|
||||
import useUsageOverview from "../(hooks)/use-usage-overview";
|
||||
import UsageDayChart from "../(partials)/usage-day-chart";
|
||||
import UsageHarnessBreakdown from "../(partials)/usage-harness-breakdown";
|
||||
import UsageModelsTable from "../(partials)/usage-models-table";
|
||||
import UsageRangePicker from "../(partials)/usage-range-picker";
|
||||
import UsageTotals from "../(partials)/usage-totals";
|
||||
|
||||
/**
|
||||
* The usage page's state: which period is totalled, kept in the address so
|
||||
* a range can be linked to. The default period comes with the page;
|
||||
* another one loads through the API.
|
||||
*/
|
||||
export default function UsageSection() {
|
||||
const { pageProps } = useContext(AppContext);
|
||||
const page_usage = pageProps?.usage;
|
||||
const [days, set_days] = useState(page_usage?.window.days || UsageRanges[1]!.days);
|
||||
const { usage, loading, error } = useUsageOverview({ days, page_usage });
|
||||
|
||||
const changeRange = (next_days: number) => {
|
||||
set_days(next_days);
|
||||
|
||||
const url = new URL(usageHref({ days: next_days }), window.location.origin);
|
||||
|
||||
history.pushState(null, "", url);
|
||||
};
|
||||
|
||||
return (
|
||||
<>
|
||||
<Row className="w-full pt-6 gap-4! items-center justify-between flex-wrap">
|
||||
<Span className="text-[13px] leading-[18px] text-ink-secondary">
|
||||
Every model call the crew made, whether it answered in chat, read an image,
|
||||
spoke or made one.
|
||||
</Span>
|
||||
<UsageRangePicker days={days} loading={loading} onChange={changeRange} />
|
||||
</Row>
|
||||
{error ? (
|
||||
<Center className="p-16">
|
||||
<Span size="small">{error}</Span>
|
||||
</Center>
|
||||
) : !usage ? (
|
||||
<Center className="p-16">
|
||||
<Span size="small">Reading the usage...</Span>
|
||||
</Center>
|
||||
) : (
|
||||
<UsageTotals usage={usage} />
|
||||
)}
|
||||
{usage ? <UsageDayChart days={usage.days} /> : null}
|
||||
{usage ? <UsageHarnessBreakdown rows={usage.by_harness} /> : null}
|
||||
{usage ? <UsageModelsTable rows={usage.models} days={usage.window.days} /> : null}
|
||||
</>
|
||||
);
|
||||
}
|
||||
@@ -0,0 +1,7 @@
|
||||
import type { PagePropsType } from "@/src/types";
|
||||
import type { BunextPageServerFn } from "@moduletrace/bunext/types";
|
||||
import grabUsageServerProps from "./(functions)/grab-usage-server-props";
|
||||
|
||||
export const server: BunextPageServerFn<PagePropsType> = async (params) => {
|
||||
return await grabUsageServerProps(params);
|
||||
};
|
||||
@@ -0,0 +1,23 @@
|
||||
import type { BunextPageModuleMetaFn } from "@moduletrace/bunext/types";
|
||||
import { SiteData } from "@/src/data/site-data";
|
||||
import SettingsShell from "@/src/pages/admin/settings/(partials)/settings-shell";
|
||||
import UsageSection from "./(sections)/usage-section";
|
||||
|
||||
export default function AdminUsagePage() {
|
||||
return (
|
||||
<SettingsShell
|
||||
title="Usage"
|
||||
description="What the crew's model calls spent: in total, day by day, and per model, so a model's cost and the work of a day are both readable."
|
||||
wide
|
||||
>
|
||||
<UsageSection />
|
||||
</SettingsShell>
|
||||
);
|
||||
}
|
||||
|
||||
export const meta: BunextPageModuleMetaFn = async () => {
|
||||
return {
|
||||
title: `Usage | ${SiteData["SiteName"]}`,
|
||||
description: "What the crew's model calls spent.",
|
||||
};
|
||||
};
|
||||
@@ -1,6 +1,7 @@
|
||||
import userAuth from "@/src/functions/backend/auth/user-auth";
|
||||
import grabSpeakableMessage from "@/src/functions/backend/db/bots/grab-speakable-message";
|
||||
import speakText from "@/src/harness/audio/speak-text";
|
||||
import withUsageScope from "@/src/harness/usage/with-usage-scope";
|
||||
import HarnessError from "@/src/harness/harness-error";
|
||||
import { SpeakBotMessageRequestSchema } from "@/src/schemas/bot-schema";
|
||||
import AppError from "@/src/utils/app-error";
|
||||
@@ -47,11 +48,26 @@ export const handler: BunextAPIRouteHandler<APIResponseObject> = async ({ req, q
|
||||
throw new AppError(`There's nothing to read in that message.`);
|
||||
}
|
||||
|
||||
const audio = await speakText({
|
||||
text,
|
||||
bot_id: conversation.bot.id,
|
||||
workspace_id: conversation.workspace_id,
|
||||
signal: req.signal,
|
||||
/**
|
||||
* Reading a message aloud is billed like any other model call, so
|
||||
* it is recorded against the message it reads.
|
||||
*/
|
||||
const audio = await withUsageScope({
|
||||
scope: {
|
||||
user_id: user.id,
|
||||
bot_id: conversation.bot.id,
|
||||
workspace_id: conversation.workspace_id,
|
||||
session_id: message.session_id || null,
|
||||
job_id: null,
|
||||
message_id: message.id || null,
|
||||
},
|
||||
task: () =>
|
||||
speakText({
|
||||
text,
|
||||
bot_id: conversation.bot.id,
|
||||
workspace_id: conversation.workspace_id,
|
||||
signal: req.signal,
|
||||
}),
|
||||
});
|
||||
|
||||
return new Response(audio.stream, {
|
||||
|
||||
@@ -1,6 +1,8 @@
|
||||
import userAuth from "@/src/functions/backend/auth/user-auth";
|
||||
import grabActiveBotSession from "@/src/functions/backend/db/bot-sessions/grab-active-bot-session";
|
||||
import grabUserConversation from "@/src/functions/backend/db/bot-sessions/grab-user-conversation";
|
||||
import transcribeAudio from "@/src/harness/audio/transcribe-audio";
|
||||
import withUsageScope from "@/src/harness/usage/with-usage-scope";
|
||||
import { TranscribeVoiceRequestSchema } from "@/src/schemas/bot-schema";
|
||||
import HarnessError from "@/src/harness/harness-error";
|
||||
import AppError from "@/src/utils/app-error";
|
||||
@@ -60,12 +62,33 @@ export const handler: BunextAPIRouteHandler<
|
||||
throw new AppError(`Nothing was recorded.`);
|
||||
}
|
||||
|
||||
const text = await transcribeAudio({
|
||||
audio: { data: new Uint8Array(buffer), format: AudioFormats[type] || type },
|
||||
/**
|
||||
* The dictation is work done for this user in this conversation,
|
||||
* even though no queued job or message is behind it.
|
||||
*/
|
||||
const session = await grabActiveBotSession({
|
||||
user_id: user.id,
|
||||
bot_id: conversation.bot.id,
|
||||
workspace_id: conversation.workspace_id,
|
||||
});
|
||||
|
||||
const text = await withUsageScope({
|
||||
scope: {
|
||||
user_id: user.id,
|
||||
bot_id: conversation.bot.id,
|
||||
workspace_id: conversation.workspace_id,
|
||||
session_id: session?.id || null,
|
||||
job_id: null,
|
||||
message_id: null,
|
||||
},
|
||||
task: () =>
|
||||
transcribeAudio({
|
||||
audio: { data: new Uint8Array(buffer), format: AudioFormats[type] || type },
|
||||
bot_id: conversation.bot.id,
|
||||
workspace_id: conversation.workspace_id,
|
||||
}),
|
||||
});
|
||||
|
||||
if (!text) {
|
||||
throw new AppError(`No speech was heard. Try again a little closer to the mic.`);
|
||||
}
|
||||
|
||||
@@ -0,0 +1,40 @@
|
||||
import userAuth from "@/src/functions/backend/auth/user-auth";
|
||||
import grabUsageOverview from "@/src/functions/backend/db/usage/grab-usage-overview";
|
||||
import { UsageRequestSchema } from "@/src/schemas/usage-schema";
|
||||
import type { UsageOverview } from "@/src/types";
|
||||
import canManageCrew from "@/src/utils/rbac/can-manage-crew";
|
||||
import parseWithSchema from "@/src/utils/parse-with-schema";
|
||||
import safeErrorMessage from "@/src/utils/safe-error-message";
|
||||
import type { APIResponseObject, BunextAPIRouteHandler } from "@moduletrace/bunext/types";
|
||||
|
||||
/**
|
||||
* What the install's models spent over a period: in total, per day, per
|
||||
* harness and per model. Super admins and admins only, since it is the
|
||||
* whole install's work rather than one user's conversation.
|
||||
*/
|
||||
export const handler: BunextAPIRouteHandler<APIResponseObject<UsageOverview>> = async ({ req, query }) => {
|
||||
try {
|
||||
if (req.method != "GET") {
|
||||
return { success: false, msg: `Method not allowed` };
|
||||
}
|
||||
|
||||
const { user, user_types } = await userAuth({ req });
|
||||
|
||||
if (!user?.logged_in_status || !user.id) {
|
||||
return { success: false, msg: `Unauthorized`, logoutUser: true };
|
||||
}
|
||||
|
||||
if (!canManageCrew({ user_types })) {
|
||||
return { success: false, msg: `You don't manage the crew.` };
|
||||
}
|
||||
|
||||
const request = parseWithSchema({ schema: UsageRequestSchema, data: query });
|
||||
|
||||
return {
|
||||
success: true,
|
||||
singleRes: await grabUsageOverview({ days: request.days }),
|
||||
};
|
||||
} catch (error: unknown) {
|
||||
return { success: false, msg: safeErrorMessage({ error }) };
|
||||
}
|
||||
};
|
||||
@@ -0,0 +1,18 @@
|
||||
import { z } from "zod";
|
||||
import { AppData } from "@/src/data/app-data";
|
||||
|
||||
/**
|
||||
* `GET api/admin/usage`: the period the usage page totals, in days back
|
||||
* from today. Left out, it is the page's default; past the limit, the
|
||||
* limit.
|
||||
*/
|
||||
export const UsageRequestSchema = z.object({
|
||||
days: z.coerce
|
||||
.number()
|
||||
.int()
|
||||
.positive()
|
||||
.max(AppData.UsageMaxDays, "That is too long a period")
|
||||
.optional(),
|
||||
});
|
||||
|
||||
export type UsageRequest = z.infer<typeof UsageRequestSchema>;
|
||||
@@ -32,6 +32,7 @@ import type { HarnessProviderKind } from "../dict/harness-provider-dict";
|
||||
import type { EnvParadigm, KnownEnv } from "../dict/env-dict";
|
||||
import type { ContainerStatus } from "../dict/computer-dict";
|
||||
import type { BotFace, BotMessageReactionEmoji } from "../dict/bot-dict";
|
||||
import type { TokenUsageSummary } from "../harness/usage/types";
|
||||
import type { McpServerStatus, McpTransport } from "../dict/mcp-dict";
|
||||
import type {
|
||||
CustomToolHttpMethod,
|
||||
@@ -875,6 +876,36 @@ export type DatabaseTableColumn = {
|
||||
*/
|
||||
export type DatabaseSearchMode = "every_word" | "phrase";
|
||||
|
||||
/**
|
||||
* The period the usage page totals, in epoch ms.
|
||||
*/
|
||||
export type UsageWindow = {
|
||||
days: number;
|
||||
start: number;
|
||||
end: number;
|
||||
};
|
||||
|
||||
/**
|
||||
* Everything the usage page shows for one window: what the install's
|
||||
* model calls spent, in total, per day, per harness and per model, with
|
||||
* the day before the window and the last seven days for comparison.
|
||||
*/
|
||||
export type UsageOverview = {
|
||||
window: UsageWindow;
|
||||
totals: TokenUsageSummary;
|
||||
by_harness: (TokenUsageSummary & { harness: string })[];
|
||||
days: (TokenUsageSummary & { at: number; label: string })[];
|
||||
models: (TokenUsageSummary & {
|
||||
model_slug: string;
|
||||
model_name: string | null;
|
||||
provider_name: string;
|
||||
provider_id: number | null;
|
||||
harness: string;
|
||||
})[];
|
||||
previous: TokenUsageSummary;
|
||||
week: TokenUsageSummary;
|
||||
};
|
||||
|
||||
/**
|
||||
* One table of the database, as the database browser lists it.
|
||||
*/
|
||||
@@ -988,6 +1019,11 @@ export type BotMessageWithAttachments = BUN_SQLITE_COBBLE_BOT_MESSAGES & {
|
||||
attachments?: BotMessageAttachment[];
|
||||
reactions?: BotMessageReaction[];
|
||||
reply_to?: BotMessageReply | null;
|
||||
/**
|
||||
* What the model calls behind this message spent (`token_usages`),
|
||||
* summed. Only on bot messages that took at least one call.
|
||||
*/
|
||||
usage?: TokenUsageSummary | null;
|
||||
};
|
||||
|
||||
/**
|
||||
@@ -1177,6 +1213,11 @@ export type PagePropsType = {
|
||||
*/
|
||||
database_tables?: DatabaseTableSummary[] | null;
|
||||
database_records?: DatabaseRecordsPayload | null;
|
||||
/**
|
||||
* What the install's models spent, on the usage page (crew managers
|
||||
* only).
|
||||
*/
|
||||
usage?: UsageOverview | null;
|
||||
/**
|
||||
* The open conversation's files (a bot's own, or a workspace's shared
|
||||
* ones), for the files modal and the chips that name one.
|
||||
|
||||
@@ -0,0 +1,35 @@
|
||||
import { describe, expect, test } from "bun:test";
|
||||
import formatTokens from "./format-tokens";
|
||||
|
||||
describe("formatTokens", () => {
|
||||
test("counts under a thousand are exact", () => {
|
||||
expect(formatTokens({ tokens: 0 })).toBe("0");
|
||||
expect(formatTokens({ tokens: 842 })).toBe("842");
|
||||
expect(formatTokens({ tokens: 999 })).toBe("999");
|
||||
});
|
||||
|
||||
test("counts under ten thousand keep a decimal, without a trailing zero", () => {
|
||||
expect(formatTokens({ tokens: 1000 })).toBe("1K");
|
||||
expect(formatTokens({ tokens: 1240 })).toBe("1.2K");
|
||||
expect(formatTokens({ tokens: 1240.7 })).toBe("1.2K");
|
||||
expect(formatTokens({ tokens: 9999 })).toBe("10K");
|
||||
});
|
||||
|
||||
test("counts under a million are whole thousands", () => {
|
||||
expect(formatTokens({ tokens: 12_400 })).toBe("12K");
|
||||
expect(formatTokens({ tokens: 841_600 })).toBe("842K");
|
||||
expect(formatTokens({ tokens: 999_999 })).toBe("1000K");
|
||||
});
|
||||
|
||||
test("counts past a million are millions", () => {
|
||||
expect(formatTokens({ tokens: 1_000_000 })).toBe("1M");
|
||||
expect(formatTokens({ tokens: 1_240_000 })).toBe("1.2M");
|
||||
expect(formatTokens({ tokens: 24_800_000 })).toBe("24.8M");
|
||||
});
|
||||
|
||||
test("a missing count reads as nothing spent", () => {
|
||||
expect(formatTokens({ tokens: null })).toBe("0");
|
||||
expect(formatTokens({ tokens: undefined })).toBe("0");
|
||||
expect(formatTokens({ tokens: Number.NaN })).toBe("0");
|
||||
});
|
||||
});
|
||||
@@ -0,0 +1,19 @@
|
||||
type Params = {
|
||||
tokens: number | null | undefined;
|
||||
};
|
||||
|
||||
/**
|
||||
* A token count as a person reads it: exact up to a thousand, then
|
||||
* rounded to a decimal under ten thousand ("12.4K"), whole thousands up to
|
||||
* a million ("842K") and millions after it ("1.2M"). A long run's total is
|
||||
* never a wall of digits.
|
||||
*/
|
||||
export default function formatTokens({ tokens }: Params) {
|
||||
const count = Number(tokens) || 0;
|
||||
|
||||
if (count < 1000) return String(count);
|
||||
if (count < 10_000) return `${(count / 1000).toFixed(1).replace(/\.0$/, "")}K`;
|
||||
if (count < 1_000_000) return `${Math.round(count / 1000)}K`;
|
||||
|
||||
return `${(count / 1_000_000).toFixed(1).replace(/\.0$/, "")}M`;
|
||||
}
|
||||
@@ -0,0 +1,85 @@
|
||||
import { describe, expect, test } from "bun:test";
|
||||
import canSeeRailItem from "./can-see-rail-item";
|
||||
import canManageCrew from "./can-manage-crew";
|
||||
import canListUsers from "./can-list-users";
|
||||
import { RailPages, UsagePath } from "@/src/dict/rail-pages-dict";
|
||||
|
||||
function types(...user_types: string[]) {
|
||||
return user_types.map((user_type) => ({ user_type })) as never;
|
||||
}
|
||||
|
||||
const super_admin = types("super_admin");
|
||||
const admin = types("member", "admin");
|
||||
const member = types("member");
|
||||
const read_only = types("read_only");
|
||||
|
||||
describe("canSeeRailItem", () => {
|
||||
test("a page anyone may open shows to everyone", () => {
|
||||
const page = {};
|
||||
|
||||
expect(canSeeRailItem({ item: page, is_super_admin: true, user_types: super_admin })).toBe(true);
|
||||
expect(canSeeRailItem({ item: page, is_super_admin: false, user_types: member })).toBe(true);
|
||||
expect(canSeeRailItem({ item: page, is_super_admin: false, user_types: read_only })).toBe(true);
|
||||
});
|
||||
|
||||
test("a super admin page shows to super admins only", () => {
|
||||
const page = { super_admin_only: true };
|
||||
|
||||
expect(canSeeRailItem({ item: page, is_super_admin: true, user_types: super_admin })).toBe(true);
|
||||
expect(canSeeRailItem({ item: page, is_super_admin: false, user_types: admin })).toBe(false);
|
||||
expect(canSeeRailItem({ item: page, is_super_admin: false, user_types: member })).toBe(false);
|
||||
});
|
||||
|
||||
test("an RBAC gate is asked, not guessed from super admin", () => {
|
||||
const page = { can_view: canManageCrew };
|
||||
|
||||
expect(canSeeRailItem({ item: page, is_super_admin: false, user_types: super_admin })).toBe(true);
|
||||
expect(canSeeRailItem({ item: page, is_super_admin: false, user_types: admin })).toBe(true);
|
||||
expect(canSeeRailItem({ item: page, is_super_admin: false, user_types: member })).toBe(false);
|
||||
});
|
||||
|
||||
test("a can_view rule wins over super_admin_only, so a gate is never widened by it", () => {
|
||||
/**
|
||||
* The two together should follow the narrower rule the code asked
|
||||
* for. No page sets both today, and this keeps a future one from
|
||||
* quietly showing to every super admin when its rule says no.
|
||||
*/
|
||||
const page = { can_view: canListUsers, super_admin_only: true };
|
||||
|
||||
expect(canSeeRailItem({ item: page, is_super_admin: false, user_types: member })).toBe(false);
|
||||
expect(canSeeRailItem({ item: page, is_super_admin: false, user_types: read_only })).toBe(false);
|
||||
expect(canSeeRailItem({ item: page, is_super_admin: false, user_types: admin })).toBe(true);
|
||||
});
|
||||
});
|
||||
|
||||
describe("RailPages", () => {
|
||||
test("usage is a rail page, gated to the crew's managers", () => {
|
||||
const usage = RailPages.find((page) => page.url == UsagePath);
|
||||
|
||||
expect(usage?.title).toBe("Usage");
|
||||
expect(usage?.can_view).toBeDefined();
|
||||
expect(usage?.super_admin_only).toBeFalsy();
|
||||
});
|
||||
|
||||
test("activity stays super admin only", () => {
|
||||
const activity = RailPages.find((page) => page.url == "/admin/logs");
|
||||
|
||||
expect(activity?.super_admin_only).toBe(true);
|
||||
});
|
||||
|
||||
test("a member is offered routines, files and help but not usage or activity", () => {
|
||||
const shown = RailPages.filter((page) =>
|
||||
canSeeRailItem({ item: page, is_super_admin: false, user_types: member }),
|
||||
).map((page) => page.title);
|
||||
|
||||
expect(shown).toEqual(["Routines", "Files", "Help"]);
|
||||
});
|
||||
|
||||
test("an admin is offered usage but still not activity", () => {
|
||||
const shown = RailPages.filter((page) =>
|
||||
canSeeRailItem({ item: page, is_super_admin: false, user_types: admin }),
|
||||
).map((page) => page.title);
|
||||
|
||||
expect(shown).toEqual(["Routines", "Files", "Usage", "Help"]);
|
||||
});
|
||||
});
|
||||
@@ -0,0 +1,29 @@
|
||||
import type { BUN_SQLITE_COBBLE_USER_TYPES } from "@/db/types/db";
|
||||
|
||||
type Params = {
|
||||
item: {
|
||||
/**
|
||||
* Who sees it, when that's an RBAC rule rather than super admins only.
|
||||
*/
|
||||
can_view?: (params: {
|
||||
user_types?: Pick<BUN_SQLITE_COBBLE_USER_TYPES, "user_type">[] | null;
|
||||
}) => boolean;
|
||||
super_admin_only?: boolean;
|
||||
};
|
||||
is_super_admin: boolean;
|
||||
user_types?: Pick<BUN_SQLITE_COBBLE_USER_TYPES, "user_type">[] | null;
|
||||
};
|
||||
|
||||
/**
|
||||
* Whether a rail destination (a settings sub page, a popover page) shows
|
||||
* to this user: its own RBAC rule when it has one, else super admins only
|
||||
* when it's a super admin page, else everyone.
|
||||
*
|
||||
* Shared by the rail lists and the user popover, so a page gated in one is
|
||||
* never offered in the other.
|
||||
*/
|
||||
export default function canSeeRailItem({ item, is_super_admin, user_types }: Params): boolean {
|
||||
if (item.can_view) return item.can_view({ user_types });
|
||||
|
||||
return is_super_admin || !item.super_admin_only;
|
||||
}
|
||||
Reference in New Issue
Block a user