Merge branch 'dev' into feature/cog-2734-cognee-feedbacks-interactions-poc-to-prod

cognee-frontend/src/modules/chat/hooks/useChat.ts

@@ -14,7 +14,7 @@ const fetchMessages = () => {
     .then(response => response.json());
 };
 
-const sendMessage = (message: string, searchType: string) => {
+const sendMessage = (message: string, searchType: string, topK: number) => {
   return fetch("/v1/search/", {
     method: "POST",
     headers: {
@@ -24,6 +24,7 @@ const sendMessage = (message: string, searchType: string) => {
       query: message,
       searchType,
       datasets: ["main_dataset"],
+      top_k: topK,
     }),
   })
     .then(response => response.json());
@@ -45,7 +46,7 @@ export default function useChat(dataset: Dataset) {
     return setMessages(data);
   }, []);
 
-  const handleMessageSending = useCallback((message: string, searchType: string) => {
+  const handleMessageSending = useCallback((message: string, searchType: string, topK: number) => {
     const sentMessageId = v4();
 
     setMessages((messages) => [
@@ -59,7 +60,7 @@ export default function useChat(dataset: Dataset) {
 
     disableSearchRun();
 
-    return sendMessage(message, searchType)
+    return sendMessage(message, searchType, topK)
       .then(newMessages => {
         setMessages((messages) => [
           ...messages,

cognee-frontend/src/ui/Partials/SearchView/SearchView.tsx

@@ -4,7 +4,7 @@ import classNames from "classnames";
 import { useCallback, useEffect, useRef, useState } from "react";
 
 import { LoadingIndicator } from "@/ui/App";
-import { CTAButton, Select, TextArea } from "@/ui/elements";
+import { CTAButton, Select, TextArea, Input } from "@/ui/elements";
 import useChat from "@/modules/chat/hooks/useChat";
 
 import styles from "./SearchView.module.css";
@@ -59,17 +59,28 @@ export default function SearchView() {
   }, [refreshChat, scrollToBottom]);
 
   const [searchInputValue, setSearchInputValue] = useState("");
+  // Add state for top_k
+  const [topK, setTopK] = useState(10);
 
   const handleSearchInputChange = useCallback((value: string) => {
     setSearchInputValue(value);
   }, []);
 
+  // Add handler for top_k input
+  const handleTopKChange = useCallback((e: React.ChangeEvent<HTMLInputElement>) => {
+    let value = parseInt(e.target.value, 10);
+    if (isNaN(value)) value = 10;
+    if (value < 1) value = 1;
+    if (value > 100) value = 100;
+    setTopK(value);
+  }, []);
+
   const handleChatMessageSubmit = useCallback((event: React.FormEvent<SearchFormPayload>) => {
     event.preventDefault();
 
     const formElements = event.currentTarget;
-
     const searchType = formElements.searchType.value;
+
     const chatInput = searchInputValue.trim();
 
     if (chatInput === "") {
@@ -79,10 +90,11 @@ export default function SearchView() {
     scrollToBottom();
 
     setSearchInputValue("");
-
+    
-    sendMessage(chatInput, searchType)
+    // Pass topK to sendMessage
+    sendMessage(chatInput, searchType, topK)
       .then(scrollToBottom)
-  }, [scrollToBottom, sendMessage, searchInputValue]);
+  }, [scrollToBottom, sendMessage, searchInputValue, topK]);
 
   const chatFormRef = useRef<HTMLFormElement>(null);
 
@@ -125,13 +137,30 @@ export default function SearchView() {
             className="resize-none min-h-14 max-h-96 overflow-y-auto"
           />
           <div className="flex flex-row items-center justify-between gap-4">
-            <div className="flex flex-row items-center gap-2">
+            <div className="flex flex-row items-center gap-4">
-              <label className="text-gray-600 whitespace-nowrap">Search type:</label>
+              <div className="flex flex-row items-center gap-2">
-              <Select name="searchType" defaultValue={searchOptions[0].value} className="max-w-2xs">
+                <label className="text-gray-600 whitespace-nowrap">Search type:</label>
-                {searchOptions.map((option) => (
+                <Select name="searchType" defaultValue={searchOptions[0].value} className="max-w-2xs">
-                  <option key={option.value} value={option.value}>{option.label}</option>
+                  {searchOptions.map((option) => (
-                ))}
+                    <option key={option.value} value={option.value}>{option.label}</option>
-              </Select>
+                  ))}
+                </Select>
+              </div>
+              <div className="flex flex-row items-center gap-2">
+                <label className="text-gray-600 whitespace-nowrap" title="Controls how many results to return. Smaller = focused, larger = broader graph exploration.">
+                  Max results:
+                </label>
+                <Input
+                  type="number"
+                  name="topK"
+                  min={1}
+                  max={100}
+                  value={topK}
+                  onChange={handleTopKChange}
+                  className="w-20"
+                  title="Controls how many results to return. Smaller = focused, larger = broader graph exploration."
+                />
+              </div>
             </div>
             <CTAButton disabled={isSearchRunning} type="submit">
               {isSearchRunning? "Searching..." : "Search"}
@@ -142,4 +171,4 @@ export default function SearchView() {
       </form>
     </div>
   );
-}
+}

cognee/infrastructure/databases/graph/kuzu/adapter.py

@@ -138,8 +138,9 @@ class KuzuAdapter(GraphDBInterface):
 
             s3_file_storage = S3FileStorage("")
 
-            async with self.KUZU_ASYNC_LOCK:
+            if self.connection:
-                self.connection.execute("CHECKPOINT;")
+                async with self.KUZU_ASYNC_LOCK:
+                    self.connection.execute("CHECKPOINT;")
 
             s3_file_storage.s3.put(self.temp_graph_file, self.db_path, recursive=True)