Skip to content

Commit c18c97a

Browse files
author
Raushen
committed
Passing additionalInfo
1 parent 91af18f commit c18c97a

3 files changed

Lines changed: 17 additions & 8 deletions

File tree

packages/devextreme/js/__internal/grids/grid_core/ai_column/ai_column.integration.test.ts

Lines changed: 15 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1540,14 +1540,16 @@ describe('API Methods', () => {
15401540
describe('API Handlers', () => {
15411541
const columnSendRequestStarted = jest.fn();
15421542
const columnSendRequestResolved = jest.fn();
1543-
const sendRequestSpy = jest.fn();
1543+
const sendRequestPromptSpy = jest.fn();
1544+
const sendRequestDataSpy = jest.fn();
15441545
const abortSpy = jest.fn();
15451546

15461547
beforeEach(() => {
15471548
beforeTest();
15481549
columnSendRequestStarted.mockClear();
15491550
columnSendRequestResolved.mockClear();
1550-
sendRequestSpy.mockClear();
1551+
sendRequestPromptSpy.mockClear();
1552+
sendRequestDataSpy.mockClear();
15511553
abortSpy.mockClear();
15521554
});
15531555

@@ -1568,8 +1570,9 @@ describe('API Handlers', () => {
15681570
},
15691571
});
15701572
const columnAiIntegration = new AIIntegration({
1571-
sendRequest({ prompt }): RequestResult {
1572-
sendRequestSpy(prompt);
1573+
sendRequest({ prompt, data }): RequestResult {
1574+
sendRequestPromptSpy(prompt);
1575+
sendRequestDataSpy(data);
15731576
return aiIntegrationResult();
15741577
},
15751578
});
@@ -1694,7 +1697,7 @@ describe('API Handlers', () => {
16941697
jest.advanceTimersByTime(10000);
16951698
expect(columnSendRequestStarted).toHaveBeenCalledTimes(1);
16961699
expect(columnSendRequestResolved).toHaveBeenCalledTimes(1);
1697-
expect(sendRequestSpy).toHaveBeenCalledWith(expect.objectContaining({
1700+
expect(sendRequestPromptSpy).toHaveBeenCalledWith(expect.objectContaining({
16981701
user: expect.stringContaining('Data: {"2":{"id":2,"name":"Name 2","value":20}}'),
16991702
}));
17001703

@@ -1734,13 +1737,18 @@ describe('API Handlers', () => {
17341737
jest.advanceTimersByTime(10000);
17351738
expect(columnSendRequestStarted).toHaveBeenCalledTimes(1);
17361739
expect(columnSendRequestResolved).toHaveBeenCalledTimes(1);
1737-
expect(sendRequestSpy).toHaveBeenCalledWith(expect.objectContaining({
1738-
user: expect.stringContaining('Data: {"2":{"id":2,"name":"Name 2","value":20}}'),
1740+
expect(sendRequestDataSpy).toHaveBeenCalledWith(expect.objectContaining({
1741+
additionalInfo: { customData: 'My custom data' },
17391742
}));
17401743

17411744
await Promise.resolve();
17421745
expect(abortSpy).toHaveBeenCalledTimes(1);
17431746
});
1747+
1748+
// TODO: Implement after cache is done
1749+
// if('should take into account useCache property', async () => {
1750+
1751+
// });
17441752
});
17451753

17461754
// describe('onAIRequestCompleted', () => {

packages/devextreme/js/__internal/grids/grid_core/ai_column/m_ai_column_integration_controller.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ export class AiColumnIntegrationController extends Controller {
9090
{
9191
text: prompt,
9292
data: reducedData,
93-
params: args.additionalInfo,
93+
additionalInfo: args.additionalInfo,
9494
},
9595
this.getAICommandCallbacks<GenerateGridColumnCommandResult>(columnName, cachedResponse),
9696
);

packages/devextreme/js/common/ai-integration.d.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -163,6 +163,7 @@ export type SmartPasteCommandParams = {
163163
export type GenerateGridColumnCommandParams = {
164164
text: string;
165165
data: Record<PropertyKey, unknown>;
166+
additionalInfo?: Record<string, unknown>;
166167
};
167168

168169
/**

0 commit comments

Comments
 (0)