feat(index): start the ollama server in the background

.github/workflows/test.yml

@@ -38,9 +38,6 @@ jobs:
       - name: Run action
         uses: ./
 
-      - name: Start Ollama
-        run: ollama serve &
-
       - name: Get version
         run: ollama --version
 
@@ -62,10 +59,6 @@ jobs:
           version: ${{ matrix.version }}
           name: ollama-cli
 
-      - name: Start Ollama
-        shell: bash
-        run: ollama-cli serve &
-
       - name: Locate binary
         run: command -v ollama-cli
 

README.md

@@ -19,9 +19,6 @@ jobs:
       - name: Setup Ollama
         uses: ai-action/setup-ollama@v1
 
-      - name: Start Ollama
-        run: ollama serve &
-
       - name: Run LLM
         run: ollama run llama3.2 'Explain the basics of machine learning.'
 ```
@@ -31,13 +28,13 @@ jobs:
 
 ## Usage
 
-Install Ollama CLI:
+Set up Ollama CLI and start server:
 
 ```yaml
 - uses: ai-action/setup-ollama@v1
 ```
 
-See [action.yml](action.yml)
+See [action.yml](action.yml).
 
 ## Inputs
 

src/index.test.ts

@@ -1,19 +1,22 @@
 import * as core from '@actions/core';
 import * as exec from '@actions/exec';
 import * as tc from '@actions/tool-cache';
+import { type ChildProcess, spawn } from 'child_process';
 import os from 'os';
 
 import { run } from '.';
 
 jest.mock('@actions/core');
 jest.mock('@actions/exec');
 jest.mock('@actions/tool-cache');
+jest.mock('child_process');
 jest.mock('os');
 
 const mockedCore = jest.mocked(core);
 const mockedExec = jest.mocked(exec);
-const mockedTc = jest.mocked(tc);
 const mockedOs = jest.mocked(os);
+const mockedSpawn = jest.mocked(spawn);
+const mockedTc = jest.mocked(tc);
 
 beforeEach(() => {
   jest.resetAllMocks();
@@ -48,6 +51,8 @@ describe.each(['darwin', 'win32', 'linux'])('when OS is %p', (os) => {
     mockedTc.downloadTool.mockResolvedValueOnce(pathToTarball);
     const extract = os === 'win32' ? mockedTc.extractZip : mockedTc.extractTar;
     extract.mockResolvedValueOnce(pathToCLI);
+    const unref = jest.fn();
+    mockedSpawn.mockReturnValueOnce({ unref } as unknown as ChildProcess);
 
     await run();
 
@@ -70,6 +75,12 @@ describe.each(['darwin', 'win32', 'linux'])('when OS is %p', (os) => {
     );
 
     expect(mockedCore.addPath).toHaveBeenCalledWith(binPath);
+
+    expect(mockedSpawn).toHaveBeenCalledWith(cliName, ['serve'], {
+      detached: true,
+      stdio: 'ignore',
+    });
+    expect(unref).toHaveBeenCalledTimes(1);
   });
 });
 

src/index.ts

@@ -7,6 +7,7 @@ import {
   extractZip,
   find,
 } from '@actions/tool-cache';
+import { spawn } from 'child_process';
 import path from 'path';
 
 import { getBinaryPath, getDownloadObject } from './utils';
@@ -50,6 +51,13 @@ export async function run() {
     // Expose the tool by adding it to the PATH
     addPath(path.dirname(binaryPath));
 
+    // Start the Ollama server in the background
+    const subprocess = spawn(cliName, ['serve'], {
+      detached: true,
+      stdio: 'ignore',
+    });
+    subprocess.unref();
+
     // Cache the tool
     if (!isCached) {
       await cacheFile(binaryPath, cliName, toolName, cliVersion);