Browse Source

feat: support systemPrompt in createSession and updateSession

Co-authored-by: CodeFree <codefree@chinatelcom.cn>
feat/agent-independence
npmrun 2 months ago
parent
commit
76280a25a1
  1. 4
      server/service/agent/session.ts

4
server/service/agent/session.ts

@ -10,6 +10,7 @@ export async function createSession(params: {
modelId?: number | null; modelId?: number | null;
enableThinking?: number | null; enableThinking?: number | null;
enableTools?: number | null; enableTools?: number | null;
systemPrompt?: string | null;
expiresAt?: Date | null; expiresAt?: Date | null;
}): Promise<AgentSessionRow> { }): Promise<AgentSessionRow> {
const id = `as_${Date.now().toString(36)}_${Math.random().toString(36).slice(2, 10)}`; const id = `as_${Date.now().toString(36)}_${Math.random().toString(36).slice(2, 10)}`;
@ -23,6 +24,7 @@ export async function createSession(params: {
modelId: params.modelId ?? null, modelId: params.modelId ?? null,
enableThinking: params.enableThinking ?? null, enableThinking: params.enableThinking ?? null,
enableTools: params.enableTools ?? null, enableTools: params.enableTools ?? null,
systemPrompt: params.systemPrompt ?? null,
expiresAt: params.expiresAt ?? null, expiresAt: params.expiresAt ?? null,
}) })
.returning(); .returning();
@ -107,6 +109,7 @@ export async function updateSession(
modelId?: number | null; modelId?: number | null;
enableThinking?: number | null; enableThinking?: number | null;
enableTools?: number | null; enableTools?: number | null;
systemPrompt?: string | null;
lastActiveAt?: Date; lastActiveAt?: Date;
}, },
): Promise<void> { ): Promise<void> {
@ -116,6 +119,7 @@ export async function updateSession(
if (data.modelId !== undefined) updates.modelId = data.modelId; if (data.modelId !== undefined) updates.modelId = data.modelId;
if (data.enableThinking !== undefined) updates.enableThinking = data.enableThinking; if (data.enableThinking !== undefined) updates.enableThinking = data.enableThinking;
if (data.enableTools !== undefined) updates.enableTools = data.enableTools; if (data.enableTools !== undefined) updates.enableTools = data.enableTools;
if (data.systemPrompt !== undefined) updates.systemPrompt = data.systemPrompt;
if (data.lastActiveAt !== undefined) updates.lastActiveAt = data.lastActiveAt; if (data.lastActiveAt !== undefined) updates.lastActiveAt = data.lastActiveAt;
if (Object.keys(updates).length === 0) return; if (Object.keys(updates).length === 0) return;
await dbGlobal.update(agentSessions).set(updates).where(eq(agentSessions.id, id)); await dbGlobal.update(agentSessions).set(updates).where(eq(agentSessions.id, id));

Loading…
Cancel
Save