diff --git a/src/OllamaControlApi.php b/src/OllamaControlApi.php
index 13b66dc16c88fbb4ee23f32ecb5f1ad2a5924adf..f3aca32649a8553f224fab29293162528a084847 100644
--- a/src/OllamaControlApi.php
+++ b/src/OllamaControlApi.php
@@ -62,8 +62,8 @@ class OllamaControlApi {
    *   The response.
    */
   public function embeddings($text, $model) {
-    $result = json_decode($this->makeRequest("api/embeddings", [], 'POST', [
-      'prompt' => $text,
+    $result = json_decode($this->makeRequest("api/embed", [], 'POST', [
+      'input' => $text,
       'model' => $model,
     ]), TRUE);
     return $result;
diff --git a/src/Plugin/AiProvider/OllamaProvider.php b/src/Plugin/AiProvider/OllamaProvider.php
index a7b9bb14d959cb9b8ee43c6ce17051efc61c9c26..3fd8cbe6a95d7953ad0bb2d1976b99a50f010e54 100644
--- a/src/Plugin/AiProvider/OllamaProvider.php
+++ b/src/Plugin/AiProvider/OllamaProvider.php
@@ -285,7 +285,7 @@ class OllamaProvider extends AiProviderClientBase implements
     }
     $response = $this->controlApi->embeddings($input, $model_id);
 
-    return new EmbeddingsOutput($response['embedding'], $response, []);
+    return new EmbeddingsOutput($response['embeddings'][0], $response, []);
   }
 
   /**