Added Chat functionality

Author: prasad89

examples/chat/main.go

@@ -0,0 +1,62 @@
+package main
+
+import (
+	"fmt"
+	"time"
+
+	"github.com/prasad89/golamify/pkg/golamify"
+)
+
+func main() {
+	// Optional config; pass nil for defaults
+	config := golamify.Config{
+		OllamaHost: "http://localhost:11434",
+		Timeout:    30 * time.Second,
+	}
+
+	client, err := golamify.NewClient(&config)
+	if err != nil {
+		fmt.Println("Error creating client:", err)
+		return
+	}
+
+	// Required parameters for Chat; others optional
+	payload := golamify.ChatPayload{
+		Model: "llama3.2:1b",
+		Messages: []golamify.Message{
+			{
+				Role:    "user",
+				Content: "What we discussed earlier?",
+			},
+		},
+	}
+
+	responseChannel, errorChannel := golamify.Chat(client, &payload)
+
+	for {
+		select {
+		case response, ok := <-responseChannel:
+			if !ok {
+				return
+			}
+
+			// Check to safely extract "message" and "content"
+			if messMap, exists := response["message"].(map[string]interface{}); exists {
+				if content, exists := messMap["content"].(string); exists {
+					fmt.Print(content)
+				} else {
+					fmt.Println("Content key missing or not a string")
+				}
+			} else {
+				fmt.Println("Message key missing or not a valid map")
+			}
+
+		case err, ok := <-errorChannel:
+			if ok && err != nil {
+				fmt.Println("Error:", err)
+			} else if !ok {
+				return
+			}
+		}
+	}
+}

pkg/golamify/chat.go

@@ -0,0 +1,97 @@
+package golamify
+
+import (
+	"bufio"
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+)
+
+type ChatPayload struct {
+	Model     string    `json:"model" validate:"required"`
+	Messages  []Message `json:"messages" validate:"required,dive"`
+	Tools     []string  `json:"tools,omitempty"`
+	Format    string    `json:"format,omitempty" validate:"omitempty,oneof=json"`
+	Stream    *bool     `json:"stream,omitempty"`
+	KeepAlive string    `json:"keep_alive,omitempty"`
+}
+
+type Message struct {
+	Role      string   `json:"role" validate:"required,oneof=system user assistant tool"`
+	Content   string   `json:"content" validate:"required"`
+	Images    []string `json:"images,omitempty"`
+	ToolCalls []string `json:"tool_calls,omitempty"`
+}
+
+func Chat(c *Client, payload *ChatPayload) (<-chan map[string]interface{}, <-chan error) {
+	responseChannel := make(chan map[string]interface{})
+	errorChannel := make(chan error, 1)
+
+	go func() {
+		defer close(responseChannel)
+		defer close(errorChannel)
+
+		statusCode, err := ShowModel(c, payload.Model)
+		if err != nil {
+			errorChannel <- fmt.Errorf("error showing model: %w", err)
+			return
+		}
+
+		if statusCode == http.StatusNotFound {
+			err := PullModel(c, payload.Model)
+			if err != nil {
+				errorChannel <- fmt.Errorf("failed to pull model: %w", err)
+				return
+			}
+		}
+
+		body, err := json.Marshal(payload)
+		if err != nil {
+			errorChannel <- fmt.Errorf("failed to encode request payload: %w", err)
+			return
+		}
+
+		req, err := http.NewRequest("POST", c.config.OllamaHost+"/api/chat", bytes.NewBuffer(body))
+		if err != nil {
+			errorChannel <- fmt.Errorf("failed to create request: %w", err)
+			return
+		}
+		req.Header.Set("User-Agent", c.userAgent)
+		req.Header.Set("Content-Type", "application/json")
+
+		resp, err := c.httpClient.Do(req)
+		if err != nil {
+			errorChannel <- fmt.Errorf("failed to connect to generate endpoint: %w", err)
+			return
+		}
+		defer resp.Body.Close()
+
+		if resp.StatusCode != http.StatusOK {
+			respBody, _ := io.ReadAll(resp.Body)
+			errorChannel <- fmt.Errorf("generate endpoint is not reachable, received status: %s, body: %s", resp.Status, string(respBody))
+			return
+		}
+
+		scanner := bufio.NewScanner(resp.Body)
+		for scanner.Scan() {
+			var generated map[string]interface{}
+			err := json.Unmarshal(scanner.Bytes(), &generated)
+			if err != nil {
+				errorChannel <- fmt.Errorf("error parsing JSON: %w", err)
+				continue
+			}
+			responseChannel <- generated
+			if done, exists := generated["done"].(bool); exists && done {
+				break
+			}
+		}
+
+		if err := scanner.Err(); err != nil {
+			errorChannel <- fmt.Errorf("error reading response body: %w", err)
+		}
+	}()
+
+	return responseChannel, errorChannel
+}