reply func(http.ResponseWriter, ModelFakeCall)
}
+func newModelFake(t *testing.T) *ModelFake {
+ t.Helper()
+
+ f := &ModelFake{
+ reply: func(w http.ResponseWriter, _ ModelFakeCall) {
+ modelWriteEvents(w, modelCompletedEvent([]any{modelMessageItem("a line")}, 100, 0, 5))
+ },
+ }
+
+ f.Server = httptest.NewServer(http.HandlerFunc(f.serve))
+ t.Cleanup(f.Close)
+
+ return f
+}
+
// Call returns call number n.
func (f *ModelFake) Call(n int) ModelFakeCall {
f.mutex.Lock()
reply(w, call)
}
-func newModelFake(t *testing.T) *ModelFake {
- t.Helper()
-
- f := &ModelFake{
- reply: func(w http.ResponseWriter, _ ModelFakeCall) {
- modelWriteEvents(w, modelCompletedEvent([]any{modelMessageItem("a line")}, 100, 0, 5))
- },
- }
-
- f.Server = httptest.NewServer(http.HandlerFunc(f.serve))
- t.Cleanup(f.Close)
-
- return f
-}
-
func modelTestRequest() ModelRequest {
return ModelRequest{
Model: "gpt-test",