From a453c7d5c277efdf6acaed50e2b02dabf66c078f Mon Sep 17 00:00:00 2001 From: Jeremy Price Date: Mon, 19 May 2025 17:24:08 +0200 Subject: [PATCH] Add search-generative test --- .../docs/mainpkg/search-aggregation_test.go | 1 + .../go/docs/mainpkg/search-generative_test.go | 30 ++++++++++++++++++- 2 files changed, 30 insertions(+), 1 deletion(-) diff --git a/_includes/code/howto/go/docs/mainpkg/search-aggregation_test.go b/_includes/code/howto/go/docs/mainpkg/search-aggregation_test.go index 49a8ca957e..7b149c2289 100644 --- a/_includes/code/howto/go/docs/mainpkg/search-aggregation_test.go +++ b/_includes/code/howto/go/docs/mainpkg/search-aggregation_test.go @@ -223,6 +223,7 @@ func TestNearTextWithDistance(t *testing.T) { t.Logf("%s", jsonResponse) points := response.Data["Aggregate"].(map[string]interface{})["JeopardyQuestion"].([]interface{})[0].(map[string]interface{})["points"].(map[string]interface{}) + assert.NotNil(t, points["sum"]) assert.Greater(t, points["sum"].(float64), float64(0)) } diff --git a/_includes/code/howto/go/docs/mainpkg/search-generative_test.go b/_includes/code/howto/go/docs/mainpkg/search-generative_test.go index e537b4f262..aa93b232fa 100644 --- a/_includes/code/howto/go/docs/mainpkg/search-generative_test.go +++ b/_includes/code/howto/go/docs/mainpkg/search-generative_test.go @@ -12,6 +12,34 @@ import ( "github.com/weaviate/weaviate-go-client/v4/weaviate/graphql" ) +func TestDynamicRag(t *testing.T) { +// START DynamicRag +//const reviews = client.collections.use("WineReviewNV") + + client := setupClient() + ctx := context.Background() + generatePrompt := "Translate this into German: {review_body}" + gs := graphql.NewGenerativeSearch().SingleResult(generatePrompt).GroupedResult("Summarize these reviews") + + response, err := client.GraphQL().Get(). + WithClassName("WineReviewNV"). + WithGenerativeSearch(gs). + WithNearText((&graphql.NearTextArgumentBuilder{}). + WithConcepts([]string{"a sweet German white wine"}).WithTargetVectors("title_country")). + WithLimit(2). + Do(ctx) + require.NoError(t, err) + require.NotNil(t, response) +// END DynamicRag + +// Test results +_, ok := response.Data["Get"].(map[string]interface{})["WineReviewNV"] +assert.True(t, ok) +//assert "title" in response.objects[0].properties.keys() +// End test + +} + // ===================================== // ===== SINGLE GENERATIVE EXAMPLE ===== // ===================================== @@ -192,4 +220,4 @@ func TestGroupedGenerativeProperties(t *testing.T) { assert.True(t, strings.Contains(generate["groupedResult"].(string), "Australia")) } } -} \ No newline at end of file +}