From 9c521af2c7170d566b4c3eb964716b6892263f3d Mon Sep 17 00:00:00 2001
From: unintendedfraud <briceunintendedfraud@fastmail.com>
Date: Sat, 6 May 2023 12:39:09 +0200
Subject: [PATCH] d

---
 handlers/message_created.go | 10 +++++++---
 openai/client.go            |  6 ++++++
 2 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/handlers/message_created.go b/handlers/message_created.go
index 0768693..fe0ff9d 100644
--- a/handlers/message_created.go
+++ b/handlers/message_created.go
@@ -1,6 +1,7 @@
 package handlers
 
 import (
+	"fmt"
 	"hal/env"
 	"hal/openai"
 	"log"
@@ -20,9 +21,12 @@ func OnMessageCreated(s *discordgo.Session, m *discordgo.MessageCreate) {
 			log.Panicf("failed to query open ai with the following prompt [%s]. Error: %s", m.Content, err.Error())
 		}
 
-		aiResponse := res.Choices[0].Text
-		if _, err = s.ChannelMessageSend(m.ChannelID, aiResponse); err != nil {
-			log.Panicf("failed to send the response [%s] to the discord channel [%s]", aiResponse, err.Error())
+		if len(res.Choices) > 0 {
+			if _, err = s.ChannelMessageSend(m.ChannelID, aiResponse); err != nil {
+				log.Panicf("failed to send the response [%s] to the discord channel [%s]", aiResponse, err.Error())
+			}
+		} else {
+			fmt.Println("### no response ##")
 		}
 
 		return
diff --git a/openai/client.go b/openai/client.go
index 9d5d93b..a1527e0 100644
--- a/openai/client.go
+++ b/openai/client.go
@@ -41,6 +41,12 @@ func (c Client) Completions(prompt string) (*CompletionResponse, error) {
 		return nil, fmt.Errorf("failed to query openai. Error: %s", err.Error())
 	}
 
+	var test interface{}
+	if err := json.Unmarshal(res.Body(), &test); err != nil {
+		return nil, fmt.Errorf("failed to parse the openai response. Error: %s", err.Error())
+	}
+	fmt.Println("### RESPONSE INTERFCE ###", test)
+
 	var r CompletionResponse
 	if err := json.Unmarshal(res.Body(), &r); err != nil {
 		return nil, fmt.Errorf("failed to parse the openai response. Error: %s", err.Error())