Function useChat

ui/src/composables/useChat.ts:18–595

(config: ChatConfig)

import { useWorkflowStore } from "@/stores/workflow";

export function useChat(config: ChatConfig) {
  // 初始化 Pinia Stores
  const chatStore = useChatStore();
  const thinkingStore = useThinkingStore();
  const toolsStore = useToolsStore();
  const todosStore = useTodosStore();
  const approvalStore = useApprovalStore();
  const workflowStore = useWorkflowStore();

  // 保留部分本地状态 (不适合放在 store 中的)
  const currentInput = ref("");
  const demoConnection = ref(true);
  const isDemoMode = config.demoMode ?? true;
  const pendingAskUser = ref<{ requestId: string; questions: Question[] } | null>(null);
  const agent = ref<Agent>({
    id: config.agentId || "demo-agent",
    name: config.agentProfile?.name || "Aster Copilot",
    description: config.agentProfile?.description || "多模态执行、自动规划、符合企业安全的 Agent",
    avatar: config.agentProfile?.avatar,
    status: "idle",
    metadata: {
      model: "aster:builder",
    },
  });
  const demoCursor = ref(0);

  const apiUrl = config.apiUrl || import.meta.env.VITE_API_URL || "http://localhost:8080";
  const wsUrlOverride = config.wsUrl || import.meta.env.VITE_WS_URL;

  const { client } = useAsterClient({
    baseUrl: apiUrl,
    apiKey: config.apiKey,
    wsUrl: wsUrlOverride,
  });

  const { connect, getInstance, isConnected: wsConnected } = useWebSocket();
  // connectionState 用于组件中判断连接状态
  const connectionState = computed(() => (isDemoMode ? demoConnection.value : wsConnected.value));

  // 初始化 WebSocket 连接
  onMounted(async () => {
    if (!isDemoMode) {
      const wsUrl = wsUrlOverride || apiUrl.replace(/^http/, "ws") + "/v1/ws";
      console.log("🚀 Initializing WebSocket connection to:", wsUrl);
      try {
        await connect(wsUrl);
        console.log("✅ WebSocket initialized in useChat");
      } catch (error) {
        console.error("❌ Failed to initialize WebSocket:", error);
      }
    }
  });

  const fallbackResponses = [
    "我已经为你生成了一个新的多 Agent 工作流，包含大纲、评价器和部署策略。",
    "Aster 的沙箱已准备好，所有写入都被限制在 /workspace 目录，你可以放心执行指令。",
    "我为这个会话自动挂载了上下文记忆，后续可以直接引用历史工单。",
    "Streaming 模式已打开，等待后端返回 token，平均延迟 220ms。",