Added llm logic and config

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}"
+        )