Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions docs/bub-agent.md
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
# Agent integration with Bub main

Dohnuts provides one decision tool through Bub. The `agent` extra pins Bub main at
`9bf70488e22a44eeb875a9d55d9a7d352b82e381`. Integration uses its
[Python SDK](https://github.com/bubbuild/bub/blob/9bf70488e22a44eeb875a9d55d9a7d352b82e381/website/src/content/docs/docs/build/sdk.md)
`6f23fee36a8aa55b71d5be015529a4b486b31d89`. Integration uses its
[Python SDK](https://github.com/bubbuild/bub/blob/6f23fee36a8aa55b71d5be015529a4b486b31d89/website/src/content/docs/docs/build/sdk.md)
to provide one batch decision tool and persistent session tapes.

Follow the [runtime and checkpoint setup](inference.md) first.
Expand Down
2 changes: 1 addition & 1 deletion docs/upstream-alignment.md
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

References are pinned to Laya `d113dca2512fb3eaca313534bc54c7162d87c1d4`,
Laya Vision `86ccec115ef3d72d1851168fc9b7194e8dbed35a`, and Bub main
`9bf70488e22a44eeb875a9d55d9a7d352b82e381`. The training mixture includes upstream public tasks and soft teacher
`6f23fee36a8aa55b71d5be015529a4b486b31d89`. The training mixture includes upstream public tasks and soft teacher
labels. Existing Laya weights are benchmark references, not training targets.

## Publicly documented training coverage
Expand Down
42 changes: 21 additions & 21 deletions pdm.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ dependencies = [

[project.optional-dependencies]
train = ["pyarrow>=25", "scikit-learn>=1.9", "pandas>=3.0", "datasets>=4"]
agent = ["bub @ git+https://github.com/bubbuild/bub.git@9bf70488e22a44eeb875a9d55d9a7d352b82e381"]
agent = ["bub @ git+https://github.com/bubbuild/bub.git@6f23fee36a8aa55b71d5be015529a4b486b31d89"]

[dependency-groups]
quality = ["ruff==0.16.8", "ty==0.0.82", "prek>=0.3.1,<1", "pytest>=9,<10"]
Expand Down
2 changes: 1 addition & 1 deletion src/dohnuts/bub_agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@

from PIL import Image

BUB_REVISION = "9bf70488e22a44eeb875a9d55d9a7d352b82e381"
BUB_REVISION = "6f23fee36a8aa55b71d5be015529a4b486b31d89"


def create_agent(predictor, workspace: Path, tape_directory: Path):
Expand Down
3 changes: 1 addition & 2 deletions tests/acceptance.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
"""Accept an exported checkpoint through prediction, reload, and the real Bub SDK."""

import argparse
import ast
import asyncio
import gc
import json
Expand Down Expand Up @@ -118,7 +117,7 @@ async def verify(args):
command += " image_path=" + shlex.quote(str(image_path.resolve()))
stream = await agent.run_stream(session_id="acceptance-" + label, prompt=command)
events = [event async for event in stream]
actual = ast.literal_eval(events[-1].data["text"])
actual = json.loads(events[-1].data["text"])
assert actual == reference, "Bub changed the decision response"
responses.append(actual)
report = {
Expand Down
Loading