@@ -20,26 +20,36 @@ func main() {
20
20
log .Fatalf ("Couldn't load config: %v" , err )
21
21
}
22
22
23
- if persistentConfig .OpenAISession == "" {
24
- token , err := session .GetSession ()
25
- if err != nil {
26
- log .Fatalf ("Couldn't get OpenAI session: %v" , err )
27
- }
23
+ err = godotenv .Load ()
24
+ if err != nil {
25
+ log .Printf ("Couldn't load .env file: %v. Using shell exposed env variables..." , err )
26
+ }
27
+
28
+
29
+ if config .OpenAISession == "" {
30
+ if os .Getenv ("OPENAI_SESSION" ) == "" {
31
+ session , err := session .GetSession ()
32
+ if err != nil {
33
+ log .Fatalf ("Couldn't get OpenAI session: %v" , err )
34
+ }
28
35
29
- if err = persistentConfig .SetSessionToken (token ); err != nil {
30
- log .Fatalf ("Couldn't save OpenAI session: %v" , err )
36
+ err = config .Set ("OpenAISession" , session )
37
+ if err != nil {
38
+ log .Fatalf ("Couldn't save OpenAI session: %v" , err )
39
+ }
40
+ } else
41
+ {
42
+ err = config .Set ("OpenAISession" , os .Getenv ("OPENAI_SESSION" ))
43
+ if err != nil {
44
+ log .Fatalf ("Couldn't save OpenAI session: %v" , err )
45
+ }
31
46
}
32
47
}
33
48
34
49
chatGPT := chatgpt .Init (persistentConfig )
35
50
log .Println ("Started ChatGPT" )
36
51
37
- envConfig , err := config .LoadEnvConfig (".env" )
38
- if err != nil {
39
- log .Printf ("Couldn't load .env file: %v. Using shell exposed env variables..." , err )
40
- }
41
-
42
- bot , err := tgbot .New (envConfig .TelegramToken , time .Duration (envConfig .EditWaitSeconds ))
52
+ bot , err := tgbotapi .NewBotAPI (os .Getenv ("TELEGRAM_TOKEN" ))
43
53
if err != nil {
44
54
log .Fatalf ("Couldn't start Telegram bot: %v" , err )
45
55
}
0 commit comments