testlangchain.go 596 B

12345678910111213141516171819202122232425
  1. package main
  2. import (
  3. "context"
  4. "fmt"
  5. "log"
  6. "github.com/tmc/langchaingo/llms/ollama"
  7. )
  8. func main() {
  9. llm, err := ollama.New(ollama.WithModel("nomic-embed-text:v1.5"))
  10. if err != nil {
  11. log.Fatal(err)
  12. }
  13. ctx := context.Background()
  14. inputText := "The sky is blue because of Rayleigh scattering"
  15. result, err := llm.CreateEmbedding(ctx, []string{inputText})
  16. if err != nil {
  17. log.Fatal(err)
  18. }
  19. fmt.Printf("%#v\n", result)
  20. fmt.Printf("%d\n", len(result[0]))
  21. }