rename proxy handler

mostlygeek · Dec 18, 2024 · 9b4e3f3 · 9b4e3f3
1 parent 6fe37c3
commit 9b4e3f3
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/proxy/proxymanager.go b/proxy/proxymanager.go
@@ -47,13 +47,13 @@ func New(config *Config) *ProxyManager {
 	}
 
 	// Set up routes using the Gin engine
-	pm.ginEngine.POST("/v1/chat/completions", pm.proxyChatRequestHandler)
+	pm.ginEngine.POST("/v1/chat/completions", pm.proxyOAIHandler)
 
 	// Support embeddings
-	pm.ginEngine.POST("/v1/embeddings", pm.proxyChatRequestHandler)
+	pm.ginEngine.POST("/v1/embeddings", pm.proxyOAIHandler)
 
 	// Support legacy /v1/completions api, see issue #12
-	pm.ginEngine.POST("/v1/completions", pm.proxyChatRequestHandler)
+	pm.ginEngine.POST("/v1/completions", pm.proxyOAIHandler)
 
 	pm.ginEngine.GET("/v1/models", pm.listModelsHandler)
 
@@ -235,7 +235,7 @@ func (pm *ProxyManager) upstreamIndex(c *gin.Context) {
 	c.String(http.StatusOK, html.String())
 }
 
-func (pm *ProxyManager) proxyChatRequestHandler(c *gin.Context) {
+func (pm *ProxyManager) proxyOAIHandler(c *gin.Context) {
 	bodyBytes, err := io.ReadAll(c.Request.Body)
 	if err != nil {
 		c.AbortWithError(http.StatusBadRequest, fmt.Errorf("invalid JSON"))