UniqAI/bolt.diy
1
0
Fork 0
mirror of https://github.com/stackblitz-labs/bolt.diy synced 2025-03-09 21:50:36 +00:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

fix: support for multiple artifacts to support newer llm

Browse Source
This commit is contained in:
Anirban Kar 2025-03-02 23:22:41 +05:30
parent 4404f4a5c0
commit 67ebc54ac1
4 changed files with 38 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
app/components/chat/Artifact.tsx
View File

@ -23,15 +23,16 @@ if (import.meta.hot) {
interface ArtifactProps {
messageId: string;
artifactId: string;
}
export const Artifact = memo(({ messageId }: ArtifactProps) => {
export const Artifact = memo(({ artifactId }: ArtifactProps) => {
const userToggledActions = useRef(false);
const [showActions, setShowActions] = useState(false);
const [allActionFinished, setAllActionFinished] = useState(false);
const artifacts = useStore(workbenchStore.artifacts);
const artifact = artifacts[messageId];
const artifact = artifacts[artifactId];
const actions = useStore(
computed(artifact.runner.actions, (actions) => {

7
app/components/chat/Markdown.tsx
View File

@ -25,12 +25,17 @@ export const Markdown = memo(({ children, html = false, limitedMarkdown = false
div: ({ className, children, node, ...props }) => {
if (className?.includes('__boltArtifact__')) {
const messageId = node?.properties.dataMessageId as string;
const artifactId = node?.properties.dataArtifactId as string;
if (!messageId) {
logger.error(`Invalid message id ${messageId}`);
}
return <Artifact messageId={messageId} />;
if (!artifactId) {
logger.error(`Invalid artifact id ${artifactId}`);
}
return <Artifact messageId={messageId} artifactId={artifactId} />;
}
if (className?.includes('__boltThought__')) {

22
app/lib/runtime/message-parser.ts
View File

@ -12,6 +12,7 @@ const logger = createScopedLogger('MessageParser');
export interface ArtifactCallbackData extends BoltArtifactData {
messageId: string;
artifactId: string;
}
export interface ActionCallbackData {
@ -34,6 +35,7 @@ export interface ParserCallbacks {
interface ElementFactoryProps {
messageId: string;
artifactId: string;
}
type ElementFactory = (props: ElementFactoryProps) => string;
@ -47,6 +49,7 @@ interface MessageState {
position: number;
insideArtifact: boolean;
insideAction: boolean;
artifactCounter: number;
currentArtifact?: BoltArtifactData;
currentAction: BoltActionData;
actionId: number;
@ -81,6 +84,7 @@ export class StreamingMessageParser {
position: 0,
insideAction: false,
insideArtifact: false,
artifactCounter: 0,
currentAction: { content: '' },
actionId: 0,
};
@ -187,7 +191,11 @@ export class StreamingMessageParser {
break;
}
} else if (artifactCloseIndex !== -1) {
this._options.callbacks?.onArtifactClose?.({ messageId, ...currentArtifact });
this._options.callbacks?.onArtifactClose?.({
messageId,
artifactId: currentArtifact.id,
...currentArtifact,
});
state.insideArtifact = false;
state.currentArtifact = undefined;
@ -220,7 +228,9 @@ export class StreamingMessageParser {
const artifactTitle = this.#extractAttribute(artifactTag, 'title') as string;
const type = this.#extractAttribute(artifactTag, 'type') as string;
const artifactId = this.#extractAttribute(artifactTag, 'id') as string;
// const artifactId = this.#extractAttribute(artifactTag, 'id') as string;
const artifactId = `${messageId}-${state.artifactCounter++}`;
if (!artifactTitle) {
logger.warn('Artifact title missing');
@ -240,11 +250,15 @@ export class StreamingMessageParser {
state.currentArtifact = currentArtifact;
this._options.callbacks?.onArtifactOpen?.({ messageId, ...currentArtifact });
this._options.callbacks?.onArtifactOpen?.({
messageId,
artifactId: currentArtifact.id,
...currentArtifact,
});
const artifactFactory = this._options.artifactElement ?? createArtifactElement;
output += artifactFactory({ messageId });
output += artifactFactory({ messageId, artifactId });
i = openTagEnd + 1;
} else {

22
app/lib/stores/workbench.ts
View File

@ -255,17 +255,17 @@ export class WorkbenchStore {
}
addArtifact({ messageId, title, id, type }: ArtifactCallbackData) {
const artifact = this.#getArtifact(messageId);
const artifact = this.#getArtifact(id);
if (artifact) {
return;
}
if (!this.artifactIdList.includes(messageId)) {
this.artifactIdList.push(messageId);
if (!this.artifactIdList.includes(id)) {
this.artifactIdList.push(id);
}
this.artifacts.setKey(messageId, {
this.artifacts.setKey(id, {
id,
title,
closed: false,
@ -284,14 +284,14 @@ export class WorkbenchStore {
});
}
updateArtifact({ messageId }: ArtifactCallbackData, state: Partial<ArtifactUpdateState>) {
const artifact = this.#getArtifact(messageId);
updateArtifact({ artifactId }: ArtifactCallbackData, state: Partial<ArtifactUpdateState>) {
const artifact = this.#getArtifact(artifactId);
if (!artifact) {
return;
}
this.artifacts.setKey(messageId, { ...artifact, ...state });
this.artifacts.setKey(artifactId, { ...artifact, ...state });
}
addAction(data: ActionCallbackData) {
// this._addAction(data);
@ -299,9 +299,9 @@ export class WorkbenchStore {
this.addToExecutionQueue(() => this._addAction(data));
}
async _addAction(data: ActionCallbackData) {
const { messageId } = data;
const { artifactId } = data;
const artifact = this.#getArtifact(messageId);
const artifact = this.#getArtifact(artifactId);
if (!artifact) {
unreachable('Artifact not found');
@ -318,9 +318,9 @@ export class WorkbenchStore {
}
}
async _runAction(data: ActionCallbackData, isStreaming: boolean = false) {
const { messageId } = data;
const { artifactId } = data;
const artifact = this.#getArtifact(messageId);
const artifact = this.#getArtifact(artifactId);
if (!artifact) {
unreachable('Artifact not found');