#29: gofmt

.gitignore

@@ -6,3 +6,4 @@ bin
 glide
 tmp
 coverage.txt
+precommit.txt

pkg/providers/openai/openai_test.go

@@ -1,34 +1,32 @@
 package openai
 
 import (
-	"testing"
 	"encoding/json"
 	"fmt"
+	"testing"
 )
 
-
-
 func TestOpenAIClient(t *testing.T) {
 	// Initialize the OpenAI client
 
 	poolName := "default"
 	modelName := "gpt-3.5-turbo"
 
 	payload := map[string]interface{}{
-        "message": []map[string]string{
-            {
-                "role":    "system",
-                "content": "You are a helpful assistant.",
-            },
-            {
-                "role":    "user",
-                "content": "tell me a joke",
-            },
-        },
-        "messageHistory": []string{"Hello there", "How are you?", "I'm good, how about you?"},
-    }
-
-    payloadBytes, _ := json.Marshal(payload)
+		"message": []map[string]string{
+			{
+				"role":    "system",
+				"content": "You are a helpful assistant.",
+			},
+			{
+				"role":    "user",
+				"content": "tell me a joke",
+			},
+		},
+		"messageHistory": []string{"Hello there", "How are you?", "I'm good, how about you?"},
+	}
+
+	payloadBytes, _ := json.Marshal(payload)
 
 	c := &Client{}
 
@@ -37,4 +35,4 @@ func TestOpenAIClient(t *testing.T) {
 	respJSON, _ := json.Marshal(resp)
 
 	fmt.Println(string(respJSON))
-}
+}

pkg/providers/openai/openaiclient.go

@@ -18,8 +18,8 @@ import (
 )
 
 const (
-	defaultBaseURL              = "https://api.openai.com/v1"
-	defaultOrganization         = ""
+	defaultBaseURL      = "https://api.openai.com/v1"
+	defaultOrganization = ""
 )
 
 // ErrEmptyResponse is returned when the OpenAI API returns an empty response.
@@ -41,6 +41,7 @@ const (
 	APITypeAzure   APIType = "AZURE"
 	APITypeAzureAD APIType = "AZURE_AD"
 )
+
 // Client is a client for the OpenAI API.
 type Client struct {
 	Provider     providers.Provider
@@ -53,7 +54,6 @@ type Client struct {
 	apiVersion string
 }
 
-
 func (c *Client) Run(poolName string, modelName string, payload []byte) (*ChatResponse, error) {
 
 	c, err := c.NewClient(poolName, modelName)
@@ -88,7 +88,7 @@ func (c *Client) NewClient(poolName string, modelName string) (*Client, error) {
 
 	// Read the YAML file
 	data, err := os.ReadFile("/Users/max/code/Glide/config.yaml")
-	
+
 	if err != nil {
 		slog.Error("Failed to read file: %v", err)
 		return nil, err
@@ -156,4 +156,3 @@ func HttpClient() *client.Client {
 	return c
 
 }
-

pkg/providers/types.go

@@ -8,23 +8,23 @@ type PoolsConfig struct {
 }
 
 type Pool struct {
-	Name       string      `yaml:"pool"`
-	Balancing  string      `yaml:"balancing"`
-	Providers  []Provider  `yaml:"providers"`
+	Name      string     `yaml:"pool"`
+	Balancing string     `yaml:"balancing"`
+	Providers []Provider `yaml:"providers"`
 }
 
 type Provider struct {
-	Provider       string                 `yaml:"provider"`
-	Model          string                 `yaml:"model"`
-	ApiKey         string                 `yaml:"api_key"`
-	TimeoutMs      int                    `yaml:"timeout_ms,omitempty"`
-	DefaultParams  map[string]interface{} `yaml:"default_params,omitempty"`
+	Provider      string                 `yaml:"provider"`
+	Model         string                 `yaml:"model"`
+	ApiKey        string                 `yaml:"api_key"`
+	TimeoutMs     int                    `yaml:"timeout_ms,omitempty"`
+	DefaultParams map[string]interface{} `yaml:"default_params,omitempty"`
 }
 
 type RequestBody struct {
-	Message        []struct {
+	Message []struct {
 		Role    string `json:"role"`
 		Content string `json:"content"`
 	} `json:"message"`
 	MessageHistory []string `json:"messageHistory"`
-}
+}