feat: manually mirror opencoze's code from bytedance

Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf
This commit is contained in:
fanlv
2025-07-20 17:36:12 +08:00
commit 890153324f
14811 changed files with 1923430 additions and 0 deletions

View File

@@ -0,0 +1,31 @@
import { mergeConfig } from 'vite';
import svgr from 'vite-plugin-svgr';
/** @type { import('@storybook/react-vite').StorybookConfig } */
const config = {
stories: ['../stories/**/*.mdx', '../stories/**/*.stories.tsx'],
addons: [
'@storybook/addon-links',
'@storybook/addon-essentials',
'@storybook/addon-onboarding',
'@storybook/addon-interactions',
],
framework: {
name: '@storybook/react-vite',
options: {},
},
docs: {
autodocs: 'tag',
},
viteFinal: config =>
mergeConfig(config, {
plugins: [
svgr({
svgrOptions: {
native: false,
},
}),
],
}),
};
export default config;

View File

@@ -0,0 +1,14 @@
/** @type { import('@storybook/react').Preview } */
const preview = {
parameters: {
actions: { argTypesRegex: "^on[A-Z].*" },
controls: {
matchers: {
color: /(background|color)$/i,
date: /Date$/i,
},
},
},
};
export default preview;

View File

@@ -0,0 +1,5 @@
const { defineConfig } = require('@coze-arch/stylelint-config');
module.exports = defineConfig({
extends: [],
});

View File

@@ -0,0 +1,16 @@
# @coze-common/chat-area-plugin-resume
chat area中断续聊插件
## Features
- [x] eslint & ts
- [x] esm bundle
- [x] umd bundle
- [x] storybook
## Commands
- init: `rush update`
- dev: `npm run dev`
- build: `npm run build`

View File

@@ -0,0 +1,77 @@
/*
* Copyright 2025 coze-dev Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
import { renderHook, act } from '@testing-library/react-hooks';
import { useGetPosition } from '../../src/hooks/use-get-position';
vi.mock('@coze-arch/coze-design', () => ({
Toast: {
error: vi.fn(),
},
}));
const mockPosition = {
coords: {
latitude: 10,
longitude: 20,
},
};
const mockGetPositionSuccess = vi.fn();
describe('useGetPosition', () => {
it('should set loading to true when getSysPosition is called', () => {
const { result } = renderHook(() =>
useGetPosition({ getPositionSuccess: vi.fn() }),
);
vi.stubGlobal('navigator', {
geolocation: {
getCurrentPosition: vi.fn(),
},
});
expect(result.current.loading).toBe(false);
act(() => {
result.current.getSysPosition();
});
expect(result.current.loading).toBe(true);
});
it('should call getPositionSuccess with the position when geolocation is available', () => {
vi.stubGlobal('navigator', {
geolocation: {
getCurrentPosition: vi.fn(successCallback => {
successCallback(mockPosition);
}),
},
});
const { result } = renderHook(() =>
useGetPosition({ getPositionSuccess: mockGetPositionSuccess }),
);
act(() => {
result.current.getSysPosition();
});
expect(result.current.loading).toBe(false);
expect(mockGetPositionSuccess).toHaveBeenCalledWith(mockPosition);
});
});

View File

@@ -0,0 +1,12 @@
{
"operationSettings": [
{
"operationName": "test:cov",
"outputFolderNames": ["coverage"]
},
{
"operationName": "ts-check",
"outputFolderNames": ["./dist"]
}
]
}

View File

@@ -0,0 +1,7 @@
const { defineConfig } = require('@coze-arch/eslint-config');
module.exports = defineConfig({
packageRoot: __dirname,
preset: 'web',
rules: {},
});

View File

@@ -0,0 +1,49 @@
{
"name": "@coze-common/chat-area-plugin-resume",
"version": "0.0.1",
"description": "resume scence",
"license": "Apache-2.0",
"author": "yuwenbinjie@bytedance.com",
"maintainers": [],
"main": "src/index.tsx",
"scripts": {
"build": "exit 0",
"lint": "eslint ./ --cache",
"test": "vitest --run --passWithNoTests",
"test:cov": "npm run test -- --coverage"
},
"dependencies": {
"@coze-arch/bot-semi": "workspace:*",
"@coze-arch/coze-design": "0.0.6-alpha.346d77",
"@coze-common/chat-area": "workspace:*",
"@coze-common/chat-uikit": "workspace:*",
"classnames": "^2.3.2",
"immer": "^10.0.3",
"zustand": "^4.4.7"
},
"devDependencies": {
"@coze-arch/bot-typings": "workspace:*",
"@coze-arch/eslint-config": "workspace:*",
"@coze-arch/i18n": "workspace:*",
"@coze-arch/stylelint-config": "workspace:*",
"@coze-arch/ts-config": "workspace:*",
"@coze-arch/vitest-config": "workspace:*",
"@testing-library/jest-dom": "^6.1.5",
"@testing-library/react": "^14.1.2",
"@testing-library/react-hooks": "^8.0.1",
"@types/react": "18.2.37",
"@types/react-dom": "18.2.15",
"@vitest/coverage-v8": "~3.0.5",
"react": "~18.2.0",
"react-dom": "~18.2.0",
"stylelint": "^15.11.0",
"vite-plugin-svgr": "~3.3.0",
"vitest": "~3.0.5"
},
"peerDependencies": {
"react": ">=18.2.0",
"react-dom": ">=18.2.0"
},
"// deps": "immer@^10.0.3 为脚本自动补齐,请勿改动"
}

View File

@@ -0,0 +1,3 @@
.interrupt-message-box {
margin-top: 8px;
}

View File

@@ -0,0 +1,57 @@
/*
* Copyright 2025 coze-dev Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
import { useState } from 'react';
import classNames from 'classnames';
import { MessageBox as UIKitMessageBox } from '@coze-common/chat-uikit';
import { type CustomComponent } from '@coze-common/chat-area';
import { InterruptMessageContent } from './interrupt-message-content';
import styles from './index.module.less';
export const InterruptMessageBox: CustomComponent['MessageBox'] = props => {
// 用户操作后文案,前端维护暂时状态,刷新消失
const [actionText, setActionText] = useState('');
const { message, meta } = props;
// 不展示逻辑: 为历史消息、无action且不在最后一个group
if (message._fromHistory || (!actionText && !meta.isFromLatestGroup)) {
return null;
}
return (
<div className={classNames(styles['interrupt-message-box'])}>
<UIKitMessageBox
{...props}
messageId={message.message_id}
senderInfo={{ id: '' }}
showUserInfo={false}
theme={actionText ? 'none' : 'border'}
>
<InterruptMessageContent
interruptMessage={message}
actionText={actionText}
setActionText={setActionText}
/>
</UIKitMessageBox>
</div>
);
};
InterruptMessageBox.displayName = 'ChatAreaFunctionCallMessageBox';

View File

@@ -0,0 +1,34 @@
.interrupt-message-box {
user-select: none;
overflow: hidden;
display: flex;
flex-direction: column;
box-sizing: border-box;
width: 100%;
font-weight: 400;
line-height: 16px;
word-break: break-word;
// 用户点击授权后文案样式,无背景
.interrupt-message-action {
padding: 6px;
color: var(--fg-coze-fg-secondary, rgba(6, 7, 9, 50%));
}
// 中断消息内容
.interrupt-message-content {
padding: 12px;
color: var(--fg-coze-fg-primary, rgba(6, 7, 9, 80%));
background-color: var(--bg-coze-bg-max, #FFF);
.interrupt-message-content-btns {
display: flex;
justify-content: flex-end;
margin-top: 16px;
}
}
}

View File

@@ -0,0 +1,127 @@
/*
* Copyright 2025 coze-dev Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
import classNames from 'classnames';
import {
PluginName,
useWriteablePlugin,
type Message,
} from '@coze-common/chat-area';
import { I18n } from '@coze-arch/i18n';
import { Button } from '@coze-arch/coze-design';
import { Space } from '@coze-arch/bot-semi';
import { useGetPosition } from '../../../hooks/use-get-position';
import styles from './index.module.less';
export const InterruptMessageContent: React.FC<{
interruptMessage: Message;
actionText?: string;
setActionText: (actionText: string) => void;
}> = ({ interruptMessage, actionText, setActionText }) => {
const plugin = useWriteablePlugin<unknown>(PluginName.Resume);
const { sendResumeMessage, stopResponding } =
plugin.chatAreaPluginContext?.writeableAPI.message ?? {};
// 获取中断场景、续聊id
const toolCall =
interruptMessage.required_action?.submit_tool_outputs?.tool_calls?.[0];
const { loading, getSysPosition: handleAllowOnce } = useGetPosition({
getPositionSuccess: position => {
setActionText(
I18n.t('chat_geolocation_auth_allow_tip', {
plugin: toolCall?.require_info?.name ?? 'plugin',
}),
);
sendResumeMessage?.({
replyId: interruptMessage.reply_id,
options: {
extendFiled: {
interrupt_message_id: interruptMessage.message_id,
resume_message_id: interruptMessage.reply_id,
tool_outputs: [
{
tool_call_id: toolCall?.id,
output: JSON.stringify({
coordinates: {
longitude: String(position.coords.longitude),
latitude: String(position.coords.latitude),
},
}),
},
],
},
},
});
},
});
const handleReject = () => {
setActionText(
I18n.t('chat_geolocation_auth_decline_tip', {
plugin: toolCall?.require_info?.name ?? 'plugin',
}),
);
stopResponding?.();
};
return (
<div className={classNames(styles['interrupt-message-box'])}>
{actionText ? (
<div className={classNames(styles['interrupt-message-action'])}>
{actionText}
</div>
) : (
<div className={classNames(styles['interrupt-message-content'])}>
{toolCall?.require_info?.require_fields?.includes('coordinates') ? (
<>
{I18n.t('chat_geolocation_auth_request_message', {
plugin_name: toolCall?.require_info?.name ?? 'plugin',
})}
<Space
className={classNames(styles['interrupt-message-content-btns'])}
>
<Button
color="highlight"
size="small"
loading={loading}
onClick={handleAllowOnce}
>
{I18n.t('chat_geolocation_auth_request_message_allow_button')}
</Button>
<Button
color="primary"
size="small"
disabled={loading}
onClick={handleReject}
>
{I18n.t(
'chat_geolocation_auth_request_message_decline_button',
)}
</Button>
</Space>
</>
) : null}
</div>
)}
</div>
);
};
InterruptMessageContent.displayName = 'ChatAreaFunctionCallMessageContent';

View File

@@ -0,0 +1,72 @@
/*
* Copyright 2025 coze-dev Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
import { useState } from 'react';
import { I18n } from '@coze-arch/i18n';
import { Toast } from '@coze-arch/coze-design';
interface GetPositionProps {
getPositionSuccess: (position: GeolocationPosition) => void;
}
export const useGetPosition = ({ getPositionSuccess }: GetPositionProps) => {
// 位置授权按钮loading态
const [loading, setLoading] = useState(false);
const getSysPosition = () => {
setLoading(true);
/** 获取系统地理位置信息 */
if (navigator.geolocation) {
navigator.geolocation.getCurrentPosition(
position => {
setLoading(false);
getPositionSuccess?.(position);
},
error => {
setLoading(false);
switch (error.code) {
case error.PERMISSION_DENIED:
Toast.error(
I18n.t('bot_ide_user_declines_geolocation_auth_toast'),
);
break;
case error.POSITION_UNAVAILABLE:
Toast.error(I18n.t('bot_ide_geolocation_not_usable_toast'));
break;
case error.TIMEOUT:
Toast.error(I18n.t('bot_ide_geolocation_request_timeout_toast'));
break;
default:
Toast.error(
I18n.t('bot_ide_geolocation_request_unknown_error_toast'),
);
break;
}
},
);
} else {
setLoading(false);
Toast.error(I18n.t('bot_ide_browser_not_support_geolocation_toast'));
}
};
return {
loading,
getSysPosition,
};
};

View File

@@ -0,0 +1,26 @@
/*
* Copyright 2025 coze-dev Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
import { type PluginRegistryEntry } from '@coze-common/chat-area';
import { ResumePlugin } from './plugin';
export const ResumePluginRegistry: PluginRegistryEntry<unknown> = {
createPluginBizContext() {
return;
},
Plugin: ResumePlugin,
};

View File

@@ -0,0 +1,35 @@
/*
* Copyright 2025 coze-dev Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
import {
isRequireInfoInterruptMessage,
WriteableMessageLifeCycleService,
type OnBeforeDistributeMessageIntoMemberSetContent,
} from '@coze-common/chat-area';
export class ResumeMessageLifeCycleService extends WriteableMessageLifeCycleService<unknown> {
onBeforeDistributeMessageIntoMemberSet(
ctx: OnBeforeDistributeMessageIntoMemberSetContent,
): OnBeforeDistributeMessageIntoMemberSetContent {
const { message } = ctx;
if (isRequireInfoInterruptMessage(message)) {
return { ...ctx, memberSetType: 'llm' };
}
return ctx;
}
}

View File

@@ -0,0 +1,38 @@
/*
* Copyright 2025 coze-dev Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
import {
isRequireInfoInterruptMessage,
type OnMessageBoxRenderContext,
WriteableRenderLifeCycleService,
} from '@coze-common/chat-area';
import { InterruptMessageBox } from '../custom-components/interrupt-message';
export class ResumeRenderLifeCycleService extends WriteableRenderLifeCycleService<unknown> {
onMessageBoxRender(ctx: OnMessageBoxRenderContext) {
const { message } = ctx;
if (isRequireInfoInterruptMessage(message)) {
return {
...ctx,
MessageBox: InterruptMessageBox,
};
}
return ctx;
}
}

View File

@@ -0,0 +1,34 @@
/*
* Copyright 2025 coze-dev Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
import {
PluginMode,
PluginName,
WriteableChatAreaPlugin,
} from '@coze-common/chat-area';
import { ResumeRenderLifeCycleService } from './life-cycle-service/render-life-cycle-service';
import { ResumeMessageLifeCycleService } from './life-cycle-service/message-life-cycle-service';
export class ResumePlugin extends WriteableChatAreaPlugin<unknown> {
public pluginMode: PluginMode = PluginMode.Writeable;
public pluginName: PluginName = PluginName.Resume;
public lifeCycleServices = {
messageLifeCycleService: new ResumeMessageLifeCycleService(this),
renderLifeCycleService: new ResumeRenderLifeCycleService(this),
};
}

View File

@@ -0,0 +1,17 @@
/*
* Copyright 2025 coze-dev Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
/// <reference types='@coze-arch/bot-typings' />

View File

@@ -0,0 +1,34 @@
import { Meta } from "@storybook/blocks";
<Meta title="Hello world" />
<div className="sb-container">
<div className='sb-section-title'>
# Hello world
Hello world
</div>
</div>
<style>
{`
.sb-container {
margin-bottom: 48px;
}
.sb-section {
width: 100%;
display: flex;
flex-direction: row;
gap: 20px;
}
img {
object-fit: cover;
}
.sb-section-title {
margin-bottom: 32px;
}
`}
</style>

View File

@@ -0,0 +1,42 @@
{
"$schema": "https://json.schemastore.org/tsconfig",
"extends": "@coze-arch/ts-config/tsconfig.web.json",
"compilerOptions": {
"types": [],
"strictNullChecks": true,
"noImplicitAny": true,
"rootDir": "./src",
"outDir": "./dist",
"tsBuildInfoFile": "./dist/tsconfig.build.tsbuildinfo"
},
"include": ["src"],
"references": [
{
"path": "../../../arch/bot-typings/tsconfig.build.json"
},
{
"path": "../../../arch/i18n/tsconfig.build.json"
},
{
"path": "../chat-area/tsconfig.build.json"
},
{
"path": "../chat-uikit/tsconfig.build.json"
},
{
"path": "../../../components/bot-semi/tsconfig.build.json"
},
{
"path": "../../../../config/eslint-config/tsconfig.build.json"
},
{
"path": "../../../../config/stylelint-config/tsconfig.build.json"
},
{
"path": "../../../../config/ts-config/tsconfig.build.json"
},
{
"path": "../../../../config/vitest-config/tsconfig.build.json"
}
]
}

View File

@@ -0,0 +1,15 @@
{
"$schema": "https://json.schemastore.org/tsconfig",
"compilerOptions": {
"composite": true
},
"references": [
{
"path": "./tsconfig.build.json"
},
{
"path": "./tsconfig.misc.json"
}
],
"exclude": ["**/*"]
}

View File

@@ -0,0 +1,18 @@
{
"extends": "@coze-arch/ts-config/tsconfig.web.json",
"$schema": "https://json.schemastore.org/tsconfig",
"include": ["__tests__", "stories", "vitest.config.ts", "tailwind.config.ts"],
"exclude": ["./dist"],
"references": [
{
"path": "./tsconfig.build.json"
}
],
"compilerOptions": {
"rootDir": "./",
"outDir": "./dist",
"types": ["vitest/globals"],
"strictNullChecks": true,
"noImplicitAny": true
}
}

View File

@@ -0,0 +1,22 @@
/*
* Copyright 2025 coze-dev Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
import { defineConfig } from '@coze-arch/vitest-config';
export default defineConfig({
dirname: __dirname,
preset: 'web',
});