Merge pull request #19 from gbfansheng/main

intitni · web-flow · commit ea9597fd960d · 2024-12-06T00:12:20.000+08:00
fix: correct stream completion detection in OpenAI response
diff --git a/Core/Sources/CodeCompletionService/API/OpenAIService.swift b/Core/Sources/CodeCompletionService/API/OpenAIService.swift
@@ -170,7 +170,7 @@ extension OpenAIService {
                     ChatCompletionsStreamDataChunk.self,
                     from: text.data(using: .utf8) ?? Data()
                 )
-                return .init(chunk: chunk, done: chunk.choices?.first?.delta?.content != nil)
+                return .init(chunk: chunk, done: chunk.choices?.first?.finish_reason != nil)
             } catch {
                 print(error)
                 throw error

Original file line number	Diff line number	Diff line change
`@@ -170,7 +170,7 @@ extension OpenAIService {`
`170`	`170`	`ChatCompletionsStreamDataChunk.self,`
`171`	`171`	`from: text.data(using: .utf8) ?? Data()`
`172`	`172`	`)`
`173`		`- return .init(chunk: chunk, done: chunk.choices?.first?.delta?.content != nil)`
	`173`	`+ return .init(chunk: chunk, done: chunk.choices?.first?.finish_reason != nil)`
`174`	`174`	`} catch {`
`175`	`175`	`print(error)`
`176`	`176`	`throw error`