feat: Add DeepSeek provider

2026-05-30 02:49:12 -04:00 · 2026-05-29 12:41:18 +02:00
parent ea67ba0e2a
commit 033c0e8652
17 changed files with 212 additions and 10 deletions
--- a/providers/DeepSeekProvider.cpp
+++ b/providers/DeepSeekProvider.cpp
@@ -0,0 +1,110 @@
+// Copyright (C) 2024-2026 Petr Mironychev
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+#include "DeepSeekProvider.hpp"
+
+#include <LLMQore/ToolsManager.hpp>
+#include "logger/Logger.hpp"
+#include "settings/ChatAssistantSettings.hpp"
+#include "settings/CodeCompletionSettings.hpp"
+#include "settings/GeneralSettings.hpp"
+#include "settings/ProviderSettings.hpp"
+#include "settings/QuickRefactorSettings.hpp"
+#include "tools/ToolsRegistration.hpp"
+
+#include <QJsonArray>
+#include <QJsonDocument>
+#include <QJsonObject>
+
+namespace QodeAssist::Providers {
+
+DeepSeekProvider::DeepSeekProvider(QObject *parent)
+    : PluginLLMCore::Provider(parent)
+    , m_client(new ::LLMQore::OpenAIClient(QString(), QString(), QString(), this))
+{
+    Tools::registerQodeAssistTools(m_client->tools());
+}
+
+QString DeepSeekProvider::name() const
+{
+    return "DeepSeek";
+}
+
+QString DeepSeekProvider::apiKey() const
+{
+    return Settings::providerSettings().deepSeekApiKey();
+}
+
+QString DeepSeekProvider::url() const
+{
+    return "https://api.deepseek.com";
+}
+
+QFuture<QList<QString>> DeepSeekProvider::getInstalledModels(const QString &url)
+{
+    m_client->setUrl(url);
+    m_client->setApiKey(apiKey());
+    return m_client->listModels();
+}
+
+PluginLLMCore::ProviderID DeepSeekProvider::providerID() const
+{
+    return PluginLLMCore::ProviderID::DeepSeek;
+}
+
+PluginLLMCore::ProviderCapabilities DeepSeekProvider::capabilities() const
+{
+    return PluginLLMCore::ProviderCapability::Tools
+           | PluginLLMCore::ProviderCapability::Thinking
+           | PluginLLMCore::ProviderCapability::ModelListing;
+}
+
+void DeepSeekProvider::prepareRequest(
+    QJsonObject &request,
+    PluginLLMCore::PromptTemplate *prompt,
+    PluginLLMCore::ContextData context,
+    PluginLLMCore::RequestType type,
+    bool isToolsEnabled,
+    bool isThinkingEnabled)
+{
+    if (!prompt->isSupportProvider(providerID())) {
+        LOG_MESSAGE(QString("Template %1 doesn't support %2 provider").arg(name(), prompt->name()));
+    }
+
+    prompt->prepareRequest(request, context);
+
+    auto applyModelParams = [&request](const auto &settings) {
+        request["max_tokens"] = settings.maxTokens();
+        request["temperature"] = settings.temperature();
+
+        if (settings.useTopP())
+            request["top_p"] = settings.topP();
+        if (settings.useFrequencyPenalty())
+            request["frequency_penalty"] = settings.frequencyPenalty();
+        if (settings.usePresencePenalty())
+            request["presence_penalty"] = settings.presencePenalty();
+    };
+
+    if (type == PluginLLMCore::RequestType::CodeCompletion) {
+        applyModelParams(Settings::codeCompletionSettings());
+    } else if (type == PluginLLMCore::RequestType::QuickRefactoring) {
+        applyModelParams(Settings::quickRefactorSettings());
+    } else {
+        applyModelParams(Settings::chatAssistantSettings());
+    }
+
+    if (isToolsEnabled) {
+        auto toolsDefinitions = m_client->tools()->getToolsDefinitions();
+        if (!toolsDefinitions.isEmpty()) {
+            request["tools"] = toolsDefinitions;
+            LOG_MESSAGE(QString("Added %1 tools to DeepSeek request").arg(toolsDefinitions.size()));
+        }
+    }
+}
+
+::LLMQore::BaseClient *DeepSeekProvider::client() const
+{
+    return m_client;
+}
+
+} // namespace QodeAssist::Providers