fixed tests for recent PR

brainlid · Jan 30, 2025 · 3c1260c · 3c1260c
1 parent abf059e
commit 3c1260c
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/test/chat_models/chat_open_ai_test.exs b/test/chat_models/chat_open_ai_test.exs
@@ -120,7 +120,8 @@ defmodule LangChain.ChatModels.ChatOpenAITest do
       assert data.model == @test_model
       assert data.temperature == 1
       assert data.frequency_penalty == 0.5
-      assert data.response_format == %{"type" => "text"}
+      # NOTE: %{"type" => "text"} is the default when not specified
+      assert data[:response_format] == nil
     end
 
     test "generates a map for an API call with JSON response set to true" do
@@ -2178,7 +2179,8 @@ defmodule LangChain.ChatModels.ChatOpenAITest do
 
       data = ChatOpenAI.for_api(openai, [], [])
 
-      assert data.response_format == %{"type" => "text"}
+      # NOTE: %{"type" => "text"} is the default
+      assert data[:response_format] == nil
     end
 
     test "generates a map for an API call with json_object format when json_response is true and no schema" do