28 lines
645 B

package model
import (
"encoding/json"
"godo/libs"
"net/http"
)
func ChatHandler(w http.ResponseWriter, r *http.Request) {
url := GetOllamaUrl() + "/v1/chat/completions"
var request interface{}
err := json.NewDecoder(r.Body).Decode(&request)
if err != nil {
libs.ErrorMsg(w, err.Error())
return
}
ForwardHandler(w, r, request, url, "POST")
}
func EmbeddingHandler(w http.ResponseWriter, r *http.Request) {
url := GetOllamaUrl() + "/api/embeddings"
var request interface{}
err := json.NewDecoder(r.Body).Decode(&request)
if err != nil {
libs.ErrorMsg(w, err.Error())
return
}
ForwardHandler(w, r, request, url, "POST")
}