opencode-morphllm 0.0.5 → 0.0.6

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.
@@ -0,0 +1,51 @@
1
+ import { describe, it, expect, vi } from 'bun:test';
2
+
3
+ vi.mock('../shared/config', () => ({
4
+ API_KEY: 'test-api-key-123',
5
+ }));
6
+
7
+ import { createBuiltinMcps } from './mcps';
8
+
9
+ describe('mcps.ts', () => {
10
+ describe('createBuiltinMcps', () => {
11
+ it('should create morph_mcp configuration', () => {
12
+ const mcps = createBuiltinMcps();
13
+
14
+ expect(mcps).toHaveProperty('morph_mcp');
15
+ expect(mcps.morph_mcp.type).toBe('local');
16
+ expect(mcps.morph_mcp.enabled).toBe(true);
17
+ });
18
+
19
+ it('should set correct command for morph_mcp', () => {
20
+ const mcps = createBuiltinMcps();
21
+
22
+ expect(mcps.morph_mcp.command).toEqual([
23
+ 'npx',
24
+ '-y',
25
+ '@morphllm/morphmcp',
26
+ ]);
27
+ });
28
+
29
+ it('should set MORPH_API_KEY in environment', () => {
30
+ const mcps = createBuiltinMcps();
31
+ const env = mcps.morph_mcp.environment;
32
+
33
+ expect(env).toBeDefined();
34
+ expect(env?.MORPH_API_KEY).toBe('test-api-key-123');
35
+ });
36
+
37
+ it('should set ENABLED_TOOLS environment variable', () => {
38
+ const mcps = createBuiltinMcps();
39
+ const env = mcps.morph_mcp.environment;
40
+
41
+ expect(env).toBeDefined();
42
+ expect(env?.ENABLED_TOOLS).toBe('edit_file,warpgrep_codebase_search');
43
+ });
44
+
45
+ it('should only create morph_mcp key', () => {
46
+ const mcps = createBuiltinMcps();
47
+
48
+ expect(Object.keys(mcps)).toEqual(['morph_mcp']);
49
+ });
50
+ });
51
+ });
@@ -1,5 +1,5 @@
1
1
  import type { McpLocalConfig } from '@opencode-ai/sdk';
2
- import { API_KEY } from './config';
2
+ import { API_KEY } from '../shared/config';
3
3
 
4
4
  export function createBuiltinMcps(): Record<string, McpLocalConfig> {
5
5
  return {
@@ -0,0 +1,266 @@
1
+ import { describe, it, expect, vi } from 'bun:test';
2
+ import type { Part } from '@opencode-ai/sdk';
3
+
4
+ vi.mock('@morphllm/morphsdk', () => ({
5
+ MorphClient: vi.fn().mockImplementation(() => ({
6
+ routers: {
7
+ raw: {
8
+ classify: vi.fn(),
9
+ },
10
+ },
11
+ })),
12
+ }));
13
+
14
+ vi.mock('../shared/config', () => ({
15
+ API_KEY: 'sk-test-api-key-123',
16
+ MORPH_MODEL_EASY: 'easy/easy',
17
+ MORPH_MODEL_MEDIUM: 'medium/medium',
18
+ MORPH_MODEL_HARD: 'hard/hard',
19
+ MORPH_MODEL_DEFAULT: 'default/default',
20
+ MORPH_ROUTER_ONLY_FIRST_MESSAGE: false,
21
+ }));
22
+
23
+ import { createModelRouterHook, extractPromptText } from './router';
24
+
25
+ describe('router.ts', () => {
26
+ describe('extractPromptText', () => {
27
+ it('should extract text from parts', () => {
28
+ const parts: Part[] = [
29
+ { type: 'text', text: 'Hello' },
30
+ { type: 'text', text: 'World' },
31
+ ] as any;
32
+ expect(extractPromptText(parts)).toBe('Hello World');
33
+ });
34
+
35
+ it('should handle empty parts', () => {
36
+ const parts: Part[] = [];
37
+ expect(extractPromptText(parts)).toBe('');
38
+ });
39
+
40
+ it('should filter out non-text parts', () => {
41
+ const parts: Part[] = [
42
+ { type: 'text', text: 'Hello' },
43
+ { type: 'other', text: 'Ignore' },
44
+ { type: 'text', text: 'World' },
45
+ ] as any;
46
+ expect(extractPromptText(parts)).toBe('Hello World');
47
+ });
48
+
49
+ it('should handle parts with no text', () => {
50
+ const parts: Part[] = [
51
+ { type: 'text', text: 'Hello' },
52
+ { type: 'text' },
53
+ { type: 'text', text: 'World' },
54
+ ] as any;
55
+ expect(extractPromptText(parts)).toBe('Hello World');
56
+ });
57
+
58
+ it('should handle all non-text parts', () => {
59
+ const parts: Part[] = [
60
+ { type: 'image', data: 'base64...' },
61
+ { type: 'tool_use', name: 'bash' },
62
+ ] as any;
63
+ expect(extractPromptText(parts)).toBe('');
64
+ });
65
+ });
66
+
67
+ describe('createModelRouterHook', () => {
68
+ it('should return a chat.message hook', () => {
69
+ const hook = createModelRouterHook();
70
+ expect('chat.message' in hook).toBe(true);
71
+ expect(typeof hook['chat.message']).toBe('function');
72
+ });
73
+
74
+ it('should call the classifier with the correct input', async () => {
75
+ const hook = createModelRouterHook();
76
+ const classify = vi.fn().mockResolvedValue({ difficulty: 'easy' });
77
+ const input = {
78
+ sessionID: '123',
79
+ classify,
80
+ };
81
+ const output = {
82
+ message: {} as any,
83
+ parts: [{ type: 'text', text: 'test prompt' }],
84
+ } as any;
85
+ await hook['chat.message'](input as any, output);
86
+ expect(classify).toHaveBeenCalledWith({ input: 'test prompt' });
87
+ });
88
+
89
+ it('should assign the correct model based on difficulty', async () => {
90
+ const hook = createModelRouterHook();
91
+ const classify = vi.fn().mockResolvedValue({ difficulty: 'hard' });
92
+ const input = {
93
+ sessionID: '123',
94
+ classify,
95
+ };
96
+ const output = {
97
+ message: {} as any,
98
+ parts: [{ type: 'text', text: 'test prompt' }],
99
+ } as any;
100
+ await hook['chat.message'](input as any, output);
101
+ expect(input).toHaveProperty('model');
102
+ expect((input as any).model.providerID).toBe('hard');
103
+ expect((input as any).model.modelID).toBe('hard');
104
+ });
105
+
106
+ it('should use the default model if no difficulty is returned', async () => {
107
+ const hook = createModelRouterHook();
108
+ const classify = vi.fn().mockResolvedValue({});
109
+ const input = {
110
+ sessionID: '123',
111
+ classify,
112
+ };
113
+ const output = {
114
+ message: {} as any,
115
+ parts: [{ type: 'text', text: 'test prompt' }],
116
+ } as any;
117
+ await hook['chat.message'](input as any, output);
118
+ expect(input).toHaveProperty('model');
119
+ expect((input as any).model.providerID).toBe('default');
120
+ expect((input as any).model.modelID).toBe('default');
121
+ });
122
+
123
+ it('should use default model when router returns nothing', async () => {
124
+ const hook = createModelRouterHook();
125
+ const classify = vi.fn().mockResolvedValue({});
126
+ const input = {
127
+ sessionID: '123',
128
+ classify,
129
+ };
130
+ const output = {
131
+ message: {} as any,
132
+ parts: [{ type: 'text', text: 'test prompt' }],
133
+ } as any;
134
+ await hook['chat.message'](input as any, output);
135
+ expect(input).toHaveProperty('model');
136
+ expect((input as any).model.providerID).toBe('default');
137
+ expect((input as any).model.modelID).toBe('default');
138
+ });
139
+
140
+ it('should handle classifier throwing an error gracefully', async () => {
141
+ const hook = createModelRouterHook();
142
+ const classify = vi.fn().mockRejectedValue(new Error('API Error'));
143
+ const input = {
144
+ sessionID: '123',
145
+ classify,
146
+ };
147
+ const output = {
148
+ message: {} as any,
149
+ parts: [{ type: 'text', text: 'test prompt' }],
150
+ } as any;
151
+
152
+ await expect(hook['chat.message'](input as any, output)).rejects.toThrow(
153
+ 'API Error'
154
+ );
155
+ });
156
+
157
+ it('should handle classifier returning null', async () => {
158
+ const hook = createModelRouterHook();
159
+ const classify = vi.fn().mockResolvedValue(null);
160
+ const input = {
161
+ sessionID: '123',
162
+ classify,
163
+ };
164
+ const output = {
165
+ message: {} as any,
166
+ parts: [{ type: 'text', text: 'test prompt' }],
167
+ } as any;
168
+ await hook['chat.message'](input as any, output);
169
+ expect((input as any).model.providerID).toBe('default');
170
+ expect((input as any).model.modelID).toBe('default');
171
+ });
172
+
173
+ it('should handle classifier returning undefined difficulty', async () => {
174
+ const hook = createModelRouterHook();
175
+ const classify = vi.fn().mockResolvedValue({ difficulty: undefined });
176
+ const input = {
177
+ sessionID: '123',
178
+ classify,
179
+ };
180
+ const output = {
181
+ message: {} as any,
182
+ parts: [{ type: 'text', text: 'test prompt' }],
183
+ } as any;
184
+ await hook['chat.message'](input as any, output);
185
+ expect((input as any).model.providerID).toBe('default');
186
+ expect((input as any).model.modelID).toBe('default');
187
+ });
188
+
189
+ it('should default to default model for invalid difficulty values', async () => {
190
+ const hook = createModelRouterHook();
191
+ const classify = vi.fn().mockResolvedValue({ difficulty: 'ultra-hard' });
192
+ const input = {
193
+ sessionID: '123',
194
+ classify,
195
+ };
196
+ const output = {
197
+ message: {} as any,
198
+ parts: [{ type: 'text', text: 'test prompt' }],
199
+ } as any;
200
+ await hook['chat.message'](input as any, output);
201
+ expect((input as any).model.providerID).toBe('default');
202
+ expect((input as any).model.modelID).toBe('default');
203
+ });
204
+
205
+ it('should default to default model for empty string difficulty', async () => {
206
+ const hook = createModelRouterHook();
207
+ const classify = vi.fn().mockResolvedValue({ difficulty: '' });
208
+ const input = {
209
+ sessionID: '123',
210
+ classify,
211
+ };
212
+ const output = {
213
+ message: {} as any,
214
+ parts: [{ type: 'text', text: 'test prompt' }],
215
+ } as any;
216
+ await hook['chat.message'](input as any, output);
217
+ expect((input as any).model.providerID).toBe('default');
218
+ expect((input as any).model.modelID).toBe('default');
219
+ });
220
+
221
+ it('should handle case-insensitive difficulty matching', async () => {
222
+ const hook = createModelRouterHook();
223
+ const classify = vi.fn().mockResolvedValue({ difficulty: 'HARD' });
224
+ const input = {
225
+ sessionID: '123',
226
+ classify,
227
+ };
228
+ const output = {
229
+ message: {} as any,
230
+ parts: [{ type: 'text', text: 'test prompt' }],
231
+ } as any;
232
+ await hook['chat.message'](input as any, output);
233
+ expect((input as any).model.providerID).toBe('hard');
234
+ expect((input as any).model.modelID).toBe('hard');
235
+ });
236
+
237
+ it('should route all messages when MORPH_ROUTER_ONLY_FIRST_MESSAGE is disabled', async () => {
238
+ const hook = createModelRouterHook();
239
+ const classify = vi
240
+ .fn()
241
+ .mockResolvedValueOnce({ difficulty: 'hard' })
242
+ .mockResolvedValueOnce({ difficulty: 'easy' });
243
+
244
+ const sessionID = 'session-456';
245
+ const input1 = { sessionID, classify };
246
+ const output1 = {
247
+ message: {} as any,
248
+ parts: [{ type: 'text', text: 'first message' }],
249
+ } as any;
250
+
251
+ await hook['chat.message'](input1 as any, output1);
252
+ expect(classify).toHaveBeenCalledTimes(1);
253
+ expect((input1 as any).model.providerID).toBe('hard');
254
+
255
+ const input2 = { sessionID, classify };
256
+ const output2 = {
257
+ message: {} as any,
258
+ parts: [{ type: 'text', text: 'second message' }],
259
+ } as any;
260
+
261
+ await hook['chat.message'](input2 as any, output2);
262
+ expect(classify).toHaveBeenCalledTimes(2);
263
+ expect((input2 as any).model.providerID).toBe('easy');
264
+ });
265
+ });
266
+ });
@@ -6,7 +6,8 @@ import {
6
6
  MORPH_MODEL_MEDIUM,
7
7
  MORPH_MODEL_HARD,
8
8
  MORPH_MODEL_DEFAULT,
9
- } from './config';
9
+ MORPH_ROUTER_ONLY_FIRST_MESSAGE,
10
+ } from '../shared/config';
10
11
  import type { Part, UserMessage } from '@opencode-ai/sdk';
11
12
  import type {
12
13
  RouterInput,
@@ -14,7 +15,17 @@ import type {
14
15
  ComplexityLevel,
15
16
  } from '@morphllm/morphsdk';
16
17
 
17
- const morph = new MorphClient({ apiKey: API_KEY });
18
+ // Lazy initialization to allow mocking in tests
19
+ let morph: MorphClient | null = null;
20
+
21
+ const sessionsWithModelSelected = new Set<string>();
22
+
23
+ function getMorphClient(): MorphClient {
24
+ if (!morph) {
25
+ morph = new MorphClient({ apiKey: API_KEY });
26
+ }
27
+ return morph;
28
+ }
18
29
 
19
30
  function parseModel(s?: string): { providerID: string; modelID: string } {
20
31
  if (!s) return { providerID: '', modelID: '' };
@@ -54,11 +65,17 @@ export function createModelRouterHook() {
54
65
  ): Promise<void> => {
55
66
  input.model = input.model ?? { providerID: '', modelID: '' };
56
67
 
68
+ if (MORPH_ROUTER_ONLY_FIRST_MESSAGE) {
69
+ if (sessionsWithModelSelected.has(input.sessionID)) {
70
+ return;
71
+ }
72
+ }
73
+
57
74
  const promptText = extractPromptText(output.parts);
58
75
 
59
76
  const classifier =
60
77
  input.classify ??
61
- ((args: RouterInput) => morph.routers.raw.classify(args));
78
+ ((args: RouterInput) => getMorphClient().routers.raw.classify(args));
62
79
 
63
80
  const classification: RawRouterResult = await classifier({
64
81
  input: promptText,
@@ -71,6 +88,10 @@ export function createModelRouterHook() {
71
88
 
72
89
  input.model.providerID = finalProviderID;
73
90
  input.model.modelID = finalModelID;
91
+
92
+ if (MORPH_ROUTER_ONLY_FIRST_MESSAGE) {
93
+ sessionsWithModelSelected.add(input.sessionID);
94
+ }
74
95
  },
75
96
  };
76
97
  }
@@ -0,0 +1,245 @@
1
+ import {
2
+ describe,
3
+ it,
4
+ expect,
5
+ beforeEach,
6
+ afterEach,
7
+ vi,
8
+ beforeAll,
9
+ afterAll,
10
+ } from 'bun:test';
11
+ import {
12
+ existsSync,
13
+ writeFileSync,
14
+ readFileSync,
15
+ rmSync,
16
+ mkdirSync,
17
+ } from 'node:fs';
18
+ import { join } from 'node:path';
19
+ import { tmpdir } from 'node:os';
20
+
21
+ // Mock the opencode-config-dir module before importing config
22
+ const mockConfigDir = join(tmpdir(), 'mock-opencode-config-test');
23
+ const mockMorphJson = join(mockConfigDir, 'morph.json');
24
+ const mockMorphJsonc = join(mockConfigDir, 'morph.jsonc');
25
+
26
+ vi.mock('./opencode-config-dir', () => ({
27
+ getOpenCodeConfigDir: vi.fn(() => mockConfigDir),
28
+ }));
29
+
30
+ import {
31
+ getMorphPluginConfigPath,
32
+ loadMorphPluginConfig,
33
+ loadMorphPluginConfigWithProjectOverride,
34
+ } from './config';
35
+
36
+ describe('config.ts', () => {
37
+ beforeAll(() => {
38
+ // Create mock config directory
39
+ if (!existsSync(mockConfigDir)) {
40
+ mkdirSync(mockConfigDir, { recursive: true });
41
+ }
42
+ });
43
+
44
+ afterAll(() => {
45
+ // Cleanup
46
+ try {
47
+ if (existsSync(mockMorphJson)) rmSync(mockMorphJson);
48
+ if (existsSync(mockMorphJsonc)) rmSync(mockMorphJsonc);
49
+ if (existsSync(mockConfigDir)) rmSync(mockConfigDir, { recursive: true });
50
+ } catch {
51
+ // Ignore cleanup errors
52
+ }
53
+ });
54
+
55
+ beforeEach(() => {
56
+ // Clean up any existing config files before each test
57
+ try {
58
+ if (existsSync(mockMorphJson)) rmSync(mockMorphJson);
59
+ if (existsSync(mockMorphJsonc)) rmSync(mockMorphJsonc);
60
+ } catch {
61
+ // Ignore cleanup errors
62
+ }
63
+ });
64
+
65
+ describe('getMorphPluginConfigPath', () => {
66
+ it('should return path to morph.json in config directory', () => {
67
+ const path = getMorphPluginConfigPath();
68
+ expect(path).toContain('morph.json');
69
+ expect(path).toContain(mockConfigDir);
70
+ });
71
+ });
72
+
73
+ describe('loadMorphPluginConfig', () => {
74
+ it('should return null when no config files exist', () => {
75
+ const result = loadMorphPluginConfig();
76
+ expect(result).toBeNull();
77
+ });
78
+
79
+ it('should load config from .jsonc file', () => {
80
+ const content = `{
81
+ "MORPH_API_KEY": "test-key-123",
82
+ "MORPH_ROUTER_ENABLED": false
83
+ }`;
84
+ writeFileSync(mockMorphJsonc, content);
85
+
86
+ const result = loadMorphPluginConfig();
87
+ expect(result).not.toBeNull();
88
+ expect(result?.MORPH_API_KEY).toBe('test-key-123');
89
+ expect(result?.MORPH_ROUTER_ENABLED).toBe(false);
90
+ });
91
+
92
+ it('should load config from .json file', () => {
93
+ const content = JSON.stringify({
94
+ MORPH_API_KEY: 'json-key-456',
95
+ MORPH_MODEL_EASY: 'provider/model-easy',
96
+ });
97
+ writeFileSync(mockMorphJson, content);
98
+
99
+ const result = loadMorphPluginConfig();
100
+ expect(result).not.toBeNull();
101
+ expect(result?.MORPH_API_KEY).toBe('json-key-456');
102
+ expect(result?.MORPH_MODEL_EASY).toBe('provider/model-easy');
103
+ });
104
+
105
+ it('should prefer .jsonc over .json when both exist', () => {
106
+ const jsoncContent = '{"MORPH_API_KEY": "from-jsonc"}';
107
+ const jsonContent = '{"MORPH_API_KEY": "from-json"}';
108
+
109
+ writeFileSync(mockMorphJsonc, jsoncContent);
110
+ writeFileSync(mockMorphJson, jsonContent);
111
+
112
+ const result = loadMorphPluginConfig();
113
+ expect(result?.MORPH_API_KEY).toBe('from-jsonc');
114
+ });
115
+
116
+ it('should return null for invalid .jsonc content', () => {
117
+ writeFileSync(mockMorphJsonc, 'invalid json content');
118
+
119
+ const result = loadMorphPluginConfig();
120
+ expect(result).toBeNull();
121
+ });
122
+
123
+ it('should return null for invalid .json content', () => {
124
+ writeFileSync(mockMorphJson, 'invalid json content');
125
+
126
+ const result = loadMorphPluginConfig();
127
+ expect(result).toBeNull();
128
+ });
129
+
130
+ it('should handle all MORPH config fields', () => {
131
+ const content = JSON.stringify({
132
+ MORPH_API_KEY: 'api-key',
133
+ MORPH_ROUTER_ENABLED: true,
134
+ MORPH_MODEL_EASY: 'easy/provider/model',
135
+ MORPH_MODEL_MEDIUM: 'medium/provider/model',
136
+ MORPH_MODEL_HARD: 'hard/provider/model',
137
+ MORPH_MODEL_DEFAULT: 'default/provider/model',
138
+ });
139
+ writeFileSync(mockMorphJson, content);
140
+
141
+ const result = loadMorphPluginConfig();
142
+ expect(result).toEqual({
143
+ MORPH_API_KEY: 'api-key',
144
+ MORPH_ROUTER_ENABLED: true,
145
+ MORPH_MODEL_EASY: 'easy/provider/model',
146
+ MORPH_MODEL_MEDIUM: 'medium/provider/model',
147
+ MORPH_MODEL_HARD: 'hard/provider/model',
148
+ MORPH_MODEL_DEFAULT: 'default/provider/model',
149
+ });
150
+ });
151
+
152
+ it('should handle nested MORPH_ROUTER_CONFIGS', () => {
153
+ const content = JSON.stringify({
154
+ MORPH_API_KEY: 'api-key',
155
+ MORPH_ROUTER_CONFIGS: {
156
+ MORPH_ROUTER_ENABLED: true,
157
+ MORPH_MODEL_EASY: 'easy/provider/model',
158
+ MORPH_MODEL_MEDIUM: 'medium/provider/model',
159
+ MORPH_MODEL_HARD: 'hard/provider/model',
160
+ MORPH_MODEL_DEFAULT: 'default/provider/model',
161
+ },
162
+ });
163
+ writeFileSync(mockMorphJson, content);
164
+
165
+ const result = loadMorphPluginConfig();
166
+ expect(result?.MORPH_API_KEY).toBe('api-key');
167
+ expect(result?.MORPH_ROUTER_CONFIGS).toEqual({
168
+ MORPH_ROUTER_ENABLED: true,
169
+ MORPH_MODEL_EASY: 'easy/provider/model',
170
+ MORPH_MODEL_MEDIUM: 'medium/provider/model',
171
+ MORPH_MODEL_HARD: 'hard/provider/model',
172
+ MORPH_MODEL_DEFAULT: 'default/provider/model',
173
+ });
174
+ });
175
+ });
176
+
177
+ describe('loadMorphPluginConfigWithProjectOverride', () => {
178
+ it('should return empty object when no config exists', () => {
179
+ const result =
180
+ loadMorphPluginConfigWithProjectOverride('/non-existent-path');
181
+ expect(result).toEqual({});
182
+ });
183
+
184
+ it('should merge user config with project config', () => {
185
+ // Create user config
186
+ writeFileSync(
187
+ mockMorphJson,
188
+ JSON.stringify({
189
+ MORPH_API_KEY: 'user-api-key',
190
+ MORPH_MODEL_EASY: 'user-easy-model',
191
+ })
192
+ );
193
+
194
+ // Create project config directory
195
+ const projectDir = join(tmpdir(), 'test-project');
196
+ const projectConfigDir = join(projectDir, '.opencode');
197
+ const projectConfigPath = join(projectConfigDir, 'morph.json');
198
+
199
+ mkdirSync(projectConfigDir, { recursive: true });
200
+ writeFileSync(
201
+ projectConfigPath,
202
+ JSON.stringify({
203
+ MORPH_API_KEY: 'project-api-key',
204
+ MORPH_MODEL_HARD: 'project-hard-model',
205
+ })
206
+ );
207
+
208
+ const result = loadMorphPluginConfigWithProjectOverride(projectDir);
209
+
210
+ // Project config should override user config
211
+ expect(result.MORPH_API_KEY).toBe('project-api-key');
212
+ expect(result.MORPH_MODEL_EASY).toBe('user-easy-model');
213
+ expect(result.MORPH_MODEL_HARD).toBe('project-hard-model');
214
+ });
215
+
216
+ it('should handle project .jsonc config', () => {
217
+ // Create user config
218
+ writeFileSync(
219
+ mockMorphJson,
220
+ JSON.stringify({
221
+ MORPH_API_KEY: 'user-key',
222
+ })
223
+ );
224
+
225
+ // Create project config with .jsonc
226
+ const projectDir = join(tmpdir(), 'test-project2');
227
+ const projectConfigDir = join(projectDir, '.opencode');
228
+ const projectConfigPath = join(projectConfigDir, 'morph.jsonc');
229
+
230
+ mkdirSync(projectConfigDir, { recursive: true });
231
+ const jsoncContent = `
232
+ // Project config with comments
233
+ {
234
+ "MORPH_MODEL_MEDIUM": "project-medium-model"
235
+ }
236
+ `;
237
+ writeFileSync(projectConfigPath, jsoncContent);
238
+
239
+ const result = loadMorphPluginConfigWithProjectOverride(projectDir);
240
+
241
+ expect(result.MORPH_API_KEY).toBe('user-key');
242
+ expect(result.MORPH_MODEL_MEDIUM).toBe('project-medium-model');
243
+ });
244
+ });
245
+ });
@@ -1,6 +1,6 @@
1
1
  import { existsSync, readFileSync } from 'node:fs';
2
2
  import { join } from 'node:path';
3
- import { getOpenCodeConfigDir } from './shared/opencode-config-dir';
3
+ import { getOpenCodeConfigDir } from './opencode-config-dir';
4
4
 
5
5
  const MORPH_PLUGIN_NAME = 'morph';
6
6
 
@@ -9,9 +9,21 @@ export function getMorphPluginConfigPath(): string {
9
9
  return join(configDir, `${MORPH_PLUGIN_NAME}.json`);
10
10
  }
11
11
 
12
+ interface MorphRouterConfigs {
13
+ MORPH_ROUTER_ENABLED?: boolean;
14
+ MORPH_ROUTER_ONLY_FIRST_MESSAGE?: boolean;
15
+ MORPH_MODEL_EASY?: string;
16
+ MORPH_MODEL_MEDIUM?: string;
17
+ MORPH_MODEL_HARD?: string;
18
+ MORPH_MODEL_DEFAULT?: string;
19
+ }
20
+
12
21
  interface MorphConfig {
13
22
  MORPH_API_KEY?: string;
23
+ MORPH_ROUTER_CONFIGS?: MorphRouterConfigs;
24
+ // Legacy fields for backward compatibility
14
25
  MORPH_ROUTER_ENABLED?: boolean;
26
+ MORPH_ROUTER_ONLY_FIRST_MESSAGE?: boolean;
15
27
  MORPH_MODEL_EASY?: string;
16
28
  MORPH_MODEL_MEDIUM?: string;
17
29
  MORPH_MODEL_HARD?: string;
@@ -85,10 +97,22 @@ export function loadMorphPluginConfigWithProjectOverride(
85
97
 
86
98
  const config = loadMorphPluginConfigWithProjectOverride();
87
99
 
100
+ const routerConfigs = config.MORPH_ROUTER_CONFIGS || {};
101
+
88
102
  export const API_KEY = config.MORPH_API_KEY || '';
89
- export const MORPH_MODEL_EASY = config.MORPH_MODEL_EASY || '';
90
- export const MORPH_MODEL_MEDIUM = config.MORPH_MODEL_MEDIUM || '';
91
- export const MORPH_MODEL_HARD = config.MORPH_MODEL_HARD || '';
103
+ export const MORPH_MODEL_EASY =
104
+ routerConfigs.MORPH_MODEL_EASY || config.MORPH_MODEL_EASY || '';
105
+ export const MORPH_MODEL_MEDIUM =
106
+ routerConfigs.MORPH_MODEL_MEDIUM || config.MORPH_MODEL_MEDIUM || '';
107
+ export const MORPH_MODEL_HARD =
108
+ routerConfigs.MORPH_MODEL_HARD || config.MORPH_MODEL_HARD || '';
92
109
  export const MORPH_MODEL_DEFAULT =
93
- config.MORPH_MODEL_DEFAULT || MORPH_MODEL_MEDIUM;
94
- export const MORPH_ROUTER_ENABLED = config.MORPH_ROUTER_ENABLED ?? true;
110
+ routerConfigs.MORPH_MODEL_DEFAULT ||
111
+ config.MORPH_MODEL_DEFAULT ||
112
+ MORPH_MODEL_MEDIUM;
113
+ export const MORPH_ROUTER_ENABLED =
114
+ routerConfigs.MORPH_ROUTER_ENABLED ?? config.MORPH_ROUTER_ENABLED ?? true;
115
+ export const MORPH_ROUTER_ONLY_FIRST_MESSAGE =
116
+ routerConfigs.MORPH_ROUTER_ONLY_FIRST_MESSAGE ??
117
+ config.MORPH_ROUTER_ONLY_FIRST_MESSAGE ??
118
+ false;
File without changes
File without changes