Skip to content

Commit afe63ed

Browse files
committed
Fix comments from code review
1 parent bc32628 commit afe63ed

3 files changed

Lines changed: 11 additions & 14 deletions

File tree

typescript/src/ai-sdk/__tests__/essentials.test.ts

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -144,9 +144,9 @@ describe('CommercetoolsAgentEssentials with Admin tools', () => {
144144
toolFormat
145145
);
146146

147-
expect(Object.keys(agentEssentials.tools)).toContain('tool1');
148-
expect(Object.keys(agentEssentials.tools)).toContain('tool2');
149-
expect(Object.keys(agentEssentials.tools)).not.toContain('tool3');
147+
expect(Object.keys(agentEssentials.getTools())).toContain('tool1');
148+
expect(Object.keys(agentEssentials.getTools())).toContain('tool2');
149+
expect(Object.keys(agentEssentials.getTools())).not.toContain('tool3');
150150
});
151151

152152
it('should return all created tools via getTools method', () => {
@@ -193,6 +193,6 @@ describe('CommercetoolsAgentEssentials with Admin tools', () => {
193193

194194
expect(isToolAllowed).toHaveBeenCalledTimes(tools.length);
195195
expect(CommercetoolsTool).not.toHaveBeenCalled();
196-
expect(Object.keys(agentEssentials.tools).length).toBe(0);
196+
expect(Object.keys(agentEssentials.getTools()).length).toBe(0);
197197
});
198198
});

typescript/src/mastra/__tests__/essentials.test.ts

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -145,9 +145,9 @@ describe('CommercetoolsAgentEssentials with Mastra', () => {
145145
toolFormat
146146
);
147147

148-
expect(Object.keys(agentEssentials.tools)).toContain('tool1');
149-
expect(Object.keys(agentEssentials.tools)).toContain('tool2');
150-
expect(Object.keys(agentEssentials.tools)).not.toContain('tool3');
148+
expect(Object.keys(agentEssentials.getTools())).toContain('tool1');
149+
expect(Object.keys(agentEssentials.getTools())).toContain('tool2');
150+
expect(Object.keys(agentEssentials.getTools())).not.toContain('tool3');
151151
});
152152

153153
it('should return all created tools via getTools method', () => {
@@ -194,7 +194,7 @@ describe('CommercetoolsAgentEssentials with Mastra', () => {
194194

195195
expect(isToolAllowed).toHaveBeenCalledTimes(tools.length);
196196
expect(CommercetoolsTool).not.toHaveBeenCalled();
197-
expect(Object.keys(agentEssentials.tools).length).toBe(0);
197+
expect(Object.keys(agentEssentials.getTools()).length).toBe(0);
198198
});
199199

200200
it('should store tools as an object with method names as keys', () => {

typescript/src/mastra/essentials.ts

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,7 @@ import {AuthConfig} from '../types/auth';
1212

1313
class CommercetoolsAgentEssentials {
1414
private _commercetools: CommercetoolsAPI;
15-
16-
tools: {[key: string]: Tool};
15+
private _tools: {[key: string]: Tool} = {};
1716

1817
constructor({
1918
authConfig,
@@ -29,12 +28,10 @@ class CommercetoolsAgentEssentials {
2928
processedConfiguration.context
3029
);
3130

32-
this.tools = {};
33-
3431
contextToTools(processedConfiguration.context)
3532
.filter((tool) => isToolAllowed(tool, processedConfiguration))
3633
.forEach((tool) => {
37-
this.tools[tool.method] = CommercetoolsTool(
34+
this._tools[tool.method] = CommercetoolsTool(
3835
this._commercetools,
3936
tool.method,
4037
tool.description,
@@ -45,7 +42,7 @@ class CommercetoolsAgentEssentials {
4542
}
4643

4744
getTools(): {[key: string]: Tool} {
48-
return this.tools;
45+
return this._tools;
4946
}
5047
}
5148

0 commit comments

Comments
 (0)