diff --git a/cmd/wisdom/main.go b/cmd/wisdom/main.go index 6b3d151..ff8a985 100644 --- 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 index e38391c..83c62cd 100644 --- 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