Ryan/docmost
1
0
Fork 0
mirror of https://github.com/docmost/docmost.git synced 2025-11-10 08:52:05 +10:00
Code Issues Packages Projects Releases Wiki Activity

reset ai answer on query change

Browse Source
This commit is contained in:
Philipinho
2025-10-26 01:09:34 +01:00
parent 70d151056b
commit 2710bcb879
2 changed files with 18 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
apps/client/src/ee/ai/hooks/use-ai-search.ts
View File

@ -1,5 +1,5 @@
import { useMutation, UseMutationResult } from "@tanstack/react-query";
import { useState } from "react";
import { useState, useCallback } from "react";
import { askAi, IAiSearchResponse } from "@/ee/ai/services/ai-search-service.ts";
import { IPageSearchParams } from "@/features/search/types/search.types.ts";
@ -7,12 +7,18 @@ import { IPageSearchParams } from "@/features/search/types/search.types.ts";
interface UseAiSearchResult extends UseMutationResult<IAiSearchResponse, Error, IPageSearchParams> {
streamingAnswer: string;
streamingSources: any[];
clearStreaming: () => void;
}
export function useAiSearch(): UseAiSearchResult {
const [streamingAnswer, setStreamingAnswer] = useState("");
const [streamingSources, setStreamingSources] = useState<any[]>([]);
const clearStreaming = useCallback(() => {
setStreamingAnswer("");
setStreamingSources([]);
}, []);
const mutation = useMutation({
mutationFn: async (params: IPageSearchParams & { contentType?: string }) => {
setStreamingAnswer("");
@ -35,5 +41,6 @@ export function useAiSearch(): UseAiSearchResult {
...mutation,
streamingAnswer,
streamingSources,
clearStreaming,
};
}

11
apps/client/src/features/search/components/search-spotlight.tsx
View File

@ -1,7 +1,7 @@
import { Spotlight } from "@mantine/spotlight";
import { IconSearch, IconSparkles } from "@tabler/icons-react";
import { Group, Button } from "@mantine/core";
import React, { useState, useMemo } from "react";
import React, { useState, useMemo, useEffect } from "react";
import { useDebouncedValue } from "@mantine/hooks";
import { useTranslation } from "react-i18next";
import { searchSpotlightStore } from "../constants.ts";
@ -55,10 +55,19 @@ export function SearchSpotlight({ spaceId }: SearchSpotlightProps) {
isPending: isAiLoading,
//@ts-ignore
mutate: triggerAiSearchMutation,
//@ts-ignore
reset: resetAiMutation,
streamingAnswer,
streamingSources,
clearStreaming,
} = useAiSearch();
// Clear streaming state and mutation data when query changes (user is typing a new query)
useEffect(() => {
clearStreaming();
resetAiMutation();
}, [query, clearStreaming, resetAiMutation]);
// Determine result type for rendering
const isAttachmentSearch =
filters.contentType === "attachment" && (hasLicenseKey || isCloud());