chore: default to ONNX_PATH env

BREAKING

fastembed.go

@@ -55,7 +55,6 @@ type InitOptions struct {
 	MaxLength            int
 	CacheDir             string
 	ShowDownloadProgress *bool
-	OnnxPath             string
 }
 
 // Struct to represent FastEmbed model information
@@ -84,8 +83,8 @@ func NewFlagEmbedding(options *InitOptions) (*FlagEmbedding, error) {
 		options.ShowDownloadProgress = &showDownloadProgress
 	}
 
-	if options.OnnxPath != "" {
-		ort.SetSharedLibraryPath(options.OnnxPath)
+	if onnxPath := os.Getenv("ONNX_PATH"); onnxPath != "" {
+		ort.SetSharedLibraryPath(onnxPath)
 	}
 
 	if !ort.IsInitialized() {

fastembed_test.go

@@ -2,29 +2,21 @@ package fastembed
 
 import (
 	"fmt"
-	"os"
 	"reflect"
 	"testing"
 )
 
 func TestNewFlagEmbedding(t *testing.T) {
 	// Test with default options
-	options := &InitOptions{
-		OnnxPath: os.Getenv("ONNX_PATH"),
-	}
-	_, err := NewFlagEmbedding(options)
+	_, err := NewFlagEmbedding(&InitOptions{})
 	if err != nil {
 		t.Fatalf("Expected no error, got %v", err)
 	}
 }
 
 func TestEmbed(t *testing.T) {
 	// Test with a single input
-	options := &InitOptions{
-		OnnxPath: os.Getenv("ONNX_PATH"),
-		Model:    AllMiniLML6V2,
-	}
-	fe, err := NewFlagEmbedding(options)
+	fe, err := NewFlagEmbedding(&InitOptions{})
 	defer fe.Destroy()
 	if err != nil {
 		t.Fatalf("Expected no error, got %v", err)
@@ -42,10 +34,7 @@ func TestEmbed(t *testing.T) {
 
 func TestQueryEmbed(t *testing.T) {
 	// Test with a single input
-	options := &InitOptions{
-		OnnxPath: os.Getenv("ONNX_PATH"),
-	}
-	fe, err := NewFlagEmbedding(options)
+	fe, err := NewFlagEmbedding(&InitOptions{})
 	if err != nil {
 		t.Fatalf("Expected no error, got %v", err)
 	}
@@ -61,10 +50,7 @@ func TestQueryEmbed(t *testing.T) {
 
 func TestPassageEmbed(t *testing.T) {
 	// Test with a single input
-	options := &InitOptions{
-		OnnxPath: os.Getenv("ONNX_PATH"),
-	}
-	fe, err := NewFlagEmbedding(options)
+	fe, err := NewFlagEmbedding(&InitOptions{})
 	if err != nil {
 		t.Fatalf("Expected no error, got %v", err)
 	}