16 lines
591 B
Python
16 lines
591 B
Python
class PromptConstants:
|
|
'''String constants related to prompt engineering.'''
|
|
|
|
# Prefix for user messages.
|
|
USER_PREFIX = "You"
|
|
|
|
# Global target word count. The word count is chosen in such a way that we
|
|
# can fit all the required prompt trickery into the model's input, but still
|
|
# leave enough space for the user's input message and the infernce result.
|
|
TARGET_WORD_COUNT_PER_EPISODE = 1536
|
|
|
|
@staticmethod
|
|
def pdm_prefix_for(name: str) -> str:
|
|
'''Builds the Persona Dialogue Module prefix for a given `name`.'''
|
|
return f"{name}'s Persona"
|