OrgTJWater/TJWaterAgent
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

revert: remove unused Agent PostgreSQL persistence

Browse Source
This commit is contained in:
TJWater CI
2026-08-07 18:06:08 +08:00
parent 649af949c5
commit 2f267af7a3
39 changed files with 1330 additions and 2760 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.opencode/tools/dynamic_http_call.ts
+12 -9
View File
@@ -1,11 +1,14 @@
import { tool } from "@opencode-ai/plugin";
import { ToolSessionContextStore } from "../../src/session/toolContextStore.js";
const internalBaseUrl = process.env.TJWATER_AGENT_INTERNAL_BASE_URL ?? "http://127.0.0.1:8787";
const internalToken = process.env.TJWATER_AGENT_INTERNAL_TOKEN ?? "";
const toolContextStore = new ToolSessionContextStore();
const initializePromise = toolContextStore.initialize();
export default tool({
description:
"通过本地 Agent 桥接调用 TJWater 后端 API。支持 query 参数，且在 POST/PUT/PATCH/DELETE 时可选传递 JSON body。",
"通过本地 Agent 桥接调用 TJWater 后端 API。需提供 API 路径、可选的请求方法以及查询参数。",
args: {
reason: tool.schema
.string()
@@ -18,14 +21,15 @@ export default tool({
arguments: tool.schema
.record(tool.schema.string(), tool.schema.unknown())
.optional()
.describe("Query arguments object. Encoded into URL query string."),
body: tool.schema
.unknown()
.optional()
.describe("Optional JSON body for POST/PUT/PATCH/DELETE requests."),
.describe("Query arguments object."),
},
async execute(args, context) {
// 工具本身不直接持有用户 token；通过 runtimeSessionId 回调 Agent 服务，由服务侧补齐用户上下文。
await initializePromise;
const sessionContext = await toolContextStore.read(context.sessionID);
if (!sessionContext) {
throw new Error(`session context not found for ${context.sessionID}`);
}
// 工具本身不直接持有用户 token；通过 sessionID 回调 Agent 服务，由服务侧补齐用户上下文。
const response = await fetch(`${internalBaseUrl}/internal/tools/dynamic-http-call`, {
method: "POST",
headers: {
@@ -33,12 +37,11 @@ export default tool({
"x-agent-internal-token": internalToken,
},
body: JSON.stringify({
runtimeSessionId: context.sessionID,
sessionScopeKey: sessionContext.sessionScopeKey,
reason: args.reason,
path: args.path,
method: args.method,
arguments: args.arguments,
body: args.body,
}),
});
.opencode/tools/fetch_result_ref.ts
+9 -1
View File
@@ -1,7 +1,10 @@
import { tool } from "@opencode-ai/plugin";
import { ToolSessionContextStore } from "../../src/session/toolContextStore.js";
const internalBaseUrl = process.env.TJWATER_AGENT_INTERNAL_BASE_URL ?? "http://127.0.0.1:8787";
const internalToken = process.env.TJWATER_AGENT_INTERNAL_TOKEN ?? "";
const toolContextStore = new ToolSessionContextStore();
const initializePromise = toolContextStore.initialize();
export default tool({
description:
@@ -19,6 +22,11 @@ export default tool({
.describe("Optional maximum number of top-level items or fields to return."),
},
async execute(args, context) {
await initializePromise;
const sessionContext = await toolContextStore.read(context.sessionID);
if (!sessionContext) {
throw new Error(`session context not found for ${context.sessionID}`);
}
const response = await fetch(`${internalBaseUrl}/internal/tools/fetch-result-ref`, {
method: "POST",
headers: {
@@ -26,7 +34,7 @@ export default tool({
"x-agent-internal-token": internalToken,
},
body: JSON.stringify({
runtimeSessionId: context.sessionID,
sessionScopeKey: sessionContext.sessionScopeKey,
result_ref: args.result_ref,
max_items: args.max_items,
}),
.opencode/tools/memory_manager.ts
+6 -6
View File
@@ -1,12 +1,12 @@
import { tool } from "@opencode-ai/plugin";
import { MemoryStore } from "../../src/memory/store.js";
import { RuntimeSessionStore } from "../../src/session/runtimeSessionStore.js";
import { ToolSessionContextStore } from "../../src/session/toolContextStore.js";
const memoryStore = new MemoryStore();
const runtimeSessionStore = new RuntimeSessionStore();
const toolContextStore = new ToolSessionContextStore();
const initializePromise = Promise.all([
memoryStore.initialize(),
runtimeSessionStore.initialize(),
toolContextStore.initialize(),
]);
export default tool({
@@ -37,7 +37,7 @@ export default tool({
},
async execute(args, context) {
await initializePromise;
const sessionContext = await runtimeSessionStore.read(context.sessionID);
const sessionContext = await toolContextStore.read(context.sessionID);
if (!sessionContext) {
throw new Error(`session context not found for ${context.sessionID}`);
}
@@ -80,7 +80,7 @@ export default tool({
if (args.action === "add") {
const result = await memoryStore.upsert(scope, scopeKey, {
content: args.content ?? "",
sessionId: sessionContext.sessionId,
sessionId: sessionContext.clientSessionId,
source: "tool",
traceId: sessionContext.traceId,
});
@@ -105,7 +105,7 @@ export default tool({
if (args.action === "replace") {
const result = await memoryStore.replace(scope, scopeKey, args.target_id ?? "", {
content: args.content ?? "",
sessionId: sessionContext.sessionId,
sessionId: sessionContext.clientSessionId,
source: "tool",
traceId: sessionContext.traceId,
});
.opencode/tools/session_search.ts
+9 -1
View File
@@ -1,8 +1,11 @@
import { tool } from "@opencode-ai/plugin";
import { ToolSessionContextStore } from "../../src/session/toolContextStore.js";
const internalBaseUrl =
process.env.TJWATER_AGENT_INTERNAL_BASE_URL ?? "http://127.0.0.1:8787";
const internalToken = process.env.TJWATER_AGENT_INTERNAL_TOKEN ?? "";
const toolContextStore = new ToolSessionContextStore();
const initializePromise = toolContextStore.initialize();
export default tool({
description:
@@ -22,6 +25,11 @@ export default tool({
.describe("Optional maximum number of hits to return."),
},
async execute(args, context) {
await initializePromise;
const sessionContext = await toolContextStore.read(context.sessionID);
if (!sessionContext) {
throw new Error(`session context not found for ${context.sessionID}`);
}
const response = await fetch(`${internalBaseUrl}/internal/tools/session-search`, {
method: "POST",
headers: {
@@ -31,7 +39,7 @@ export default tool({
body: JSON.stringify({
max_results: args.max_results,
query: args.query,
runtimeSessionId: context.sessionID,
sessionScopeKey: sessionContext.sessionScopeKey,
}),
});
const text = await response.text();
.opencode/tools/skill_manager.ts
+4 -4
View File
@@ -1,10 +1,10 @@
import { tool } from "@opencode-ai/plugin";
import { SkillStore } from "../../src/skills/store.js";
import { RuntimeSessionStore } from "../../src/session/runtimeSessionStore.js";
import { ToolSessionContextStore } from "../../src/session/toolContextStore.js";
const runtimeSessionStore = new RuntimeSessionStore();
const initializePromise = runtimeSessionStore.initialize();
const toolContextStore = new ToolSessionContextStore();
const initializePromise = toolContextStore.initialize();
const skillStore = new SkillStore();
export default tool({
@@ -51,7 +51,7 @@ export default tool({
},
async execute(args, context) {
await initializePromise;
const sessionContext = await runtimeSessionStore.read(context.sessionID);
const sessionContext = await toolContextStore.read(context.sessionID);
if (!sessionContext) {
throw new Error(`session context not found for ${context.sessionID}`);
}
.opencode/tools/store_render_ref.ts
+9 -1
View File
@@ -1,7 +1,10 @@
import { tool } from "@opencode-ai/plugin";
import { ToolSessionContextStore } from "../../src/session/toolContextStore.js";
const internalBaseUrl = process.env.TJWATER_AGENT_INTERNAL_BASE_URL ?? "http://127.0.0.1:8787";
const internalToken = process.env.TJWATER_AGENT_INTERNAL_TOKEN ?? "";
const toolContextStore = new ToolSessionContextStore();
const initializePromise = toolContextStore.initialize();
export default tool({
description:
@@ -17,6 +20,11 @@ export default tool({
),
},
async execute(args, context) {
await initializePromise;
const sessionContext = await toolContextStore.read(context.sessionID);
if (!sessionContext) {
throw new Error(`session context not found for ${context.sessionID}`);
}
const response = await fetch(`${internalBaseUrl}/internal/tools/store-render-ref`, {
method: "POST",
headers: {
@@ -24,7 +32,7 @@ export default tool({
"x-agent-internal-token": internalToken,
},
body: JSON.stringify({
runtimeSessionId: context.sessionID,
sessionScopeKey: sessionContext.sessionScopeKey,
file_path: args.file_path,
}),
});