Building

Author: nang-dev

package-lock.json

@@ -381,6 +381,7 @@
 		"pretty-bytes": "^6.1.1",
 		"puppeteer-chromium-resolver": "^23.0.0",
 		"puppeteer-core": "^23.4.0",
+		"react-tooltip": "^5.28.0",
 		"reconnecting-eventsource": "^1.6.4",
 		"say": "^0.16.0",
 		"serialize-error": "^11.0.3",

package.json

@@ -15,109 +15,7 @@ export class DeepSeekHandler extends OpenAiHandler {
 		})
 	}
 
-	private get baseUrl(): string {
-		return this.options.deepSeekBaseUrl ?? "https://api.deepseek.com/v1"
-	}
-
-	async *createMessage(systemPrompt: string, messages: Anthropic.Messages.MessageParam[]): ApiStream {
-		const modelInfo = this.getModel().info
-		const modelId = this.options.apiModelId ?? deepSeekDefaultModelId
-		const isReasoner = modelId.includes("deepseek-reasoner")
-
-		const systemMessage = { role: "system", content: systemPrompt }
-		const formattedMessages = isReasoner
-			? convertToR1Format([{ role: "user", content: systemPrompt }, ...messages])
-			: [systemMessage, ...convertToOpenAiMessages(messages)]
-
-		const response = await fetch(`${this.baseUrl}/chat/completions`, {
-			method: "POST",
-			headers: {
-				"Content-Type": "application/json",
-				Authorization: `Bearer ${this.options.deepSeekApiKey}`,
-			},
-			body: JSON.stringify({
-				model: modelId,
-				messages: formattedMessages,
-				temperature: 0,
-				stream: true,
-				max_tokens: modelInfo.maxTokens,
-			}),
-		})
-
-		if (!response.ok) {
-			throw new Error(`DeepSeek API error: ${response.statusText}`)
-		}
-
-		if (!response.body) {
-			throw new Error("No response body received from DeepSeek API")
-		}
-
-		const reader = response.body.getReader()
-		const decoder = new TextDecoder()
-		let buffer = ""
-
-		try {
-			while (true) {
-				const { done, value } = await reader.read()
-				if (done) break
-
-				buffer += decoder.decode(value, { stream: true })
-				const lines = buffer.split("\n")
-				buffer = lines.pop() || ""
-
-				for (const line of lines) {
-					if (line.trim() === "") continue
-					if (!line.startsWith("data: ")) continue
-
-					const data = line.slice(6)
-					if (data === "[DONE]") continue
-
-					try {
-						const chunk = JSON.parse(data)
-						// Handle regular delta format
-						const delta = chunk.choices[0]?.delta ?? {}
-						if (delta.type === "ui") {
-							yield {
-								type: "text",
-								text: delta.metadata?.content || "",
-								metadata: delta.metadata,
-							}
-						} else if (delta.content) {
-							yield {
-								type: "text",
-								text: delta.content,
-							}
-						}
-
-						if ("reasoning_content" in delta && delta.reasoning_content) {
-							yield {
-								type: "reasoning",
-								text: delta.reasoning_content,
-							}
-						}
-
-						if (chunk.usage) {
-							const usage = chunk.usage as DeepSeekUsage
-							let inputTokens = (usage.prompt_tokens || 0) - (usage.prompt_cache_hit_tokens || 0)
-							yield {
-								type: "usage",
-								inputTokens: inputTokens,
-								outputTokens: usage.completion_tokens || 0,
-								cacheReadTokens: usage.prompt_cache_hit_tokens || 0,
-								cacheWriteTokens: usage.prompt_cache_miss_tokens || 0,
-							}
-						}
-					} catch (error) {
-						console.error("Error parsing DeepSeek response:", error)
-					}
-				}
-			}
-		} finally {
-			reader.releaseLock()
-		}
-	}
-
-	getModel(): { id: string; info: ModelInfo } {
+	override getModel(): { id: string; info: ModelInfo } {
 		const modelId = this.options.apiModelId ?? deepSeekDefaultModelId
 		const info = deepSeekModels[modelId as keyof typeof deepSeekModels] || deepSeekModels[deepSeekDefaultModelId]
 
@@ -138,34 +36,4 @@ export class DeepSeekHandler extends OpenAiHandler {
 			cacheReadTokens: usage?.prompt_tokens_details?.cached_tokens,
 		}
 	}
-
-	async completePrompt(prompt: string): Promise<string> {
-		try {
-			const response = await fetch(`${this.baseUrl}/chat/completions`, {
-				method: "POST",
-				headers: {
-					"Content-Type": "application/json",
-					Authorization: `Bearer ${this.options.deepSeekApiKey}`,
-				},
-				body: JSON.stringify({
-					model: this.getModel().id,
-					messages: [{ role: "user", content: prompt }],
-					temperature: 0,
-					stream: false,
-				}),
-			})
-
-			if (!response.ok) {
-				throw new Error(`DeepSeek API error: ${response.statusText}`)
-			}
-
-			const data = await response.json()
-			return data.choices[0]?.message?.content || ""
-		} catch (error) {
-			if (error instanceof Error) {
-				throw new Error(`DeepSeek completion error: ${error.message}`)
-			}
-			throw error
-		}
-	}
 }

src/api/providers/deepseek.ts

@@ -2,6 +2,7 @@ import * as vscode from "vscode"
 import { ApiHandlerOptions, PEARAI_URL, ModelInfo } from "../../shared/api"
 import { AnthropicHandler } from "./anthropic"
 import { DeepSeekHandler } from "./deepseek"
+import Anthropic from "@anthropic-ai/sdk"
 
 interface PearAiModelsResponse {
 	models: {
@@ -150,4 +151,8 @@ export class PearAiHandler {
 	async completePrompt(prompt: string): Promise<string> {
 		return this.handler.completePrompt(prompt)
 	}
+
+	async countTokens(content: Array<Anthropic.Messages.ContentBlockParam>): Promise<number> {
+		return this.handler.countTokens(content)
+	}
 }