feat: remove partner-focus

Author: splinesreticulating

src/ambient.d.ts

@@ -1,6 +1,6 @@
 declare namespace App {
     interface PrivateEnv {
-        PARTNER_PHONE: string
+        CONTACT_PHONE: string
         KHOJ_API_URL?: string
         KHOJ_AGENT?: string
         LOG_LEVEL?: 'fatal' | 'error' | 'warn' | 'info' | 'debug' | 'trace' | 'silent'

src/lib/config.ts

@@ -20,9 +20,9 @@ await db.exec(`CREATE TABLE IF NOT EXISTS settings (
 )`)
 
 const defaultSettings: Record<string, { value?: string; description: string }> = {
-    PARTNER_PHONE: {
-        value: env.PARTNER_PHONE,
-        description: "Your conversation partner's phone number in the Messages app",
+    CONTACT_PHONE: {
+        value: env.CONTACT_PHONE,
+        description: "The phone number of the person you're having a conversation with in the Messages app",
     },
     HISTORY_LOOKBACK_HOURS: {
         value: env.HISTORY_LOOKBACK_HOURS,
@@ -31,7 +31,7 @@ const defaultSettings: Record<string, { value?: string; description: string }> =
     CUSTOM_CONTEXT: {
         value: env.CUSTOM_CONTEXT,
         description:
-            'To guide the AI\'s personality and behavior (Example: "Act as my therapist suggesting replies to my partner" or "You are a helpful assistant")',
+            'To guide the AI\'s personality and behavior (Example: "Act as my therapist suggesting thoughtful replies" or "You are a helpful assistant")',
     },
     OPENAI_API_KEY: {
         value: env.OPENAI_API_KEY,

src/lib/iMessages.ts

@@ -5,14 +5,14 @@ import path from 'node:path'
 import { open } from 'sqlite'
 import sqlite3 from 'sqlite3'
 import { logger } from './logger'
-import { hasPartnerMessages, isoToAppleNanoseconds } from './utils'
+import { hasContactMessages, isoToAppleNanoseconds } from './utils'
 
 const CHAT_DB_PATH = path.join(os.homedir(), 'Library', 'Messages', 'chat.db')
 
 const buildQuery = (startDate: string, endDate: string) => {
     logger.debug({ startDate, endDate }, 'Getting messages')
     const params = [
-        settings.PARTNER_PHONE,
+        settings.CONTACT_PHONE,
         isoToAppleNanoseconds(startDate),
         isoToAppleNanoseconds(endDate),
     ]
@@ -39,8 +39,8 @@ const formatMessages = (rows: MessageRow[]) => {
         .map((row) => ({
             sender: row.is_from_me
                 ? 'me'
-                : row.contact_id === settings.PARTNER_PHONE
-                  ? 'partner'
+                : row.contact_id === settings.CONTACT_PHONE
+                  ? 'them'
                   : 'unknown',
             text: row.text,
             timestamp: new Date(
@@ -51,8 +51,8 @@ const formatMessages = (rows: MessageRow[]) => {
 }
 
 export const queryMessagesDb = async (startDate: string, endDate: string) => {
-    if (!settings.PARTNER_PHONE) {
-        logger.warn('PARTNER_PHONE setting not configured')
+    if (!settings.CONTACT_PHONE) {
+        logger.warn('CONTACT_PHONE setting not configured')
         return { messages: [] }
     }
 
@@ -61,12 +61,12 @@ export const queryMessagesDb = async (startDate: string, endDate: string) => {
 
     try {
         const rows = (await db.all(query, params)) as MessageRow[]
-        logger.info({ count: rows.length, handleId: settings.PARTNER_PHONE }, 'Fetched messages')
+        logger.info({ count: rows.length, handleId: settings.CONTACT_PHONE }, 'Fetched messages')
 
         const formattedMessages = formatMessages(rows)
 
-        // Return empty array if all messages are from me (no partner messages)
-        return { messages: hasPartnerMessages(formattedMessages) ? formattedMessages : [] }
+        // Return empty array if there are no messages from the contact
+        return { messages: hasContactMessages(formattedMessages) ? formattedMessages : [] }
     } catch (error) {
         logger.error({ error }, 'Error querying messages database')
         return { messages: [] }

src/lib/prompts.ts

@@ -6,7 +6,7 @@ const coreContext = [
     'Analyze messages attributed to "me:" to mimic my vocabulary and tone when suggesting replies.',
     'Additional context about recent conversation history is provided below. Use this to understand the current situation and tone, but focus your reply on the most recent messages.',
     'Do not summarize the history - it is only for context.',
-    'I will also provide a desired tone and may include extra context. Always incorporate that tone and context when crafting replies.',
+    'I will provide a desired tone and may include extra context. Always incorporate that tone and context when crafting replies.',
 ].join('\n')
 
 const instructions = [
@@ -37,15 +37,15 @@ export const openAiPrompt = (tone: string, context: string): string => buildProm
 export const khojPrompt = (messages: Message[], tone: ToneType, context: string): string =>
     [
         systemContext(),
-        'Here are some text messages between my partner and I:\n' + formatMessagesAsText(messages),
+        'Here is a conversation between me and someone else:\n' + formatMessagesAsText(messages),
         buildPrompt(tone, context),
         responseFormat,
     ].join('\n')
 
 export const anthropicPrompt = (messages: Message[], tone: ToneType, context: string): string =>
     [
         systemContext(),
-        'Here are some text messages between my partner and I:\n' + formatMessagesAsText(messages),
+        'Here is a conversation between me and someone else:\n' + formatMessagesAsText(messages),
         buildPrompt(tone, context),
         responseFormat,
     ].join('\n')

src/lib/utils.ts

@@ -13,8 +13,8 @@ export const parseSummaryToHumanReadable = (rawOutput: string): string => {
     return match[2].trim().replace(/\s*###\s*$/, '')
 }
 
-export const hasPartnerMessages = (formattedRows: Message[]) =>
-    formattedRows.some((msg) => msg.sender !== 'me')
+export const hasContactMessages = (formattedRows: Message[]) =>
+    formattedRows.some((msg) => msg.sender === 'them')
 
 const cleanReplyText = (text: string): string => {
     let cleaned = text.trim()