Skip to content

Commit

Permalink
[FIX] Add system prompt to enable usage of judge in GenerateColumnFro…
Browse files Browse the repository at this point in the history
…mTemplate

GitOrigin-RevId: 53fe842ec70b29e7727cc05d63fecfac4a40682c
  • Loading branch information
lipikaramaswamy committed Nov 15, 2024
1 parent c7e2a07 commit e2dacb0
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions src/gretel_client/navigator/data_designer/prompt_templates.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,13 @@ def get_prompt_template_keywords(template: str) -> set[str]:
### Response ###
"""

DATA_DESIGNER_JUDGE_SYSTEM_PROMPT = """\
You are a fair and experienced judge. \
You are particularly adept at writing natural language. \
You must judge in English only."""

system_prompt_dict = {
"natural_language": DATA_DESIGNER_NL_SYSTEM_PROMPT,
"code": DATA_DESIGNER_CODE_SYSTEM_PROMPT,
"judge": DATA_DESIGNER_JUDGE_SYSTEM_PROMPT,
}

0 comments on commit e2dacb0

Please sign in to comment.