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

enhancement: introduce Ruff for Python linter for reordering and removing unused imports with automated pre-commit and sytle check (#2366)

Browse source
This commit is contained in:
Bowen Liang 2024-02-06 13:21:13 +08:00 • committed by GitHub
commit 843280f82b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
333 changed files with 2109 additions and 1050 deletions
Download patch file Download diff file Expand all files Collapse all files

8
api/core/tools/tool/dataset_retriever/dataset_multi_retriever_tool.py
View file

@ -1,7 +1,10 @@
import json
import threading
from typing import List, Optional, Type
from flask import Flask, current_app
from langchain.tools import BaseTool
from pydantic import BaseModel, Field
from core.callback_handler.index_tool_callback_handler import DatasetIndexToolCallbackHandler
from core.embedding.cached_embedding import CacheEmbedding
from core.errors.error import LLMBadRequestError, ProviderTokenNotInitError
@ -10,10 +13,7 @@ from core.model_manager import ModelManager
from core.model_runtime.entities.model_entities import ModelType
from core.rerank.rerank import RerankRunner
from extensions.ext_database import db
from flask import Flask, current_app
from langchain.tools import BaseTool
from models.dataset import Dataset, Document, DocumentSegment
from pydantic import BaseModel, Field
from services.retrieval_service import RetrievalService
default_retrieval_model = {

7
api/core/tools/tool/dataset_retriever/dataset_retriever_tool.py
View file

@ -1,6 +1,10 @@
import threading
from typing import List, Optional, Type
from flask import current_app
from langchain.tools import BaseTool
from pydantic import BaseModel, Field
from core.callback_handler.index_tool_callback_handler import DatasetIndexToolCallbackHandler
from core.embedding.cached_embedding import CacheEmbedding
from core.index.keyword_table_index.keyword_table_index import KeywordTableConfig, KeywordTableIndex
@ -9,10 +13,7 @@ from core.model_runtime.entities.model_entities import ModelType
from core.model_runtime.errors.invoke import InvokeAuthorizationError
from core.rerank.rerank import RerankRunner
from extensions.ext_database import db
from flask import current_app
from langchain.tools import BaseTool
from models.dataset import Dataset, Document, DocumentSegment
from pydantic import BaseModel, Field
from services.retrieval_service import RetrievalService
default_retrieval_model = {