hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-16 14:26:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: mypy for all type check (#10921)

Browse Source
This commit is contained in:
yihong
2024-12-24 18:38:51 +08:00
committed by GitHub
parent c91e8b1737
commit 56e15d09a9
584 changed files with 3975 additions and 2826 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/tools/provider/builtin/aws/tools/apply_guardrail.py
View File

@@ -2,8 +2,8 @@ import json
import logging
from typing import Any, Union
import boto3
from botocore.exceptions import BotoCoreError
import boto3 # type: ignore
from botocore.exceptions import BotoCoreError # type: ignore
from pydantic import BaseModel, Field
from core.tools.entities.tool_entities import ToolInvokeMessage

2
api/core/tools/provider/builtin/aws/tools/lambda_translate_utils.py
View File

@@ -1,7 +1,7 @@
import json
from typing import Any, Union
import boto3
import boto3 # type: ignore
from core.tools.entities.tool_entities import ToolInvokeMessage
from core.tools.tool.builtin_tool import BuiltinTool

2
api/core/tools/provider/builtin/aws/tools/lambda_yaml_to_json.py
View File

@@ -2,7 +2,7 @@ import json
import logging
from typing import Any, Union
import boto3
import boto3 # type: ignore
from core.tools.entities.tool_entities import ToolInvokeMessage
from core.tools.tool.builtin_tool import BuiltinTool

6
api/core/tools/provider/builtin/aws/tools/sagemaker_text_rerank.py
View File

@@ -2,7 +2,7 @@ import json
import operator
from typing import Any, Union
import boto3
import boto3 # type: ignore
from core.tools.entities.tool_entities import ToolInvokeMessage
from core.tools.tool.builtin_tool import BuiltinTool
@@ -10,8 +10,8 @@ from core.tools.tool.builtin_tool import BuiltinTool
class SageMakerReRankTool(BuiltinTool):
sagemaker_client: Any = None
sagemaker_endpoint: str = None
topk: int = None
sagemaker_endpoint: str | None = None
topk: int | None = None
def _sagemaker_rerank(self, query_input: str, docs: list[str], rerank_endpoint: str):
inputs = [query_input] * len(docs)

4
api/core/tools/provider/builtin/aws/tools/sagemaker_tts.py
View File

@@ -2,7 +2,7 @@ import json
from enum import Enum
from typing import Any, Optional, Union
import boto3
import boto3 # type: ignore
from core.tools.entities.tool_entities import ToolInvokeMessage
from core.tools.tool.builtin_tool import BuiltinTool
@@ -17,7 +17,7 @@ class TTSModelType(Enum):
class SageMakerTTSTool(BuiltinTool):
sagemaker_client: Any = None
sagemaker_endpoint: str = None
sagemaker_endpoint: str | None = None
s3_client: Any = None
comprehend_client: Any = None