admin/ersteller
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add mocked content field

Browse Source
This commit is contained in:
Achim Rohn
2025-12-08 23:41:47 +01:00
parent 049fefed75
commit dcb2cabc7c
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
llm/mistral.go
+9 -3
View File
@@ -4,9 +4,10 @@ import (
"bytes"
"encoding/json"
"fmt"
"git.gorlug.de/code/ersteller"
"io"
"net/http"
"git.gorlug.de/code/ersteller"
)
type ChatCompletionRequestResponseFormatType string
@@ -148,15 +149,20 @@ type MistralClient interface {
CreateChatCompletion(req *ChatCompletionRequest) (*ChatCompletionResponse, error)
}
type MistralClientMock struct{}
type MistralClientMock struct {
MockedContent string
}
func (m MistralClientMock) CreateChatCompletion(req *ChatCompletionRequest) (*ChatCompletionResponse, error) {
ersteller.Debug("mocking the mistral client")
if m.MockedContent == "" {
m.MockedContent = `{"name":"Meetup app","description":"An app to make notes for meetups","entities":[{"name":"Meetup","fields":[{"name":"name","type":"string"},{"name":"description","type":"string","longText":true},{"name":"date","type":"time"}],"entities":[{"name":"Notes","fields":[{"name":"title","type":"string"},{"name":"description","type":"string","longText":true}]}]}]}`
}
response := ChatCompletionResponse{
Choices: []ChatCompletionChoice{
{
Message: Message{
Content: `{"name":"Meetup app","description":"An app to make notes for meetups","entities":[{"name":"Meetup","fields":[{"name":"name","type":"string"},{"name":"description","type":"string","longText":true},{"name":"date","type":"time"}],"entities":[{"name":"Notes","fields":[{"name":"title","type":"string"},{"name":"description","type":"string","longText":true}]}]}]}`,
Content: m.MockedContent,
},
},
},