test: fix missing function on OpenAI mock

Author: realdavidvega

src/commonTest/kotlin/com/xebia/functional/chains/ChainTestUtils.kt

@@ -1,5 +1,7 @@
 package com.xebia.functional.chains
 
+import com.xebia.functional.llm.openai.ChatCompletionRequest
+import com.xebia.functional.llm.openai.ChatCompletionResponse
 import com.xebia.functional.llm.openai.CompletionChoice
 import com.xebia.functional.llm.openai.CompletionRequest
 import com.xebia.functional.llm.openai.EmbeddingRequest
@@ -19,7 +21,11 @@ val testLLM = object : OpenAIClient {
             else -> listOf(CompletionChoice("foo", 1, "bar"))
         }
 
-    override suspend fun createEmbeddings(request: EmbeddingRequest): EmbeddingResult = TODO()
+    override suspend fun createChatCompletion(request: ChatCompletionRequest): ChatCompletionResponse =
+        TODO()
+
+    override suspend fun createEmbeddings(request: EmbeddingRequest): EmbeddingResult =
+        TODO()
 }
 
 val testContext = """foo foo foo

src/commonTest/kotlin/com/xebia/functional/chains/LLMChainSpec.kt

@@ -1,11 +1,6 @@
 package com.xebia.functional.chains
 
 import arrow.core.raise.either
-import com.xebia.functional.llm.openai.CompletionChoice
-import com.xebia.functional.llm.openai.CompletionRequest
-import com.xebia.functional.llm.openai.EmbeddingRequest
-import com.xebia.functional.llm.openai.EmbeddingResult
-import com.xebia.functional.llm.openai.OpenAIClient
 import com.xebia.functional.prompt.PromptTemplate
 import io.kotest.assertions.arrow.core.shouldBeLeft
 import io.kotest.assertions.arrow.core.shouldBeRight
@@ -63,20 +58,3 @@ class LLMChainSpec : StringSpec({
         } shouldBeLeft Chain.InvalidInputs("The expected inputs are more than one: {name}, {age}")
     }
 })
-
-val llm = object : OpenAIClient {
-    override suspend fun createCompletion(request: CompletionRequest): List<CompletionChoice> =
-        when(request.prompt) {
-            "Tell me a joke." ->
-                listOf(CompletionChoice("I'm not good at jokes", 1, "foo"))
-            "My name is foo and I'm 28 years old" ->
-                listOf(CompletionChoice("Hello there! Nice to meet you foo", 1, "foo"))
-            else -> listOf(CompletionChoice("foo", 1, "bar"))
-        }
-
-    override suspend fun createChatCompletion(request: ChatCompletionRequest): ChatCompletionResponse {
-        TODO("Not yet implemented")
-    }
-
-    override suspend fun createEmbeddings(request: EmbeddingRequest): EmbeddingResult = TODO()
-}