Skip to content

Commit b2c9e64

Browse files
authored
Merge pull request #2362 from netbrah/fix/deepseek-r1-output-limit
fix(core): add deepseek-r1 to output token limit patterns
2 parents 9665760 + 4990a56 commit b2c9e64

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

packages/core/src/core/tokenLimits.test.ts

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -284,6 +284,8 @@ describe('tokenLimit with output type', () => {
284284
describe('other output limits', () => {
285285
it('should return correct output limits for DeepSeek', () => {
286286
expect(tokenLimit('deepseek-reasoner', 'output')).toBe(65536);
287+
expect(tokenLimit('deepseek-r1', 'output')).toBe(65536);
288+
expect(tokenLimit('deepseek-r1-0528', 'output')).toBe(65536);
287289
expect(tokenLimit('deepseek-chat', 'output')).toBe(8192);
288290
});
289291

packages/core/src/core/tokenLimits.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -169,6 +169,7 @@ const OUTPUT_PATTERNS: Array<[RegExp, TokenCount]> = [
169169

170170
// DeepSeek
171171
[/^deepseek-reasoner/, LIMITS['64k']],
172+
[/^deepseek-r1/, LIMITS['64k']],
172173
[/^deepseek-chat/, LIMITS['8k']],
173174

174175
// Zhipu GLM

0 commit comments

Comments
 (0)