update

pbl.js

@@ -47,7 +47,7 @@ async function calculateVectors(content) {
     }
 }
 
-async function findMatchingContents(content) {
+async function findMatchingContents(discipline, content) {
     const contentVector = await calculateVectors(content);
     const formattedContentVector = contentVector.length > 0 ? `[` + contentVector.join(', ') + `]` : null;
 
@@ -55,25 +55,23 @@ async function findMatchingContents(content) {
     try {
         // 使用余弦相似度进行内容向量匹配
         const query = `
-            SELECT title, id, content,
-                   1 - (content_vector <#> $1::vector) AS contentSimilarity
-            FROM knowledgefiles
+            SELECT id, content,
+                      1 - (content_vector <#> $1::vector) AS contentSimilarity
+            FROM (select id, content, content_vector from knowledgefiles as a where top_dir_id = (select id from directory_files where folder_name = $2 limit 1)) as a
             ORDER BY contentSimilarity DESC
             LIMIT 5;
         `;
 
-        const result = await client.query(query, [formattedContentVector]);
-
+        const result = await client.query(query, [formattedContentVector, discipline]);
         /*
-        console.log(result.rows);
-        // 将查询结果转换为匹配对象
-        const matches = result.rows.map(row => ({
-            id: row.id,
-            contentSimilarity: row.contentsimilarity
-        }));
-        console.log(matches);
-        */
-
+                    console.log(result.rows);
+                    // 将查询结果转换为匹配对象
+                    const matches = result.rows.map(row => ({
+                        id: row.id,
+                        contentSimilarity: row.contentsimilarity
+                    }));
+                    console.log(matches);
+                    */
         return result.rows; // 返回匹配结果
     } catch (error) {
         console.error('Error finding matching contents:', error);
@@ -86,7 +84,7 @@ async function findMatchingContents(content) {
 /*检索文件*/
 router.route("/findMatchingContents").all(async (req, res, next) => {
     var json = queryString(req.url);
-    const titleMatches = await findMatchingContents(json["string"]);
+    const titleMatches = await findMatchingContents(json["type"], json["string"]);
     res.end(JSON.stringify(titleMatches));
 });