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

chore: apply ruff's pyupgrade linter rules to modernize Python code with targeted version (#2419)

Browse source
This commit is contained in:
Bowen Liang 2024-02-09 15:21:33 +08:00 • committed by GitHub
commit 063191889d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
246 changed files with 912 additions and 937 deletions
Download patch file Download diff file Expand all files Collapse all files

6
api/core/rerank/rerank.py
View file

@ -1,4 +1,4 @@
from typing import List, Optional
from typing import Optional
from langchain.schema import Document
@ -9,8 +9,8 @@ class RerankRunner:
def __init__(self, rerank_model_instance: ModelInstance) -> None:
self.rerank_model_instance = rerank_model_instance
def run(self, query: str, documents: List[Document], score_threshold: Optional[float] = None,
top_n: Optional[int] = None, user: Optional[str] = None) -> List[Document]:
def run(self, query: str, documents: list[Document], score_threshold: Optional[float] = None,
top_n: Optional[int] = None, user: Optional[str] = None) -> list[Document]:
"""
Run rerank model
:param query: search query