51 lines
1.4 KiB
TypeScript
51 lines
1.4 KiB
TypeScript
export type RuntimeSessionContext = {
|
|
accessToken?: string;
|
|
actorKey: string;
|
|
authExpired?: {
|
|
message: string;
|
|
reason: "access_token_expired" | "access_token_rejected";
|
|
};
|
|
allowLearningWrite?: boolean;
|
|
clientSessionId: string;
|
|
learningMode?: "interactive" | "review";
|
|
memoryListReadScopes?: Partial<Record<"user" | "workspace", boolean>>;
|
|
network?: string;
|
|
projectId?: string;
|
|
projectKey: string;
|
|
sessionId: string;
|
|
tokenExpiresAt?: string;
|
|
traceId: string;
|
|
};
|
|
|
|
const contexts = new Map<string, RuntimeSessionContext>();
|
|
|
|
export const setRuntimeSessionContext = (context: RuntimeSessionContext) => {
|
|
contexts.set(context.sessionId, { ...context });
|
|
};
|
|
|
|
export const getRuntimeSessionContext = (sessionId: string) => {
|
|
const context = contexts.get(sessionId);
|
|
return context ? { ...context } : null;
|
|
};
|
|
|
|
export const markRuntimeSessionAuthExpired = (
|
|
sessionId: string,
|
|
reason: NonNullable<RuntimeSessionContext["authExpired"]>["reason"],
|
|
message = "登录态已过期，请刷新登录后重试",
|
|
) => {
|
|
const context = contexts.get(sessionId);
|
|
if (!context) {
|
|
return null;
|
|
}
|
|
const nextContext: RuntimeSessionContext = {
|
|
...context,
|
|
authExpired: { message, reason },
|
|
};
|
|
contexts.set(sessionId, nextContext);
|
|
return { ...nextContext };
|
|
};
|
|
|
|
export const removeRuntimeSessionContext = (sessionId: string) => {
|
|
contexts.delete(sessionId);
|
|
};
|