| 77 | |
| 78 | |
| 79 | class CustomSecret(BaseModel): |
| 80 | secret: SecretStr = Field(default_factory=lambda: SecretStr('')) |
| 81 | description: str = Field(default='') |
| 82 | |
| 83 | model_config = ConfigDict( |
| 84 | frozen=True, # Makes the entire model immutable |
| 85 | validate_assignment=True, |
| 86 | ) |
| 87 | |
| 88 | @classmethod |
| 89 | def from_value(cls, secret_value: CustomSecret | dict[str, str]) -> CustomSecret: |
| 90 | """Factory method to create a ProviderToken from various input types""" |
| 91 | if isinstance(secret_value, CustomSecret): |
| 92 | return secret_value |
| 93 | elif isinstance(secret_value, dict): |
| 94 | secret = secret_value.get('secret', '') |
| 95 | description = secret_value.get('description', '') |
| 96 | return cls(secret=SecretStr(secret), description=description) |
| 97 | |
| 98 | else: |
| 99 | raise ValueError('Unsupport Provider token type') |
| 100 | |
| 101 | |
| 102 | PROVIDER_TOKEN_TYPE = Mapping[ProviderType, ProviderToken] |
no outgoing calls