Skip to content

Commit

Permalink
openai: add make it possible to set the "user" in the gpt completion …
Browse files Browse the repository at this point in the history
…call
  • Loading branch information
brainexe committed Nov 8, 2023
1 parent 2e5c7ff commit a63dbed
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 0 deletions.
1 change: 1 addition & 0 deletions command/openai/client.go
Original file line number Diff line number Diff line change
Expand Up @@ -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.APILogUser,
Stream: stream,
Messages: inputMessages,
})
Expand Down
9 changes: 9 additions & 0 deletions command/openai/config.go
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
package openai

import (
"os"
"time"

"github.com/innogames/slack-bot/v2/bot/config"
Expand All @@ -27,6 +28,10 @@ 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.
// use {hostname} to set the hostname of the machine
APILogUser string `mapstructure:"api_log_user"`
}

// IsEnabled checks if token is set
Expand All @@ -46,5 +51,9 @@ func loadConfig(config *config.Config) Config {
cfg := defaultConfig
_ = config.LoadCustom("openai", &cfg)

if cfg.APILogUser == "{hostname}" {
cfg.APILogUser, _ = os.Hostname()
}

return cfg
}

0 comments on commit a63dbed

Please sign in to comment.