ollama-haskell-0.3.0.0: examples/ToolCalling.hs

module Main (main) where

import Data.List.NonEmpty (NonEmpty ((:|)))
import Ollama

calculatorTool :: Tool
calculatorTool =
  Tool
    { toolType = "function"
    , toolFunction =
        FunctionDef
          { fnName = "add"
          , fnDescription = Just "Add two numbers"
          , fnParameters =
              Just
                FunctionParameters
                  { fpType = "object"
                  , fpProperties = Nothing
                  , fpRequired = Just ["a", "b"]
                  , fpAdditionalProperties = Nothing
                  , fpDescription = Nothing
                  , fpEnum = Nothing
                  }
          , fnStrict = Just True
          }
    }

main :: IO ()
main = do
  client <- defaultClient
  let opts = Just (defaultOptions {optNumPredict = Just 20})
      req =
        (chatRequest "qwen3.5:2b" (userMessage "What is 25 + 17?" :| []))
          { chatTools = Just [calculatorTool]
          , chatOptions = opts
          , chatThink = Just ThinkDisabled
          }
  res <- chat client req
  case res of
    Left err -> putStrLn $ "Error: " <> show err
    Right resp -> print (crMessage resp)