add_openai

Author: Gekko0114

go.mod

@@ -1,3 +1,5 @@
 module golangchain
 
 go 1.22.4
+
+require github.com/jarcoal/httpmock v1.3.1 // indirect

go.sum

@@ -0,0 +1,2 @@
+github.com/jarcoal/httpmock v1.3.1 h1:iUx3whfZWVf3jT01hQTO/Eo5sAYtB2/rqaUuOtpInww=
+github.com/jarcoal/httpmock v1.3.1/go.mod h1:3yb8rc4BI7TCBhFY8ng0gjuLKJNquuDNiPaZjnENuYg=

pkg/main.go

@@ -1,5 +1,25 @@
 package main
 
+import (
+	"fmt"
+	"golangchain/pkg/openai"
+)
+
 func main() {
-	println("aaa")
+	llm, err := openai.NewChatOpenAI("gpt-3.5-turbo")
+	if err != nil {
+		fmt.Println(err)
+	}
+
+	msg := []openai.Message{
+		{Role: "system", Content: ""},
+		{Role: "user", Content: "こんにちは"},
+	}
+
+	response, err := llm.SendMessage(msg)
+	if err != nil {
+		fmt.Println(err)
+	}
+	fmt.Println("Response:", response.Choices[0].Message.Content)
+
 }

pkg/openai/openai.go

@@ -0,0 +1,70 @@
+package openai
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+	"os"
+)
+
+const openaiURL = "https://api.openai.com/v1/chat/completions"
+
+type ChatOpenAI struct {
+	apiKey     string
+	model      string
+	httpClient *http.Client
+}
+
+func NewChatOpenAI(model string) (*ChatOpenAI, error) {
+	apiKey := os.Getenv("OPENAI_API_KEY")
+	if apiKey == "" {
+		return nil, fmt.Errorf("OPENAI_API_KEY environment variable is required")
+	}
+	return &ChatOpenAI{
+		apiKey,
+		model,
+		&http.Client{},
+	}, nil
+}
+
+func (c *ChatOpenAI) SendMessage(message []Message) (*Response, error) {
+
+	requestBody := Request{
+		Model:    c.model,
+		Messages: message,
+	}
+
+	requestBodyJSON, err := json.Marshal(requestBody)
+	if err != nil {
+		return nil, fmt.Errorf("Error marshaling request body: %w", err)
+	}
+
+	// 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)
+	}
+
+	// Set the necessary headers
+	req.Header.Set("Content-Type", "application/json")
+	req.Header.Set("Authorization", "Bearer "+c.apiKey)
+
+	resp, err := c.httpClient.Do(req)
+	if err != nil {
+		return nil, fmt.Errorf("Error sending HTTP request: %w", err)
+	}
+	defer resp.Body.Close()
+
+	// Read the response body
+	respBody, err := io.ReadAll(resp.Body)
+
+	// Unmarshal the response body to the Response struct
+	var response Response
+	err = json.Unmarshal(respBody, &response)
+	if err != nil {
+		return nil, fmt.Errorf("Error unmarshaling response body: %w", err)
+	}
+	return &response, nil
+}

pkg/openai/openai_test.go

@@ -0,0 +1,59 @@
+package openai
+
+import (
+	"net/http"
+	"reflect"
+	"testing"
+
+	"github.com/jarcoal/httpmock"
+)
+
+func TestSendMessage(t *testing.T) {
+	tests := []struct {
+		name     string
+		message  []Message
+		expected *Response
+	}{
+		{
+			name: "normal condition",
+			message: []Message{
+				{Role: "system", Content: "mock test server"},
+				{Role: "user", Content: "mock test question"},
+			},
+			expected: &Response{
+				Choices: []Choice{
+					{Message: Message{Role: "assistant", Content: "This is the test response"}},
+				},
+			},
+		},
+	}
+	// mock
+	httpmock.Activate()
+	defer httpmock.DeactivateAndReset()
+	httpmock.RegisterResponder("POST", openaiURL,
+		func(req *http.Request) (*http.Response, error) {
+
+			response := Response{Choices: []Choice{
+				{Message: Message{Role: "assistant", Content: "This is the test response"}},
+			},
+			}
+			resp, err := httpmock.NewJsonResponse(200, response)
+			if err != nil {
+				return httpmock.NewStringResponse(500, ""), nil
+			}
+			return resp, nil
+		})
+
+	client, _ := NewChatOpenAI("gpt-3.5-turbo")
+	for _, tc := range tests {
+		t.Run(tc.name, func(t *testing.T) {
+			have, err := client.SendMessage(tc.message)
+			if err != nil {
+				t.Fatalf("Error happens: %v", err)
+			}
+			if !reflect.DeepEqual(have, tc.expected) {
+				t.Fatalf("unexpected entries: %v != %v", have, tc.expected)
+			}
+		})
+	}
+}

pkg/openai/struct.go

@@ -0,0 +1,23 @@
+package openai
+
+// message represents a single message in the chat
+type Message struct {
+	Role    string `json:"role"`
+	Content string `json:"content"`
+}
+
+// Request represents the OpenAI API request payload
+type Request struct {
+	Model    string    `json:"model"`
+	Messages []Message `json:"messages"`
+}
+
+// Response represents the OpenAI API response
+type Response struct {
+	Choices []Choice `json:"choices"`
+}
+
+// Choice represents a single choice in the OpenAI API response
+type Choice struct {
+	Message Message `json:"message"`
+}