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
}