feat: refactor sgclaw around zeroclaw compat runtime
This commit is contained in:
229
tools/browser_smoke/fake_deepseek_server.mjs
Normal file
229
tools/browser_smoke/fake_deepseek_server.mjs
Normal file
@@ -0,0 +1,229 @@
|
||||
import http from 'node:http'
|
||||
|
||||
const DEFAULT_DEEPSEEK_MODEL = 'deepseek-chat'
|
||||
const DEFAULT_DEEPSEEK_API_KEY = 'sgclaw-smoke-test-key'
|
||||
|
||||
const BAIDU_URL = 'https://www.baidu.com'
|
||||
const BAIDU_DOMAIN = 'www.baidu.com'
|
||||
const BAIDU_INPUT_SELECTOR = '#kw'
|
||||
const BAIDU_BUTTON_SELECTOR = '#su'
|
||||
const ZHIHU_URL = 'https://www.zhihu.com/search'
|
||||
const ZHIHU_DOMAIN = 'www.zhihu.com'
|
||||
|
||||
export function buildSmokeEnv(baseUrl, baseEnv = process.env) {
|
||||
return {
|
||||
...baseEnv,
|
||||
DEEPSEEK_API_KEY: DEFAULT_DEEPSEEK_API_KEY,
|
||||
DEEPSEEK_BASE_URL: baseUrl,
|
||||
DEEPSEEK_MODEL: DEFAULT_DEEPSEEK_MODEL,
|
||||
}
|
||||
}
|
||||
|
||||
export function buildChatCompletionResponse(requestBody) {
|
||||
const instruction = latestUserInstruction(requestBody?.messages ?? [])
|
||||
const plan = instructionPlan(instruction)
|
||||
const isFinalRound = (requestBody?.messages ?? []).some((message) => message?.role === 'tool')
|
||||
|
||||
if (isFinalRound) {
|
||||
return {
|
||||
id: `chatcmpl-smoke-final-${plan.key}`,
|
||||
object: 'chat.completion',
|
||||
created: unixTimeSeconds(),
|
||||
model: DEFAULT_DEEPSEEK_MODEL,
|
||||
choices: [{
|
||||
index: 0,
|
||||
message: {
|
||||
role: 'assistant',
|
||||
content: plan.summary,
|
||||
},
|
||||
finish_reason: 'stop',
|
||||
}],
|
||||
usage: {
|
||||
prompt_tokens: 32,
|
||||
completion_tokens: 8,
|
||||
total_tokens: 40,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
return {
|
||||
id: `chatcmpl-smoke-tools-${plan.key}`,
|
||||
object: 'chat.completion',
|
||||
created: unixTimeSeconds(),
|
||||
model: DEFAULT_DEEPSEEK_MODEL,
|
||||
choices: [{
|
||||
index: 0,
|
||||
message: {
|
||||
role: 'assistant',
|
||||
content: '',
|
||||
tool_calls: plan.toolCalls,
|
||||
},
|
||||
finish_reason: 'tool_calls',
|
||||
}],
|
||||
usage: {
|
||||
prompt_tokens: 24,
|
||||
completion_tokens: 12,
|
||||
total_tokens: 36,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
export async function startFakeDeepSeekServer() {
|
||||
const requests = []
|
||||
const server = http.createServer(async (req, res) => {
|
||||
try {
|
||||
if (req.method !== 'POST') {
|
||||
res.writeHead(405, {'content-type': 'application/json'})
|
||||
res.end(JSON.stringify({error: 'method not allowed'}))
|
||||
return
|
||||
}
|
||||
|
||||
const body = await readJsonBody(req)
|
||||
requests.push({
|
||||
method: req.method,
|
||||
url: req.url || '/',
|
||||
body,
|
||||
})
|
||||
|
||||
const response = buildChatCompletionResponse(body)
|
||||
res.writeHead(200, {'content-type': 'application/json'})
|
||||
res.end(JSON.stringify(response))
|
||||
} catch (error) {
|
||||
res.writeHead(500, {'content-type': 'application/json'})
|
||||
res.end(JSON.stringify({
|
||||
error: error instanceof Error ? error.message : String(error),
|
||||
}))
|
||||
}
|
||||
})
|
||||
|
||||
await new Promise((resolve, reject) => {
|
||||
server.once('error', reject)
|
||||
server.listen(0, '127.0.0.1', resolve)
|
||||
})
|
||||
|
||||
const address = server.address()
|
||||
if (!address || typeof address === 'string') {
|
||||
server.close()
|
||||
throw new Error('failed to bind fake DeepSeek server')
|
||||
}
|
||||
|
||||
return {
|
||||
baseUrl: `http://127.0.0.1:${address.port}`,
|
||||
requests,
|
||||
close: () => new Promise((resolve, reject) => {
|
||||
server.close((error) => error ? reject(error) : resolve())
|
||||
}),
|
||||
}
|
||||
}
|
||||
|
||||
function latestUserInstruction(messages) {
|
||||
for (let index = messages.length - 1; index >= 0; index -= 1) {
|
||||
const message = messages[index]
|
||||
if (message?.role === 'user' && typeof message.content === 'string' && message.content.trim()) {
|
||||
return normalizeSmokeInstruction(message.content)
|
||||
}
|
||||
}
|
||||
|
||||
throw new Error('missing user instruction in DeepSeek smoke request')
|
||||
}
|
||||
|
||||
export function normalizeSmokeInstruction(rawInstruction) {
|
||||
return rawInstruction.trim().replace(/^(?:\[[^\]]+\]\s*)+/, '')
|
||||
}
|
||||
|
||||
function instructionPlan(instruction) {
|
||||
const baiduQuery = extractQuery(instruction, ['打开百度搜索', '打开百度并搜索'])
|
||||
if (baiduQuery) {
|
||||
return {
|
||||
key: 'baidu',
|
||||
summary: `已在百度搜索${baiduQuery}`,
|
||||
toolCalls: [
|
||||
browserToolCall('call_baidu_1', {
|
||||
action: 'navigate',
|
||||
expected_domain: BAIDU_DOMAIN,
|
||||
url: BAIDU_URL,
|
||||
}),
|
||||
browserToolCall('call_baidu_2', {
|
||||
action: 'type',
|
||||
expected_domain: BAIDU_DOMAIN,
|
||||
selector: BAIDU_INPUT_SELECTOR,
|
||||
text: baiduQuery,
|
||||
clear_first: true,
|
||||
}),
|
||||
browserToolCall('call_baidu_3', {
|
||||
action: 'click',
|
||||
expected_domain: BAIDU_DOMAIN,
|
||||
selector: BAIDU_BUTTON_SELECTOR,
|
||||
}),
|
||||
],
|
||||
}
|
||||
}
|
||||
|
||||
const zhihuQuery = extractQuery(instruction, ['打开知乎搜索', '打开知乎并搜索'])
|
||||
if (zhihuQuery) {
|
||||
const url = new URL(ZHIHU_URL)
|
||||
url.searchParams.set('type', 'content')
|
||||
url.searchParams.set('q', zhihuQuery)
|
||||
|
||||
return {
|
||||
key: 'zhihu',
|
||||
summary: `已在知乎搜索${zhihuQuery}`,
|
||||
toolCalls: [
|
||||
browserToolCall('call_zhihu_1', {
|
||||
action: 'navigate',
|
||||
expected_domain: ZHIHU_DOMAIN,
|
||||
url: url.toString(),
|
||||
}),
|
||||
],
|
||||
}
|
||||
}
|
||||
|
||||
throw new Error(`unsupported smoke instruction: ${instruction}`)
|
||||
}
|
||||
|
||||
function extractQuery(instruction, prefixes) {
|
||||
for (const prefix of prefixes) {
|
||||
if (instruction.startsWith(prefix)) {
|
||||
const query = instruction.slice(prefix.length).trim()
|
||||
if (!query) {
|
||||
throw new Error(`missing search query in instruction: ${instruction}`)
|
||||
}
|
||||
return query
|
||||
}
|
||||
}
|
||||
|
||||
return null
|
||||
}
|
||||
|
||||
function browserToolCall(id, args) {
|
||||
return {
|
||||
id,
|
||||
type: 'function',
|
||||
function: {
|
||||
name: 'browser_action',
|
||||
arguments: JSON.stringify(args),
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
function readJsonBody(req) {
|
||||
return new Promise((resolve, reject) => {
|
||||
let raw = ''
|
||||
req.setEncoding('utf8')
|
||||
req.on('data', (chunk) => {
|
||||
raw += chunk
|
||||
})
|
||||
req.on('end', () => {
|
||||
try {
|
||||
resolve(JSON.parse(raw || '{}'))
|
||||
} catch (error) {
|
||||
reject(error)
|
||||
}
|
||||
})
|
||||
req.on('error', reject)
|
||||
})
|
||||
}
|
||||
|
||||
function unixTimeSeconds() {
|
||||
return Math.floor(Date.now() / 1000)
|
||||
}
|
||||
Reference in New Issue
Block a user