refactor(rag): fix issues: missing cookie & timestamp format (#797)

refactor(rag): fix issues: missing cookie in knowledge page requests and incorrect timestamp format

Author: ch-liuzhide

client/app/hooks/useRAG.ts

@@ -14,10 +14,9 @@ import {
   useQuery,
   UseQueryOptions,
 } from '@tanstack/react-query';
-import { useMutation, UseMutationOptions } from '@tanstack/react-query';
-
-
-
+import { useMutation } from '@tanstack/react-query';
+import axios from 'axios';
+axios.defaults.withCredentials = true;
 
 export function useKnowledgeList(
   params: PageParams<RAGKnowledge>,

client/app/knowledge/components/KnowledgeList.tsx

@@ -24,8 +24,7 @@ export default function KnowledgeList({ repo_name }: { repo_name: string }) {
     },
   });
 
-  const { data, isLoading, error, isFetching, refetch } =
-    useKnowledgeList(pageParams);
+  const { data, isLoading } = useKnowledgeList(pageParams);
   return (
     <div className="w-full flex-1 px-4 py-6 flex flex-col min-h-[calc(100vh-200px)]">
       <MySpinner loading={isLoading}>

server/rag/router.py

@@ -106,7 +106,7 @@ async def get_knowledge_list(
             token=get_env_variable("WHISKER_API_KEY"),
         )
         res = await api_client.knowledge.get_knowledge_list(**params.model_dump())
-        return res
+        return res.model_dump()
     except Exception as e:
         return json.dumps({"success": False, "message": str(e)})
 
@@ -119,7 +119,7 @@ async def get_chunk_list(params: PageParams[Chunk]):
             token=get_env_variable("WHISKER_API_KEY"),
         )
         res = await api_client.chunk.get_chunk_list(**params.model_dump())
-        return res
+        return res.model_dump()
     except Exception as e:
         return json.dumps({"success": False, "message": str(e)})
 
@@ -139,7 +139,7 @@ async def get_rag_task(
             token=get_env_variable("WHISKER_API_KEY"),
         )
         res = await api_client.task.get_task_list(**params.model_dump())
-        return res
+        return res.model_dump()
     except Exception as e:
         return json.dumps({"success": False, "message": str(e)})
 
@@ -159,7 +159,7 @@ async def restart_rag_task(
             token=get_env_variable("WHISKER_API_KEY"),
         )
         res = await api_client.task.restart_task(params.task_id_list)
-        return res
+        return res.model_dump()
     except Exception as e:
         return json.dumps({"success": False, "message": str(e)})
 
@@ -192,4 +192,4 @@ async def retrievalBot(
             metadata_filter=params.metadata_filter,
         )
     )
-    return retrieval_res
+    return retrieval_res

server/requirements.txt

@@ -26,4 +26,4 @@ requests
 httpx==0.27.2
 urllib3>=2.2.2
 toolz
-whiskerrag>=0.0.27
+whiskerrag>=0.0.29