feat: improve ai filter

koishijs · Mar 4, 2024 · fe780f6 · fe780f6
1 parent 7548c54
commit fe780f6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/packages/pixiv/src/index.ts b/packages/pixiv/src/index.ts
@@ -29,6 +29,7 @@ class PixivImageSource extends ImageSource<PixivImageSource.Config> {
     const params: PixivAppApi.SearchParams = {
       word: query.tags.join(' '),
       search_target: 'partial_match_for_tags',
+      search_ai_type: this.config.ai === 0 ? PixivAppApi.SearchAIType.Filtered : PixivAppApi.SearchAIType.All,
       sort: 'date_desc', // TODO: Pixiv member could use 'popular_desc'
       filter: 'for_ios',
     }

diff --git a/packages/pixiv/src/types.ts b/packages/pixiv/src/types.ts
@@ -20,13 +20,17 @@ export namespace PixivAppApi {
     | 'week_r18g'
 
   export type SearchTarget = 'partial_match_for_tags' | 'exact_match_for_tags' | 'title_and_caption' | 'keyword'
-
+  export enum SearchAIType {
+    Filtered = 0,
+    All = 1,
+  }
   export type Sort = 'date_desc' | 'date_asc' | 'popular_desc'
   export type Duration = 'within_last_day' | 'within_last_week' | 'within_last_month'
 
   export interface SearchParams {
     word: string
     search_target: SearchTarget
+    search_ai_type?: SearchAIType
     sort?: Sort
     filter: Filter
     duration?: Duration