From 84c94ecea34753f246bdfd51f6ff989281e873e3 Mon Sep 17 00:00:00 2001 From: Grail Finder Date: Sat, 1 Feb 2025 16:32:36 +0300 Subject: Feat: switch between completion and chat api --- bot.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'bot.go') diff --git a/bot.go b/bot.go index c256f58..e60c39f 100644 --- a/bot.go +++ b/bot.go @@ -91,7 +91,7 @@ func fetchModelName() { // func sendMsgToLLM(body io.Reader) (*models.LLMRespChunk, error) { func sendMsgToLLM(body io.Reader) { // nolint - resp, err := httpClient.Post(cfg.APIURL, "application/json", body) + resp, err := httpClient.Post(cfg.CurrentAPI, "application/json", body) if err != nil { logger.Error("llamacpp api", "error", err) streamDone <- true @@ -128,7 +128,7 @@ func sendMsgToLLM(body io.Reader) { line = line[6:] content, stop, err := chunkParser.ParseChunk(line) if err != nil { - logger.Error("error parsing response body", "error", err, "line", string(line), "url", cfg.APIURL) + logger.Error("error parsing response body", "error", err, "line", string(line), "url", cfg.CurrentAPI) streamDone <- true break } -- cgit v1.2.3