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.
81 lines
2.2 KiB
Go
81 lines
2.2 KiB
Go
package context
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/yaoapp/gou/plan"
|
|
"github.com/yaoapp/yao/openapi/oauth/authorized"
|
|
)
|
|
|
|
// NewGin create a new context from gin context
|
|
func NewGin(c *gin.Context) Context {
|
|
// Get authorized information
|
|
authInfo := authorized.GetInfo(c)
|
|
|
|
// Extract parameters from query and route
|
|
chatID := c.Query("chat_id")
|
|
assistantID := c.Param("assistant_id") // Get from route parameter
|
|
locale := c.Query("locale")
|
|
theme := c.Query("theme")
|
|
referer := c.Query("referer")
|
|
accept := c.Query("accept")
|
|
|
|
// Parse client information from User-Agent header
|
|
userAgent := c.GetHeader("User-Agent")
|
|
clientType := parseClientType(userAgent)
|
|
clientIP := c.ClientIP()
|
|
|
|
// Create base context
|
|
ctx := Context{
|
|
Context: c.Request.Context(),
|
|
Space: plan.NewMemorySharedSpace(),
|
|
Authorized: authInfo,
|
|
ChatID: chatID,
|
|
AssistantID: assistantID,
|
|
Locale: locale,
|
|
Theme: theme,
|
|
Client: Client{
|
|
Type: clientType,
|
|
UserAgent: userAgent,
|
|
IP: clientIP,
|
|
},
|
|
}
|
|
|
|
// Get Referer from query parameter, header, or default
|
|
ctx.Referer = getValidatedValue(referer, c.GetHeader("X-Yao-Referer"), RefererAPI, validateReferer)
|
|
|
|
// Get Accept from query parameter, header, or default
|
|
ctx.Accept = getValidatedAccept(accept, c.GetHeader("X-Yao-Accept"), clientType)
|
|
|
|
return ctx
|
|
}
|
|
|
|
// parseClientType parses the client type from User-Agent header
|
|
func parseClientType(userAgent string) string {
|
|
if userAgent == "" {
|
|
return "web" // Default to web
|
|
}
|
|
|
|
ua := strings.ToLower(userAgent)
|
|
|
|
// Check for specific client types
|
|
switch {
|
|
case strings.Contains(ua, "yao-agent") || strings.Contains(ua, "agent"):
|
|
return "agent"
|
|
case strings.Contains(ua, "yao-jssdk") || strings.Contains(ua, "jssdk"):
|
|
return "jssdk"
|
|
case strings.Contains(ua, "android"):
|
|
return "android"
|
|
case strings.Contains(ua, "iphone") || strings.Contains(ua, "ipad") || strings.Contains(ua, "ipod"):
|
|
return "ios"
|
|
case strings.Contains(ua, "windows"):
|
|
return "windows"
|
|
case strings.Contains(ua, "mac os x") || strings.Contains(ua, "macintosh"):
|
|
return "macos"
|
|
case strings.Contains(ua, "linux"):
|
|
return "linux"
|
|
default:
|
|
return "web"
|
|
}
|
|
}
|