feat(js/plugins/ollama): migrate ollama plugin to v2 plugin API

js/plugins/ollama/src/embeddings.ts

@@ -13,11 +13,12 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-import type { Document, EmbedderAction, Genkit } from 'genkit';
+import type { Document, EmbedderAction } from 'genkit';
+import { embedder } from 'genkit/plugin';
 import type { EmbedRequest, EmbedResponse } from 'ollama';
 import type { DefineOllamaEmbeddingParams, RequestHeaders } from './types.js';
 
-async function toOllamaEmbedRequest(
+export async function toOllamaEmbedRequest(
   modelName: string,
   dimensions: number,
   documents: Document[],
@@ -59,11 +60,13 @@ async function toOllamaEmbedRequest(
   };
 }
 
-export function defineOllamaEmbedder(
-  ai: Genkit,
-  { name, modelName, dimensions, options }: DefineOllamaEmbeddingParams
-): EmbedderAction<any> {
-  return ai.defineEmbedder(
+export function defineOllamaEmbedder({
+  name,
+  modelName,
+  dimensions,
+  options,
+}: DefineOllamaEmbeddingParams): EmbedderAction<any> {
+  return embedder(
     {
       name: `ollama/${name}`,
       info: {
@@ -76,12 +79,12 @@ export function defineOllamaEmbedder(
       },
     },
     async (input, config) => {
-      const serverAddress = config?.serverAddress || options.serverAddress;
+      const serverAddress = options.serverAddress || 'http://localhost:11434';
 
       const { url, requestPayload, headers } = await toOllamaEmbedRequest(
         modelName,
         dimensions,
-        input,
+        input.input,
         serverAddress,
         options.requestHeaders
       );