Skip to content

Commit 3a07cf3

Browse files
committed
fix recommend doc fields
1 parent 24354a2 commit 3a07cf3

File tree

1 file changed

+14
-14
lines changed

1 file changed

+14
-14
lines changed

tests/v2_tests/test_recommend.py

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -117,19 +117,19 @@ def test_recommend_rerank_depth_behavior(self):
117117
"""API-level test that rerank_depth affects recommender results according to expected behavior."""
118118
self.test_cases = [(CloudTestIndex.structured_text, self.structured_index_name)]
119119
docs = [
120-
{"_id": "doc_0", "title": "Project Overview",
121-
"content": "Summary of the project's goals and deliverables."},
122-
{"_id": "doc_1", "title": "Team Roles", "content": "Descriptions of each team member's responsibilities."},
123-
{"_id": "doc_2", "title": "Timeline", "content": "Key milestones and deadlines for the project."},
124-
{"_id": "doc_3", "title": "Budget Estimate", "content": "Projected costs and resource allocation."},
125-
{"_id": "doc_4", "title": "Tech Stack", "content": "Overview of technologies and tools being used."},
126-
{"_id": "doc_5", "title": "Risk Assessment", "content": "Potential risks and mitigation strategies."},
127-
{"_id": "doc_6", "title": "Client Feedback", "content": "Summary of feedback received from stakeholders."},
128-
{"_id": "doc_7", "title": "Testing Plan", "content": "Details on testing strategies and coverage."},
129-
{"_id": "doc_8", "title": "Deployment Guide",
130-
"content": "Steps and procedures for deploying the application."},
131-
{"_id": "doc_9", "title": "Post-Mortem",
132-
"content": "Analysis of what went well and areas for improvement."},
120+
{"_id": "doc_0", "text_field_1": "Project Overview",
121+
"text_field_2": "Summary of the project's goals and deliverables."},
122+
{"_id": "doc_1", "text_field_1": "Team Roles", "text_field_2": "Descriptions of each team member's responsibilities."},
123+
{"_id": "doc_2", "text_field_1": "Timeline", "text_field_2": "Key milestones and deadlines for the project."},
124+
{"_id": "doc_3", "text_field_1": "Budget Estimate", "text_field_2": "Projected costs and resource allocation."},
125+
{"_id": "doc_4", "text_field_1": "Tech Stack", "text_field_2": "Overview of technologies and tools being used."},
126+
{"_id": "doc_5", "text_field_1": "Risk Assessment", "text_field_2": "Potential risks and mitigation strategies."},
127+
{"_id": "doc_6", "text_field_1": "Client Feedback", "text_field_2": "Summary of feedback received from stakeholders."},
128+
{"_id": "doc_7", "text_field_1": "Testing Plan", "text_field_2": "Details on testing strategies and coverage."},
129+
{"_id": "doc_8", "text_field_1": "Deployment Guide",
130+
"text_field_2": "Steps and procedures for deploying the application."},
131+
{"_id": "doc_9", "text_field_1": "Post-Mortem",
132+
"text_field_2": "Analysis of what went well and areas for improvement."},
133133
]
134134

135135
for cloud_test_index_to_use, open_source_test_index_name in self.test_cases:
@@ -176,7 +176,7 @@ def test_recommend_rerank_depth_behavior(self):
176176
with self.subTest(case="ef_search_limits_rerank_pool"):
177177
res = self.client.index(test_index_name).recommend(
178178
documents=["doc_0", "doc_1"], limit=10, offset=0, rerank_depth=5, ef_search=3,
179-
searchable_attributes=['title']
179+
searchable_attributes=['text_field_1']
180180
)
181181
self.assertLessEqual(len(res["hits"]), 3)
182182

0 commit comments

Comments
 (0)