scope-client 1.4.911__py3-none-any.whl → 1.4.913__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (28) hide show
  1. scope_client/api_bindings/__init__.py +19 -0
  2. scope_client/api_bindings/api/__init__.py +1 -0
  3. scope_client/api_bindings/api/tasks_v1_api.py +1757 -0
  4. scope_client/api_bindings/models/__init__.py +18 -0
  5. scope_client/api_bindings/models/config.py +185 -0
  6. scope_client/api_bindings/models/config1.py +185 -0
  7. scope_client/api_bindings/models/example_config.py +89 -0
  8. scope_client/api_bindings/models/examples_config.py +102 -0
  9. scope_client/api_bindings/models/keywords_config.py +87 -0
  10. scope_client/api_bindings/models/metrics_query_result.py +2 -2
  11. scope_client/api_bindings/models/new_rule_request.py +104 -0
  12. scope_client/api_bindings/models/patch_task_request.py +101 -0
  13. scope_client/api_bindings/models/permission_name.py +6 -0
  14. scope_client/api_bindings/models/pii_config.py +106 -0
  15. scope_client/api_bindings/models/post_task_request.py +105 -0
  16. scope_client/api_bindings/models/put_task_state_cache_request.py +91 -0
  17. scope_client/api_bindings/models/regex_config.py +87 -0
  18. scope_client/api_bindings/models/rule_response.py +121 -0
  19. scope_client/api_bindings/models/rule_scope.py +37 -0
  20. scope_client/api_bindings/models/rule_type.py +44 -0
  21. scope_client/api_bindings/models/task_mutation_response.py +87 -0
  22. scope_client/api_bindings/models/task_read_response.py +106 -0
  23. scope_client/api_bindings/models/task_response.py +103 -0
  24. scope_client/api_bindings/models/toxicity_config.py +100 -0
  25. {scope_client-1.4.911.dist-info → scope_client-1.4.913.dist-info}/METADATA +1 -1
  26. {scope_client-1.4.911.dist-info → scope_client-1.4.913.dist-info}/RECORD +28 -9
  27. {scope_client-1.4.911.dist-info → scope_client-1.4.913.dist-info}/WHEEL +0 -0
  28. {scope_client-1.4.911.dist-info → scope_client-1.4.913.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,87 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ Arthur Scope
5
+
6
+ No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
+
8
+ The version of the OpenAPI document: 0.1.0
9
+ Generated by OpenAPI Generator (https://openapi-generator.tech)
10
+
11
+ Do not edit the class manually.
12
+ """ # noqa: E501
13
+
14
+
15
+ from __future__ import annotations
16
+ import pprint
17
+ import re # noqa: F401
18
+ import json
19
+
20
+ from pydantic import BaseModel, ConfigDict, Field, StrictStr
21
+ from typing import Any, ClassVar, Dict, List
22
+ from typing import Optional, Set
23
+ from typing_extensions import Self
24
+
25
+ class RegexConfig(BaseModel):
26
+ """
27
+ RegexConfig
28
+ """ # noqa: E501
29
+ regex_patterns: List[StrictStr] = Field(description="List of Regex patterns to be used for validation. Be sure to encode requests in JSON and account for escape characters.")
30
+ __properties: ClassVar[List[str]] = ["regex_patterns"]
31
+
32
+ model_config = ConfigDict(
33
+ populate_by_name=True,
34
+ validate_assignment=True,
35
+ protected_namespaces=(),
36
+ )
37
+
38
+
39
+ def to_str(self) -> str:
40
+ """Returns the string representation of the model using alias"""
41
+ return pprint.pformat(self.model_dump(by_alias=True))
42
+
43
+ def to_json(self) -> str:
44
+ """Returns the JSON representation of the model using alias"""
45
+ # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
46
+ return json.dumps(self.to_dict())
47
+
48
+ @classmethod
49
+ def from_json(cls, json_str: str) -> Optional[Self]:
50
+ """Create an instance of RegexConfig from a JSON string"""
51
+ return cls.from_dict(json.loads(json_str))
52
+
53
+ def to_dict(self) -> Dict[str, Any]:
54
+ """Return the dictionary representation of the model using alias.
55
+
56
+ This has the following differences from calling pydantic's
57
+ `self.model_dump(by_alias=True)`:
58
+
59
+ * `None` is only added to the output dict for nullable fields that
60
+ were set at model initialization. Other fields with value `None`
61
+ are ignored.
62
+ """
63
+ excluded_fields: Set[str] = set([
64
+ ])
65
+
66
+ _dict = self.model_dump(
67
+ by_alias=True,
68
+ exclude=excluded_fields,
69
+ exclude_none=True,
70
+ )
71
+ return _dict
72
+
73
+ @classmethod
74
+ def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
75
+ """Create an instance of RegexConfig from a dict"""
76
+ if obj is None:
77
+ return None
78
+
79
+ if not isinstance(obj, dict):
80
+ return cls.model_validate(obj)
81
+
82
+ _obj = cls.model_validate({
83
+ "regex_patterns": obj.get("regex_patterns")
84
+ })
85
+ return _obj
86
+
87
+
@@ -0,0 +1,121 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ Arthur Scope
5
+
6
+ No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
+
8
+ The version of the OpenAPI document: 0.1.0
9
+ Generated by OpenAPI Generator (https://openapi-generator.tech)
10
+
11
+ Do not edit the class manually.
12
+ """ # noqa: E501
13
+
14
+
15
+ from __future__ import annotations
16
+ import pprint
17
+ import re # noqa: F401
18
+ import json
19
+
20
+ from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictInt, StrictStr
21
+ from typing import Any, ClassVar, Dict, List, Optional
22
+ from scope_client.api_bindings.models.config1 import Config1
23
+ from scope_client.api_bindings.models.rule_scope import RuleScope
24
+ from scope_client.api_bindings.models.rule_type import RuleType
25
+ from typing import Optional, Set
26
+ from typing_extensions import Self
27
+
28
+ class RuleResponse(BaseModel):
29
+ """
30
+ RuleResponse
31
+ """ # noqa: E501
32
+ id: StrictStr = Field(description="ID of the Rule")
33
+ name: StrictStr = Field(description="Name of the Rule")
34
+ type: RuleType = Field(description="Type of Rule")
35
+ apply_to_prompt: StrictBool = Field(description="Rule applies to prompt")
36
+ apply_to_response: StrictBool = Field(description="Rule applies to response")
37
+ enabled: Optional[StrictBool] = None
38
+ scope: RuleScope = Field(description="Scope of the rule. The rule can be set at default level or task level.")
39
+ created_at: StrictInt = Field(description="Time the rule was created in unix milliseconds")
40
+ updated_at: StrictInt = Field(description="Time the rule was updated in unix milliseconds")
41
+ config: Optional[Config1] = None
42
+ __properties: ClassVar[List[str]] = ["id", "name", "type", "apply_to_prompt", "apply_to_response", "enabled", "scope", "created_at", "updated_at", "config"]
43
+
44
+ model_config = ConfigDict(
45
+ populate_by_name=True,
46
+ validate_assignment=True,
47
+ protected_namespaces=(),
48
+ )
49
+
50
+
51
+ def to_str(self) -> str:
52
+ """Returns the string representation of the model using alias"""
53
+ return pprint.pformat(self.model_dump(by_alias=True))
54
+
55
+ def to_json(self) -> str:
56
+ """Returns the JSON representation of the model using alias"""
57
+ # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
58
+ return json.dumps(self.to_dict())
59
+
60
+ @classmethod
61
+ def from_json(cls, json_str: str) -> Optional[Self]:
62
+ """Create an instance of RuleResponse from a JSON string"""
63
+ return cls.from_dict(json.loads(json_str))
64
+
65
+ def to_dict(self) -> Dict[str, Any]:
66
+ """Return the dictionary representation of the model using alias.
67
+
68
+ This has the following differences from calling pydantic's
69
+ `self.model_dump(by_alias=True)`:
70
+
71
+ * `None` is only added to the output dict for nullable fields that
72
+ were set at model initialization. Other fields with value `None`
73
+ are ignored.
74
+ """
75
+ excluded_fields: Set[str] = set([
76
+ ])
77
+
78
+ _dict = self.model_dump(
79
+ by_alias=True,
80
+ exclude=excluded_fields,
81
+ exclude_none=True,
82
+ )
83
+ # override the default output from pydantic by calling `to_dict()` of config
84
+ if self.config:
85
+ _dict['config'] = self.config.to_dict()
86
+ # set to None if enabled (nullable) is None
87
+ # and model_fields_set contains the field
88
+ if self.enabled is None and "enabled" in self.model_fields_set:
89
+ _dict['enabled'] = None
90
+
91
+ # set to None if config (nullable) is None
92
+ # and model_fields_set contains the field
93
+ if self.config is None and "config" in self.model_fields_set:
94
+ _dict['config'] = None
95
+
96
+ return _dict
97
+
98
+ @classmethod
99
+ def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
100
+ """Create an instance of RuleResponse from a dict"""
101
+ if obj is None:
102
+ return None
103
+
104
+ if not isinstance(obj, dict):
105
+ return cls.model_validate(obj)
106
+
107
+ _obj = cls.model_validate({
108
+ "id": obj.get("id"),
109
+ "name": obj.get("name"),
110
+ "type": obj.get("type"),
111
+ "apply_to_prompt": obj.get("apply_to_prompt"),
112
+ "apply_to_response": obj.get("apply_to_response"),
113
+ "enabled": obj.get("enabled"),
114
+ "scope": obj.get("scope"),
115
+ "created_at": obj.get("created_at"),
116
+ "updated_at": obj.get("updated_at"),
117
+ "config": Config1.from_dict(obj["config"]) if obj.get("config") is not None else None
118
+ })
119
+ return _obj
120
+
121
+
@@ -0,0 +1,37 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ Arthur Scope
5
+
6
+ No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
+
8
+ The version of the OpenAPI document: 0.1.0
9
+ Generated by OpenAPI Generator (https://openapi-generator.tech)
10
+
11
+ Do not edit the class manually.
12
+ """ # noqa: E501
13
+
14
+
15
+ from __future__ import annotations
16
+ import json
17
+ from enum import Enum
18
+ from typing_extensions import Self
19
+
20
+
21
+ class RuleScope(str, Enum):
22
+ """
23
+ RuleScope
24
+ """
25
+
26
+ """
27
+ allowed enum values
28
+ """
29
+ DEFAULT = 'default'
30
+ TASK = 'task'
31
+
32
+ @classmethod
33
+ def from_json(cls, json_str: str) -> Self:
34
+ """Create an instance of RuleScope from a JSON string"""
35
+ return cls(json.loads(json_str))
36
+
37
+
@@ -0,0 +1,44 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ Arthur Scope
5
+
6
+ No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
+
8
+ The version of the OpenAPI document: 0.1.0
9
+ Generated by OpenAPI Generator (https://openapi-generator.tech)
10
+
11
+ Do not edit the class manually.
12
+ """ # noqa: E501
13
+
14
+
15
+ from __future__ import annotations
16
+ import json
17
+ from enum import Enum
18
+ from typing_extensions import Self
19
+
20
+
21
+ class RuleType(str, Enum):
22
+ """
23
+ RuleType
24
+ """
25
+
26
+ """
27
+ allowed enum values
28
+ """
29
+ KEYWORDRULE = 'KeywordRule'
30
+ MODELHALLUCINATIONRULE = 'ModelHallucinationRule'
31
+ MODELHALLUCINATIONRULEV2 = 'ModelHallucinationRuleV2'
32
+ MODELHALLUCINATIONRULEV3 = 'ModelHallucinationRuleV3'
33
+ MODELSENSITIVEDATARULE = 'ModelSensitiveDataRule'
34
+ PIIDATARULE = 'PIIDataRule'
35
+ PROMPTINJECTIONRULE = 'PromptInjectionRule'
36
+ REGEXRULE = 'RegexRule'
37
+ TOXICITYRULE = 'ToxicityRule'
38
+
39
+ @classmethod
40
+ def from_json(cls, json_str: str) -> Self:
41
+ """Create an instance of RuleType from a JSON string"""
42
+ return cls(json.loads(json_str))
43
+
44
+
@@ -0,0 +1,87 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ Arthur Scope
5
+
6
+ No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
+
8
+ The version of the OpenAPI document: 0.1.0
9
+ Generated by OpenAPI Generator (https://openapi-generator.tech)
10
+
11
+ Do not edit the class manually.
12
+ """ # noqa: E501
13
+
14
+
15
+ from __future__ import annotations
16
+ import pprint
17
+ import re # noqa: F401
18
+ import json
19
+
20
+ from pydantic import BaseModel, ConfigDict, Field, StrictStr
21
+ from typing import Any, ClassVar, Dict, List
22
+ from typing import Optional, Set
23
+ from typing_extensions import Self
24
+
25
+ class TaskMutationResponse(BaseModel):
26
+ """
27
+ TaskMutationResponse
28
+ """ # noqa: E501
29
+ job_id: StrictStr = Field(description="Job ID executing the task mutation and retrieval.")
30
+ __properties: ClassVar[List[str]] = ["job_id"]
31
+
32
+ model_config = ConfigDict(
33
+ populate_by_name=True,
34
+ validate_assignment=True,
35
+ protected_namespaces=(),
36
+ )
37
+
38
+
39
+ def to_str(self) -> str:
40
+ """Returns the string representation of the model using alias"""
41
+ return pprint.pformat(self.model_dump(by_alias=True))
42
+
43
+ def to_json(self) -> str:
44
+ """Returns the JSON representation of the model using alias"""
45
+ # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
46
+ return json.dumps(self.to_dict())
47
+
48
+ @classmethod
49
+ def from_json(cls, json_str: str) -> Optional[Self]:
50
+ """Create an instance of TaskMutationResponse from a JSON string"""
51
+ return cls.from_dict(json.loads(json_str))
52
+
53
+ def to_dict(self) -> Dict[str, Any]:
54
+ """Return the dictionary representation of the model using alias.
55
+
56
+ This has the following differences from calling pydantic's
57
+ `self.model_dump(by_alias=True)`:
58
+
59
+ * `None` is only added to the output dict for nullable fields that
60
+ were set at model initialization. Other fields with value `None`
61
+ are ignored.
62
+ """
63
+ excluded_fields: Set[str] = set([
64
+ ])
65
+
66
+ _dict = self.model_dump(
67
+ by_alias=True,
68
+ exclude=excluded_fields,
69
+ exclude_none=True,
70
+ )
71
+ return _dict
72
+
73
+ @classmethod
74
+ def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
75
+ """Create an instance of TaskMutationResponse from a dict"""
76
+ if obj is None:
77
+ return None
78
+
79
+ if not isinstance(obj, dict):
80
+ return cls.model_validate(obj)
81
+
82
+ _obj = cls.model_validate({
83
+ "job_id": obj.get("job_id")
84
+ })
85
+ return _obj
86
+
87
+
@@ -0,0 +1,106 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ Arthur Scope
5
+
6
+ No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
+
8
+ The version of the OpenAPI document: 0.1.0
9
+ Generated by OpenAPI Generator (https://openapi-generator.tech)
10
+
11
+ Do not edit the class manually.
12
+ """ # noqa: E501
13
+
14
+
15
+ from __future__ import annotations
16
+ import pprint
17
+ import re # noqa: F401
18
+ import json
19
+
20
+ from datetime import datetime
21
+ from pydantic import BaseModel, ConfigDict, Field, StrictStr
22
+ from typing import Any, ClassVar, Dict, List, Optional
23
+ from scope_client.api_bindings.models.task_response import TaskResponse
24
+ from typing import Optional, Set
25
+ from typing_extensions import Self
26
+
27
+ class TaskReadResponse(BaseModel):
28
+ """
29
+ TaskReadResponse
30
+ """ # noqa: E501
31
+ task: Optional[TaskResponse] = None
32
+ last_synced_at: Optional[datetime] = None
33
+ scope_model_id: StrictStr = Field(description="The ID of the corresponding scope model for this task.")
34
+ __properties: ClassVar[List[str]] = ["task", "last_synced_at", "scope_model_id"]
35
+
36
+ model_config = ConfigDict(
37
+ populate_by_name=True,
38
+ validate_assignment=True,
39
+ protected_namespaces=(),
40
+ )
41
+
42
+
43
+ def to_str(self) -> str:
44
+ """Returns the string representation of the model using alias"""
45
+ return pprint.pformat(self.model_dump(by_alias=True))
46
+
47
+ def to_json(self) -> str:
48
+ """Returns the JSON representation of the model using alias"""
49
+ # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
50
+ return json.dumps(self.to_dict())
51
+
52
+ @classmethod
53
+ def from_json(cls, json_str: str) -> Optional[Self]:
54
+ """Create an instance of TaskReadResponse from a JSON string"""
55
+ return cls.from_dict(json.loads(json_str))
56
+
57
+ def to_dict(self) -> Dict[str, Any]:
58
+ """Return the dictionary representation of the model using alias.
59
+
60
+ This has the following differences from calling pydantic's
61
+ `self.model_dump(by_alias=True)`:
62
+
63
+ * `None` is only added to the output dict for nullable fields that
64
+ were set at model initialization. Other fields with value `None`
65
+ are ignored.
66
+ """
67
+ excluded_fields: Set[str] = set([
68
+ ])
69
+
70
+ _dict = self.model_dump(
71
+ by_alias=True,
72
+ exclude=excluded_fields,
73
+ exclude_none=True,
74
+ )
75
+ # override the default output from pydantic by calling `to_dict()` of task
76
+ if self.task:
77
+ _dict['task'] = self.task.to_dict()
78
+ # set to None if task (nullable) is None
79
+ # and model_fields_set contains the field
80
+ if self.task is None and "task" in self.model_fields_set:
81
+ _dict['task'] = None
82
+
83
+ # set to None if last_synced_at (nullable) is None
84
+ # and model_fields_set contains the field
85
+ if self.last_synced_at is None and "last_synced_at" in self.model_fields_set:
86
+ _dict['last_synced_at'] = None
87
+
88
+ return _dict
89
+
90
+ @classmethod
91
+ def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
92
+ """Create an instance of TaskReadResponse from a dict"""
93
+ if obj is None:
94
+ return None
95
+
96
+ if not isinstance(obj, dict):
97
+ return cls.model_validate(obj)
98
+
99
+ _obj = cls.model_validate({
100
+ "task": TaskResponse.from_dict(obj["task"]) if obj.get("task") is not None else None,
101
+ "last_synced_at": obj.get("last_synced_at"),
102
+ "scope_model_id": obj.get("scope_model_id")
103
+ })
104
+ return _obj
105
+
106
+
@@ -0,0 +1,103 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ Arthur Scope
5
+
6
+ No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
+
8
+ The version of the OpenAPI document: 0.1.0
9
+ Generated by OpenAPI Generator (https://openapi-generator.tech)
10
+
11
+ Do not edit the class manually.
12
+ """ # noqa: E501
13
+
14
+
15
+ from __future__ import annotations
16
+ import pprint
17
+ import re # noqa: F401
18
+ import json
19
+
20
+ from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr
21
+ from typing import Any, ClassVar, Dict, List
22
+ from scope_client.api_bindings.models.rule_response import RuleResponse
23
+ from typing import Optional, Set
24
+ from typing_extensions import Self
25
+
26
+ class TaskResponse(BaseModel):
27
+ """
28
+ TaskResponse
29
+ """ # noqa: E501
30
+ id: StrictStr = Field(description=" ID of the task")
31
+ name: StrictStr = Field(description="Name of the task")
32
+ created_at: StrictInt = Field(description="Time the task was created in unix milliseconds")
33
+ updated_at: StrictInt = Field(description="Time the task was created in unix milliseconds")
34
+ rules: List[RuleResponse] = Field(description="List of all the rules for the task.")
35
+ __properties: ClassVar[List[str]] = ["id", "name", "created_at", "updated_at", "rules"]
36
+
37
+ model_config = ConfigDict(
38
+ populate_by_name=True,
39
+ validate_assignment=True,
40
+ protected_namespaces=(),
41
+ )
42
+
43
+
44
+ def to_str(self) -> str:
45
+ """Returns the string representation of the model using alias"""
46
+ return pprint.pformat(self.model_dump(by_alias=True))
47
+
48
+ def to_json(self) -> str:
49
+ """Returns the JSON representation of the model using alias"""
50
+ # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
51
+ return json.dumps(self.to_dict())
52
+
53
+ @classmethod
54
+ def from_json(cls, json_str: str) -> Optional[Self]:
55
+ """Create an instance of TaskResponse from a JSON string"""
56
+ return cls.from_dict(json.loads(json_str))
57
+
58
+ def to_dict(self) -> Dict[str, Any]:
59
+ """Return the dictionary representation of the model using alias.
60
+
61
+ This has the following differences from calling pydantic's
62
+ `self.model_dump(by_alias=True)`:
63
+
64
+ * `None` is only added to the output dict for nullable fields that
65
+ were set at model initialization. Other fields with value `None`
66
+ are ignored.
67
+ """
68
+ excluded_fields: Set[str] = set([
69
+ ])
70
+
71
+ _dict = self.model_dump(
72
+ by_alias=True,
73
+ exclude=excluded_fields,
74
+ exclude_none=True,
75
+ )
76
+ # override the default output from pydantic by calling `to_dict()` of each item in rules (list)
77
+ _items = []
78
+ if self.rules:
79
+ for _item_rules in self.rules:
80
+ if _item_rules:
81
+ _items.append(_item_rules.to_dict())
82
+ _dict['rules'] = _items
83
+ return _dict
84
+
85
+ @classmethod
86
+ def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
87
+ """Create an instance of TaskResponse from a dict"""
88
+ if obj is None:
89
+ return None
90
+
91
+ if not isinstance(obj, dict):
92
+ return cls.model_validate(obj)
93
+
94
+ _obj = cls.model_validate({
95
+ "id": obj.get("id"),
96
+ "name": obj.get("name"),
97
+ "created_at": obj.get("created_at"),
98
+ "updated_at": obj.get("updated_at"),
99
+ "rules": [RuleResponse.from_dict(_item) for _item in obj["rules"]] if obj.get("rules") is not None else None
100
+ })
101
+ return _obj
102
+
103
+
@@ -0,0 +1,100 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ Arthur Scope
5
+
6
+ No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
+
8
+ The version of the OpenAPI document: 0.1.0
9
+ Generated by OpenAPI Generator (https://openapi-generator.tech)
10
+
11
+ Do not edit the class manually.
12
+ """ # noqa: E501
13
+
14
+
15
+ from __future__ import annotations
16
+ import pprint
17
+ import re # noqa: F401
18
+ import json
19
+
20
+ from pydantic import BaseModel, ConfigDict, Field, StrictFloat, StrictInt
21
+ from typing import Any, ClassVar, Dict, List, Optional, Union
22
+ from typing import Optional, Set
23
+ from typing_extensions import Self
24
+
25
+ class ToxicityConfig(BaseModel):
26
+ """
27
+ ToxicityConfig
28
+ """ # noqa: E501
29
+ threshold: Optional[Union[StrictFloat, StrictInt]] = Field(default=0.5, description="Optional. Float (0, 1) indicating the level of tolerable toxicity to consider the rule passed or failed. Min: 0 (no toxic language) Max: 1 (very toxic language). Default: 0.5")
30
+ additional_properties: Dict[str, Any] = {}
31
+ __properties: ClassVar[List[str]] = ["threshold"]
32
+
33
+ model_config = ConfigDict(
34
+ populate_by_name=True,
35
+ validate_assignment=True,
36
+ protected_namespaces=(),
37
+ )
38
+
39
+
40
+ def to_str(self) -> str:
41
+ """Returns the string representation of the model using alias"""
42
+ return pprint.pformat(self.model_dump(by_alias=True))
43
+
44
+ def to_json(self) -> str:
45
+ """Returns the JSON representation of the model using alias"""
46
+ # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
47
+ return json.dumps(self.to_dict())
48
+
49
+ @classmethod
50
+ def from_json(cls, json_str: str) -> Optional[Self]:
51
+ """Create an instance of ToxicityConfig from a JSON string"""
52
+ return cls.from_dict(json.loads(json_str))
53
+
54
+ def to_dict(self) -> Dict[str, Any]:
55
+ """Return the dictionary representation of the model using alias.
56
+
57
+ This has the following differences from calling pydantic's
58
+ `self.model_dump(by_alias=True)`:
59
+
60
+ * `None` is only added to the output dict for nullable fields that
61
+ were set at model initialization. Other fields with value `None`
62
+ are ignored.
63
+ * Fields in `self.additional_properties` are added to the output dict.
64
+ """
65
+ excluded_fields: Set[str] = set([
66
+ "additional_properties",
67
+ ])
68
+
69
+ _dict = self.model_dump(
70
+ by_alias=True,
71
+ exclude=excluded_fields,
72
+ exclude_none=True,
73
+ )
74
+ # puts key-value pairs in additional_properties in the top level
75
+ if self.additional_properties is not None:
76
+ for _key, _value in self.additional_properties.items():
77
+ _dict[_key] = _value
78
+
79
+ return _dict
80
+
81
+ @classmethod
82
+ def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
83
+ """Create an instance of ToxicityConfig from a dict"""
84
+ if obj is None:
85
+ return None
86
+
87
+ if not isinstance(obj, dict):
88
+ return cls.model_validate(obj)
89
+
90
+ _obj = cls.model_validate({
91
+ "threshold": obj.get("threshold") if obj.get("threshold") is not None else 0.5
92
+ })
93
+ # store additional fields in additional_properties
94
+ for _key in obj.keys():
95
+ if _key not in cls.__properties:
96
+ _obj.additional_properties[_key] = obj.get(_key)
97
+
98
+ return _obj
99
+
100
+