langchain-hs-0.0.5.0: test/Test/Langchain/Callback/CallbackManagerSpec.hs

{-# LANGUAGE OverloadedStrings #-}

module Test.Langchain.Callback.CallbackManagerSpec (tests) where

import Control.Concurrent.STM (readTVarIO)
import Data.Time.Clock (getCurrentTime)
import Test.Tasty
import Test.Tasty.HUnit

import Langchain.Callback.Manager

tests :: TestTree
tests =
  testGroup
    "Langchain.Callback.CallbackManagerSpec"
    [ testCase "CallbackManager registers handler and dispatches events" $ do
        mgr <- newCallbackManager
        (handler, logsVar) <- newLoggingCallbackHandler "TestHandler"
        registerHandler mgr handler

        now <- getCurrentTime
        dispatchEvent mgr (OnLLMStart "qwen2.5:7b" ["Hello"] now)
        dispatchEvent mgr (OnLLMEnd "qwen2.5:7b" "Hi there!" 1500 now)

        logged <- readTVarIO logsVar
        length logged @?= 2
    ]