pacnpal/Roo-Code
1
0
Fork 0
mirror of https://github.com/pacnpal/Roo-Code.git synced 2025-12-21 12:51:17 -05:00
Code Issues Packages Projects Releases Wiki Activity

Deepseek fixes

Browse Source
This commit is contained in:
Matt Rubens
2025-01-20 11:02:09 -05:00
parent eff636aef0
commit a13da25dd1
5 changed files with 25 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/api/providers/__tests__/deepseek.test.ts
View File

@@ -71,7 +71,7 @@ describe("DeepSeekHandler", () => {
beforeEach(() => {
mockOptions = {
deepSeekApiKey: "test-api-key",
deepSeekModelId: "deepseek-chat",
apiModelId: "deepseek-chat",
deepSeekBaseUrl: "https://api.deepseek.com/v1",
}
handler = new DeepSeekHandler(mockOptions)
@@ -81,7 +81,7 @@ describe("DeepSeekHandler", () => {
describe("constructor", () => {
it("should initialize with provided options", () => {
expect(handler).toBeInstanceOf(DeepSeekHandler)
expect(handler.getModel().id).toBe(mockOptions.deepSeekModelId)
expect(handler.getModel().id).toBe(mockOptions.apiModelId)
})
it("should throw error if API key is missing", () => {
@@ -96,7 +96,7 @@ describe("DeepSeekHandler", () => {
it("should use default model ID if not provided", () => {
const handlerWithoutModel = new DeepSeekHandler({
...mockOptions,
deepSeekModelId: undefined,
apiModelId: undefined,
})
expect(handlerWithoutModel.getModel().id).toBe(deepSeekDefaultModelId)
})
@@ -144,7 +144,7 @@ describe("DeepSeekHandler", () => {
describe("getModel", () => {
it("should return model info for valid model ID", () => {
const model = handler.getModel()
expect(model.id).toBe(mockOptions.deepSeekModelId)
expect(model.id).toBe(mockOptions.apiModelId)
expect(model.info).toBeDefined()
expect(model.info.maxTokens).toBe(8192)
expect(model.info.contextWindow).toBe(64_000)
@@ -155,7 +155,7 @@ describe("DeepSeekHandler", () => {
it("should return provided model ID with default model info if model does not exist", () => {
const handlerWithInvalidModel = new DeepSeekHandler({
...mockOptions,
deepSeekModelId: "invalid-model",
apiModelId: "invalid-model",
})
const model = handlerWithInvalidModel.getModel()
expect(model.id).toBe("invalid-model") // Returns provided ID
@@ -166,7 +166,7 @@ describe("DeepSeekHandler", () => {
it("should return default model if no model ID is provided", () => {
const handlerWithoutModel = new DeepSeekHandler({
...mockOptions,
deepSeekModelId: undefined,
apiModelId: undefined,
})
const model = handlerWithoutModel.getModel()
expect(model.id).toBe(deepSeekDefaultModelId)