Improved examples [skip ci]

Author: ankane

examples/hybrid/main.go

@@ -51,7 +51,7 @@ func main() {
 		"The cat is purring",
 		"The bear is growling",
 	}
-	embeddings, err := FetchEmbeddings(input)
+	embeddings, err := Embed(input)
 	if err != nil {
 		panic(err)
 	}
@@ -87,7 +87,7 @@ ORDER BY score DESC
 LIMIT 5
 	`
 	query := "growling bear"
-	queryEmbedding, err := FetchEmbeddings([]string{query})
+	queryEmbedding, err := Embed([]string{query})
 	if err != nil {
 		panic(err)
 	}
@@ -118,7 +118,7 @@ type apiRequest struct {
 	Model string   `json:"model"`
 }
 
-func FetchEmbeddings(input []string) ([][]float32, error) {
+func Embed(input []string) ([][]float32, error) {
 	url := "http://localhost:11434/api/embed"
 	data := &apiRequest{
 		Input: input,

examples/openai/main.go

@@ -53,7 +53,7 @@ func main() {
 		"The cat is purring",
 		"The bear is growling",
 	}
-	embeddings, err := FetchEmbeddings(input, apiKey)
+	embeddings, err := Embed(input, apiKey)
 	if err != nil {
 		panic(err)
 	}
@@ -65,8 +65,13 @@ func main() {
 		}
 	}
 
-	documentId := 1
-	rows, err := conn.Query(ctx, "SELECT id, content FROM documents WHERE id != $1 ORDER BY embedding <=> (SELECT embedding FROM documents WHERE id = $1) LIMIT 5", documentId)
+	query := "forest"
+	queryEmbedding, err := Embed([]string{query}, apiKey)
+	if err != nil {
+		panic(err)
+	}
+
+	rows, err := conn.Query(ctx, "SELECT id, content FROM documents ORDER BY embedding <=> $1 LIMIT 5", pgvector.NewVector(queryEmbedding[0]))
 	if err != nil {
 		panic(err)
 	}
@@ -92,7 +97,7 @@ type apiRequest struct {
 	Model string   `json:"model"`
 }
 
-func FetchEmbeddings(input []string, apiKey string) ([][]float32, error) {
+func Embed(input []string, apiKey string) ([][]float32, error) {
 	url := "https://api.openai.com/v1/embeddings"
 	data := &apiRequest{
 		Input: input,

examples/sparse/main.go

@@ -53,7 +53,7 @@ func main() {
 		"The cat is purring",
 		"The bear is growling",
 	}
-	embeddings, err := FetchEmbeddings(input)
+	embeddings, err := Embed(input)
 	if err != nil {
 		panic(err)
 	}
@@ -66,7 +66,7 @@ func main() {
 	}
 
 	query := "forest"
-	queryEmbeddings, err := FetchEmbeddings([]string{query})
+	queryEmbeddings, err := Embed([]string{query})
 	if err != nil {
 		panic(err)
 	}
@@ -94,7 +94,7 @@ type apiRequest struct {
 	Inputs []string `json:"inputs"`
 }
 
-func FetchEmbeddings(inputs []string) ([]map[int32]float32, error) {
+func Embed(inputs []string) ([]map[int32]float32, error) {
 	url := "http://localhost:3000/embed_sparse"
 	data := &apiRequest{
 		Inputs: inputs,