test: Update model used in OpenAI tests to gpt-4o-mini (davidmigloz#539)

KennethKnudsen97 · Oct 1, 2024 · 3739397 · 3739397
1 parent 910aec1
commit 3739397
Show file tree

Hide file tree

Showing 5 changed files with 3 additions and 155 deletions.
diff --git a/packages/langchain_openai/test/agents/tools_test.dart b/packages/langchain_openai/test/agents/tools_test.dart
@@ -24,7 +24,6 @@ void main() {
       final llm = ChatOpenAI(
         apiKey: openaiApiKey,
         defaultOptions: const ChatOpenAIOptions(
-          model: 'gpt-4-turbo',
           temperature: 0,
         ),
       );
@@ -46,7 +45,6 @@ void main() {
       final llm = ChatOpenAI(
         apiKey: openaiApiKey,
         defaultOptions: const ChatOpenAIOptions(
-          model: 'gpt-4-turbo',
           temperature: 0,
         ),
       );
@@ -135,7 +133,6 @@ void main() {
     final model = ChatOpenAI(
       apiKey: openaiApiKey,
       defaultOptions: const ChatOpenAIOptions(
-        model: 'gpt-4-turbo',
         temperature: 0,
       ),
     ).bind(ChatOpenAIOptions(tools: [tool]));

diff --git a/packages/langchain_openai/test/chains/qa_with_sources_test.dart b/packages/langchain_openai/test/chains/qa_with_sources_test.dart
@@ -125,7 +125,6 @@ Question: {question}
       final chatModel = ChatOpenAI(
         apiKey: openaiApiKey,
         defaultOptions: const ChatOpenAIOptions(
-          model: 'gpt-4o-mini',
           temperature: 0,
         ),
       );

diff --git a/packages/langchain_openai/test/chat_models/anyscale_test.dart b/packages/langchain_openai/test/chat_models/anyscale_test.dart
diff --git a/packages/langchain_openai/test/chat_models/chat_openai_test.dart b/packages/langchain_openai/test/chat_models/chat_openai_test.dart
@@ -208,7 +208,6 @@ void main() {
 
     test('Test countTokens messages', () async {
       final models = [
-        'gpt-3.5-turbo-16k-0613',
         'gpt-4-0314',
         'gpt-4-0613',
       ];
@@ -328,7 +327,7 @@ void main() {
       final llm = ChatOpenAI(
         apiKey: openaiApiKey,
         defaultOptions: const ChatOpenAIOptions(
-          model: 'gpt-4-turbo',
+          model: defaultModel,
           temperature: 0,
           seed: 12345,
         ),
@@ -455,7 +454,7 @@ void main() {
       final chatModel = ChatOpenAI(
         apiKey: openaiApiKey,
         defaultOptions: const ChatOpenAIOptions(
-          model: 'gpt-4-turbo',
+          model: defaultModel,
         ),
       );
 
@@ -482,7 +481,7 @@ void main() {
       final chatModel = ChatOpenAI(
         apiKey: openaiApiKey,
         defaultOptions: const ChatOpenAIOptions(
-          model: 'gpt-4-turbo',
+          model: defaultModel,
         ),
       );
 

diff --git a/packages/langchain_openai/test/embeddings/anyscale_embeddings_test.dart b/packages/langchain_openai/test/embeddings/anyscale_embeddings_test.dart