testnlp.go 1.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. package main
  2. import (
  3. //"fmt"
  4. //"time"
  5. "git.wecise.com/wecise/common/nlp"
  6. "git.wecise.com/wecise/odbserver/odb/test"
  7. "gitee.com/wecisecode/util/logger"
  8. )
  9. func main() {
  10. option := &Option{Cache: CacheAll, Keyspace: "matrix", DisableInitialHostLookup: true}
  11. g, err := test.NewG(option)
  12. if err != nil {
  13. logger.Error(err.Error())
  14. } else {
  15. defer g.Close()
  16. }
  17. logger.SetConsole(true)
  18. samples := []string{
  19. "查找{Author}上传的文档",
  20. "查找包含{File}的文档",
  21. "查找关于{Tags}文档",
  22. "谢谢,查找关于{Tags}的文档",
  23. "有关于{Tags}东西?",
  24. "查找{Author}上传的,关于{Tags}文档"}
  25. r, err := nlp.NLPNew(g, "kdb", samples)
  26. if err != nil {
  27. logger.Error(err)
  28. } else {
  29. if err := r.Update("kdb", samples); err != nil {
  30. logger.Error(err)
  31. }
  32. //if rows, meta, err := r.Answer("kdb", "#/matrix/filesystem | admin | print id as value,name as title, id,name,class") ; err != nil {
  33. if rows, meta, err := r.Answer("kdb", "admin昨天上传的文档"); err != nil {
  34. logger.Error(err)
  35. } else {
  36. logger.Info("meta ===>", meta)
  37. logger.Info("answer ==>", rows)
  38. }
  39. }
  40. }