Updated README and unit tests to properly reflect builder syntax appr…

…oach.
btfranklin · Apr 18, 2023 · 75132a3 · 75132a3
1 parent 7d7803b
commit 75132a3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 13 deletions.
diff --git a/README.md b/README.md
@@ -24,17 +24,13 @@ Initialize an OpenAIAPIConnection with your API key:
 let openAIAPIConnection = OpenAIAPIConnection(apiKey: "your_api_key_here")
 ```
 
-Create an OpenAIChatThread instance with the connection:
+Create an OpenAIChatThread instance with the connection, and
+add system, user, or assistant messages to the chat thread:
 
 ```swift
 let chatThread = OpenAIChatThread(connection: openAIAPIConnection)
-```
-
-Add system, user, or assistant messages to the chat thread:
-
-```swift
-chatThread.addSystemMessage(content: "You are a helpful assistant.")
-          .addUserMessage(content: "Who won the world series in 2020?")
+    .addSystemMessage(content: "You are a helpful assistant.")
+    .addUserMessage(content: "Who won the world series in 2020?")
 ```
 
 Generate a completion using the chat thread:

diff --git a/Tests/CleverBirdTests/OpenAIChatThreadTests.swift b/Tests/CleverBirdTests/OpenAIChatThreadTests.swift
@@ -55,9 +55,9 @@ class OpenAIChatThreadTests: XCTestCase {
 
         let openAIAPIConnection = OpenAIAPIConnection(apiKey: "fake_api_key", urlRequester: mockURLRequester)
         let chatThread = OpenAIChatThread(connection: openAIAPIConnection)
+            .addSystemMessage(content: "You are a helpful assistant.")
+            .addUserMessage(content: "Who won the world series in 2020?")
 
-        chatThread.addSystemMessage(content: "You are a helpful assistant.")
-        chatThread.addUserMessage(content: "Who won the world series in 2020?")
 
         let completionResponse = await chatThread.complete()
 
@@ -69,9 +69,8 @@ class OpenAIChatThreadTests: XCTestCase {
     func testTokenCount() {
         let openAIAPIConnection = OpenAIAPIConnection(apiKey: "fake_api_key", urlRequester: MockURLRequester(response: ""))
         let chatThread = OpenAIChatThread(connection: openAIAPIConnection)
-
-        chatThread.addSystemMessage(content: "You are a helpful assistant.")
-        chatThread.addUserMessage(content: "Who won the world series in 2020?")
+            .addSystemMessage(content: "You are a helpful assistant.")
+            .addUserMessage(content: "Who won the world series in 2020?")
 
         let tokenCount = chatThread.tokenCount()