diff --git a/src/core/message-parsing/DirectiveStreamingParser.ts b/src/core/message-parsing/DirectiveStreamingParser.ts index fe8c6a8258..9cfd874180 100644 --- a/src/core/message-parsing/DirectiveStreamingParser.ts +++ b/src/core/message-parsing/DirectiveStreamingParser.ts @@ -124,9 +124,9 @@ export class DirectiveStreamingParser { private static isInsideCodeBlock(context: ParseContext, activeHandler: DirectiveHandler | null): boolean { return ( context.codeBlockState === CodeBlockState.INSIDE || - (activeHandler && + (!!activeHandler && activeHandler instanceof ToolDirectiveHandler && - (activeHandler as ParameterCodeBlockHandler).isInsideParameterCodeBlock()) + !!(activeHandler as ParameterCodeBlockHandler).isInsideParameterCodeBlock()) ) }