mirror/ezbookkeeping
1
0
Fork 0
mirror of https://github.com/mayswind/ezbookkeeping.git synced 2026-05-14 06:57:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

add missing stream field

Browse Source
This commit is contained in:
MaysWind
2026-02-01 15:22:15 +08:00
parent 7647f4f5b9
commit 4177ac3d46
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/llm/provider/lmstudio/lm_studio_large_language_model_adapter.go
+4 -2
View File
@@ -28,6 +28,7 @@ type LMStudioLargeLanguageModelAdapter struct {
// LMStudioChatRequest defines the structure of LM Studio chat request // LMStudioChatRequest defines the structure of LM Studio chat request
type LMStudioChatRequest struct { type LMStudioChatRequest struct {
Model string `json:"model"` Model string `json:"model"`
Stream bool `json:"stream"`
SystemPrompt string `json:"system_prompt,omitempty"` SystemPrompt string `json:"system_prompt,omitempty"`
Input []*LMStudioChatRequestInput `json:"input"` Input []*LMStudioChatRequestInput `json:"input"`
} }
@@ -100,8 +101,9 @@ func (p *LMStudioLargeLanguageModelAdapter) buildJsonRequestBody(c core.Context,
} }
chatRequest := &LMStudioChatRequest{ chatRequest := &LMStudioChatRequest{
Model: p.LMStudioModelID, Model: p.LMStudioModelID,
Input: make([]*LMStudioChatRequestInput, 0, 1), Stream: request.Stream,
Input: make([]*LMStudioChatRequestInput, 0, 1),
} }
if request.SystemPrompt != "" { if request.SystemPrompt != "" {
pkg/llm/provider/lmstudio/lm_studio_large_language_model_adapter_test.go
+2 -2
View File
@@ -27,7 +27,7 @@ func TestLMStudioLargeLanguageModelAdapter_buildJsonRequestBody_TextualUserPromp
err = json.Unmarshal(bodyBytes, &body) err = json.Unmarshal(bodyBytes, &body)
assert.Nil(t, err) assert.Nil(t, err)
assert.Equal(t, "{\"model\":\"test\",\"system_prompt\":\"You are a helpful assistant.\",\"input\":[{\"type\":\"text\",\"content\":\"Hello, how are you?\"}]}", string(bodyBytes)) assert.Equal(t, "{\"model\":\"test\",\"stream\":false,\"system_prompt\":\"You are a helpful assistant.\",\"input\":[{\"type\":\"text\",\"content\":\"Hello, how are you?\"}]}", string(bodyBytes))
} }
func TestLMStudioLargeLanguageModelAdapter_buildJsonRequestBody_ImageUserPrompt(t *testing.T) { func TestLMStudioLargeLanguageModelAdapter_buildJsonRequestBody_ImageUserPrompt(t *testing.T) {
@@ -49,7 +49,7 @@ func TestLMStudioLargeLanguageModelAdapter_buildJsonRequestBody_ImageUserPrompt(
err = json.Unmarshal(bodyBytes, &body) err = json.Unmarshal(bodyBytes, &body)
assert.Nil(t, err) assert.Nil(t, err)
assert.Equal(t, "{\"model\":\"test\",\"system_prompt\":\"What's in this image?\",\"input\":[{\"type\":\"image\",\"data_url\":\"data:image/png;base64,ZmFrZWRhdGE=\"}]}", string(bodyBytes)) assert.Equal(t, "{\"model\":\"test\",\"stream\":false,\"system_prompt\":\"What's in this image?\",\"input\":[{\"type\":\"image\",\"data_url\":\"data:image/png;base64,ZmFrZWRhdGE=\"}]}", string(bodyBytes))
} }
func TestLMStudioLargeLanguageModelAdapter_ParseTextualResponse_ValidJsonResponse(t *testing.T) { func TestLMStudioLargeLanguageModelAdapter_ParseTextualResponse_ValidJsonResponse(t *testing.T) {