@ray-js/t-agent-plugin-aistream 0.2.2-beta-1 → 0.2.3-beta-1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1573,7 +1573,8 @@ export type ReceivedTextAsrPacket = ReceivedTextPacketBase<ReceivedTextPacketTyp
1573
1573
  text: string;
1574
1574
  }>;
1575
1575
  export type ReceivedTextNlgPacket = ReceivedTextPacketBase<ReceivedTextPacketType.NLG, {
1576
- content: string;
1576
+ reasoningContent?: string;
1577
+ content?: string;
1577
1578
  appendMode: 'append';
1578
1579
  finish: boolean;
1579
1580
  }>;
@@ -0,0 +1,22 @@
1
+ import { ChatMessageObject } from '@ray-js/t-agent';
2
+ import { ChatHistoryStore, ChatHistoryStoreOptions, StoredMessageObject, ChatHistoryStorePagination } from './ChatHistoryStore';
3
+ export declare class ChatHistoryLocalStore implements ChatHistoryStore {
4
+ /** 版本号,会作为内部索引的一部分 */
5
+ private readonly version;
6
+ private options;
7
+ constructor(options: ChatHistoryStoreOptions);
8
+ private itemToMessage;
9
+ private getRecordIndex;
10
+ private getQueryCondition;
11
+ query(id: number): Promise<StoredMessageObject | null>;
12
+ queryAll(pagination: ChatHistoryStorePagination): Promise<{
13
+ total: number;
14
+ records: StoredMessageObject[];
15
+ }>;
16
+ update(id: number, body: ChatMessageObject): Promise<void>;
17
+ remove(id: number): Promise<void>;
18
+ removeAll(ids?: number[]): Promise<void>;
19
+ insert(message: ChatMessageObject): Promise<{
20
+ id: number;
21
+ }>;
22
+ }
@@ -0,0 +1,191 @@
1
+ import _objectWithoutProperties from "@babel/runtime/helpers/esm/objectWithoutProperties";
2
+ import _objectSpread from "@babel/runtime/helpers/esm/objectSpread2";
3
+ import _defineProperty from "@babel/runtime/helpers/esm/defineProperty";
4
+ const _excluded = ["id"];
5
+ import "core-js/modules/es.array.sort.js";
6
+ import "core-js/modules/es.json.stringify.js";
7
+ import "core-js/modules/esnext.iterator.constructor.js";
8
+ import "core-js/modules/esnext.iterator.filter.js";
9
+ import "core-js/modules/esnext.iterator.map.js";
10
+ import { safeParseJSON } from '@ray-js/t-agent';
11
+ import logger from './utils/logger';
12
+ import { deleteRecordList, insertRecord, queryRecordList, updateRecord } from './utils';
13
+ export class ChatHistoryLocalStore {
14
+ constructor(options) {
15
+ /** 版本号,会作为内部索引的一部分 */
16
+ _defineProperty(this, "version", 'v1');
17
+ if (!options.agentId) {
18
+ throw new Error('agentId is required');
19
+ }
20
+ if (!options.bizCode) {
21
+ throw new Error('bizCode is required');
22
+ }
23
+ this.options = _objectSpread({
24
+ indexId: 'default',
25
+ homeId: 0
26
+ }, options);
27
+ }
28
+ itemToMessage(item) {
29
+ const messageInfo = safeParseJSON(item.data);
30
+ if (!messageInfo) {
31
+ logger.error('ChatHistoryLocalStore failed parse item: ', item);
32
+ return null;
33
+ }
34
+ messageInfo.message.meta.id = item.id;
35
+ return {
36
+ id: item.id,
37
+ indexId: item.index2,
38
+ agentId: item.solutionCode,
39
+ deviceId: item.devId,
40
+ homeId: item.homeId,
41
+ createdAt: messageInfo.createdAt,
42
+ updatedAt: messageInfo.updatedAt,
43
+ message: messageInfo.message
44
+ };
45
+ }
46
+ getRecordIndex() {
47
+ return {
48
+ bizCode: this.options.bizCode,
49
+ solutionCode: this.options.agentId,
50
+ homeId: this.options.homeId,
51
+ index: "t-agent-".concat(this.version),
52
+ index2: this.options.indexId,
53
+ devId: this.options.deviceId
54
+ };
55
+ }
56
+ getQueryCondition() {
57
+ const indexes = this.getRecordIndex();
58
+ const params = {
59
+ bizCode: [indexes.bizCode],
60
+ solutionCode: [indexes.solutionCode],
61
+ homeId: [indexes.homeId],
62
+ index: [indexes.index],
63
+ index2: [indexes.index2]
64
+ };
65
+ if (indexes.devId) {
66
+ params.devId = [indexes.devId];
67
+ }
68
+ return params;
69
+ }
70
+ async query(id) {
71
+ if (typeof id !== 'number') {
72
+ logger.warn('ChatHistoryLocalStore query id is not number', {
73
+ id
74
+ });
75
+ return null;
76
+ }
77
+ const queryCondition = _objectSpread(_objectSpread({}, this.getQueryCondition()), {}, {
78
+ id: [id],
79
+ offset: 0,
80
+ limit: 1,
81
+ sortType: 1
82
+ });
83
+ const {
84
+ records
85
+ } = await queryRecordList(queryCondition);
86
+ return records[0] ? this.itemToMessage(records[0]) : null;
87
+ }
88
+ async queryAll(pagination) {
89
+ const queryCondition = _objectSpread(_objectSpread({}, this.getQueryCondition()), {}, {
90
+ offset: 0,
91
+ sortType: 1
92
+ });
93
+ const {
94
+ offset,
95
+ limit
96
+ } = pagination || {};
97
+ if (offset) {
98
+ queryCondition.offset = offset;
99
+ }
100
+ if (limit) {
101
+ queryCondition.limit = limit;
102
+ }
103
+ queryCondition.sortType = pagination.sort === 'desc' ? 0 : 1;
104
+ try {
105
+ const result = await queryRecordList(queryCondition);
106
+ const list = result.records.map(this.itemToMessage).filter(item => !!item);
107
+ return {
108
+ total: result.total,
109
+ records: list
110
+ };
111
+ } catch (error) {
112
+ logger.error('ChatHistoryLocalStore queryRecordList error', error);
113
+ return {
114
+ total: 0,
115
+ records: []
116
+ };
117
+ }
118
+ }
119
+ async update(id, body) {
120
+ if (typeof id !== 'number') {
121
+ throw new Error('query id is not number');
122
+ }
123
+ const stored = await this.query(id);
124
+ if (!stored) {
125
+ throw new Error("could not find message: ".concat(id));
126
+ }
127
+ const nowTime = Date.now();
128
+ const msgObject = {
129
+ message: body,
130
+ createdAt: stored.createdAt,
131
+ updatedAt: nowTime
132
+ };
133
+ stored.message.meta.updatedAt = nowTime;
134
+ return updateRecord({
135
+ id,
136
+ data: JSON.stringify(msgObject)
137
+ });
138
+ }
139
+ async remove(id) {
140
+ if (typeof id !== 'number') {
141
+ throw new Error('query id is not number');
142
+ }
143
+ const stored = await this.query(id);
144
+ if (!stored) {
145
+ logger.warn('ChatHistoryLocalStore not find by id', {
146
+ id
147
+ });
148
+ }
149
+ await deleteRecordList({
150
+ id: [id]
151
+ });
152
+ }
153
+ async removeAll(ids) {
154
+ if (ids) {
155
+ if (!Array.isArray(ids)) {
156
+ throw new Error('ids is not array');
157
+ }
158
+ if (ids.length === 0) {
159
+ return;
160
+ }
161
+ await deleteRecordList(_objectSpread({
162
+ id: ids
163
+ }, this.getQueryCondition()));
164
+ } else {
165
+ await deleteRecordList(this.getQueryCondition());
166
+ }
167
+ }
168
+ async insert(message) {
169
+ const nowTime = Date.now();
170
+
171
+ // 先把 id 提取出来不存,id 始终是在数据库里自增的
172
+ const _message$meta = message.meta,
173
+ {
174
+ id
175
+ } = _message$meta,
176
+ meta = _objectWithoutProperties(_message$meta, _excluded);
177
+ const msgObject = {
178
+ message: _objectSpread(_objectSpread({}, message), {}, {
179
+ meta: _objectSpread(_objectSpread(_objectSpread({}, meta), this.options), {}, {
180
+ createdAt: nowTime,
181
+ updatedAt: nowTime
182
+ })
183
+ }),
184
+ createdAt: nowTime,
185
+ updatedAt: nowTime
186
+ };
187
+ return insertRecord(_objectSpread(_objectSpread({}, this.getRecordIndex()), {}, {
188
+ data: JSON.stringify(msgObject)
189
+ }));
190
+ }
191
+ }
@@ -10,7 +10,7 @@ export interface StoredMessageObject {
10
10
  homeId?: number;
11
11
  indexId?: string;
12
12
  }
13
- interface Pagination {
13
+ export interface ChatHistoryStorePagination {
14
14
  sort?: 'asc' | 'desc';
15
15
  offset: number;
16
16
  limit: number;
@@ -30,7 +30,7 @@ export interface ChatHistoryStore<Key = number> {
30
30
  /** 查询消息 */
31
31
  query(id: Key): Promise<StoredMessageObject>;
32
32
  /** 批量查询 */
33
- queryAll(pagination: Pagination): Promise<{
33
+ queryAll(pagination: ChatHistoryStorePagination): Promise<{
34
34
  total: number;
35
35
  records: StoredMessageObject[];
36
36
  }>;
@@ -45,24 +45,3 @@ export interface ChatHistoryStore<Key = number> {
45
45
  id: Key;
46
46
  }>;
47
47
  }
48
- export declare class ChatHistoryLocalStore implements ChatHistoryStore {
49
- /** 版本号,会作为内部索引的一部分 */
50
- private readonly version;
51
- private options;
52
- constructor(options: ChatHistoryStoreOptions);
53
- private itemToMessage;
54
- private getRecordIndex;
55
- private getQueryCondition;
56
- query(id: number): Promise<StoredMessageObject | null>;
57
- queryAll(pagination: Pagination): Promise<{
58
- total: number;
59
- records: StoredMessageObject[];
60
- }>;
61
- update(id: number, body: ChatMessageObject): Promise<void>;
62
- remove(id: number): Promise<void>;
63
- removeAll(ids?: number[]): Promise<void>;
64
- insert(message: ChatMessageObject): Promise<{
65
- id: number;
66
- }>;
67
- }
68
- export {};
@@ -1,191 +1 @@
1
- import _objectWithoutProperties from "@babel/runtime/helpers/esm/objectWithoutProperties";
2
- import _objectSpread from "@babel/runtime/helpers/esm/objectSpread2";
3
- import _defineProperty from "@babel/runtime/helpers/esm/defineProperty";
4
- const _excluded = ["id"];
5
- import "core-js/modules/es.array.sort.js";
6
- import "core-js/modules/es.json.stringify.js";
7
- import "core-js/modules/esnext.iterator.constructor.js";
8
- import "core-js/modules/esnext.iterator.filter.js";
9
- import "core-js/modules/esnext.iterator.map.js";
10
- import { safeParseJSON } from '@ray-js/t-agent';
11
- import { insertRecord, deleteRecordList, updateRecord, queryRecordList } from './utils';
12
- import logger from './utils/logger';
13
- export class ChatHistoryLocalStore {
14
- constructor(options) {
15
- /** 版本号,会作为内部索引的一部分 */
16
- _defineProperty(this, "version", 'v1');
17
- if (!options.agentId) {
18
- throw new Error('agentId is required');
19
- }
20
- if (!options.bizCode) {
21
- throw new Error('indexId is required');
22
- }
23
- this.options = _objectSpread({
24
- indexId: 'default',
25
- homeId: 0
26
- }, options);
27
- }
28
- itemToMessage(item) {
29
- const messageInfo = safeParseJSON(item.data);
30
- if (!messageInfo) {
31
- logger.error('ChatHistoryLocalStore failed parse item: ', item);
32
- return null;
33
- }
34
- messageInfo.message.meta.id = item.id;
35
- return {
36
- id: item.id,
37
- indexId: item.index2,
38
- agentId: item.solutionCode,
39
- deviceId: item.devId,
40
- homeId: item.homeId,
41
- createdAt: messageInfo.createdAt,
42
- updatedAt: messageInfo.updatedAt,
43
- message: messageInfo.message
44
- };
45
- }
46
- getRecordIndex() {
47
- return {
48
- bizCode: this.options.bizCode,
49
- solutionCode: this.options.agentId,
50
- homeId: this.options.homeId,
51
- index: "t-agent-".concat(this.version),
52
- index2: this.options.indexId,
53
- deviceId: this.options.deviceId
54
- };
55
- }
56
- getQueryCondition() {
57
- const indexes = this.getRecordIndex();
58
- const params = {
59
- bizCode: [indexes.bizCode],
60
- solutionCode: [indexes.solutionCode],
61
- homeId: [indexes.homeId],
62
- index: [indexes.index],
63
- index2: [indexes.index2]
64
- };
65
- if (indexes.deviceId) {
66
- params.devId = [indexes.deviceId];
67
- }
68
- return params;
69
- }
70
- async query(id) {
71
- if (typeof id !== 'number') {
72
- logger.warn('ChatHistoryLocalStore query id is not number', {
73
- id
74
- });
75
- return null;
76
- }
77
- const queryCondition = _objectSpread(_objectSpread({}, this.getQueryCondition()), {}, {
78
- id: [id],
79
- offset: 0,
80
- limit: 1,
81
- sortType: 1
82
- });
83
- const {
84
- records
85
- } = await queryRecordList(queryCondition);
86
- return records[0] ? this.itemToMessage(records[0]) : null;
87
- }
88
- async queryAll(pagination) {
89
- const queryCondition = _objectSpread(_objectSpread({}, this.getQueryCondition()), {}, {
90
- offset: 0,
91
- sortType: 1
92
- });
93
- const {
94
- offset,
95
- limit
96
- } = pagination || {};
97
- if (offset) {
98
- queryCondition.offset = offset;
99
- }
100
- if (limit) {
101
- queryCondition.limit = limit;
102
- }
103
- queryCondition.sortType = pagination.sort === 'desc' ? 0 : 1;
104
- try {
105
- const result = await queryRecordList(queryCondition);
106
- const list = result.records.map(this.itemToMessage).filter(item => !!item);
107
- return {
108
- total: result.total,
109
- records: list
110
- };
111
- } catch (error) {
112
- logger.error('ChatHistoryLocalStore queryRecordList error', error);
113
- return {
114
- total: 0,
115
- records: []
116
- };
117
- }
118
- }
119
- async update(id, body) {
120
- if (typeof id !== 'number') {
121
- throw new Error('query id is not number');
122
- }
123
- const stored = await this.query(id);
124
- if (!stored) {
125
- throw new Error("could not find message: ".concat(id));
126
- }
127
- const nowTime = Date.now();
128
- const msgObject = {
129
- message: body,
130
- createdAt: stored.createdAt,
131
- updatedAt: nowTime
132
- };
133
- stored.message.meta.updatedAt = nowTime;
134
- return updateRecord({
135
- id,
136
- data: JSON.stringify(msgObject)
137
- });
138
- }
139
- async remove(id) {
140
- if (typeof id !== 'number') {
141
- throw new Error('query id is not number');
142
- }
143
- const stored = await this.query(id);
144
- if (!stored) {
145
- logger.warn('ChatHistoryLocalStore not find by id', {
146
- id
147
- });
148
- }
149
- await deleteRecordList({
150
- id: [id]
151
- });
152
- }
153
- async removeAll(ids) {
154
- if (ids) {
155
- if (!Array.isArray(ids)) {
156
- throw new Error('ids is not array');
157
- }
158
- if (ids.length === 0) {
159
- return;
160
- }
161
- await deleteRecordList(_objectSpread({
162
- id: ids
163
- }, this.getQueryCondition()));
164
- } else {
165
- await deleteRecordList(this.getQueryCondition());
166
- }
167
- }
168
- async insert(message) {
169
- const nowTime = Date.now();
170
-
171
- // 先把 id 提取出来不存,id 始终是在数据库里自增的
172
- const _message$meta = message.meta,
173
- {
174
- id
175
- } = _message$meta,
176
- meta = _objectWithoutProperties(_message$meta, _excluded);
177
- const msgObject = {
178
- message: _objectSpread(_objectSpread({}, message), {}, {
179
- meta: _objectSpread(_objectSpread(_objectSpread({}, meta), this.options), {}, {
180
- createdAt: nowTime,
181
- updatedAt: nowTime
182
- })
183
- }),
184
- createdAt: nowTime,
185
- updatedAt: nowTime
186
- };
187
- return insertRecord(_objectSpread(_objectSpread({}, this.getRecordIndex()), {}, {
188
- data: JSON.stringify(msgObject)
189
- }));
190
- }
191
- }
1
+ export {};
@@ -41,41 +41,18 @@ export function withBuildIn() {
41
41
 
42
42
  // 关联文档
43
43
 
44
+ // 关联智能家居卡片
44
45
  (() => {
45
46
  onSkillsEnd((skills, responseMessage) => {
46
47
  if (!responseMessage) {
47
48
  return;
48
49
  }
49
- const data = {
50
- documents: []
50
+ const operateData = {
51
+ deviceInfo: [],
52
+ sceneInfo: [],
53
+ changeInfo: []
51
54
  };
52
- for (const skill of skills) {
53
- var _content$custom;
54
- if (skill.code !== BuildInSkillCode.SEARCH_KNOWLEDGE) {
55
- continue;
56
- }
57
- const content = skill;
58
- if (!((_content$custom = content.custom) !== null && _content$custom !== void 0 && (_content$custom = _content$custom.data) !== null && _content$custom !== void 0 && _content$custom.documents)) {
59
- continue;
60
- }
61
- for (const doc of content.custom.data.documents) {
62
- data.documents.push({
63
- title: doc.title,
64
- url: doc.url
65
- });
66
- }
67
- }
68
- if (data.documents.length) {
69
- responseMessage.bubble.addTile('documents', data);
70
- }
71
- });
72
- })();
73
- (() => {
74
- onSkillsEnd((skills, responseMessage) => {
75
- if (!responseMessage) {
76
- return;
77
- }
78
- const data = {
55
+ const executeData = {
79
56
  deviceInfo: [],
80
57
  sceneInfo: [],
81
58
  changeInfo: []
@@ -87,7 +64,7 @@ export function withBuildIn() {
87
64
  const content = skill;
88
65
  if (content.general.action === ReceivedSmartHomeSkillAction.QUERY_SCENE) {
89
66
  for (const scene of content.general.data.scenes) {
90
- data.sceneInfo.push({
67
+ executeData.sceneInfo.push({
91
68
  type: scene.type,
92
69
  enabled: scene.enable,
93
70
  icon: scene.icon,
@@ -101,7 +78,7 @@ export function withBuildIn() {
101
78
  }
102
79
  } else if (content.general.action === ReceivedSmartHomeSkillAction.CONTROL_SCENE) {
103
80
  for (const scene of content.general.data.scenes) {
104
- data.sceneInfo.push({
81
+ operateData.sceneInfo.push({
105
82
  type: scene.type,
106
83
  icon: scene.icon,
107
84
  name: scene.name,
@@ -114,7 +91,7 @@ export function withBuildIn() {
114
91
  }
115
92
  } else if (content.general.action === ReceivedSmartHomeSkillAction.QUERY_DEVICE) {
116
93
  for (const dev of content.general.data.devices) {
117
- data.deviceInfo.push({
94
+ executeData.deviceInfo.push({
118
95
  icon: dev.icon,
119
96
  name: dev.name,
120
97
  deviceId: dev.deviceId,
@@ -125,7 +102,7 @@ export function withBuildIn() {
125
102
  }
126
103
  } else if (content.general.action === ReceivedSmartHomeSkillAction.CONTROL_DEVICE) {
127
104
  for (const dev of content.general.data.devices) {
128
- data.deviceInfo.push({
105
+ operateData.deviceInfo.push({
129
106
  icon: dev.icon,
130
107
  name: dev.name,
131
108
  deviceId: dev.deviceId,
@@ -135,8 +112,41 @@ export function withBuildIn() {
135
112
  }
136
113
  }
137
114
  }
138
- if (data.deviceInfo.length) {
139
- responseMessage.bubble.addTile('operateCard', data);
115
+ if (executeData.deviceInfo.length || executeData.sceneInfo.length || executeData.changeInfo.length) {
116
+ responseMessage.bubble.addTile('executeCard', executeData);
117
+ }
118
+ if (operateData.deviceInfo.length || operateData.sceneInfo.length || operateData.changeInfo.length) {
119
+ responseMessage.bubble.addTile('operateCard', operateData);
120
+ }
121
+ });
122
+ })();
123
+ // 最后是关联文档
124
+ (() => {
125
+ onSkillsEnd((skills, responseMessage) => {
126
+ if (!responseMessage) {
127
+ return;
128
+ }
129
+ const data = {
130
+ documents: []
131
+ };
132
+ for (const skill of skills) {
133
+ var _content$custom;
134
+ if (skill.code !== BuildInSkillCode.SEARCH_KNOWLEDGE) {
135
+ continue;
136
+ }
137
+ const content = skill;
138
+ if (!((_content$custom = content.custom) !== null && _content$custom !== void 0 && (_content$custom = _content$custom.data) !== null && _content$custom !== void 0 && _content$custom.documents)) {
139
+ continue;
140
+ }
141
+ for (const doc of content.custom.data.documents) {
142
+ data.documents.push({
143
+ title: doc.title,
144
+ url: doc.url
145
+ });
146
+ }
147
+ }
148
+ if (data.documents.length) {
149
+ responseMessage.bubble.addTile('documents', data);
140
150
  }
141
151
  });
142
152
  })();
package/dist/index.d.ts CHANGED
@@ -3,3 +3,4 @@ export * from './AIStreamTypes';
3
3
  export * from './withAIStream';
4
4
  export * from './ChatHistoryStore';
5
5
  export * from './buildIn';
6
+ export * from './ChatHistoryLocalStore';
package/dist/index.js CHANGED
@@ -2,4 +2,5 @@ export * from './utils';
2
2
  export * from './AIStreamTypes';
3
3
  export * from './withAIStream';
4
4
  export * from './ChatHistoryStore';
5
- export * from './buildIn';
5
+ export * from './buildIn';
6
+ export * from './ChatHistoryLocalStore';
@@ -86,7 +86,6 @@ type AIStreamEventWriteChunk = {
86
86
  type AIStreamEventSource = {
87
87
  type: 'audio';
88
88
  dataChannel?: string;
89
- amplitudeCount?: number;
90
89
  userData?: Attribute[];
91
90
  } | {
92
91
  type: 'video';
@@ -8,7 +8,7 @@ import "core-js/modules/esnext.iterator.for-each.js";
8
8
  import "core-js/modules/esnext.iterator.map.js";
9
9
  import "core-js/modules/web.dom-collections.iterator.js";
10
10
  import { AIStreamAttributeType, AIStreamErrorCode, AIStreamServerErrorCode, BizTag, ConnectClientType, ConnectState, EventType, NetworkType, SessionState } from '../AIStreamTypes';
11
- import { closeSession, connect, createSession, disconnect, getCurrentHomeInfo, getNetworkType, isConnected, queryAgentToken, registerRecordAmplitudes, sendEventChatBreak, sendEventEnd, sendEventPayloadEnd, sendEventStart, sendImageData, sendTextData, startRecordAndSendAudioData, stopRecordAndSendAudioData, unregisterVoiceAmplitudes } from './ttt';
11
+ import { closeSession, connect, createSession, disconnect, getCurrentHomeInfo, getNetworkType, isConnected, queryAgentToken, sendEventChatBreak, sendEventEnd, sendEventPayloadEnd, sendEventStart, sendImageData, sendTextData, startRecordAndSendAudioData, stopRecordAndSendAudioData } from './ttt';
12
12
  import { AIStreamObserver, AIStreamObserverPool } from './observer';
13
13
  import { isAbortError } from '@ray-js/t-agent';
14
14
  import logger from './logger';
@@ -456,11 +456,6 @@ export class AIStreamEvent {
456
456
  }
457
457
  stream.started = true;
458
458
  if (source.type === 'audio') {
459
- if (source.amplitudeCount) {
460
- registerRecordAmplitudes({
461
- count: source.amplitudeCount
462
- });
463
- }
464
459
  startPromise = startRecordAndSendAudioData({
465
460
  sessionId: this.sessionId,
466
461
  dataChannel,
@@ -486,11 +481,6 @@ export class AIStreamEvent {
486
481
  startPromise = null;
487
482
  }
488
483
  if (source.type === 'audio') {
489
- if (source.amplitudeCount) {
490
- unregisterVoiceAmplitudes({
491
- count: source.amplitudeCount
492
- });
493
- }
494
484
  stopRecordAndSendAudioData({
495
485
  sessionId: this.sessionId,
496
486
  dataChannel,
@@ -22,14 +22,12 @@ export function sendBlocksToAIStream(params) {
22
22
  signal
23
23
  } = params;
24
24
  let audioEmitter = null;
25
- let amplitudeCount = 0;
26
25
  for (const block of blocks) {
27
26
  if (block.type === 'audio') {
28
27
  if (audioEmitter) {
29
28
  throw new Error('only one audio emitter is allowed');
30
29
  }
31
30
  audioEmitter = block.audio_emitter;
32
- amplitudeCount = block.amplitude_count || 0;
33
31
  }
34
32
  }
35
33
  const attribute = _objectSpread({
@@ -144,11 +142,13 @@ export function sendBlocksToAIStream(params) {
144
142
  }
145
143
  const packet = safeParseJSON(data.body.text);
146
144
  if (packet.bizType === ReceivedTextPacketType.NLG) {
147
- logger.debug('sendBlocksToAIStream Receive NLG', packet.data.content);
148
- const text = prevText + packet.data.content;
145
+ // TODO: 处理 reasoningContent 推理能力
146
+ const delta = packet.data.content || '';
147
+ logger.debug('sendBlocksToAIStream Receive NLG', delta);
148
+ const text = prevText + delta;
149
149
  enqueue({
150
150
  type: 'text',
151
- delta: packet.data.content,
151
+ delta,
152
152
  text,
153
153
  meta
154
154
  });
@@ -291,8 +291,7 @@ export function sendBlocksToAIStream(params) {
291
291
  if (audioEmitter) {
292
292
  await new Promise(resolve => {
293
293
  const s = event.stream({
294
- type: 'audio',
295
- amplitudeCount
294
+ type: 'audio'
296
295
  });
297
296
  audioEmitter.addEventListener('confirm', async () => {
298
297
  if (!canceled) {
@@ -1,6 +1,7 @@
1
- import { ChatAgent, ChatMessage, GetChatPluginHandler, InputBlock } from '@ray-js/t-agent';
1
+ import { ChatAgent, ChatCardObject, ChatMessage, ChatTile, GetChatPluginHandler, InputBlock } from '@ray-js/t-agent';
2
+ import { TTTAction } from './utils';
2
3
  import { ConnectClientType, ReceivedTextSkillPacketBody } from './AIStreamTypes';
3
- import { ChatHistoryLocalStore, StoredMessageObject } from './ChatHistoryStore';
4
+ import { ChatHistoryStore, StoredMessageObject } from './ChatHistoryStore';
4
5
  export interface AIStreamOptions {
5
6
  /** client 类型: 1-作为设备代理, 2-作为 App */
6
7
  clientType?: ConnectClientType;
@@ -24,8 +25,8 @@ export interface AIStreamOptions {
24
25
  homeId?: number;
25
26
  /** 是否在 onAgentStart 阶段就建立连接 */
26
27
  earlyStart?: boolean;
27
- /** 自定义消息存储, 返回的实例需要实现 ChatHistoryLocalStore 接口, 返回null则不存储历史聊天记录 */
28
- createChatHistoryStore?: (agent: ChatAgent) => ChatHistoryLocalStore | null;
28
+ /** 自定义消息存储, 返回的实例需要实现 ChatHistoryStore 接口, 返回null则不存储历史聊天记录 */
29
+ createChatHistoryStore?: (agent: ChatAgent) => ChatHistoryStore | null;
29
30
  /** 是否开启音频合成 */
30
31
  enableTts?: boolean;
31
32
  }
@@ -40,9 +41,15 @@ export interface AIStreamHooks {
40
41
  onSkillsEnd: (skills: ReceivedTextSkillPacketBody[], respMsg: ChatMessage, result: {
41
42
  messages: ChatMessage[];
42
43
  }) => void;
44
+ onTTTAction: (tile: ChatTile | null, result: {
45
+ action?: TTTAction;
46
+ }) => void;
47
+ onCardsReceived: (skills: ReceivedTextSkillPacketBody[], result: {
48
+ cards: ChatCardObject[];
49
+ }) => void;
43
50
  }
44
51
  export declare function withAIStream(options?: AIStreamOptions): (agent: ChatAgent) => {
45
- hooks: import("hookable").Hookable<any, string>;
52
+ hooks: import("hookable").Hookable<AIStreamHooks, import("hookable").HookKeys<AIStreamHooks>>;
46
53
  aiStream: {
47
54
  send: (blocks: InputBlock[], signal?: AbortSignal, extraOptions?: Record<string, any>) => {
48
55
  response: import("@ray-js/t-agent").StreamResponse;
@@ -56,8 +63,6 @@ export declare function withAIStream(options?: AIStreamOptions): (agent: ChatAge
56
63
  options: AIStreamOptions;
57
64
  removeMessage: (message: ChatMessage) => Promise<void>;
58
65
  clearAllMessages: () => Promise<void>;
59
- onSkillCompose: (fn: AIStreamHooks['onSkillCompose']) => () => void;
60
- onSkillsEnd: (fn: AIStreamHooks['onSkillsEnd']) => () => void;
61
66
  feedback: ({ requestId, type }: {
62
67
  requestId: string;
63
68
  type: string;
@@ -65,5 +70,9 @@ export declare function withAIStream(options?: AIStreamOptions): (agent: ChatAge
65
70
  thingjson?: any;
66
71
  data: string;
67
72
  }>;
73
+ onSkillCompose: (fn: AIStreamHooks['onSkillCompose']) => () => void;
74
+ onSkillsEnd: (fn: AIStreamHooks['onSkillsEnd']) => () => void;
75
+ onCardsReceived: (fn: AIStreamHooks['onCardsReceived']) => () => void;
76
+ onTTTAction: (fn: AIStreamHooks['onTTTAction']) => () => void;
68
77
  };
69
78
  };
@@ -11,9 +11,9 @@ import { BubbleTileStatus, ChatMessageStatus, createHooks, EmitterEvent } from '
11
11
  import { messageAppraise } from './utils/apis';
12
12
  import { getAccountInfo, getCurrentHomeInfo, runTTTAction, sendBlocksToAIStream } from './utils';
13
13
  import { BizCode, ConnectClientType } from './AIStreamTypes';
14
- import { ChatHistoryLocalStore } from './ChatHistoryStore';
15
14
  import { DEFAULT_TOKEN_API, DEFAULT_TOKEN_API_VERSION, globalAIStreamClient } from './global';
16
15
  import logger from './utils/logger';
16
+ import { ChatHistoryLocalStore } from './ChatHistoryLocalStore';
17
17
  export function withAIStream() {
18
18
  let options = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : {};
19
19
  const hooks = createHooks();
@@ -38,11 +38,24 @@ export function withAIStream() {
38
38
  const {
39
39
  deviceId,
40
40
  agentId,
41
+ clientType = ConnectClientType.APP,
41
42
  tokenOptions = {
42
43
  api: DEFAULT_TOKEN_API,
43
44
  version: DEFAULT_TOKEN_API_VERSION
44
45
  }
45
46
  } = options;
47
+ if (!agentId) {
48
+ throw new Error('agentId is required');
49
+ }
50
+ if (!tokenOptions.api || !tokenOptions.version) {
51
+ throw new Error('tokenOptions.api and tokenOptions.version are required');
52
+ }
53
+ if (clientType === ConnectClientType.DEVICE && !deviceId) {
54
+ throw new Error('deviceId is required when clientType is device');
55
+ }
56
+ if (clientType === ConnectClientType.APP && deviceId) {
57
+ throw new Error('deviceId is not allowed when clientType is app');
58
+ }
46
59
  let homeId = options.homeId;
47
60
  if (!homeId) {
48
61
  const info = await getCurrentHomeInfo();
@@ -56,7 +69,7 @@ export function withAIStream() {
56
69
  /*
57
70
  * 2. 获取历史消息
58
71
  */
59
- let historyStore = null;
72
+ let historyStore;
60
73
  if (typeof options.createChatHistoryStore === 'function') {
61
74
  historyStore = options.createChatHistoryStore(agent);
62
75
  } else {
@@ -87,21 +100,18 @@ export function withAIStream() {
87
100
 
88
101
  // 创建 streamSession
89
102
  const connection = globalAIStreamClient.getConnection({
90
- clientType: options.clientType || ConnectClientType.APP,
91
- deviceId: options.deviceId
103
+ clientType,
104
+ deviceId
92
105
  });
93
- if (options.clientType === ConnectClientType.DEVICE && !options.deviceId) {
94
- throw new Error('deviceId is required when clientType is device');
95
- }
96
106
  const streamSession = connection.createSession({
97
- ownerId: options.clientType === ConnectClientType.DEVICE ? options.deviceId : "".concat(homeId),
107
+ ownerId: clientType === ConnectClientType.DEVICE ? deviceId : "".concat(homeId),
98
108
  api: tokenOptions.api,
99
109
  apiVersion: tokenOptions.version,
100
110
  solutionCode: agentId,
101
- extParams: _objectSpread(_objectSpread({}, tokenOptions.extParams), {}, {
111
+ extParams: _objectSpread({
102
112
  needTts: !!options.enableTts,
103
- deviceId: options.deviceId || undefined
104
- })
113
+ deviceId
114
+ }, tokenOptions.extParams)
105
115
  });
106
116
  await session.set('AIStream.streamSession', streamSession);
107
117
  if (options.earlyStart) {
@@ -331,12 +341,24 @@ export function withAIStream() {
331
341
  await hooks.callHook('onSkillsEnd', skills, message, result);
332
342
  await message.update();
333
343
  }
344
+ let valid = false;
334
345
  if (message.bubble.text) {
346
+ valid = true;
335
347
  await message.persist();
336
348
  } else if (message.bubble.status === BubbleTileStatus.NORMAL) {
337
- await message.remove();
349
+ valid = false;
338
350
  } else {
351
+ valid = true;
352
+ }
353
+ if (valid) {
339
354
  await message.persist();
355
+ const [_, ...rest] = result.messages;
356
+ for (const m of rest) {
357
+ await m.show();
358
+ await m.persist();
359
+ }
360
+ } else {
361
+ await message.remove();
340
362
  }
341
363
  return [userMsg, ...result.messages];
342
364
  };
@@ -452,6 +474,43 @@ export function withAIStream() {
452
474
  }
453
475
  }
454
476
  });
477
+ const onSkillsEnd = fn => {
478
+ return hooks.hook('onSkillsEnd', fn);
479
+ };
480
+ onSkillsEnd(async (skills, respMsg, result) => {
481
+ const cards = [];
482
+ for (const skill of skills) {
483
+ var _skill$custom, _skill$general;
484
+ if ((_skill$custom = skill.custom) !== null && _skill$custom !== void 0 && (_skill$custom = _skill$custom.data) !== null && _skill$custom !== void 0 && _skill$custom.aiCards) {
485
+ for (const card of skill.custom.data.aiCards) {
486
+ cards.push(card);
487
+ }
488
+ }
489
+ if ((_skill$general = skill.general) !== null && _skill$general !== void 0 && (_skill$general = _skill$general.data) !== null && _skill$general !== void 0 && _skill$general.aiCards) {
490
+ for (const card of skill.custom.data.aiCards) {
491
+ cards.push(card);
492
+ }
493
+ }
494
+ }
495
+ const r = {
496
+ cards: cards
497
+ };
498
+ await hooks.callHook('onCardsReceived', skills, r);
499
+ for (const card of cards) {
500
+ const m = createMessage({
501
+ role: respMsg.role,
502
+ status: ChatMessageStatus.FINISH,
503
+ meta: {
504
+ eventId: respMsg.meta.eventId,
505
+ sessionId: respMsg.meta.sessionId
506
+ }
507
+ });
508
+ m.addTile('card', {
509
+ card
510
+ });
511
+ result.messages.push(m);
512
+ }
513
+ });
455
514
  return {
456
515
  hooks,
457
516
  aiStream: {
@@ -469,13 +528,17 @@ export function withAIStream() {
469
528
  await historyStore.removeAll();
470
529
  }
471
530
  },
531
+ feedback,
472
532
  onSkillCompose: fn => {
473
533
  return hooks.hook('onSkillCompose', fn);
474
534
  },
475
- onSkillsEnd: fn => {
476
- return hooks.hook('onSkillsEnd', fn);
535
+ onSkillsEnd,
536
+ onCardsReceived: fn => {
537
+ return hooks.hook('onCardsReceived', fn);
477
538
  },
478
- feedback
539
+ onTTTAction: fn => {
540
+ return hooks.hook('onTTTAction', fn);
541
+ }
479
542
  }
480
543
  };
481
544
  };
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@ray-js/t-agent-plugin-aistream",
3
- "version": "0.2.2-beta-1",
3
+ "version": "0.2.3-beta-1",
4
4
  "author": "Tuya.inc",
5
5
  "license": "MIT",
6
6
  "private": false,
@@ -35,5 +35,5 @@
35
35
  "devDependencies": {
36
36
  "@types/url-parse": "^1.4.11"
37
37
  },
38
- "gitHead": "0dd513ad93a351ad5c00a1ee8993c7b62f03640f"
38
+ "gitHead": "ed9810075246e1261af2fdaa41550070313a98dc"
39
39
  }