mirror of https://gitee.com/godoos/godoos.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
28 lines
645 B
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")
|
|
}
|
|
|