openai: add make it possible to set the "user" in the gpt completion …

…call
innogames · Nov 8, 2023 · d962a6a · d962a6a
1 parent 2e5c7ff
commit d962a6a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/command/openai/client.go b/command/openai/client.go
@@ -20,6 +20,7 @@ func CallChatGPT(cfg Config, inputMessages []ChatMessage, stream bool) (<-chan s
 		Temperature: cfg.Temperature,
 		Seed:        cfg.Seed,
 		MaxTokens:   cfg.MaxTokens,
+		User:        cfg.APIUser,
 		Stream:      stream,
 		Messages:    inputMessages,
 	})

diff --git a/command/openai/config.go b/command/openai/config.go
@@ -27,6 +27,9 @@ type Config struct {
 
 	// log all input+output text to the logger. This could include personal information, therefore disabled by default!
 	LogTexts bool `mapstructure:"log_texts"`
+
+	// user name to identify the bot which should be part of the OpenAI usage statistics
+	APIUser string `mapstructure:"api_user"`
 }
 
 // IsEnabled checks if token is set