diff options
Diffstat (limited to 'context.py')
| -rw-r--r-- | context.py | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/context.py b/context.py new file mode 100644 index 0000000..5006e3a --- /dev/null +++ b/context.py @@ -0,0 +1,24 @@ +from identity import AgentIdentity, UserProfile + +class ContextBuilder: + def __init__( + self, + agent_identity: AgentIdentity, + user_profile: UserProfile, + ) -> None: + self.agent_identity = agent_identity + self.user_profile = user_profile + + def build_system_prompt(self) -> str: + agent = self.agent_identity + user = self.user_profile + + return ( + f"You are {agent.name}, a personal AI assistant.\n" + f"Personality: {agent.personality['style']}.\n" + f"Honesty: {agent.values['honesty']}.\n" + f"Directness: {agent.values['directness']}.\n" + f"Favorite color: {agent.quirks['favorite_color']}.\n" + f"\n" + f"The user's name is {user.name}" + ) |
