jbergner
a6bc71fb3a
Some checks failed
release-tag / Resolve release metadata (push) Successful in 30s
release-tag / Build knowledge (push) Failing after 4m51s
release-tag / Build control (push) Failing after 5m0s
release-tag / Build agent (push) Failing after 5m0s
release-tag / Build agent-data-init (push) Failing after 5m5s
release-tag / Build neuroforge-worker (push) Failing after 5m7s
release-tag / Build neuroforge (push) Failing after 5m9s
84 lines
2.9 KiB
Go
84 lines
2.9 KiB
Go
package httpapi
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"mime/multipart"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestV8TextDocumentAndResearchEndpoints(t *testing.T) {
|
|
ollama := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if r.URL.Path == "/api/embed" {
|
|
_ = json.NewEncoder(w).Encode(map[string]any{"embeddings": [][]float32{{1, 0, 0, 0}}, "prompt_eval_count": 1})
|
|
return
|
|
}
|
|
http.NotFound(w, r)
|
|
}))
|
|
defer ollama.Close()
|
|
searx := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if r.URL.Query().Get("format") != "json" {
|
|
t.Errorf("format=%q", r.URL.Query().Get("format"))
|
|
}
|
|
_ = json.NewEncoder(w).Encode(map[string]any{"results": []map[string]any{{"title": "GPU", "url": "https://example.com/gpu", "content": "GPU evidence from search", "engine": "test"}}})
|
|
}))
|
|
defer searx.Close()
|
|
|
|
srv, _ := newMetricsTestServer(t)
|
|
cfg := srv.store.Config()
|
|
cfg.Ollama[0].BaseURL = ollama.URL
|
|
cfg.Routing.EmbeddingProvider = "ollama"
|
|
cfg.Brain.ExternalRelinkWorker = false
|
|
cfg.Research.Enabled = true
|
|
cfg.Research.SearXNG.Enabled = true
|
|
cfg.Research.SearXNG.BaseURL = searx.URL
|
|
cfg.Research.WebFetch.Enabled = false
|
|
if err := srv.store.UpdateConfig(cfg); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
admin := srv.store.Secrets().AdminToken
|
|
|
|
// Plain text source.
|
|
body := `{"title":"manual","text":"NVIDIA CUDA fact","trust":0.9}`
|
|
req := httptest.NewRequest(http.MethodPost, "/api/v1/ingest/text", strings.NewReader(body))
|
|
req.Header.Set("X-Admin-Token", admin)
|
|
req.Header.Set("Content-Type", "application/json")
|
|
rr := httptest.NewRecorder()
|
|
srv.Handler().ServeHTTP(rr, req)
|
|
if rr.Code != http.StatusCreated {
|
|
t.Fatalf("text ingest status=%d body=%s", rr.Code, rr.Body.String())
|
|
}
|
|
|
|
// Multipart text document.
|
|
var mb bytes.Buffer
|
|
mw := multipart.NewWriter(&mb)
|
|
fw, _ := mw.CreateFormFile("file", "note.txt")
|
|
_, _ = fw.Write([]byte("A second document fact about CUDA."))
|
|
_ = mw.WriteField("title", "note")
|
|
_ = mw.Close()
|
|
req = httptest.NewRequest(http.MethodPost, "/api/v1/ingest/document", &mb)
|
|
req.Header.Set("X-Admin-Token", admin)
|
|
req.Header.Set("Content-Type", mw.FormDataContentType())
|
|
rr = httptest.NewRecorder()
|
|
srv.Handler().ServeHTTP(rr, req)
|
|
if rr.Code != http.StatusCreated {
|
|
t.Fatalf("document ingest status=%d body=%s", rr.Code, rr.Body.String())
|
|
}
|
|
|
|
// SearXNG-backed research learns the snippet without web-page fetching.
|
|
req = httptest.NewRequest(http.MethodPost, "/api/v1/research", strings.NewReader(`{"query":"NVIDIA CUDA","learn":true,"fetch_pages":false}`))
|
|
req.Header.Set("X-Admin-Token", admin)
|
|
req.Header.Set("Content-Type", "application/json")
|
|
rr = httptest.NewRecorder()
|
|
srv.Handler().ServeHTTP(rr, req)
|
|
if rr.Code != http.StatusOK {
|
|
t.Fatalf("research status=%d body=%s", rr.Code, rr.Body.String())
|
|
}
|
|
if len(srv.store.SourcesSnapshot(10)) < 3 {
|
|
t.Fatalf("expected source records, got %#v", srv.store.SourcesSnapshot(10))
|
|
}
|
|
}
|