Browse Source

Merge pull request #724 from umi008/fix/679-guard-message-transform

fix: guard message-transform against malformed messages
Alvin 1 month ago
parent
commit
cb6a191e3d
1 changed files with 4 additions and 1 deletions
  1. 4 1
      src/index.ts

+ 4 - 1
src/index.ts

@@ -39,7 +39,7 @@ import {
   SessionLifecycle,
 } from './hooks';
 import { processImageAttachments } from './hooks/image-hook';
-import type { MessageWithParts } from './hooks/types';
+import { isMessageWithParts, type MessageWithParts } from './hooks/types';
 import { createInterviewManager } from './interview';
 import { createBuiltinMcps } from './mcp';
 import {
@@ -1120,6 +1120,9 @@ const OhMyOpenCodeLite: Plugin = async (ctx) => {
       const typedOutput = output as { messages: MessageWithParts[] };
 
       for (const message of typedOutput.messages) {
+        if (!isMessageWithParts(message)) {
+          continue;
+        }
         if (message.info.role !== 'user') {
           continue;
         }