Skip to content

Commit

Permalink
minor test cleanup
Browse files Browse the repository at this point in the history
  • Loading branch information
brainlid committed Jan 13, 2025
1 parent 3630129 commit 9ee95a7
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions test/chains/llm_chain_test.exs
Original file line number Diff line number Diff line change
Expand Up @@ -1835,9 +1835,7 @@ defmodule LangChain.Chains.LLMChainTest do
updated_chain = LLMChain.add_callback(chain, handler2)
assert updated_chain.callbacks == [handler1, handler2]
end
end

describe "add_llm_callback/2" do
test "appends a callback handler to the chain's LLM", %{chat: chat} do
handler1 = %{on_llm_new_message: fn %LLMChain{} = _chain, _msg -> IO.puts("MESSAGE 1!") end}
handler2 = %{on_llm_new_message: fn %LLMChain{} = _chain, _msg -> IO.puts("MESSAGE 2!") end}
Expand Down

0 comments on commit 9ee95a7

Please sign in to comment.