test

Author: Gekko0114

pkg/openai/openai.go

@@ -36,27 +36,29 @@ func (c *ChatOpenAI) SendMessage(message []Message) (*Response, error) {
 		Messages: message,
 	}
 
+	fmt.Println("a")
 	requestBodyJSON, err := json.Marshal(requestBody)
 	if err != nil {
 		return nil, fmt.Errorf("Error marshaling request body: %w", err)
 	}
-
+	fmt.Println("b")
 	// Create a new HTTP request
 	req, err := http.NewRequest("POST", openaiURL, bytes.NewBuffer(requestBodyJSON))
 	if err != nil {
 		return nil, fmt.Errorf("Error creating HTTP request: %w", err)
 	}
+	fmt.Println("c")
 
 	// Set the necessary headers
 	req.Header.Set("Content-Type", "application/json")
 	req.Header.Set("Authorization", "Bearer "+c.apiKey)
-
+	fmt.Println("d")
 	resp, err := c.httpClient.Do(req)
 	if err != nil {
 		return nil, fmt.Errorf("Error sending HTTP request: %w", err)
 	}
 	defer resp.Body.Close()
-
+	fmt.Println("d")
 	// Read the response body
 	respBody, err := io.ReadAll(resp.Body)
 

pkg/openai/openai_test.go

@@ -1,6 +1,7 @@
 package openai
 
 import (
+	"fmt"
 	"net/http"
 	"reflect"
 	"testing"
@@ -44,10 +45,16 @@ func TestSendMessage(t *testing.T) {
 			return resp, nil
 		})
 
-	client, _ := NewChatOpenAI("gpt-3.5-turbo")
+	client, err := NewChatOpenAI("gpt-3.5-turbo")
+	if err != nil {
+		t.Fatalf("Error happens: %v", err)
+	}
 	for _, tc := range tests {
 		t.Run(tc.name, func(t *testing.T) {
+			fmt.Println("1")
+			fmt.Println(tc.message)
 			have, err := client.SendMessage(tc.message)
+			fmt.Println("2")
 			if err != nil {
 				t.Fatalf("Error happens: %v", err)
 			}