@pontia/pi-client-plugin 0.7.2 → 0.7.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pontia/pi-client-plugin",
-  "version": "0.7.2",
+  "version": "0.7.4",
   "description": "Pi extension that connects coding-agent sessions to Pontia.",
   "keywords": [
     "pi-package",

package/src/index.ts

@@ -525,13 +525,14 @@ export function createPontiaPiExtension(
     );
   }
 
-  pi.on("before_agent_start", async (event) => {
+  pi.on("before_agent_start", async (event, ctx) => {
     pendingPrompt = event.messageType === "user" ? event.prompt : undefined;
     if (reportingDisabled || !(await confirmManagedPane())) return;
     try {
       const profilePrompt = await loadProfileSystemPrompt(
         rpcConnection,
         boundSessionContext?.sessionId,
+        ctx.sessionManager.buildSessionProjection().messages,
       );
       if (profilePrompt) {
         event.systemPromptOptions.sections.pontia_execution_profile = profilePrompt;

package/src/profile.ts

@@ -4,18 +4,20 @@ import { asRecord, optionalString } from "./values.js";
 export async function loadProfileSystemPrompt(
   connection: Pick<PiConnection, "request">,
   sessionId?: string,
+  existingMessages: readonly unknown[] = [],
 ): Promise<string | undefined> {
   if (!sessionId) return undefined;
   const sessionBody = await connection.request("session.get", { session_id: sessionId });
-  const session = asRecord(asRecord(sessionBody)?.session);
-  const profileId = optionalString(session?.execution_profile_id);
-  const profileVersion = optionalString(session?.execution_profile_version);
-  if (!profileId) return undefined;
-
-  const profileBody = await connection.request("profile.get", {
-    profile_id: profileId,
-    ...(profileVersion ? { version: profileVersion } : {}),
-  });
-  const profile = asRecord(asRecord(profileBody)?.agent_profile);
-  return optionalString(profile?.system_prompt_template);
+  const binding = asRecord(asRecord(sessionBody)?.execution_profile_binding);
+  if (binding) return optionalString(binding.system_prompt);
+  let previous: string | undefined;
+  for (const value of existingMessages) {
+    const message = asRecord(value);
+    if (message?.role !== "system") continue;
+    const sections = asRecord(message.sections);
+    if (sections && "pontia_execution_profile" in sections) {
+      previous = optionalString(sections.pontia_execution_profile);
+    }
+  }
+  return previous;
 }