@assistant-ui/react
Version:
React components for AI chat.
1 lines • 4.35 kB
Source Map (JSON)
{"version":3,"sources":["../../src/ui/assistant-message.tsx"],"sourcesContent":["\"use client\";\n\nimport { ComponentPropsWithoutRef, forwardRef, useMemo, type FC } from \"react\";\nimport { MessagePrimitive } from \"../primitives\";\nimport BranchPicker from \"./branch-picker\";\nimport { Avatar } from \"./base/avatar\";\nimport { withDefaults } from \"./utils/withDefaults\";\nimport { useThreadConfig } from \"./thread-config\";\nimport AssistantActionBar from \"./assistant-action-bar\";\nimport ContentPart from \"./content-part\";\n\nconst AssistantMessage: FC = () => {\n return (\n <AssistantMessageRoot>\n <AssistantMessageAvatar />\n <AssistantMessageContent />\n <BranchPicker />\n <AssistantActionBar />\n </AssistantMessageRoot>\n );\n};\n\nAssistantMessage.displayName = \"AssistantMessage\";\n\nconst AssistantMessageAvatar: FC = () => {\n const { assistantAvatar: avatar = { fallback: \"A\" } } = useThreadConfig();\n return <Avatar {...avatar} />;\n};\n\nconst AssistantMessageRoot = withDefaults(MessagePrimitive.Root, {\n className: \"aui-assistant-message-root\",\n});\n\nAssistantMessageRoot.displayName = \"AssistantMessageRoot\";\n\nconst AssistantMessageContentWrapper = withDefaults(\"div\", {\n className: \"aui-assistant-message-content\",\n});\n\n/**\n * @deprecated Use `AssistantMessage.Content.Props` instead. This will be removed in 0.6.\n */\nexport type AssistantMessageContentProps = AssistantMessageContent.Props;\n\nnamespace AssistantMessageContent {\n export type Element = HTMLDivElement;\n export type Props = MessagePrimitive.Content.Props &\n ComponentPropsWithoutRef<\"div\">;\n}\n\nconst AssistantMessageContent = forwardRef<\n AssistantMessageContent.Element,\n AssistantMessageContent.Props\n>(({ components: componentsProp, ...rest }, ref) => {\n const { tools, assistantMessage: { components = {} } = {} } =\n useThreadConfig();\n\n const toolsComponents = useMemo(\n () => ({\n by_name: !tools\n ? undefined\n : Object.fromEntries(\n tools.map((t) => [\n t.unstable_tool.toolName,\n t.unstable_tool.render,\n ]),\n ),\n Fallback: components.ToolFallback,\n }),\n // eslint-disable-next-line react-hooks/exhaustive-deps\n [...(tools ?? []), components.ToolFallback],\n );\n\n return (\n <AssistantMessageContentWrapper {...rest} ref={ref}>\n <MessagePrimitive.Content\n components={{\n ...componentsProp,\n Text: componentsProp?.Text ?? components.Text ?? ContentPart.Text,\n tools: toolsComponents,\n }}\n />\n </AssistantMessageContentWrapper>\n );\n});\n\nAssistantMessageContent.displayName = \"AssistantMessageContent\";\n\nconst exports = {\n Root: AssistantMessageRoot,\n Avatar: AssistantMessageAvatar,\n Content: AssistantMessageContent,\n};\n\nexport default Object.assign(\n AssistantMessage,\n exports,\n) as typeof AssistantMessage & typeof exports;\n"],"mappings":";;;AAEA,SAAmC,YAAY,eAAwB;AACvE,SAAS,wBAAwB;AACjC,OAAO,kBAAkB;AACzB,SAAS,cAAc;AACvB,SAAS,oBAAoB;AAC7B,SAAS,uBAAuB;AAChC,OAAO,wBAAwB;AAC/B,OAAO,iBAAiB;AAIpB,SACE,KADF;AAFJ,IAAM,mBAAuB,MAAM;AACjC,SACE,qBAAC,wBACC;AAAA,wBAAC,0BAAuB;AAAA,IACxB,oBAAC,2BAAwB;AAAA,IACzB,oBAAC,gBAAa;AAAA,IACd,oBAAC,sBAAmB;AAAA,KACtB;AAEJ;AAEA,iBAAiB,cAAc;AAE/B,IAAM,yBAA6B,MAAM;AACvC,QAAM,EAAE,iBAAiB,SAAS,EAAE,UAAU,IAAI,EAAE,IAAI,gBAAgB;AACxE,SAAO,oBAAC,UAAQ,GAAG,QAAQ;AAC7B;AAEA,IAAM,uBAAuB,aAAa,iBAAiB,MAAM;AAAA,EAC/D,WAAW;AACb,CAAC;AAED,qBAAqB,cAAc;AAEnC,IAAM,iCAAiC,aAAa,OAAO;AAAA,EACzD,WAAW;AACb,CAAC;AAaD,IAAM,0BAA0B,WAG9B,CAAC,EAAE,YAAY,gBAAgB,GAAG,KAAK,GAAG,QAAQ;AAClD,QAAM,EAAE,OAAO,kBAAkB,EAAE,aAAa,CAAC,EAAE,IAAI,CAAC,EAAE,IACxD,gBAAgB;AAElB,QAAM,kBAAkB;AAAA,IACtB,OAAO;AAAA,MACL,SAAS,CAAC,QACN,SACA,OAAO;AAAA,QACL,MAAM,IAAI,CAAC,MAAM;AAAA,UACf,EAAE,cAAc;AAAA,UAChB,EAAE,cAAc;AAAA,QAClB,CAAC;AAAA,MACH;AAAA,MACJ,UAAU,WAAW;AAAA,IACvB;AAAA;AAAA,IAEA,CAAC,GAAI,SAAS,CAAC,GAAI,WAAW,YAAY;AAAA,EAC5C;AAEA,SACE,oBAAC,kCAAgC,GAAG,MAAM,KACxC;AAAA,IAAC,iBAAiB;AAAA,IAAjB;AAAA,MACC,YAAY;AAAA,QACV,GAAG;AAAA,QACH,MAAM,gBAAgB,QAAQ,WAAW,QAAQ,YAAY;AAAA,QAC7D,OAAO;AAAA,MACT;AAAA;AAAA,EACF,GACF;AAEJ,CAAC;AAED,wBAAwB,cAAc;AAEtC,IAAM,UAAU;AAAA,EACd,MAAM;AAAA,EACN,QAAQ;AAAA,EACR,SAAS;AACX;AAEA,IAAO,4BAAQ,OAAO;AAAA,EACpB;AAAA,EACA;AACF;","names":[]}