diff --git a/ui/components/ChatWindow.tsx b/ui/components/ChatWindow.tsx
index b26573f..1be4ec5 100644
--- a/ui/components/ChatWindow.tsx
+++ b/ui/components/ChatWindow.tsx
@@ -1,17 +1,18 @@
'use client';
-import { useEffect, useRef, useState } from 'react';
-import { Document } from '@langchain/core/documents';
+import {useEffect, useRef, useState} from 'react';
+import {Document} from '@langchain/core/documents';
import Navbar from './Navbar';
import Chat from './Chat';
import EmptyChat from './EmptyChat';
import crypto from 'crypto';
-import { toast } from 'sonner';
-import { useSearchParams } from 'next/navigation';
-import { getSuggestions } from '@/lib/actions';
-import { Settings } from 'lucide-react';
+import {toast} from 'sonner';
+import {useSearchParams} from 'next/navigation';
+import {getSuggestions} from '@/lib/actions';
+import {Settings} from 'lucide-react';
import SettingsDialog from './SettingsDialog';
import NextError from 'next/error';
+import {Mcid} from "@/lib/mcid";
export type Message = {
messageId: string;
@@ -140,7 +141,7 @@ const useSocket = (
if (
Object.keys(chatModelProviders).length > 0 &&
(((!openAIBaseURL || !openAIPIKey) &&
- chatModelProvider === 'custom_openai') ||
+ chatModelProvider === 'custom_openai') ||
!chatModelProviders[chatModelProvider])
) {
const chatModelProvidersKeys = Object.keys(chatModelProviders);
@@ -173,7 +174,7 @@ const useSocket = (
Object.keys(chatModelProviders[chatModelProvider]).length > 0
? chatModelProvider
: Object.keys(chatModelProviders)[0]
- ],
+ ],
)[0];
localStorage.setItem('chatModel', chatModel);
}
@@ -382,10 +383,10 @@ const loadMessages = async (
setIsMessagesLoaded(true);
};
-const ChatWindow = ({ id }: { id?: string }) => {
+const ChatWindow = ({id}: { id?: string }) => {
const searchParams = useSearchParams();
const initialMessage = searchParams.get('q');
-
+ const [userId, setUserId] = useState();
const [chatId, setChatId] = useState(id);
const [newChatCreated, setNewChatCreated] = useState(false);
@@ -410,6 +411,7 @@ const ChatWindow = ({ id }: { id?: string }) => {
const [focusMode, setFocusMode] = useState('webSearch');
const [optimizationMode, setOptimizationMode] = useState('speed');
+ const [copilotEnabled, setCopilotEnabled] = useState(true);
const [isMessagesLoaded, setIsMessagesLoaded] = useState(false);
@@ -417,6 +419,33 @@ const ChatWindow = ({ id }: { id?: string }) => {
const [isSettingsOpen, setIsSettingsOpen] = useState(false);
+ useEffect(() => {
+ const initializeUserId = () => {
+ try {
+ // 从 localStorage 读取现有用户 ID
+ const storedUserId = localStorage.getItem('userId');
+
+ if (storedUserId) {
+ setUserId(storedUserId);
+ console.debug('Using existing user ID:', storedUserId);
+ } else {
+ const newUserId = new Mcid().generate().toString(); // 转换为字符串
+
+ localStorage.setItem('userId', newUserId);
+ setUserId(newUserId);
+ console.debug('Generated new user ID:', newUserId);
+ }
+ } catch (error) {
+ console.error('Error initializing user ID:', error);
+ const fallbackId = "1234567890";
+ localStorage.setItem('userId', fallbackId);
+ setUserId(fallbackId);
+ }
+ };
+
+ initializeUserId();
+ }, []);
+
useEffect(() => {
if (
chatId &&
@@ -437,7 +466,7 @@ const ChatWindow = ({ id }: { id?: string }) => {
} else if (!chatId) {
setNewChatCreated(true);
setIsMessagesLoaded(true);
- setChatId(crypto.randomBytes(20).toString('hex'));
+ setChatId(new Mcid().generate().toString());
}
// eslint-disable-next-line react-hooks/exhaustive-deps
}, []);
@@ -465,7 +494,7 @@ const ChatWindow = ({ id }: { id?: string }) => {
} else {
setIsReady(false);
}
- }, [isMessagesLoaded, isWSReady]);
+ }, [isMessagesLoaded, isWSReady, userId]);
const sendMessage = async (message: string, messageId?: string) => {
if (loading) return;
@@ -481,11 +510,12 @@ const ChatWindow = ({ id }: { id?: string }) => {
let recievedMessage = '';
let added = false;
- messageId = messageId ?? crypto.randomBytes(7).toString('hex');
+ messageId = messageId ?? new Mcid().generate().toString();
ws.send(
JSON.stringify({
type: 'message',
+ useerId:userId,
message: {
messageId: messageId,
chatId: chatId!,
@@ -556,7 +586,7 @@ const ChatWindow = ({ id }: { id?: string }) => {
setMessages((prev) =>
prev.map((message) => {
if (message.messageId === data.messageId) {
- return { ...message, content: message.content + data.data };
+ return {...message, content: message.content + data.data};
}
return message;
@@ -589,7 +619,7 @@ const ChatWindow = ({ id }: { id?: string }) => {
setMessages((prev) =>
prev.map((msg) => {
if (msg.messageId === lastMsg.messageId) {
- return { ...msg, suggestions: suggestions };
+ return {...msg, suggestions: suggestions};
}
return msg;
}),
@@ -639,19 +669,19 @@ const ChatWindow = ({ id }: { id?: string }) => {
Failed to connect to the server. Please try again later.
-
+
);
}
return isReady ? (
notFound ? (
-
+
) : (
{messages.length > 0 ? (
<>
-
+
{
void;
focusMode: string;
setFocusMode: (mode: string) => void;
+ copilotEnabled: boolean;
+ setCopilotEnabled: (enabled: boolean) => void;
optimizationMode: string;
setOptimizationMode: (mode: string) => void;
fileIds: string[];
@@ -44,6 +48,8 @@ const EmptyChat = ({
sendMessage={sendMessage}
focusMode={focusMode}
setFocusMode={setFocusMode}
+ copilotEnabled={copilotEnabled}
+ setCopilotEnabled={setCopilotEnabled}
optimizationMode={optimizationMode}
setOptimizationMode={setOptimizationMode}
fileIds={fileIds}
diff --git a/ui/components/EmptyChatMessageInput.tsx b/ui/components/EmptyChatMessageInput.tsx
index 43d1e28..195b8d3 100644
--- a/ui/components/EmptyChatMessageInput.tsx
+++ b/ui/components/EmptyChatMessageInput.tsx
@@ -11,6 +11,8 @@ const EmptyChatMessageInput = ({
sendMessage,
focusMode,
setFocusMode,
+ copilotEnabled,
+ setCopilotEnabled,
optimizationMode,
setOptimizationMode,
fileIds,
@@ -23,12 +25,13 @@ const EmptyChatMessageInput = ({
setFocusMode: (mode: string) => void;
optimizationMode: string;
setOptimizationMode: (mode: string) => void;
+ copilotEnabled:boolean
+ setCopilotEnabled:(mode: boolean) => void;
fileIds: string[];
setFileIds: (fileIds: string[]) => void;
files: File[];
setFiles: (files: File[]) => void;
}) => {
- const [copilotEnabled, setCopilotEnabled] = useState(false);
const [message, setMessage] = useState('');
const inputRef = useRef(null);
@@ -94,6 +97,7 @@ const EmptyChatMessageInput = ({
/>
+
{
return (
-
+
{children}
);
diff --git a/ui/lib/mcid.ts b/ui/lib/mcid.ts
new file mode 100644
index 0000000..1a0d0d3
--- /dev/null
+++ b/ui/lib/mcid.ts
@@ -0,0 +1,38 @@
+// utils/mcid.ts
+export class Mcid {
+ private lastTimestamp = -1;
+ private sequence = 0;
+
+ constructor(private readonly machineId: number = 0) {
+ if (machineId < 0 || machineId > 255) {
+ throw new Error('Machine ID must be between 0 and 255');
+ }
+ }
+
+ generate(): number {
+ let now = Date.now();
+
+ if (now < this.lastTimestamp) {
+ throw new Error('Clock moved backwards');
+ }
+
+ if (now === this.lastTimestamp) {
+ this.sequence = (this.sequence + 1) & 0xf;
+ if (this.sequence === 0) {
+ while (now <= this.lastTimestamp) {
+ now = Date.now();
+ }
+ }
+ } else {
+ this.sequence = 0;
+ }
+
+ this.lastTimestamp = now;
+
+ return (
+ (now * 0x1000) + // 时间戳左移 12 位
+ (this.machineId * 16) + // 机器 ID 左移 4 位
+ this.sequence // 序列号
+ );
+ }
+}