From 7e60b8fd6367e40ed935466fdafed3f0a18f2a76 Mon Sep 17 00:00:00 2001 From: Julien Breux Date: Mon, 2 Sep 2024 16:00:23 +0200 Subject: [PATCH] fix(init-go): Fix Ollama generated config --- genkit-tools/cli/src/commands/init/init-go.ts | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/genkit-tools/cli/src/commands/init/init-go.ts b/genkit-tools/cli/src/commands/init/init-go.ts index 03d276efe..1d369908d 100644 --- a/genkit-tools/cli/src/commands/init/init-go.ts +++ b/genkit-tools/cli/src/commands/init/init-go.ts @@ -69,12 +69,15 @@ const modelOptions: Record = { ollama: { label: 'Ollama (e.g. Gemma)', package: 'github.com/firebase/genkit/go/plugins/ollama', - init: `// Initialize the Ollama plugin. -\terr := ollama.Init(ctx, + init: `// Ollama plugin configuration. +\tconfig := ollama.Config{ \t\t// The address of your Ollama API server. This is often a different host \t\t// from your app backend (which runs Genkit), in order to run Ollama on \t\t// a GPU-accelerated machine. -\t\t"http://127.0.0.1:11434") +\t\tServerAddress: "http://127.0.0.1:11434", +\t} +\t// Initialize the Ollama plugin. +\terr := ollama.Init(ctx, &config) \tif err != nil { \t\tlog.Fatal(err) \t}