feat: mypy for all type check (#10921)

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

View File

@@ -72,9 +72,13 @@ class ToolConfigurationManager(BaseModel):
return a deep copy of credentials with decrypted values
"""
identity_id = ""
if self.provider_controller.identity:
identity_id = f"{self.provider_controller.provider_type.value}.{self.provider_controller.identity.name}"
cache = ToolProviderCredentialsCache(
tenant_id=self.tenant_id,
identity_id=f"{self.provider_controller.provider_type.value}.{self.provider_controller.identity.name}",
identity_id=identity_id,
cache_type=ToolProviderCredentialsCacheType.PROVIDER,
)
cached_credentials = cache.get()
@@ -95,9 +99,13 @@ class ToolConfigurationManager(BaseModel):
return credentials
def delete_tool_credentials_cache(self):
identity_id = ""
if self.provider_controller.identity:
identity_id = f"{self.provider_controller.provider_type.value}.{self.provider_controller.identity.name}"
cache = ToolProviderCredentialsCache(
tenant_id=self.tenant_id,
identity_id=f"{self.provider_controller.provider_type.value}.{self.provider_controller.identity.name}",
identity_id=identity_id,
cache_type=ToolProviderCredentialsCacheType.PROVIDER,
)
cache.delete()
@@ -199,6 +207,9 @@ class ToolParameterConfigurationManager(BaseModel):
return a deep copy of parameters with decrypted values
"""
if self.tool_runtime is None or self.tool_runtime.identity is None:
raise ValueError("tool_runtime is required")
cache = ToolParameterCache(
tenant_id=self.tenant_id,
provider=f"{self.provider_type}.{self.provider_name}",
@@ -232,6 +243,9 @@ class ToolParameterConfigurationManager(BaseModel):
return parameters
def delete_tool_parameters_cache(self):
if self.tool_runtime is None or self.tool_runtime.identity is None:
raise ValueError("tool_runtime is required")
cache = ToolParameterCache(
tenant_id=self.tenant_id,
provider=f"{self.provider_type}.{self.provider_name}",