From bb8570c6e7bc5c4ab9c91050ef1150b3ace82f2e Mon Sep 17 00:00:00 2001
From: cir9no <44470218+cir9no@users.noreply.github.com>
Date: Mon, 30 Sep 2024 17:10:11 +0800
Subject: [PATCH] feat(seasearch): add dir filter

---
 seahub/api2/views.py   | 2 +-
 seahub/search/utils.py | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/seahub/api2/views.py b/seahub/api2/views.py
index b1344f2b4b1..528ca7f2dff 100644
--- a/seahub/api2/views.py
+++ b/seahub/api2/views.py
@@ -678,7 +678,7 @@ def get(self, request, format=None):
                 repos = [(repo.id, repo.origin_repo_id, repo.origin_path, repo.name)]
 
             searched_repos, repos_map = format_repos(repos)
-            results, total = ai_search_files(keyword, searched_repos, per_page, suffixes, search_path)
+            results, total = ai_search_files(keyword, searched_repos, per_page, suffixes, search_path, obj_type)
 
             for f in results:
                 repo_id = f['repo_id']
diff --git a/seahub/search/utils.py b/seahub/search/utils.py
index ba3794f8a86..8feec4a9f67 100644
--- a/seahub/search/utils.py
+++ b/seahub/search/utils.py
@@ -201,13 +201,14 @@ def search_files(repos_map, search_path, keyword, obj_desc, start, size, org_id=
     return result, total
 
 
-def ai_search_files(keyword, searched_repos, count, suffixes, search_path):
+def ai_search_files(keyword, searched_repos, count, suffixes, search_path, obj_type):
     params = {
         'query': keyword,
         'repos': searched_repos,
         'count': count,
         'suffixes': suffixes,
         'search_path': search_path,
+        'obj_type': obj_type
     }
 
     resp = search(params)