mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-15 23:31:29 +00:00
(sap) fix after bot review
This commit is contained in:
parent
6123651b30
commit
1132a7904a
2 changed files with 3 additions and 3 deletions
|
|
@ -621,7 +621,7 @@ response = completion(model="sap/gpt-4o",
|
|||
masking=masking_config)
|
||||
print(response.choices[0].message.content)
|
||||
|
||||
###Data masking module also available for embedding
|
||||
# Data masking module also available for embedding
|
||||
response = embedding(model="sap/text-embedding-3-small",
|
||||
input=mock_cv,
|
||||
masking=masking_config)
|
||||
|
|
|
|||
|
|
@ -290,7 +290,7 @@ class DPIMethodConstant(BaseModel):
|
|||
"""
|
||||
Replaces the entity with the specified value followed by an incrementing number
|
||||
"""
|
||||
method: str = "constant"
|
||||
method: Literal["constant"] = "constant"
|
||||
value: str
|
||||
|
||||
|
||||
|
|
@ -298,7 +298,7 @@ class DPIMethodFabricatedData(BaseModel):
|
|||
"""
|
||||
Replaces the entity with a randomly generated value appropriate to its type.
|
||||
"""
|
||||
method: str = "fabricated_data"
|
||||
method: Literal["fabricated_data"] = "fabricated_data"
|
||||
|
||||
|
||||
class DPICustomEntity(BaseModel):
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue