jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use typing.Literal to replace str places (#24099)

Browse source 
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
This commit is contained in:
Zhehao Peng 2025-08-18 06:34:13 -07:00 • committed by GitHub
commit c0702aacac
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 34 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

3
api/tasks/deal_dataset_vector_index_task.py
View file

@ -1,5 +1,6 @@
import logging
import time
from typing import Literal
import click
from celery import shared_task # type: ignore
@ -13,7 +14,7 @@ from models.dataset import Document as DatasetDocument
@shared_task(queue="dataset")
def deal_dataset_vector_index_task(dataset_id: str, action: str):
def deal_dataset_vector_index_task(dataset_id: str, action: Literal["remove", "add", "update"]):
"""
Async deal dataset from index
:param dataset_id: dataset_id