增加获取用户 ID 的功能,Agent chat 请求头新增传递 userId
This commit is contained in:
@@ -1,12 +1,14 @@
|
||||
import { getAccessToken } from "@/lib/authToken";
|
||||
import { getAccessToken, getUserId } from "@/lib/authToken";
|
||||
import { useProjectStore } from "@/store/projectStore";
|
||||
|
||||
export type AuthHeaderMode = "include" | "omit";
|
||||
export type ProjectHeaderMode = "auto" | "include" | "omit";
|
||||
export type UserHeaderMode = "include" | "omit";
|
||||
|
||||
export interface AuthContextHeaderOptions {
|
||||
authHeaderMode?: AuthHeaderMode;
|
||||
projectHeaderMode?: ProjectHeaderMode;
|
||||
userHeaderMode?: UserHeaderMode;
|
||||
}
|
||||
|
||||
const shouldIncludeProjectHeader = (
|
||||
@@ -34,6 +36,13 @@ export const applyAuthContextHeaders = async (
|
||||
headers.set("Authorization", `Bearer ${accessToken}`);
|
||||
}
|
||||
|
||||
if (options.userHeaderMode === "include") {
|
||||
const userId = await getUserId();
|
||||
if (userId) {
|
||||
headers.set("X-User-Id", userId);
|
||||
}
|
||||
}
|
||||
|
||||
const projectId = useProjectStore.getState().currentProjectId;
|
||||
if (
|
||||
projectId &&
|
||||
|
||||
Reference in New Issue
Block a user