added AI auto-generated keywords to customer profile

2026-01-04 13:44:45 +03:30
parent 3381e04e05
commit 8dcda0e031
9 changed files with 435 additions and 48 deletions
@@ -5,6 +5,7 @@ import (
 	"time"
 	"tm/pkg/authorization"
 	"tm/pkg/config"
+	"tm/pkg/glm"
 	"tm/pkg/hcaptcha"
 	"tm/pkg/logger"
 	"tm/pkg/mongo"
@@ -268,3 +269,49 @@ func InitHCaptchaService(conf config.HCaptchaConfig, log logger.Logger) hcaptcha

 	return hcaptchaVerifier
 }
+
+// InitGLMService initializes the GLM service
+func InitGLMService(conf GLMConfig, log logger.Logger) *glm.SDK {
+	glmSDK, err := glm.New(
+		&glm.Config{
+			BaseURL:                 conf.BaseURL,
+			APIKey:                  conf.APIKey,
+			Timeout:                 conf.Timeout,
+			RetryAttempts:           conf.RetryAttempts,
+			RetryDelay:              conf.RetryDelay,
+			EnableLogging:           conf.EnableLogging,
+			UserAgent:               conf.UserAgent,
+			DefaultModel:            conf.DefaultModel,
+			DefaultMaxTokens:        conf.DefaultMaxTokens,
+			DefaultTopP:             conf.DefaultTopP,
+			DefaultTemperature:      conf.DefaultTemperature,
+			DefaultFrequencyPenalty: conf.DefaultFrequencyPenalty,
+			DefaultPresencePenalty:  conf.DefaultPresencePenalty,
+			EnableStreaming:         conf.EnableStreaming,
+			EnableThinking:          conf.EnableThinking,
+			TranslationOptions: glm.TranslationOption{
+				Model:            conf.TranslateOptions.Model,
+				MaxTokens:        conf.TranslateOptions.MaxTokens,
+				Temperature:      conf.TranslateOptions.Temperature,
+				FrequencyPenalty: conf.TranslateOptions.FrequencyPenalty,
+				PresencePenalty:  conf.TranslateOptions.PresencePenalty,
+				TopP:             conf.TranslateOptions.TopP,
+				EnableThinking:   conf.TranslateOptions.EnableThinking,
+			},
+		}, log)
+
+	if err != nil {
+		log.Error("Failed to initialize GLM SDK", map[string]interface{}{
+			"error": err.Error(),
+		})
+		log.Warn("GLM service will be disabled, keyword generation features will not be available", map[string]interface{}{})
+		return nil
+	}
+
+	log.Info("GLM SDK initialized successfully", map[string]interface{}{
+		"base_url": conf.BaseURL,
+		"timeout":  conf.Timeout,
+	})
+
+	return glmSDK
+}
@@ -16,11 +16,41 @@ type Config struct {
 	Notification config.NotificationConfig
 	Ollama       config.OllamaConfig
 	HCaptcha     config.HCaptchaConfig
+	GLM          GLMConfig
 	Scraper      ScraperConfig
 	UserAuth     AuthConfig
 	CustomerAuth AuthConfig
 }

+type GLMConfig struct {
+	BaseURL                 string        `env:"GLM_BASE_URL" envDefault:"https://api.z.ai"`
+	APIKey                  string        `env:"GLM_API_KEY" envDefault:""`
+	Timeout                 time.Duration `env:"GLM_TIMEOUT" envDefault:"300s"`
+	RetryAttempts           int           `env:"GLM_RETRY_ATTEMPTS" envDefault:"3"`
+	RetryDelay              time.Duration `env:"GLM_RETRY_DELAY" envDefault:"2s"`
+	EnableLogging           bool          `env:"GLM_ENABLE_LOGGING" envDefault:"true"`
+	UserAgent               string        `env:"GLM_USER_AGENT" envDefault:"Opplens-GLMSDK/1.0"`
+	DefaultModel            string        `env:"GLM_DEFAULT_MODEL" envDefault:"glm-4.5"`
+	DefaultMaxTokens        int           `env:"GLM_DEFAULT_MAX_TOKENS" envDefault:"4096"`
+	DefaultTopP             float64       `env:"GLM_DEFAULT_TOP_P" envDefault:"1.0"`
+	DefaultTemperature      float64       `env:"GLM_DEFAULT_TEMPERATURE" envDefault:"0.7"`
+	DefaultFrequencyPenalty float64       `env:"GLM_DEFAULT_FREQUENCY_PENALTY" envDefault:"0"`
+	DefaultPresencePenalty  float64       `env:"GLM_DEFAULT_PRESENCE_PENALTY" envDefault:"0"`
+	EnableStreaming         bool          `env:"GLM_ENABLE_STREAMING" envDefault:"false"`
+	EnableThinking          bool          `env:"GLM_ENABLE_THINKING" envDefault:"false"`
+	TranslateOptions        TranslateOptions
+}
+
+type TranslateOptions struct {
+	Model            string  `env:"GLM_TRANSLATION_MODEL" envDefault:"glm-4.5"`
+	MaxTokens        int     `env:"GLM_TRANSLATION_MAX_TOKENS" envDefault:"2048"`
+	Temperature      float64 `env:"GLM_TRANSLATION_TEMPERATURE" envDefault:"0.3"`
+	FrequencyPenalty float64 `env:"GLM_TRANSLATION_FREQUENCY_PENALTY" envDefault:"0.1"`
+	PresencePenalty  float64 `env:"GLM_TRANSLATION_PRESENCE_PENALTY" envDefault:"0.1"`
+	TopP             float64 `env:"GLM_TRANSLATION_TOP_P" envDefault:"1.0"`
+	EnableThinking   bool    `env:"GLM_TRANSLATION_ENABLE_THINKING" envDefault:"false"`
+}
+
 type ScraperConfig struct {
 	BaseURL string        `env:"SCRAPER_BASE_URL"`
 	Timeout time.Duration `env:"SCRAPER_TIMEOUT"`