[Community][minor]: Added prompt governance in pebblo_retrieval (#24874)

Title: [pebblo_retrieval] Identifying entities in prompts given in
PebbloRetrievalQA leading to prompt governance
Description: Implemented identification of entities in the prompt using
Pebblo prompt governance API.
Issue: NA
Dependencies: NA
Add tests and docs: NA
This commit is contained in:
Nishan Jain
2024-07-31 18:44:51 +05:30
committed by GitHub
parent a6add89bd4
commit b00c0fc558
3 changed files with 81 additions and 3 deletions

View File

@@ -133,7 +133,10 @@ class Context(BaseModel):
class Prompt(BaseModel):
data: str
data: Optional[Union[list, str]]
entityCount: Optional[int]
entities: Optional[dict]
prompt_gov_enabled: Optional[bool]
class Qa(BaseModel):