Update code

box · Aug 23, 2024 · e60dd21 · e60dd21
1 parent b7552f4
commit e60dd21
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/integration_test/__tests__/ai.test.js b/tests/integration_test/__tests__/ai.test.js
@@ -90,9 +90,9 @@ test('test AI text gen', async() => {
 		],
 		dialogue_history: dialogueHistory,
 		ai_agent: {
-			type: 'ai_agent_ask',
-			basic_text_multi: {
-				model: 'openai__gpt_3_5_turbo'
+			type: 'ai_agent_text_gen',
+			basic_gen: {
+				model: 'openai__gpt_3_5_turbo_16k'
 			}
 		}
 	});