updated test

xebia-functional · May 30, 2024 · 8f9e8e4 · 8f9e8e4
1 parent 2d1aa46
commit 8f9e8e4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/core/src/commonTest/kotlin/com/xebia/functional/xef/conversation/ConversationSpec.kt b/core/src/commonTest/kotlin/com/xebia/functional/xef/conversation/ConversationSpec.kt
@@ -52,7 +52,7 @@ class ConversationSpec :
     }
 
     """"
-      | GPT_3_5_TURBO model has 4097 max context length
+      | GPT_3_5_TURBO model has 16385 max context length
       | when the number of token in the conversation is greater than
       | the space allotted for the message history in the prompt configuration
       | the number of messages in the request must have fewer messages than
@@ -83,7 +83,7 @@ class ConversationSpec :
       val memories = vectorStore.memories(model, conversationId, totalTokens)
 
       // The messages in the request doesn't contain the message response
-      val messagesSizePlusMessageResponse = lastRequest.messages.size + 1
+      val messagesSizePlusMessageResponse = lastRequest.messages.size
 
       messagesSizePlusMessageResponse shouldBeLessThan memories.size
     }