Function useChatTitles

src/renderer/src/hooks/useChatTitles.ts:36–141

({
  pageId,
  messages,
  currentPath
}: UseChatTitlesOptions)

}

export function useChatTitles({
  pageId,
  messages,
  currentPath
}: UseChatTitlesOptions): UseChatTitlesResult {
  const [batchProgress, setBatchProgress] = useState<BatchProgress | null>(null)

  const updateTitle = useCallback(
    (messageId: string, title: string) => {
      updateMessageTitle(pageId, messageId, title)
    },
    [pageId]
  )

  const deleteTitle = useCallback(
    (messageId: string) => {
      updateMessageTitle(pageId, messageId, '')
    },
    [pageId]
  )

  // 统一的 generateTitle 方法，通过 options 决定是否使用自定义配置
  const generateTitle = useCallback(
    async (messageId: string, options?: GenerateOptions) => {
      const message = messages.find((m) => m.id === messageId)
      if (!message) return

      try {
        const generation = getSwitchGeneration()
        const result =
          options?.llmId || options?.extraRequirements || options?.modelConfigId
            ? await generateMessageTitleWithOptions({
                content: message.content,
                extraRequirements: options.extraRequirements,
                llmId: options.llmId,
                modelConfigId: options.modelConfigId
              })
            : await generateMessageTitle(message.content)

        if (generation === getSwitchGeneration() && result.success && result.title) {
          updateMessageTitle(pageId, messageId, result.title)
        } else if (result.error) {
          console.error('Failed to generate title:', result.error)
        }
      } catch (error) {
        console.error('Failed to generate title:', error)
      }
    },
    [pageId, messages]
  )

  // 统一的批量生成标题方法，消除了原来两个几乎相同的方法
  const batchGenerateTitles = useCallback(
    async (options?: GenerateOptions) => {
      // 筛选出没有标题的消息
      const messagesWithoutTitle = currentPath.filter((m) => !m.title)
      if (messagesWithoutTitle.length === 0) return