Refactor error handling in main.go and README.md to log errors without terminating the program. Update GitHub Actions workflow to support multi-OS builds and include ONNX Runtime DLL in artifacts.

Author: scriptmaster

.github/workflows/release.yml

@@ -9,8 +9,17 @@ permissions:
   contents: write
 
 jobs:
-  build-and-release:
-    runs-on: ubuntu-latest
+  build:
+    runs-on: ${{ matrix.os }}
+    strategy:
+      matrix:
+        include:
+          - os: ubuntu-latest
+            goos: linux
+            goarch: amd64
+          - os: windows-latest
+            goos: windows
+            goarch: amd64
     steps:
       - name: Checkout
         uses: actions/checkout@v4
@@ -22,19 +31,61 @@ jobs:
         with:
           go-version-file: go.mod
 
-      - name: Build artifacts
+      - name: Prepare dist
+        run: mkdir -p dist
+
+      - name: Download ONNX Runtime (windows only)
+        if: matrix.goos == 'windows'
+        shell: pwsh
+        run: |
+          $ver="1.22.0"
+          $url="https://github.com/microsoft/onnxruntime/releases/download/v$ver/onnxruntime-win-x64-$ver.zip"
+          Invoke-WebRequest -Uri $url -OutFile ort.zip
+          Expand-Archive ort.zip -DestinationPath ort
+          Copy-Item "ort\onnxruntime-win-x64-$ver\onnxruntime.dll" dist\onnxruntime.dll
+
+      - name: Build
+        shell: bash
+        env:
+          GOOS: ${{ matrix.goos }}
+          GOARCH: ${{ matrix.goarch }}
+          CGO_ENABLED: 1
+          WINDOWS: ${{ matrix.goos == 'windows' && '1' || '0' }}
         run: |
-          mkdir -p dist
-          make build-linux
-          make build-windows
+          if [ "$GOOS" = "windows" ]; then
+            make build-windows
+          else
+            make build-linux
+          fi
+
+      - name: Upload artifacts
+        uses: actions/upload-artifact@v4
+        with:
+          name: build-${{ matrix.goos }}-${{ matrix.goarch }}
+          path: |
+            dist/hf_transformers_go-${{ matrix.goos }}-${{ matrix.goarch }}*
+            dist/onnxruntime.dll
+
+  release:
+    needs: build
+    runs-on: ubuntu-latest
+    steps:
+      - name: Download artifacts
+        uses: actions/download-artifact@v4
+        with:
+          path: dist
+
+      - name: List artifacts
+        run: ls -R dist
 
       - name: Create release
         uses: softprops/action-gh-release@v2
         with:
           tag_name: ${{ github.ref_name }}
           name: ${{ github.ref_name }}
           files: |
-            dist/hf_transformers_go-linux-amd64
-            dist/hf_transformers_go-windows-amd64.exe
+            dist/**/hf_transformers_go-linux-amd64
+            dist/**/hf_transformers_go-windows-amd64.exe
+            dist/**/onnxruntime.dll
         env:
           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}

Makefile

@@ -15,11 +15,15 @@ build-linux: deps
 	GOOS=linux GOARCH=amd64 go build -o dist/hf_transformers_go-linux-amd64 .
 
 build-windows: deps
+ifneq ($(GOOS),windows)
 ifeq ($(WINDOWS),1)
 	GOOS=windows GOARCH=amd64 CGO_ENABLED=1 go build -o dist/hf_transformers_go-windows-amd64.exe .
 else
 	@echo "Skipping Windows build; set WINDOWS=1 to attempt (requires Windows toolchain and onnxruntime DLLs)"
 endif
+else
+	GOOS=windows GOARCH=amd64 CGO_ENABLED=1 go build -o dist/hf_transformers_go-windows-amd64.exe .
+endif
 
 # Build both (Linux + optional Windows)
 build: build-linux build-windows

README.md

@@ -32,13 +32,15 @@ func main() {
 		map[string]any{"dtype": "q4"},
 	)
 	if err != nil {
-		log.Fatal(err)
+		log.Println(err)
+		return
 	}
 	out, err := generator([]ChatMessage{
 		{Role: RoleUser, Content: "What is the third planet in our solar system and who inhabits it?"},
 	}, map[string]any{"max_new_tokens": 32})
 	if err != nil {
-		log.Fatal(err)
+		log.Println(err)
+		return
 	}
 	fmt.Println(out[0]["generated_text"])
 }

main.go

@@ -29,7 +29,8 @@ func main() {
 		map[string]any{"dtype": "q4"},
 	)
 	if err != nil {
-		log.Fatal(err)
+		log.Println(err)
+		return
 	}
 
 	messages := []ChatMessage{
@@ -46,7 +47,8 @@ func main() {
 		},
 	})
 	if err != nil {
-		log.Fatal(err)
+		log.Println(err)
+		return
 	}
 
 	gen := out[0]["generated_text"].([]map[string]any)