Refactor error handling and variable declarations in browser and logging modules

- Simplified variable initialization in `browser.go` for readability.
- Updated error handling in `request_logger.go` with better resource cleanup using deferred anonymous functions.

Refactor API handlers to use `GetContextWithCancel` for streamlined context creation and response handling

- Replaced redundant `context.WithCancel` and `context.WithValue` logic with the new `GetContextWithCancel` utility in all handlers.
- Centralized API response storage in the given context during cancellation.
- Updated associated cancellation calls for consistency and improved resource management.

- Replaced `apiResponseData` with `AddAPIResponseData` for centralized response recording.
- Simplified cancellation logic by switching to a boolean-based `cliCancel` method.
- Removed unused `apiResponseData` slices across handlers to reduce memory usage.
- Updated `handlers.go` to support unified response data storage per request context.

internal/api/handlers/gemini/gemini_handlers.go

@@ -262,8 +262,7 @@ func (h *GeminiAPIHandlers) handleGeminiStreamGenerateContent(c *gin.Context, ra
 	modelResult := gjson.GetBytes(rawJSON, "model")
 	modelName := modelResult.String()
 
-	backgroundCtx, cliCancel := context.WithCancel(context.Background())
-	cliCtx := context.WithValue(backgroundCtx, "gin", c)
+	cliCtx, cliCancel := h.GetContextWithCancel(c, context.Background())
 
 	var cliClient client.Client
 	defer func() {
@@ -324,25 +323,23 @@ outLoop:
 
 		// Send the message and receive response chunks and errors via channels.
 		respChan, errChan := cliClient.SendRawMessageStream(cliCtx, rawJSON, alt)
-		apiResponseData := make([]byte, 0)
 		for {
 			select {
 			// Handle client disconnection.
 			case <-c.Request.Context().Done():
 				if c.Request.Context().Err().Error() == "context canceled" {
 					log.Debugf("GeminiClient disconnected: %v", c.Request.Context().Err())
-					c.Set("API_RESPONSE", apiResponseData)
 					cliCancel() // Cancel the backend request.
 					return
 				}
 			// Process incoming response chunks.
 			case chunk, okStream := <-respChan:
 				if !okStream {
-					c.Set("API_RESPONSE", apiResponseData)
 					cliCancel()
 					return
 				}
-				apiResponseData = append(apiResponseData, chunk...)
+
+				h.AddAPIResponseData(c, chunk)
 
 				if cliClient.(*client.GeminiClient).GetGenerativeLanguageAPIKey() == "" {
 					if alt == "" {
@@ -384,8 +381,7 @@ outLoop:
 						c.Status(err.StatusCode)
 						_, _ = fmt.Fprint(c.Writer, err.Error.Error())
 						flusher.Flush()
-						c.Set("API_RESPONSE", []byte(err.Error.Error()))
-						cliCancel()
+						cliCancel(err.Error)
 					}
 					return
 				}
@@ -403,8 +399,7 @@ func (h *GeminiAPIHandlers) handleGeminiCountTokens(c *gin.Context, rawJSON []by
 	// orgrawJSON := rawJSON
 	modelResult := gjson.GetBytes(rawJSON, "model")
 	modelName := modelResult.String()
-	backgroundCtx, cliCancel := context.WithCancel(context.Background())
-	cliCtx := context.WithValue(backgroundCtx, "gin", c)
+	cliCtx, cliCancel := h.GetContextWithCancel(c, context.Background())
 
 	var cliClient client.Client
 	defer func() {
@@ -446,8 +441,7 @@ func (h *GeminiAPIHandlers) handleGeminiCountTokens(c *gin.Context, rawJSON []by
 			} else {
 				c.Status(err.StatusCode)
 				_, _ = c.Writer.Write([]byte(err.Error.Error()))
-				c.Set("API_RESPONSE", []byte(err.Error.Error()))
-				cliCancel()
+				cliCancel(err.Error)
 				// log.Debugf(err.Error.Error())
 				// log.Debugf(string(rawJSON))
 				// log.Debugf(string(orgrawJSON))
@@ -461,8 +455,7 @@ func (h *GeminiAPIHandlers) handleGeminiCountTokens(c *gin.Context, rawJSON []by
 				}
 			}
 			_, _ = c.Writer.Write(resp)
-			c.Set("API_RESPONSE", resp)
-			cliCancel()
+			cliCancel(resp)
 			break
 		}
 	}
@@ -475,8 +468,7 @@ func (h *GeminiAPIHandlers) handleGeminiGenerateContent(c *gin.Context, rawJSON
 
 	modelResult := gjson.GetBytes(rawJSON, "model")
 	modelName := modelResult.String()
-	backgroundCtx, cliCancel := context.WithCancel(context.Background())
-	cliCtx := context.WithValue(backgroundCtx, "gin", c)
+	cliCtx, cliCancel := h.GetContextWithCancel(c, context.Background())
 
 	var cliClient client.Client
 	defer func() {
@@ -538,8 +530,7 @@ func (h *GeminiAPIHandlers) handleGeminiGenerateContent(c *gin.Context, rawJSON
 			} else {
 				c.Status(err.StatusCode)
 				_, _ = c.Writer.Write([]byte(err.Error.Error()))
-				c.Set("API_RESPONSE", []byte(err.Error.Error()))
-				cliCancel()
+				cliCancel(err.Error)
 			}
 			break
 		} else {
@@ -550,8 +541,7 @@ func (h *GeminiAPIHandlers) handleGeminiGenerateContent(c *gin.Context, rawJSON
 				}
 			}
 			_, _ = c.Writer.Write(resp)
-			c.Set("API_RESPONSE", resp)
-			cliCancel()
+			cliCancel(resp)
 			break
 		}
 	}
@@ -581,8 +571,7 @@ func (h *GeminiAPIHandlers) handleCodexStreamGenerateContent(c *gin.Context, raw
 
 	modelName := gjson.GetBytes(rawJSON, "model")
 
-	backgroundCtx, cliCancel := context.WithCancel(context.Background())
-	cliCtx := context.WithValue(backgroundCtx, "gin", c)
+	cliCtx, cliCancel := h.GetContextWithCancel(c, context.Background())
 
 	var cliClient client.Client
 	defer func() {
@@ -609,8 +598,6 @@ outLoop:
 		// Send the message and receive response chunks and errors via channels.
 		respChan, errChan := cliClient.SendRawMessageStream(cliCtx, []byte(newRequestJSON), "")
 
-		apiResponseData := make([]byte, 0)
-
 		params := &translatorGeminiToCodex.ConvertCodexResponseToGeminiParams{
 			Model:             modelName.String(),
 			CreatedAt:         0,
@@ -623,18 +610,17 @@ outLoop:
 			case <-c.Request.Context().Done():
 				if c.Request.Context().Err().Error() == "context canceled" {
 					log.Debugf("CodexClient disconnected: %v", c.Request.Context().Err())
-					c.Set("API_RESPONSE", apiResponseData)
 					cliCancel() // Cancel the backend request.
 					return
 				}
 			// Process incoming response chunks.
 			case chunk, okStream := <-respChan:
 				if !okStream {
-					c.Set("API_RESPONSE", apiResponseData)
 					cliCancel()
 					return
 				}
-				apiResponseData = append(apiResponseData, chunk...)
+
+				h.AddAPIResponseData(c, chunk)
 
 				if bytes.HasPrefix(chunk, []byte("data: ")) {
 					jsonData := chunk[6:]
@@ -662,8 +648,7 @@ outLoop:
 						c.Status(err.StatusCode)
 						_, _ = fmt.Fprint(c.Writer, err.Error.Error())
 						flusher.Flush()
-						c.Set("API_RESPONSE", []byte(err.Error.Error()))
-						cliCancel()
+						cliCancel(err.Error)
 					}
 					return
 				}
@@ -683,8 +668,7 @@ func (h *GeminiAPIHandlers) handleCodexGenerateContent(c *gin.Context, rawJSON [
 
 	modelName := gjson.GetBytes(rawJSON, "model")
 
-	backgroundCtx, cliCancel := context.WithCancel(context.Background())
-	cliCtx := context.WithValue(backgroundCtx, "gin", c)
+	cliCtx, cliCancel := h.GetContextWithCancel(c, context.Background())
 
 	var cliClient client.Client
 	defer func() {
@@ -709,25 +693,23 @@ outLoop:
 
 		// Send the message and receive response chunks and errors via channels.
 		respChan, errChan := cliClient.SendRawMessageStream(cliCtx, []byte(newRequestJSON), "")
-		apiResponseData := make([]byte, 0)
 		for {
 			select {
 			// Handle client disconnection.
 			case <-c.Request.Context().Done():
 				if c.Request.Context().Err().Error() == "context canceled" {
 					log.Debugf("CodexClient disconnected: %v", c.Request.Context().Err())
-					c.Set("API_RESPONSE", apiResponseData)
 					cliCancel() // Cancel the backend request.
 					return
 				}
 			// Process incoming response chunks.
 			case chunk, okStream := <-respChan:
 				if !okStream {
-					c.Set("API_RESPONSE", apiResponseData)
 					cliCancel()
 					return
 				}
-				apiResponseData = append(apiResponseData, chunk...)
+
+				h.AddAPIResponseData(c, chunk)
 
 				if bytes.HasPrefix(chunk, []byte("data: ")) {
 					jsonData := chunk[6:]
@@ -749,8 +731,7 @@ outLoop:
 					} else {
 						c.Status(err.StatusCode)
 						_, _ = fmt.Fprint(c.Writer, err.Error.Error())
-						c.Set("API_RESPONSE", []byte(err.Error.Error()))
-						cliCancel()
+						cliCancel(err.Error)
 					}
 					return
 				}