fixes test (wrong model contxt length)

xebia-functional · Nov 25, 2023 · 066135e · 066135e
1 parent 253adba
commit 066135e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/core/src/commonMain/kotlin/com/xebia/functional/xef/llm/BaseChat.kt b/core/src/commonMain/kotlin/com/xebia/functional/xef/llm/BaseChat.kt
@@ -17,9 +17,9 @@ interface BaseChat : LLM {
           "accessing maxContextLength requires model's context length to be of type MaxIoContextLength.Combined"
         )
 
-  @Suppress("OVERRIDE_DEPRECATION") fun countTokens(text: String): Int
+  fun countTokens(text: String): Int
 
-  @Suppress("OVERRIDE_DEPRECATION") fun truncateText(text: String, maxTokens: Int): String
+  fun truncateText(text: String, maxTokens: Int): String
 
-  @Suppress("OVERRIDE_DEPRECATION") fun tokensFromMessages(messages: List<Message>): Int
+  fun tokensFromMessages(messages: List<Message>): Int
 }
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
@@ -63,7 +63,7 @@ class ConversationSpec :
         )
       val vectorStore = scope.store
 
-      val modelAda = TestModel(responses = messages)
+      val modelAda = TestModel(responses = messages, contextLength = MaxIoContextLength.Combined(2049))
 
       val totalTokens =
         modelAda.tokensFromMessages(