change request endpoint to /infer

bparees · Aug 14, 2023 · 0b12028 · 0b12028
1 parent feeeac1
commit 0b12028
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/cmd/wisdom/main.go b/cmd/wisdom/main.go
@@ -97,7 +97,7 @@ func newStartServerCommand() *cobra.Command {
 				h.BearerTokens[t] = true
 			}
 
-			r.HandleFunc("/prompt_request", h.PromptRequestHandler).Methods("POST")
+			r.HandleFunc("/infer", h.InferHandler).Methods("POST")
 
 			log.Infof("Default model provider: %s\n", h.DefaultProvider)
 			log.Infof("Default model: %s\n", h.DefaultModel)

diff --git a/pkg/server/inference_handler.go b/pkg/server/inference_handler.go
@@ -10,7 +10,7 @@ import (
 	"github.com/openshift/wisdom/pkg/model"
 )
 
-func (h *Handler) PromptRequestHandler(w http.ResponseWriter, r *http.Request) {
+func (h *Handler) InferHandler(w http.ResponseWriter, r *http.Request) {
 	if !h.hasValidBearerToken(r) {
 		http.Error(w, "No valid bearer token found", http.StatusUnauthorized)
 		return