1c502bfea8
- Updated context creation in handleChat, handleGenerateTitle, and handleGeneratePrompts to use the request context, improving context management. - Commented out unused context settings for assistant ID, silent mode, history visibility, and client type to streamline the code. - Refactored context handling in assistant methods to enhance clarity and maintainability. - Removed deprecated functions and cleaned up the context structure for better performance and readability.
104 lines
3 KiB
Go
104 lines
3 KiB
Go
package chat
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/md5"
|
|
"encoding/hex"
|
|
"fmt"
|
|
"io"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/google/uuid"
|
|
"github.com/yaoapp/yao/openapi/response"
|
|
)
|
|
|
|
// GinCreateCompletions handles POST /chat/:assistant_id/completions - Create a chat completion
|
|
func GinCreateCompletions(c *gin.Context) {
|
|
// Print request information for debugging
|
|
fmt.Println("========== Chat Completions Request ==========")
|
|
fmt.Printf("Method: %s\n", c.Request.Method)
|
|
fmt.Printf("URL: %s\n", c.Request.URL.String())
|
|
fmt.Printf("RemoteAddr: %s\n", c.Request.RemoteAddr)
|
|
|
|
// Print headers
|
|
fmt.Println("\n--- Headers ---")
|
|
for key, values := range c.Request.Header {
|
|
for _, value := range values {
|
|
fmt.Printf("%s: %s\n", key, value)
|
|
}
|
|
}
|
|
|
|
// Print path parameters
|
|
fmt.Println("\n--- Path Parameters ---")
|
|
for _, param := range c.Params {
|
|
fmt.Printf("%s: %s\n", param.Key, param.Value)
|
|
}
|
|
|
|
// Print query parameters
|
|
fmt.Println("\n--- Query Parameters ---")
|
|
for key, values := range c.Request.URL.Query() {
|
|
for _, value := range values {
|
|
fmt.Printf("%s: %s\n", key, value)
|
|
}
|
|
}
|
|
|
|
// Print request body
|
|
fmt.Println("\n--- Request Body ---")
|
|
body, err := io.ReadAll(c.Request.Body)
|
|
if err != nil {
|
|
fmt.Printf("Error reading body: %v\n", err)
|
|
} else {
|
|
fmt.Printf("%s\n", string(body))
|
|
// Restore the body for further processing
|
|
c.Request.Body = io.NopCloser(bytes.NewBuffer(body))
|
|
}
|
|
fmt.Println("===============================================")
|
|
|
|
// Handle Sid - try multiple methods for maximum compatibility
|
|
var sid string
|
|
|
|
// Method 1: Check if client sent X-Session-Id header
|
|
sid = c.GetHeader("X-Session-Id")
|
|
|
|
// Method 2: Try to read from cookie
|
|
if sid == "" {
|
|
sid, err = c.Cookie("Sid")
|
|
if err == nil && sid != "" {
|
|
fmt.Printf("Existing Sid from cookie: %s\n", sid)
|
|
}
|
|
} else {
|
|
fmt.Printf("Existing Sid from header: %s\n", sid)
|
|
}
|
|
|
|
// Method 3: For clients that can't store cookies/headers (like Electron cross-origin),
|
|
// generate a deterministic session ID based on client fingerprint
|
|
if sid == "" {
|
|
// Use Authorization token if available (most stable identifier)
|
|
authToken := c.GetHeader("Authorization")
|
|
userAgent := c.GetHeader("User-Agent")
|
|
|
|
if authToken != "" {
|
|
// Generate stable session ID from auth token
|
|
hash := md5.Sum([]byte(authToken))
|
|
sid = hex.EncodeToString(hash[:])
|
|
fmt.Printf("Generated deterministic Sid from auth token: %s\n", sid)
|
|
} else {
|
|
// Fallback: generate random UUID
|
|
sid = uuid.New().String()
|
|
fmt.Printf("Generated random Sid: %s\n", sid)
|
|
}
|
|
|
|
fmt.Printf("Client fingerprint - UserAgent: %s\n", userAgent)
|
|
}
|
|
|
|
// Try to set cookie (may not work for cross-origin, but doesn't hurt)
|
|
c.SetCookie("Sid", sid, 86400*30, "/", "", false, false)
|
|
|
|
// Return Sid in response header and body for client reference
|
|
c.Header("X-Session-Id", sid)
|
|
|
|
response.RespondWithSuccess(c, response.StatusOK, gin.H{"message": "Create Completions", "sid": sid})
|
|
}
|
|
|
|
// GinUpdateCompletions handles PUT /chat/:assistant_id/completions - Update a chat completion metadata
|
|
func GinUpdateCompletions(c *gin.Context) {}
|