hindsight-client 0.1.0__tar.gz → 0.1.1__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (138) hide show
  1. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/PKG-INFO +1 -1
  2. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client/hindsight_client.py +20 -54
  3. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/pyproject.toml +1 -1
  4. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/tests/test_main_operations.py +11 -10
  5. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/.gitignore +0 -0
  6. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/.openapi-generator/FILES +0 -0
  7. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/.openapi-generator/VERSION +0 -0
  8. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/.openapi-generator-ignore +0 -0
  9. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/README.md +0 -0
  10. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client/__init__.py +0 -0
  11. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/__init__.py +0 -0
  12. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/api/__init__.py +0 -0
  13. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/api/default_api.py +0 -0
  14. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/api/monitoring_api.py +0 -0
  15. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/api_client.py +0 -0
  16. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/api_response.py +0 -0
  17. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/configuration.py +0 -0
  18. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/AddBackgroundRequest.md +0 -0
  19. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/BackgroundResponse.md +0 -0
  20. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/BankListItem.md +0 -0
  21. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/BankListResponse.md +0 -0
  22. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/BankProfileResponse.md +0 -0
  23. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/Budget.md +0 -0
  24. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/ChunkData.md +0 -0
  25. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/ChunkIncludeOptions.md +0 -0
  26. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/ChunkResponse.md +0 -0
  27. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/CreateBankRequest.md +0 -0
  28. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/DefaultApi.md +0 -0
  29. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/DeleteResponse.md +0 -0
  30. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/DispositionTraits.md +0 -0
  31. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/DocumentResponse.md +0 -0
  32. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/EntityDetailResponse.md +0 -0
  33. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/EntityIncludeOptions.md +0 -0
  34. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/EntityListItem.md +0 -0
  35. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/EntityListResponse.md +0 -0
  36. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/EntityObservationResponse.md +0 -0
  37. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/EntityStateResponse.md +0 -0
  38. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/GraphDataResponse.md +0 -0
  39. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/HTTPValidationError.md +0 -0
  40. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/IncludeOptions.md +0 -0
  41. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/ListDocumentsResponse.md +0 -0
  42. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/ListMemoryUnitsResponse.md +0 -0
  43. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/MemoryItem.md +0 -0
  44. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/MonitoringApi.md +0 -0
  45. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/RecallRequest.md +0 -0
  46. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/RecallResponse.md +0 -0
  47. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/RecallResult.md +0 -0
  48. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/ReflectFact.md +0 -0
  49. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/ReflectIncludeOptions.md +0 -0
  50. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/ReflectRequest.md +0 -0
  51. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/ReflectResponse.md +0 -0
  52. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/RetainRequest.md +0 -0
  53. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/RetainResponse.md +0 -0
  54. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/UpdateDispositionRequest.md +0 -0
  55. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/ValidationError.md +0 -0
  56. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/docs/ValidationErrorLocInner.md +0 -0
  57. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/exceptions.py +0 -0
  58. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/__init__.py +0 -0
  59. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/add_background_request.py +0 -0
  60. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/background_response.py +0 -0
  61. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/bank_list_item.py +0 -0
  62. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/bank_list_response.py +0 -0
  63. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/bank_profile_response.py +0 -0
  64. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/budget.py +0 -0
  65. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/chunk_data.py +0 -0
  66. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/chunk_include_options.py +0 -0
  67. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/chunk_response.py +0 -0
  68. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/create_bank_request.py +0 -0
  69. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/delete_response.py +0 -0
  70. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/disposition_traits.py +0 -0
  71. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/document_response.py +0 -0
  72. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/entity_detail_response.py +0 -0
  73. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/entity_include_options.py +0 -0
  74. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/entity_list_item.py +0 -0
  75. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/entity_list_response.py +0 -0
  76. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/entity_observation_response.py +0 -0
  77. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/entity_state_response.py +0 -0
  78. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/graph_data_response.py +0 -0
  79. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/http_validation_error.py +0 -0
  80. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/include_options.py +0 -0
  81. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/list_documents_response.py +0 -0
  82. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/list_memory_units_response.py +0 -0
  83. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/memory_item.py +0 -0
  84. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/recall_request.py +0 -0
  85. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/recall_response.py +0 -0
  86. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/recall_result.py +0 -0
  87. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/reflect_fact.py +0 -0
  88. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/reflect_include_options.py +0 -0
  89. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/reflect_request.py +0 -0
  90. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/reflect_response.py +0 -0
  91. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/retain_request.py +0 -0
  92. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/retain_response.py +0 -0
  93. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/update_disposition_request.py +0 -0
  94. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/validation_error.py +0 -0
  95. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/models/validation_error_loc_inner.py +0 -0
  96. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/rest.py +0 -0
  97. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/__init__.py +0 -0
  98. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_add_background_request.py +0 -0
  99. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_background_response.py +0 -0
  100. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_bank_list_item.py +0 -0
  101. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_bank_list_response.py +0 -0
  102. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_bank_profile_response.py +0 -0
  103. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_budget.py +0 -0
  104. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_chunk_data.py +0 -0
  105. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_chunk_include_options.py +0 -0
  106. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_chunk_response.py +0 -0
  107. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_create_bank_request.py +0 -0
  108. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_default_api.py +0 -0
  109. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_delete_response.py +0 -0
  110. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_disposition_traits.py +0 -0
  111. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_document_response.py +0 -0
  112. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_entity_detail_response.py +0 -0
  113. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_entity_include_options.py +0 -0
  114. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_entity_list_item.py +0 -0
  115. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_entity_list_response.py +0 -0
  116. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_entity_observation_response.py +0 -0
  117. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_entity_state_response.py +0 -0
  118. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_graph_data_response.py +0 -0
  119. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_http_validation_error.py +0 -0
  120. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_include_options.py +0 -0
  121. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_list_documents_response.py +0 -0
  122. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_list_memory_units_response.py +0 -0
  123. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_memory_item.py +0 -0
  124. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_monitoring_api.py +0 -0
  125. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_recall_request.py +0 -0
  126. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_recall_response.py +0 -0
  127. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_recall_result.py +0 -0
  128. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_reflect_fact.py +0 -0
  129. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_reflect_include_options.py +0 -0
  130. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_reflect_request.py +0 -0
  131. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_reflect_response.py +0 -0
  132. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_retain_request.py +0 -0
  133. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_retain_response.py +0 -0
  134. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_update_disposition_request.py +0 -0
  135. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_validation_error.py +0 -0
  136. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/hindsight_client_api/test/test_validation_error_loc_inner.py +0 -0
  137. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/openapi-generator-config.yaml +0 -0
  138. {hindsight_client-0.1.0 → hindsight_client-0.1.1}/tests/__init__.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: hindsight-client
3
- Version: 0.1.0
3
+ Version: 0.1.1
4
4
  Summary: Python client for Hindsight - Semantic memory system with personality-driven thinking
5
5
  Author: Hindsight Team
6
6
  Requires-Python: >=3.10
@@ -157,7 +157,11 @@ class Hindsight:
157
157
  types: Optional[List[str]] = None,
158
158
  max_tokens: int = 4096,
159
159
  budget: str = "mid",
160
- ) -> List[RecallResult]:
160
+ trace: bool = False,
161
+ query_timestamp: Optional[str] = None,
162
+ include_entities: bool = False,
163
+ max_entity_tokens: int = 500,
164
+ ) -> RecallResponse:
161
165
  """
162
166
  Recall memories using semantic similarity.
163
167
 
@@ -167,20 +171,31 @@ class Hindsight:
167
171
  types: Optional list of fact types to filter (world, experience, opinion, observation)
168
172
  max_tokens: Maximum tokens in results (default: 4096)
169
173
  budget: Budget level for recall - "low", "mid", or "high" (default: "mid")
174
+ trace: Enable trace output (default: False)
175
+ query_timestamp: Optional ISO format date string (e.g., '2023-05-30T23:40:00')
176
+ include_entities: Include entity observations in results (default: False)
177
+ max_entity_tokens: Maximum tokens for entity observations (default: 500)
170
178
 
171
179
  Returns:
172
- List of RecallResult objects
180
+ RecallResponse with results, optional entities, and optional trace
173
181
  """
182
+ from hindsight_client_api.models import include_options, entity_include_options
183
+
184
+ include_opts = include_options.IncludeOptions(
185
+ entities=entity_include_options.EntityIncludeOptions(max_tokens=max_entity_tokens) if include_entities else None
186
+ )
187
+
174
188
  request_obj = recall_request.RecallRequest(
175
189
  query=query,
176
190
  types=types,
177
191
  budget=budget,
178
192
  max_tokens=max_tokens,
179
- trace=False,
193
+ trace=trace,
194
+ query_timestamp=query_timestamp,
195
+ include=include_opts,
180
196
  )
181
197
 
182
- response = _run_async(self._api.recall_memories(bank_id, request_obj))
183
- return response.results if hasattr(response, 'results') else []
198
+ return _run_async(self._api.recall_memories(bank_id, request_obj))
184
199
 
185
200
  def reflect(
186
201
  self,
@@ -209,55 +224,6 @@ class Hindsight:
209
224
 
210
225
  return _run_async(self._api.reflect(bank_id, request_obj))
211
226
 
212
- # Full-featured methods (expose more options)
213
-
214
- def recall_memories(
215
- self,
216
- bank_id: str,
217
- query: str,
218
- types: Optional[List[str]] = None,
219
- budget: str = "mid",
220
- max_tokens: int = 4096,
221
- trace: bool = False,
222
- query_timestamp: Optional[str] = None,
223
- include_entities: bool = True,
224
- max_entity_tokens: int = 500,
225
- ) -> RecallResponse:
226
- """
227
- Recall memories with all options (full-featured).
228
-
229
- Args:
230
- bank_id: The memory bank ID
231
- query: Search query
232
- types: Optional list of fact types to filter (world, experience, opinion, observation)
233
- budget: Budget level - "low", "mid", or "high"
234
- max_tokens: Maximum tokens in results
235
- trace: Enable trace output
236
- query_timestamp: Optional ISO format date string (e.g., '2023-05-30T23:40:00')
237
- include_entities: Include entity observations in results (default: True)
238
- max_entity_tokens: Maximum tokens for entity observations (default: 500)
239
-
240
- Returns:
241
- RecallResponse with results, optional entities, and optional trace
242
- """
243
- from hindsight_client_api.models import include_options, entity_include_options
244
-
245
- include_opts = include_options.IncludeOptions(
246
- entities=entity_include_options.EntityIncludeOptions(max_tokens=max_entity_tokens) if include_entities else None
247
- )
248
-
249
- request_obj = recall_request.RecallRequest(
250
- query=query,
251
- types=types,
252
- budget=budget,
253
- max_tokens=max_tokens,
254
- trace=trace,
255
- query_timestamp=query_timestamp,
256
- include=include_opts,
257
- )
258
-
259
- return _run_async(self._api.recall_memories(bank_id, request_obj))
260
-
261
227
  def list_memories(
262
228
  self,
263
229
  bank_id: str,
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "hindsight-client"
3
- version = "0.1.0"
3
+ version = "0.1.1"
4
4
  description = "Python client for Hindsight - Semantic memory system with personality-driven thinking"
5
5
  authors = [
6
6
  {name = "Hindsight Team"}
@@ -92,32 +92,33 @@ class TestRecall:
92
92
 
93
93
  def test_recall_basic(self, client, bank_id):
94
94
  """Test basic memory search."""
95
- results = client.recall(
95
+ response = client.recall(
96
96
  bank_id=bank_id,
97
97
  query="What does Alice like?",
98
98
  )
99
99
 
100
- assert results is not None
101
- assert len(results) > 0
100
+ assert response is not None
101
+ assert response.results is not None
102
+ assert len(response.results) > 0
102
103
 
103
104
  # Check that at least one result contains relevant information
104
- result_texts = [r.text for r in results]
105
+ result_texts = [r.text for r in response.results]
105
106
  assert any("Alice" in text or "Python" in text or "programming" in text for text in result_texts)
106
107
 
107
108
  def test_recall_with_max_tokens(self, client, bank_id):
108
109
  """Test search with token limit."""
109
- results = client.recall(
110
+ response = client.recall(
110
111
  bank_id=bank_id,
111
112
  query="outdoor activities",
112
113
  max_tokens=1024,
113
114
  )
114
115
 
115
- assert results is not None
116
- assert isinstance(results, list)
116
+ assert response is not None
117
+ assert response.results is not None
117
118
 
118
- def test_recall_memories_full_featured(self, client, bank_id):
119
- """Test recall_memories with all features."""
120
- response = client.recall_memories(
119
+ def test_recall_full_featured(self, client, bank_id):
120
+ """Test recall with all features."""
121
+ response = client.recall(
121
122
  bank_id=bank_id,
122
123
  query="What are people's hobbies?",
123
124
  types=["world"],