tests

Author: xfalcox

spec/lib/completions/dialects/dialect_spec.rb

@@ -99,7 +99,8 @@ def tokenizer
     end
 
     it "limits the system message to 60% of available tokens" do
-      prompt = DiscourseAi::Completions::Prompt.new("I'm a system message consisting of 10 tokens")
+      prompt =
+        DiscourseAi::Completions::Prompt.new("I'm a system message consisting of 10 tokens okay")
       prompt.push(type: :user, content: five_token_msg)
 
       dialect = TestDialect.new(prompt, llm_model)
@@ -109,7 +110,7 @@ def tokenizer
 
       expect(trimmed).to eq(
         [
-          { type: :system, content: "I'm a system message consisting of 10" },
+          { type: :system, content: "I'm a system message consisting of 10 tokens" },
           { type: :user, content: five_token_msg },
         ],
       )

spec/lib/completions/endpoints/open_ai_spec.rb

@@ -18,7 +18,7 @@ def response(content, tool_call: false)
       model: "gpt-3.5-turbo-0301",
       usage: {
         prompt_tokens: 8,
-        completion_tokens: 13,
+        completion_tokens: 12,
         total_tokens: 499,
       },
       choices: [