434 lines
17 KiB
C
434 lines
17 KiB
C
/* test_agent.c - agent 上下文管理（不涉及网络） */
|
||
#include "agent.h"
|
||
#include "json.h"
|
||
#include "util.h"
|
||
|
||
#include <stdio.h>
|
||
#include <stdlib.h>
|
||
#include <string.h>
|
||
|
||
static int failures = 0;
|
||
static int checks = 0;
|
||
|
||
#define CHECK(cond, ...) \
|
||
do { \
|
||
checks++; \
|
||
if (!(cond)) { \
|
||
failures++; \
|
||
printf(" ✗ %s:%d: ", __FILE__, __LINE__); \
|
||
printf(__VA_ARGS__); \
|
||
printf("\n"); \
|
||
} \
|
||
} while (0)
|
||
|
||
static size_t count_role(const Agent *a, const char *role)
|
||
{
|
||
size_t n = 0;
|
||
for (size_t i = 0; i < json_array_size(a->messages); i++) {
|
||
const char *r = json_get_string(json_array_get(a->messages, i), "role");
|
||
if (r && strcmp(r, role) == 0) n++;
|
||
}
|
||
return n;
|
||
}
|
||
|
||
/* ------------------------------------------------------------------ */
|
||
/* 上下文超限的自动恢复 */
|
||
/* ------------------------------------------------------------------ */
|
||
|
||
static void add_msg(Agent *a, const char *role, const char *content)
|
||
{
|
||
JsonValue *m = json_new_object();
|
||
json_object_set_string(m, "role", role);
|
||
json_object_set_string(m, "content", content);
|
||
json_array_push(a->messages, m);
|
||
}
|
||
|
||
/* 造一条 assistant 带 tool_calls 的消息 */
|
||
static void add_assistant_tool_call(Agent *a, const char *id, const char *name)
|
||
{
|
||
JsonValue *m = json_new_object();
|
||
json_object_set_string(m, "role", "assistant");
|
||
json_object_set_null(m, "content");
|
||
JsonValue *tcs = json_new_array();
|
||
JsonValue *tc = json_new_object();
|
||
json_object_set_string(tc, "id", id);
|
||
json_object_set_string(tc, "type", "function");
|
||
JsonValue *fn = json_new_object();
|
||
json_object_set_string(fn, "name", name);
|
||
json_object_set_string(fn, "arguments", "{}");
|
||
json_object_set(tc, "function", fn);
|
||
json_array_push(tcs, tc);
|
||
json_object_set(m, "tool_calls", tcs);
|
||
json_array_push(a->messages, m);
|
||
}
|
||
|
||
static void add_tool_result(Agent *a, const char *id, const char *content)
|
||
{
|
||
JsonValue *m = json_new_object();
|
||
json_object_set_string(m, "role", "tool");
|
||
json_object_set_string(m, "tool_call_id", id);
|
||
json_object_set_string(m, "content", content);
|
||
json_array_push(a->messages, m);
|
||
}
|
||
|
||
/* 序列化历史，便于断言 */
|
||
static char *dump_history(const Agent *a)
|
||
{
|
||
return json_dump(a->messages);
|
||
}
|
||
|
||
/* 配对不变量：任何 assistant(tool_calls) 后面必须**紧跟**同 id 的 tool 消息。
|
||
* 服务端就是这么要求的，违反了下一轮直接 400 —— 所以每次动过历史之后都要查。 */
|
||
static void check_pairing(Agent *a)
|
||
{
|
||
size_t n = json_array_size(a->messages);
|
||
for (size_t i = 0; i < n; i++) {
|
||
JsonValue *m = json_array_get(a->messages, i);
|
||
JsonValue *tcs = json_object_get(m, "tool_calls");
|
||
if (!tcs) continue;
|
||
const char *id = json_get_string(json_array_get(tcs, 0), "id");
|
||
CHECK(i + 1 < n, "tool_calls 之后必须有消息");
|
||
if (i + 1 < n) {
|
||
JsonValue *nxt = json_array_get(a->messages, i + 1);
|
||
const char *nid = json_get_string(nxt, "tool_call_id");
|
||
CHECK(nid && id && strcmp(nid, id) == 0,
|
||
"tool 结果必须紧跟其 tool_call（第 %zu 条）", i);
|
||
}
|
||
}
|
||
}
|
||
|
||
/* 一轮一轮摘下去（/back 走的就是这条路），历史在每一步都必须仍然自洽。
|
||
*
|
||
* 这里原来测的是"上下文超限自动裁剪"（agent_drop_oldest_round）—— 那套机制
|
||
* 已经**整个删掉**：它的判据只是错误文案里的关键词，却会不可逆地删掉最老的
|
||
* 几轮对话。现在超限就是如实报错、历史一个字节都不动，怎么处理由用户决定
|
||
* （/back 摘一轮、/reset 重来）。但"摘掉一整轮之后 tool 配对不能散"这条
|
||
* 不变量与由谁触发无关，所以留着继续测。 */
|
||
static void test_take_rounds_keeps_pairing(void)
|
||
{
|
||
puts("逐轮摘掉之后历史仍然自洽");
|
||
Agent a;
|
||
agent_init(&a);
|
||
agent_add_system(&a, "SYS");
|
||
for (int i = 1; i <= 4; i++) {
|
||
char buf[32];
|
||
snprintf(buf, sizeof buf, "U%d", i);
|
||
add_msg(&a, "user", buf);
|
||
add_assistant_tool_call(&a, "call_x", "read_file");
|
||
add_tool_result(&a, "call_x", "R");
|
||
add_msg(&a, "assistant", "A");
|
||
}
|
||
|
||
int rounds = 0;
|
||
for (;;) {
|
||
JsonValue *r = agent_take_last_round(&a);
|
||
if (!r) break;
|
||
json_free(r);
|
||
rounds++;
|
||
check_pairing(&a); /* 每摘一轮都查一遍配对 */
|
||
}
|
||
CHECK(rounds == 4, "共应摘掉 4 轮，实际 %d", rounds);
|
||
CHECK(agent_history_len(&a) == 1, "最后应只剩 system，实际 %zu",
|
||
agent_history_len(&a));
|
||
agent_free(&a);
|
||
}
|
||
|
||
static void test_take_last_round(void)
|
||
{
|
||
puts("摘掉最近一轮（/back 用）");
|
||
Agent a;
|
||
agent_init(&a);
|
||
agent_add_system(&a, "SYS");
|
||
add_msg(&a, "user", "U1");
|
||
add_assistant_tool_call(&a, "call_1", "run_command");
|
||
add_tool_result(&a, "call_1", "R1");
|
||
add_msg(&a, "assistant", "A1");
|
||
add_msg(&a, "user", "U2");
|
||
add_msg(&a, "assistant", "A2");
|
||
|
||
CHECK(agent_history_len(&a) == 7, "初始 7 条，实际 %zu", agent_history_len(&a));
|
||
|
||
JsonValue *r = agent_take_last_round(&a);
|
||
CHECK(r != NULL, "应摘到一轮");
|
||
CHECK(r && json_array_size(r) == 2, "最后一轮应 2 条，实际 %zu",
|
||
r ? json_array_size(r) : 0);
|
||
CHECK(agent_history_len(&a) == 5, "摘完应剩 5 条，实际 %zu",
|
||
agent_history_len(&a));
|
||
char *h = dump_history(&a);
|
||
CHECK(strstr(h, "U2") == NULL, "U2 应被摘掉");
|
||
CHECK(strstr(h, "U1") != NULL, "U1 应保留");
|
||
free(h);
|
||
json_free(r);
|
||
|
||
/* 关键：工具调用与结果必须整组一起走 */
|
||
r = agent_take_last_round(&a);
|
||
CHECK(r && json_array_size(r) == 4, "第一轮应 4 条，实际 %zu",
|
||
r ? json_array_size(r) : 0);
|
||
h = dump_history(&a);
|
||
CHECK(strstr(h, "call_1") == NULL && strstr(h, "R1") == NULL,
|
||
"工具调用与结果应一并摘掉，不能只删一半");
|
||
CHECK(strstr(h, "SYS") != NULL, "system 必须保留");
|
||
free(h);
|
||
json_free(r);
|
||
|
||
/* 只剩 system 时无可摘 */
|
||
r = agent_take_last_round(&a);
|
||
CHECK(r == NULL, "没有用户轮次时应返回 NULL");
|
||
json_free(r);
|
||
CHECK(agent_history_len(&a) == 1, "历史应只剩 system");
|
||
agent_free(&a);
|
||
|
||
/* 没有 user 消息的历史 */
|
||
Agent b;
|
||
agent_init(&b);
|
||
add_msg(&b, "assistant", "只有助手");
|
||
CHECK(agent_take_last_round(&b) == NULL, "没有 user 时不应摘任何东西");
|
||
CHECK(agent_history_len(&b) == 1, "历史不应被改动");
|
||
agent_free(&b);
|
||
}
|
||
|
||
/* ------------------------------------------------------------------ */
|
||
/* 结构体检：assistant(tool_calls) 与 tool 消息的配对 */
|
||
/* ------------------------------------------------------------------ */
|
||
|
||
/* 造一条 assistant，带 count 个工具调用（id 依次是 c0, c1, ...） */
|
||
static JsonValue *mk_assistant(size_t count)
|
||
{
|
||
JsonValue *m = json_new_object();
|
||
json_object_set_string(m, "role", "assistant");
|
||
JsonValue *tcs = json_new_array();
|
||
for (size_t i = 0; i < count; i++) {
|
||
char id[16];
|
||
snprintf(id, sizeof id, "c%zu", i);
|
||
JsonValue *tc = json_new_object();
|
||
json_object_set_string(tc, "id", id);
|
||
json_object_set_string(tc, "type", "function");
|
||
JsonValue *fn = json_new_object();
|
||
json_object_set_string(fn, "name", "run_command");
|
||
json_object_set_string(fn, "arguments", "{}");
|
||
json_object_set(tc, "function", fn);
|
||
json_array_push(tcs, tc);
|
||
}
|
||
json_object_set(m, "tool_calls", tcs);
|
||
return m;
|
||
}
|
||
|
||
static JsonValue *mk_tool(const char *id)
|
||
{
|
||
JsonValue *m = json_new_object();
|
||
json_object_set_string(m, "role", "tool");
|
||
json_object_set_string(m, "tool_call_id", id);
|
||
json_object_set_string(m, "content", "ok");
|
||
return m;
|
||
}
|
||
|
||
static JsonValue *mk_plain(const char *role, const char *text)
|
||
{
|
||
JsonValue *m = json_new_object();
|
||
json_object_set_string(m, "role", role);
|
||
json_object_set_string(m, "content", text);
|
||
return m;
|
||
}
|
||
|
||
/* 数一数某个 assistant 之后紧跟的 tool 消息覆盖了几个 id */
|
||
static const char *nth_role(const JsonValue *msgs, size_t i)
|
||
{
|
||
return json_get_string(json_array_get((JsonValue *)msgs, i), "role");
|
||
}
|
||
|
||
static int pairing_ok(const JsonValue *msgs)
|
||
{
|
||
size_t n = json_array_size((JsonValue *)msgs);
|
||
for (size_t i = 0; i < n; i++) {
|
||
JsonValue *m = json_array_get((JsonValue *)msgs, i);
|
||
const char *r = json_get_string(m, "role");
|
||
if (!r || strcmp(r, "assistant") != 0) continue;
|
||
JsonValue *tcs = json_object_get(m, "tool_calls");
|
||
if (!tcs || json_array_size(tcs) == 0) continue;
|
||
for (size_t k = 0; k < json_array_size(tcs); k++) {
|
||
const char *id = json_get_string(json_array_get(tcs, k), "id");
|
||
size_t j = i + 1;
|
||
int found = 0;
|
||
for (; j < n && strcmp(nth_role(msgs, j), "tool") == 0; j++)
|
||
if (strcmp(json_get_string(json_array_get((JsonValue *)msgs, j),
|
||
"tool_call_id"), id) == 0) found = 1;
|
||
if (!found) return 0;
|
||
}
|
||
}
|
||
return 1;
|
||
}
|
||
|
||
static void test_repair_pairing(void)
|
||
{
|
||
int fixed = 0;
|
||
|
||
puts("结构体检：本来没问题就不该动它");
|
||
JsonValue *ok = json_new_array();
|
||
json_array_push(ok, mk_plain("user", "跑一下"));
|
||
json_array_push(ok, mk_assistant(2));
|
||
json_array_push(ok, mk_tool("c0"));
|
||
json_array_push(ok, mk_tool("c1"));
|
||
json_array_push(ok, mk_plain("assistant", "做完了"));
|
||
CHECK(pairing_ok(ok), "构造出来的历史本身应是合法的");
|
||
CHECK(agent_repair_tool_pairing(ok, &fixed) == NULL,
|
||
"合法历史不该被重建（NULL = 什么都没改）");
|
||
CHECK(fixed == 0, "合法历史的修复处数应为 0");
|
||
json_free(ok);
|
||
|
||
puts("缺 tool 消息：补齐，且补在原位");
|
||
/* 复现真实报错：assistant 带两个调用，中间夹了一条 user（图省事直接
|
||
* 模仿当年那个 bug 的形状），第二个调用没人回应 */
|
||
JsonValue *bad = json_new_array();
|
||
json_array_push(bad, mk_plain("user", "看看图"));
|
||
json_array_push(bad, mk_assistant(2));
|
||
json_array_push(bad, mk_tool("c0"));
|
||
json_array_push(bad, mk_plain("user", "图片附件"));
|
||
json_array_push(bad, mk_tool("c1"));
|
||
fixed = 0;
|
||
JsonValue *r = agent_repair_tool_pairing(bad, &fixed);
|
||
CHECK(r != NULL, "对不上时应该返回重建好的数组");
|
||
if (r) {
|
||
CHECK(pairing_ok(r), "修完之后配对必须成立");
|
||
/* 两处：给 c1 补一条占位（它没被回应），再把错位的那条 c1 结果丢掉
|
||
* （它被 user 消息隔开，已经不在 tool 段里了 —— 留着就是孤儿）。
|
||
* 丢掉真实结果听着可惜，但结构坏了只能二选一：要么丢它，要么留一段
|
||
* 服务端必然拒绝的历史。模型会从占位消息里知道该重跑。 */
|
||
CHECK(fixed == 2, "应补 1 处、丢 1 处孤儿，实际 %d", fixed);
|
||
CHECK(json_array_size(r) == 5,
|
||
"补一条、丢一条之后仍是 5 条，实际 %zu", json_array_size(r));
|
||
/* 占位消息必须紧跟在 assistant 的 tool 段里，而不是被塞到最后 */
|
||
CHECK(strcmp(nth_role(r, 2), "tool") == 0 &&
|
||
strcmp(nth_role(r, 3), "tool") == 0,
|
||
"两条 tool 消息应紧跟在 assistant 之后");
|
||
const char *c = json_get_string(json_array_get(r, 3), "content");
|
||
CHECK(c && strstr(c, "没有执行") == NULL && strstr(c, "没有得到结果") != NULL,
|
||
"占位消息要说明它没拿到结果，实际 %s", c ? c : "(null)");
|
||
json_free(r);
|
||
}
|
||
json_free(bad);
|
||
|
||
puts("末尾悬空：assistant 之后再没有消息");
|
||
JsonValue *tail = json_new_array();
|
||
json_array_push(tail, mk_plain("user", "跑一下"));
|
||
json_array_push(tail, mk_assistant(1));
|
||
fixed = 0;
|
||
r = agent_repair_tool_pairing(tail, &fixed);
|
||
CHECK(r && pairing_ok(r) && fixed == 1, "结尾缺的也要补上（fixed=%d）", fixed);
|
||
json_free(r);
|
||
json_free(tail);
|
||
|
||
puts("孤儿 tool 消息：没有对应的 tool_calls，要丢掉");
|
||
JsonValue *orphan = json_new_array();
|
||
json_array_push(orphan, mk_plain("user", "你好"));
|
||
json_array_push(orphan, mk_tool("谁也不是"));
|
||
JsonValue *r2 = agent_repair_tool_pairing(orphan, &fixed);
|
||
CHECK(fixed == 1, "孤儿 tool 消息应被计为 1 处，实际 %d", fixed);
|
||
CHECK(r2 != NULL && json_array_size(r2) == 1, "孤儿应被丢掉，只剩 user");
|
||
json_free(r2);
|
||
json_free(orphan);
|
||
|
||
puts("空输入与边界");
|
||
fixed = 7;
|
||
CHECK(agent_repair_tool_pairing(NULL, &fixed) == NULL && fixed == 0, "NULL 输入");
|
||
JsonValue *empty = json_new_array();
|
||
CHECK(agent_repair_tool_pairing(empty, &fixed) == NULL, "空历史没什么可修");
|
||
json_free(empty);
|
||
/* out_fixed 可以为 NULL */
|
||
JsonValue *b2 = json_new_array();
|
||
json_array_push(b2, mk_assistant(1));
|
||
JsonValue *r3 = agent_repair_tool_pairing(b2, NULL);
|
||
json_free(r3);
|
||
json_free(b2);
|
||
CHECK(1, "out_fixed 为 NULL 时不崩");
|
||
}
|
||
|
||
/* ------------------------------------------------------------------ */
|
||
/* 面向用户的历史口径：不含 system */
|
||
/* ------------------------------------------------------------------ */
|
||
|
||
static void test_conversation_len(void)
|
||
{
|
||
puts("历史条数：system 是配置，不该算进对话");
|
||
Agent a;
|
||
agent_init(&a);
|
||
|
||
CHECK(agent_history_len(&a) == 0, "空 agent 的数组长度是 0");
|
||
CHECK(agent_conversation_len(&a) == 0, "空 agent 的对话长度也是 0");
|
||
|
||
/* 只有 system：数组里 1 条，但对话是空的 */
|
||
agent_add_system(&a, "你是助手");
|
||
CHECK(agent_history_len(&a) == 1, "只有 system 时数组长度 1");
|
||
CHECK(agent_conversation_len(&a) == 0,
|
||
"只有 system 时**对话**长度必须是 0（用户报的就是这里）");
|
||
|
||
add_msg(&a, "user", "hi");
|
||
add_msg(&a, "assistant", "你好");
|
||
CHECK(agent_history_len(&a) == 3, "system + 一问一答 = 3");
|
||
CHECK(agent_conversation_len(&a) == 2, "对话长度 2");
|
||
|
||
/* 没有 system 的历史（比如裸构造的）不该被误减 */
|
||
Agent b;
|
||
agent_init(&b);
|
||
add_msg(&b, "user", "hi");
|
||
CHECK(agent_history_len(&b) == 1 && agent_conversation_len(&b) == 1,
|
||
"没有 system 时两个口径一致");
|
||
|
||
/* reset 之后回到"只有 system" */
|
||
agent_reset(&a);
|
||
CHECK(agent_history_len(&a) == 1 && agent_conversation_len(&a) == 0,
|
||
"reset 之后数组剩 system、对话为空");
|
||
|
||
agent_free(&a);
|
||
agent_free(&b);
|
||
}
|
||
|
||
int main(void)
|
||
{
|
||
Agent a;
|
||
agent_init(&a);
|
||
CHECK(a.messages && a.messages->type == JSON_ARRAY, "messages 应为数组");
|
||
|
||
agent_add_system(&a, "S1");
|
||
CHECK(agent_history_len(&a) == 1, "应只有 1 条 system");
|
||
CHECK(count_role(&a, "system") == 1, "system 计数");
|
||
|
||
agent_add_user(&a, "U1");
|
||
CHECK(agent_history_len(&a) == 2, "应 2 条");
|
||
|
||
/* 关键不变量：重复设置 system 应替换而不是追加 */
|
||
agent_add_system(&a, "S2");
|
||
CHECK(count_role(&a, "system") == 1, "重复设置 system 不应叠加，实际 %zu 条",
|
||
count_role(&a, "system"));
|
||
CHECK(strcmp(json_get_string(json_array_get(a.messages, 0), "content"), "S2") == 0,
|
||
"新 system 应在第 0 条");
|
||
CHECK(strcmp(json_get_string(json_array_get(a.messages, 1), "content"), "U1") == 0,
|
||
"原有对话应保留");
|
||
|
||
agent_add_user(&a, "U2");
|
||
CHECK(agent_history_len(&a) == 3, "应 3 条");
|
||
|
||
agent_reset(&a);
|
||
CHECK(agent_history_len(&a) == 1, "reset 后只保留 system，实际 %zu",
|
||
agent_history_len(&a));
|
||
CHECK(count_role(&a, "system") == 1, "reset 后 system 仍在");
|
||
CHECK(strcmp(json_get_string(json_array_get(a.messages, 0), "content"), "S2") == 0,
|
||
"reset 保留的是最新 system");
|
||
|
||
/* 历史里没有 system 时 reset 应得到空历史 */
|
||
Agent b;
|
||
agent_init(&b);
|
||
agent_add_user(&b, "only user");
|
||
agent_reset(&b);
|
||
CHECK(agent_history_len(&b) == 0, "无 system 时 reset 应为空，实际 %zu",
|
||
agent_history_len(&b));
|
||
|
||
agent_free(&a);
|
||
agent_free(&b);
|
||
test_take_rounds_keeps_pairing();
|
||
test_take_last_round();
|
||
test_repair_pairing();
|
||
test_conversation_len();
|
||
printf("\n%d 项检查，%d 项失败\n", checks, failures);
|
||
return failures ? 1 : 0;
|
||
}
|