Make document call component clickable (#558)

2024-11-19 15:46:09 +03:00 · 2024-11-19 15:46:09 +03:00 · 9dcdb7b50b
commit 9dcdb7b50b
parent f360ff96cc
3 changed files with 51 additions and 15 deletions
--- a/components/message.tsx
+++ b/components/message.tsx
@ -104,13 +104,22 @@ export const PreviewMessage = ({
                    {toolName === 'getWeather' ? (
                      <Weather />
                    ) : toolName === 'createDocument' ? (
-                      <DocumentToolCall type="create" args={args} />
+                      <DocumentToolCall
+                        type="create"
+                        args={args}
+                        setBlock={setBlock}
+                      />
                    ) : toolName === 'updateDocument' ? (
-                      <DocumentToolCall type="update" args={args} />
+                      <DocumentToolCall
+                        type="update"
+                        args={args}
+                        setBlock={setBlock}
+                      />
                    ) : toolName === 'requestSuggestions' ? (
                      <DocumentToolCall
                        type="request-suggestions"
                        args={args}
+                        setBlock={setBlock}
                      />
                    ) : null}
                  </div>