mirror of
https://github.com/vxcontrol/langchaingo.git
synced 2026-07-21 00:45:22 -04:00
Eli Bendersky
357a3b5d3b
* embeddings: add example of proper usage, and tweak doc.go accordingly For #379 * Appease lint and add comment
32 lines
551 B
Go
32 lines
551 B
Go
package embeddings_test
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
|
|
"github.com/tmc/langchaingo/embeddings"
|
|
"github.com/tmc/langchaingo/llms/openai"
|
|
)
|
|
|
|
func Example() { //nolint:testableexamples
|
|
llm, err := openai.New()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
// Create a new Embedder from the given LLM.
|
|
embedder, err := embeddings.NewEmbedder(llm)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
docs := []string{"doc 1", "another doc"}
|
|
embs, err := embedder.EmbedDocuments(context.Background(), docs)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
// Consume embs
|
|
_ = embs
|
|
}
|