Add models_deepseek() function for listing available DeepSeek models

NAMESPACE

@@ -86,6 +86,7 @@ export(live_console)
 export(models_anthropic)
 export(models_aws_bedrock)
 export(models_claude)
+export(models_deepseek)
 export(models_github)
 export(models_google_gemini)
 export(models_google_vertex)

NEWS.md

@@ -3,6 +3,7 @@
 * ellmer will now distinguish text content from thinking content while streaming, allowing downstream packages like shinychat to provide specific UI for thinking content (@simonpcouch, #909).
 * `chat_github()` now uses `chat_openai_compatible()` for improved compatibility, and `models_github()` now supports custom `base_url` configuration (@D-M4rk, #877).
 * `chat_ollama()` now contains a slot for `top_k` within the `params` argument (@frankiethull).
+* `models_deepseek()` lists available models for `chat_deepseek()` (@jcrodriguez1989, #919).
 
 # ellmer 0.4.0
 

R/provider-deepseek.R

@@ -127,3 +127,37 @@ method(as_json, list(ProviderDeepSeek, Turn)) <- function(provider, x, ...) {
 }
 
 deepseek_key <- function() key_get("DEEPSEEK_API_KEY")
+
+#' @export
+#' @rdname chat_deepseek
+models_deepseek <- function(
+  base_url = "https://api.deepseek.com",
+  api_key = NULL,
+  credentials = NULL
+) {
+  credentials <- as_credentials(
+    "models_deepseek",
+    \() deepseek_key(),
+    credentials = credentials,
+    api_key = api_key
+  )
+
+  provider <- ProviderDeepSeek(
+    name = "DeepSeek",
+    model = "",
+    base_url = base_url,
+    credentials = credentials
+  )
+
+  req <- base_request(provider)
+  req <- req_url_path_append(req, "/models")
+  resp <- req_perform(req)
+
+  json <- resp_body_json(resp)
+
+  id <- map_chr(json$data, "[[", "id")
+  owned_by <- map_chr(json$data, "[[", "owned_by")
+
+  df <- data.frame(id = id, owned_by = owned_by)
+  cbind(df, match_prices(provider@name, df$id))
+}

tests/testthat/test-provider-deepseek.R

@@ -13,6 +13,10 @@ test_that("can make simple streaming request", {
   expect_match(paste0(unlist(resp), collapse = ""), "2")
 })
 
+test_that("can list models", {
+  test_models(models_deepseek)
+})
+
 # Common provider interface -----------------------------------------------
 
 test_that("defaults are reported", {