apps

Documentation

Index

• Variables

Variables

var ChatCmd = &cobra.Command{
	Use:   "chat",
	Short: "chat with llm base on knowledge",
	Run: func(cmd *cobra.Command, args []string) {
		if len(args) <= 1 {
			panic("dirId and history is needed.")
		}
		dirIdStr := args[0]
		dirId, err := strconv.Atoi(dirIdStr)
		if err != nil {
			panic(err)
		}

		historyStr := fmt.Sprint(strings.Join(args[1:], " "))

		history := make([]map[string]string, 0)
		err = json.Unmarshal([]byte(historyStr), &history)
		if err != nil {
			panic(err)
		}

		if err := chat(int64(dirId), history); err != nil {
			panic(err)
		}
	},
}

var IngestCmd = &cobra.Command{
	Use:   "ingest",
	Short: "ingest knowledge",
	Run: func(cmd *cobra.Command, args []string) {
		if len(args) == 0 {
			panic("ingest path is needed")
		}
		ps := args[0]

		if err := ingest(ps); err != nil {
			panic(err)
		}
	},
}

var KeywordsCmd = &cobra.Command{
	Use:   "keywords",
	Short: "Extract keywords",
	Run: func(cmd *cobra.Command, args []string) {
		ps := args[0]

		if err := keywords(ps); err != nil {
			panic(err)
		}
	},
}

var QuestionCmd = &cobra.Command{
	Use:   "question",
	Short: "question base on knowledge",
	Run: func(cmd *cobra.Command, args []string) {
		question := fmt.Sprint(strings.Join(args, " "))

		if err := run(question); err != nil {
			panic(err)
		}
	},
}

var SummaryCmd = &cobra.Command{
	Use:   "summary",
	Short: "Summarize an article in short words",
	Run: func(cmd *cobra.Command, args []string) {
		ps := args[0]

		if err := summary(ps); err != nil {
			panic(err)
		}
	},
}

var WeChatCmd = &cobra.Command{
	Use:   "chat",
	Short: "conclusion base on chat",
	Run: func(cmd *cobra.Command, args []string) {
		ps := fmt.Sprint(strings.Join(args, " "))

		if err := wechat(ps); err != nil {
			panic(err)
		}
	},
}