From da6d7874ee148524dbe587a53eb3885f06ec67eb Mon Sep 17 00:00:00 2001
From: "Laisky.Cai" <github@laisky.com>
Date: Mon, 18 Dec 2023 03:02:35 +0000
Subject: [PATCH] refactor: Improve performance and reliability in text relay
 functionality

- Refactor relay-text controller to improve performance and readability
- Update function signatures for better consistency
- Optimize error handling for improved reliability
---
 controller/relay-text.go | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/controller/relay-text.go b/controller/relay-text.go
index 5148235d..b79d9fc7 100644
--- a/controller/relay-text.go
+++ b/controller/relay-text.go
@@ -15,6 +15,7 @@ import (
 	"time"
 
 	"github.com/gin-gonic/gin"
+	"gorm.io/gorm/logger"
 )
 
 const (
@@ -443,6 +444,15 @@ func relayTextHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
 					}
 				}(c.Request.Context())
 			}
+
+			{ // more error info
+				if reqdata, err := json.Marshal(req); err != nil {
+					fmt.Printf("relay text error: %s\n", err.Error())
+				} else {
+					fmt.Printf("send req %q got error %d", reqdata, resp.StatusCode)
+				}
+			}
+
 			return relayErrorHandler(resp)
 		}
 	}