{
  "tool": "<tên công cụ>",
  "tool_input": { "tham số": "đúng định dạng" }
}

{
  "tool": "getCurrentWeather",
  "tool_input": {
    "location": "New York City",
    "unit": "fahrenheit"
  }
}

{ "temperature": 78, "unit": "F" }

package main

import (
  "context"
  "encoding/json"
  "fmt"
  "log"
  "github.com/ollama/ollama/api"
)

const model = "llama3.2:3b"

func getCurrentWeather(input map[string]any) (string, error) {
  location, ok := input["location"].(string)
  if !ok {
    log.Fatalf("bad args: %v", input)
  }
  weatherInfo := map[string]any{
    "location":    location,
    "temperature": "80",
    "unit":        "fahrenheit",
    "forecast":    []string{"sunny", "windy"},
  }
  b, err := json.Marshal(weatherInfo)
  if err != nil {
    return "", err
  }
  return string(b), nil
}

func main() {
  client, err := api.ClientFromEnvironment()
  if err != nil {
    log.Fatal(err)
  }

  messages := []api.Message{
    {Role: "user", Content: "What is the weather like in New York City?"},
  }

  ctx := context.Background()

  req := &api.ChatRequest{
    Model:    model,
    Messages: messages,
    Tools: api.Tools{
      api.Tool{
        Type: "function",
        Function: api.ToolFunction{
          Name:        "getCurrentWeather",
          Description: "Get the current weather in a given location",
          Parameters: api.ToolFunctionParameters{
            Type: "object",
            Required: []string{"location"},
            Properties: map[string]api.ToolFunctionProperty{
              "location": {Type: api.PropertyType{"string"}, Description: "City and state, e.g. San Francisco, CA"},
            },
          },
        },
      },
    },
  }

  handler := func(resp api.ChatResponse) error {
    if len(resp.Message.ToolCalls) == 0 {
      fmt.Print(resp.Message.Content)
      return nil
    }

    tc := resp.Message.ToolCalls[0].Function
    switch tc.Name {
    case "getCurrentWeather":
      output, err := getCurrentWeather(tc.Arguments)
      if err != nil {
        log.Fatal(err)
      }
      messages = append(messages, api.Message{Role: "tool", Content: output})
    default:
      log.Fatal(fmt.Errorf("invalid function: %q", tc.Name))
    }
    return nil
  }

  if err := client.Chat(ctx, req, handler); err != nil {
    log.Fatal(err)
  }

  req.Messages = messages

  if err := client.Chat(ctx, req, handler); err != nil {
    log.Fatal(err)
  }
}