lukaszraczylo/claude-mnemonic
1
0
Fork 0
mirror of https://github.com/lukaszraczylo/claude-mnemonic.git synced 2026-10-12 04:55:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add windows embeddings as well.

Browse Source
This commit is contained in:
lukaszraczylo
2025-12-16 12:38:53 +00:00
parent 31d67ceb2a
commit e5f4a100ab
3 changed files with 34 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/embedding/lib_windows.go
+8
View File
@@ -0,0 +1,8 @@
//go:build windows
package embedding
// Windows doesn't need the providers shared library
var onnxRuntimeProvidersLib []byte
const onnxRuntimeProvidersLibName = ""
internal/embedding/lib_windows_amd64.go
+12
View File
@@ -0,0 +1,12 @@
//go:build windows && amd64
package embedding
import (
_ "embed"
)
//go:embed assets/lib/windows-amd64/onnxruntime.dll
var onnxRuntimeLib []byte
const onnxRuntimeLibName = "onnxruntime.dll"
scripts/download-onnx-libs.sh
+14 -2
View File
@@ -16,6 +16,7 @@ if [ "$PLATFORM" = "auto" ]; then
case "$OS" in case "$OS" in
darwin) OS="darwin" ;; darwin) OS="darwin" ;;
linux) OS="linux" ;; linux) OS="linux" ;;
mingw*|msys*|cygwin*) OS="windows" ;;
*) echo "Unsupported OS: $OS"; exit 1 ;; *) echo "Unsupported OS: $OS"; exit 1 ;;
esac esac
case "$ARCH" in case "$ARCH" in
@@ -64,12 +65,21 @@ download_linux_arm64() {
cp "${TEMP_DIR}/onnxruntime-linux-aarch64-${ONNX_VERSION}/lib/libonnxruntime_providers_shared.so" "${ASSETS_DIR}/linux-arm64/libonnxruntime_providers_shared.so" 2>/dev/null || true cp "${TEMP_DIR}/onnxruntime-linux-aarch64-${ONNX_VERSION}/lib/libonnxruntime_providers_shared.so" "${ASSETS_DIR}/linux-arm64/libonnxruntime_providers_shared.so" 2>/dev/null || true
} }
download_windows_amd64() {
echo "Downloading windows-amd64..."
mkdir -p "${ASSETS_DIR}/windows-amd64"
curl -sSL "https://github.com/microsoft/onnxruntime/releases/download/v${ONNX_VERSION}/onnxruntime-win-x64-${ONNX_VERSION}.zip" -o "${TEMP_DIR}/windows-amd64.zip"
unzip -q "${TEMP_DIR}/windows-amd64.zip" -d "${TEMP_DIR}"
cp "${TEMP_DIR}/onnxruntime-win-x64-${ONNX_VERSION}/lib/onnxruntime.dll" "${ASSETS_DIR}/windows-amd64/onnxruntime.dll"
}
# Check if library already exists for a platform # Check if library already exists for a platform
lib_exists() { lib_exists() {
local plat="$1" local plat="$1"
case "$plat" in case "$plat" in
darwin-*) [ -f "${ASSETS_DIR}/${plat}/libonnxruntime.dylib" ] ;; darwin-*) [ -f "${ASSETS_DIR}/${plat}/libonnxruntime.dylib" ] ;;
linux-*) [ -f "${ASSETS_DIR}/${plat}/libonnxruntime.so" ] ;; linux-*) [ -f "${ASSETS_DIR}/${plat}/libonnxruntime.so" ] ;;
windows-*) [ -f "${ASSETS_DIR}/${plat}/onnxruntime.dll" ] ;;
*) return 1 ;; *) return 1 ;;
esac esac
} }
@@ -86,13 +96,14 @@ download_if_needed() {
darwin-arm64) download_darwin_arm64 ;; darwin-arm64) download_darwin_arm64 ;;
linux-amd64) download_linux_amd64 ;; linux-amd64) download_linux_amd64 ;;
linux-arm64) download_linux_arm64 ;; linux-arm64) download_linux_arm64 ;;
windows-amd64) download_windows_amd64 ;;
esac esac
} }
echo "ONNX Runtime v${ONNX_VERSION} - Platform: ${PLATFORM}" echo "ONNX Runtime v${ONNX_VERSION} - Platform: ${PLATFORM}"
case "$PLATFORM" in case "$PLATFORM" in
darwin-amd64|darwin-arm64|linux-amd64|linux-arm64) darwin-amd64|darwin-arm64|linux-amd64|linux-arm64|windows-amd64)
download_if_needed "$PLATFORM" download_if_needed "$PLATFORM"
;; ;;
all) all)
@@ -100,10 +111,11 @@ case "$PLATFORM" in
download_if_needed darwin-arm64 download_if_needed darwin-arm64
download_if_needed linux-amd64 download_if_needed linux-amd64
download_if_needed linux-arm64 download_if_needed linux-arm64
download_if_needed windows-amd64
;; ;;
*) *)
echo "Unknown platform: $PLATFORM" echo "Unknown platform: $PLATFORM"
echo "Supported: darwin-amd64, darwin-arm64, linux-amd64, linux-arm64, all, auto" echo "Supported: darwin-amd64, darwin-arm64, linux-amd64, linux-arm64, windows-amd64, all, auto"
exit 1 exit 1
;; ;;
esac esac