Skip to content

Commit 4d6667b

Browse files
jrhannesrudolph
authored andcommitted
fix: get model id from ApiHandler (#3512)
Not all handlers/providers use apiModelId, but they do all return a model id from getModel().
1 parent 4228c8b commit 4d6667b

File tree

2 files changed

+4
-5
lines changed

2 files changed

+4
-5
lines changed

src/core/environment/__tests__/getEnvironmentDetails.test.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,6 @@ describe("getEnvironmentDetails", () => {
6969
maxOpenTabsContext: 10,
7070
mode: "code",
7171
customModes: [],
72-
apiModelId: "test-model",
7372
experiments: {},
7473
customInstructions: "test instructions",
7574
language: "en",
@@ -102,7 +101,7 @@ describe("getEnvironmentDetails", () => {
102101
} as unknown as RooIgnoreController,
103102
clineMessages: [],
104103
api: {
105-
getModel: jest.fn().mockReturnValue({ info: { contextWindow: 100000 } }),
104+
getModel: jest.fn().mockReturnValue({ id: "test-model", info: { contextWindow: 100000 } }),
106105
createMessage: jest.fn(),
107106
countTokens: jest.fn(),
108107
} as unknown as ApiHandler,
@@ -144,6 +143,7 @@ describe("getEnvironmentDetails", () => {
144143
expect(result).toContain("# Current Context Size (Tokens)")
145144
expect(result).toContain("# Current Cost")
146145
expect(result).toContain("# Current Mode")
146+
expect(result).toContain("<model>test-model</model>")
147147

148148
expect(mockProvider.getState).toHaveBeenCalled()
149149

src/core/environment/getEnvironmentDetails.ts

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -190,7 +190,7 @@ export async function getEnvironmentDetails(cline: Task, includeFileDetails: boo
190190

191191
// Add context tokens information.
192192
const { contextTokens, totalCost } = getApiMetrics(cline.clineMessages)
193-
const modelInfo = cline.api.getModel().info
193+
const { id: modelId, info: modelInfo } = cline.api.getModel()
194194
const contextWindow = modelInfo.contextWindow
195195

196196
const contextPercentage =
@@ -203,7 +203,6 @@ export async function getEnvironmentDetails(cline: Task, includeFileDetails: boo
203203
const {
204204
mode,
205205
customModes,
206-
apiModelId,
207206
customModePrompts,
208207
experiments = {} as Record<ExperimentId, boolean>,
209208
customInstructions: globalCustomInstructions,
@@ -221,7 +220,7 @@ export async function getEnvironmentDetails(cline: Task, includeFileDetails: boo
221220
details += `\n\n# Current Mode\n`
222221
details += `<slug>${currentMode}</slug>\n`
223222
details += `<name>${modeDetails.name}</name>\n`
224-
details += `<model>${apiModelId}</model>\n`
223+
details += `<model>${modelId}</model>\n`
225224

226225
if (Experiments.isEnabled(experiments ?? {}, EXPERIMENT_IDS.POWER_STEERING)) {
227226
details += `<role>${modeDetails.roleDefinition}</role>\n`

0 commit comments

Comments
 (0)