fixed test case

landing-ai · Apr 15, 2024 · 9f8b7e2 · 9f8b7e2
1 parent 7f2140f
commit 9f8b7e2
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_llm.py b/tests/test_llm.py
@@ -18,7 +18,7 @@ def test_generate_with_mock(openai_llm_mock): # noqa: F811
  response = llm.generate("test prompt")
  assert response == "mocked response"
  openai_llm_mock.chat.completions.create.assert_called_once_with(
- model="gpt-4-turbo-preview",
+ model="gpt-4-turbo",
  messages=[{"role": "user", "content": "test prompt"}],
  )
 
@@ -31,7 +31,7 @@ def test_chat_with_mock(openai_llm_mock): # noqa: F811
  response = llm.chat([{"role": "user", "content": "test prompt"}])
  assert response == "mocked response"
  openai_llm_mock.chat.completions.create.assert_called_once_with(
- model="gpt-4-turbo-preview",
+ model="gpt-4-turbo",
  messages=[{"role": "user", "content": "test prompt"}],
  )
 
@@ -44,14 +44,14 @@ def test_call_with_mock(openai_llm_mock): # noqa: F811
  response = llm("test prompt")
  assert response == "mocked response"
  openai_llm_mock.chat.completions.create.assert_called_once_with(
- model="gpt-4-turbo-preview",
+ model="gpt-4-turbo",
  messages=[{"role": "user", "content": "test prompt"}],
  )
 
  response = llm([{"role": "user", "content": "test prompt"}])
  assert response == "mocked response"
  openai_llm_mock.chat.completions.create.assert_called_with(
- model="gpt-4-turbo-preview",
+ model="gpt-4-turbo",
  messages=[{"role": "user", "content": "test prompt"}],
  )