mirror of
https://github.com/open-jarvis/OpenJarvis.git
synced 2026-07-30 19:02:16 +00:00
Jon Saad-FalconandClaude Opus 4.6
a4c4081ff4
- Rewrite .github/workflows/desktop.yml: 2-job pipeline (validate + build-and-release) with rolling desktop-latest pre-release on push to main and stable desktop-v* releases - Add UpdateChecker component: checks for updates on startup + every 30 min, background download with progress bar, one-click relaunch - Configure Tauri updater: endpoints pointing to desktop-latest release, pubkey placeholder - Add tauri-plugin-process for relaunch support (Cargo.toml, lib.rs, package.json) - Add macOS Entitlements.plist for notarization (network + file access, no sandbox) - Add scripts/bump-desktop-version.sh for atomic version bumps across 3 config files - Add desktop/README.md with dev setup, auto-update architecture, signing docs - Update .gitignore for desktop/node_modules, dist, target - Configure macOS minimumSystemVersion, Windows timestampUrl - Include all Phase 14-21 work: agent hardening, RBAC, taint tracking, workflows, skills, knowledge graph, sessions, A2A, MCP templates, WASM sandbox, TUI dashboard, production tools, CLI expansion, API expansion, learning productionization, Tauri desktop app, and 10 new channels Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
93 lines
3.1 KiB
Python
93 lines
3.1 KiB
Python
"""Tests for continuation handling (Phase 14.2)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any, Dict, List
|
|
|
|
from openjarvis.agents._stubs import AgentResult, BaseAgent
|
|
|
|
|
|
class MockEngine:
|
|
"""Mock engine that simulates finish_reason='length'."""
|
|
|
|
def __init__(self, responses: List[Dict[str, Any]]):
|
|
self._responses = list(responses)
|
|
self._call_count = 0
|
|
|
|
def generate(self, messages, **kwargs):
|
|
if self._call_count < len(self._responses):
|
|
resp = self._responses[self._call_count]
|
|
else:
|
|
resp = {"content": "done", "finish_reason": "stop"}
|
|
self._call_count += 1
|
|
return resp
|
|
|
|
def list_models(self):
|
|
return ["test-model"]
|
|
|
|
def health(self):
|
|
return True
|
|
|
|
|
|
class ContinuationAgent(BaseAgent):
|
|
"""Agent that exercises _check_continuation."""
|
|
|
|
agent_id = "test_continuation"
|
|
|
|
def run(self, input, context=None, **kwargs):
|
|
messages = self._build_messages(input, context)
|
|
result = self._generate(messages)
|
|
content = self._check_continuation(result, messages)
|
|
return AgentResult(content=content, turns=1)
|
|
|
|
|
|
class TestContinuation:
|
|
def test_no_continuation_needed(self):
|
|
engine = MockEngine([
|
|
{"content": "Hello world", "finish_reason": "stop"},
|
|
])
|
|
agent = ContinuationAgent(engine, "test-model")
|
|
result = agent.run("Hi")
|
|
assert result.content == "Hello world"
|
|
|
|
def test_single_continuation(self):
|
|
engine = MockEngine([
|
|
{"content": "Part 1...", "finish_reason": "length"},
|
|
{"content": " Part 2.", "finish_reason": "stop"},
|
|
])
|
|
agent = ContinuationAgent(engine, "test-model")
|
|
result = agent.run("Hi")
|
|
assert result.content == "Part 1... Part 2."
|
|
|
|
def test_multiple_continuations(self):
|
|
engine = MockEngine([
|
|
{"content": "A", "finish_reason": "length"},
|
|
{"content": "B", "finish_reason": "length"},
|
|
{"content": "C", "finish_reason": "stop"},
|
|
])
|
|
agent = ContinuationAgent(engine, "test-model")
|
|
result = agent.run("Hi")
|
|
assert result.content == "ABC"
|
|
|
|
def test_max_continuations_respected(self):
|
|
engine = MockEngine([
|
|
{"content": "A", "finish_reason": "length"},
|
|
{"content": "B", "finish_reason": "length"},
|
|
{"content": "C", "finish_reason": "length"}, # 3rd continuation
|
|
{"content": "D", "finish_reason": "stop"},
|
|
])
|
|
agent = ContinuationAgent(engine, "test-model")
|
|
# Default max_continuations=2, so should stop after 2 continuations
|
|
messages = agent._build_messages("Hi")
|
|
result_dict = agent._generate(messages)
|
|
content = agent._check_continuation(result_dict, messages, max_continuations=2)
|
|
assert content == "ABC" # A + B + C, but not D
|
|
|
|
def test_empty_finish_reason(self):
|
|
engine = MockEngine([
|
|
{"content": "Done", "finish_reason": ""},
|
|
])
|
|
agent = ContinuationAgent(engine, "test-model")
|
|
result = agent.run("Hi")
|
|
assert result.content == "Done"
|