1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

package main

import (
    "bytes"
    "encoding/json"
)

type ollamaRequest struct {
    Model  string `json:"model"`
    Prompt string `json:"prompt"`
    Stream bool   `json:"stream"`
}

type ollamaResponse struct {
    Response string `json:"response"`
    Done     bool   `json:"done"`
}

func (c *RemoteChatApp) callOllama(prompt, model string) (string, error) {
    payload, _ := json.Marshal(ollamaRequest{
        Model:  model,
        Prompt: prompt,
        Stream: false,
    })

    resp := c.ApiPost(c.config.Endpoint, bytes.NewBuffer(payload).String())

    return resp, nil
}