vrr/ProxyAI
2
0
Fork 0
mirror of https://github.com/carlrobertoh/ProxyAI.git synced 2026-05-12 14:10:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: ollama stream requests (fixes #646)

Browse source
This commit is contained in:
Carl-Robert Linnupuu 2024-09-05 12:29:49 +03:00
parent a81bfa1005
commit 2bc3d8d645
2 changed files with 13 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/test/kotlin/ee/carlrobert/codegpt/completions/DefaultCompletionRequestHandlerTest.kt
View file

@ -146,9 +146,18 @@ class DefaultCompletionRequestHandlerTest : IntegrationTest() {
true
)
listOf(
jsonMapResponse("message", jsonMap(e("content", "Hel"), e("role", "assistant"))),
jsonMapResponse("message", jsonMap(e("content", "lo"), e("role", "assistant"))),
jsonMapResponse("message", jsonMap(e("content", "!"), e("role", "assistant")))
jsonMapResponse(
e("message", jsonMap(e("content", "Hel"), e("role", "assistant"))),
e("done", false)),
jsonMapResponse(
e("message", jsonMap(e("content", "lo"), e("role", "assistant"))),
e("done", false)),
jsonMapResponse(
e("message", jsonMap(e("content", "!"), e("role", "assistant"))),
e("done", false)),
jsonMapResponse(
e("message", jsonMap(e("content", ""), e("role", "assistant"))),
e("done", true))
)
})