Skip to content

Commit 3baf780

Browse files
arvinxxarvinxx
authored
🐛 fix: fix memory schema (lobehub#11645)
* fix memory schema * fix tests * improve memory
1 parent f6988e7 commit 3baf780

File tree

20 files changed

+416
-30
lines changed

20 files changed

+416
-30
lines changed

‎packages/builtin-tool-memory/src/client/Render/AddPreferenceMemory/index.tsx‎

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
'use client';
2+
3+
import type { BuiltinRenderProps } from '@lobechat/types';
4+
import { memo } from 'react';
5+
6+
import type { AddPreferenceMemoryParams, AddPreferenceMemoryState } from '../../../types';
7+
import { PreferenceMemoryCard } from '../../components';
8+
9+
const AddPreferenceMemoryRender = memo<
10+
BuiltinRenderProps<AddPreferenceMemoryParams, AddPreferenceMemoryState>
11+
>(({ args }) => {
12+
return <PreferenceMemoryCard data={args} />;
13+
});
14+
15+
AddPreferenceMemoryRender.displayName = 'AddPreferenceMemoryRender';
16+
17+
export default AddPreferenceMemoryRender;

‎packages/builtin-tool-memory/src/client/Render/index.ts‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@ import type { BuiltinRender } from '@lobechat/types';
22

33
import { MemoryApiName } from '../../types';
44
import AddExperienceMemoryRender from './AddExperienceMemory';
5+
import AddPreferenceMemoryRender from './AddPreferenceMemory';
56
import SearchUserMemoryRender from './SearchUserMemory';
67

78
/**
@@ -11,5 +12,6 @@ import SearchUserMemoryRender from './SearchUserMemory';
1112
*/
1213
export const MemoryRenders: Record<string, BuiltinRender> = {
1314
[MemoryApiName.addExperienceMemory]: AddExperienceMemoryRender as BuiltinRender,
15+
[MemoryApiName.addPreferenceMemory]: AddPreferenceMemoryRender as BuiltinRender,
1416
[MemoryApiName.searchUserMemory]: SearchUserMemoryRender as BuiltinRender,
1517
};

‎packages/builtin-tool-memory/src/client/Streaming/AddPreferenceMemory/index.tsx‎

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
'use client';
2+
3+
import type { BuiltinStreamingProps } from '@lobechat/types';
4+
import { memo } from 'react';
5+
6+
import type { AddPreferenceMemoryParams } from '../../../types';
7+
import { PreferenceMemoryCard } from '../../components';
8+
9+
export const AddPreferenceMemoryStreaming = memo<BuiltinStreamingProps<AddPreferenceMemoryParams>>(
10+
({ args }) => {
11+
return <PreferenceMemoryCard data={args} loading />;
12+
},
13+
);
14+
15+
AddPreferenceMemoryStreaming.displayName = 'AddPreferenceMemoryStreaming';
16+
17+
export default AddPreferenceMemoryStreaming;

‎packages/builtin-tool-memory/src/client/Streaming/index.ts‎

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@ import { type BuiltinStreaming } from '@lobechat/types';
22

33
import { MemoryApiName } from '../../types';
44
import { AddExperienceMemoryStreaming } from './AddExperienceMemory';
5+
import { AddPreferenceMemoryStreaming } from './AddPreferenceMemory';
56

67
/**
78
* Memory Streaming Components Registry
@@ -11,8 +12,8 @@ import { AddExperienceMemoryStreaming } from './AddExperienceMemory';
1112
*/
1213
export const MemoryStreamings: Record<string, BuiltinStreaming> = {
1314
[MemoryApiName.addExperienceMemory]: AddExperienceMemoryStreaming as BuiltinStreaming,
15+
[MemoryApiName.addPreferenceMemory]: AddPreferenceMemoryStreaming as BuiltinStreaming,
1416
};
1517

16-
17-
18-
export {AddExperienceMemoryStreaming} from './AddExperienceMemory';
18+
export { AddExperienceMemoryStreaming } from './AddExperienceMemory';
19+
export { AddPreferenceMemoryStreaming } from './AddPreferenceMemory';

0 commit comments

Comments
 (0)