hindsight-client 0.1.13__py3-none-any.whl → 0.1.15__py3-none-any.whl

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 (114) hide show
  1. hindsight_client/hindsight_client.py +11 -10
  2. {hindsight_client-0.1.13.dist-info → hindsight_client-0.1.15.dist-info}/METADATA +1 -1
  3. hindsight_client-0.1.15.dist-info/RECORD +109 -0
  4. hindsight_client_api/__init__.py +21 -3
  5. hindsight_client_api/api/__init__.py +5 -1
  6. hindsight_client_api/api/banks_api.py +2045 -0
  7. hindsight_client_api/api/documents_api.py +1234 -0
  8. hindsight_client_api/api/entities_api.py +921 -0
  9. hindsight_client_api/api/memory_api.py +1901 -0
  10. hindsight_client_api/api/monitoring_api.py +1 -1
  11. hindsight_client_api/api/operations_api.py +610 -0
  12. hindsight_client_api/api_client.py +1 -1
  13. hindsight_client_api/configuration.py +2 -2
  14. hindsight_client_api/docs/BankListItem.md +2 -2
  15. hindsight_client_api/docs/BankStatsResponse.md +39 -0
  16. hindsight_client_api/docs/BanksApi.md +517 -0
  17. hindsight_client_api/docs/CancelOperationResponse.md +32 -0
  18. hindsight_client_api/docs/DeleteDocumentResponse.md +33 -0
  19. hindsight_client_api/docs/DeleteResponse.md +2 -0
  20. hindsight_client_api/docs/DocumentsApi.md +313 -0
  21. hindsight_client_api/docs/EntitiesApi.md +230 -0
  22. hindsight_client_api/docs/MemoryApi.md +499 -0
  23. hindsight_client_api/docs/OperationResponse.md +36 -0
  24. hindsight_client_api/docs/OperationsApi.md +154 -0
  25. hindsight_client_api/docs/OperationsListResponse.md +31 -0
  26. hindsight_client_api/exceptions.py +1 -1
  27. hindsight_client_api/models/__init__.py +6 -1
  28. hindsight_client_api/models/add_background_request.py +1 -1
  29. hindsight_client_api/models/background_response.py +1 -1
  30. hindsight_client_api/models/bank_list_item.py +13 -3
  31. hindsight_client_api/models/bank_list_response.py +1 -1
  32. hindsight_client_api/models/bank_profile_response.py +1 -1
  33. hindsight_client_api/models/bank_stats_response.py +105 -0
  34. hindsight_client_api/models/budget.py +1 -1
  35. hindsight_client_api/models/cancel_operation_response.py +91 -0
  36. hindsight_client_api/models/chunk_data.py +1 -1
  37. hindsight_client_api/models/chunk_include_options.py +1 -1
  38. hindsight_client_api/models/chunk_response.py +1 -1
  39. hindsight_client_api/models/create_bank_request.py +1 -1
  40. hindsight_client_api/models/delete_document_response.py +93 -0
  41. hindsight_client_api/models/delete_response.py +19 -5
  42. hindsight_client_api/models/disposition_traits.py +1 -1
  43. hindsight_client_api/models/document_response.py +1 -1
  44. hindsight_client_api/models/entity_detail_response.py +1 -1
  45. hindsight_client_api/models/entity_include_options.py +1 -1
  46. hindsight_client_api/models/entity_list_item.py +1 -1
  47. hindsight_client_api/models/entity_list_response.py +1 -1
  48. hindsight_client_api/models/entity_observation_response.py +1 -1
  49. hindsight_client_api/models/entity_state_response.py +1 -1
  50. hindsight_client_api/models/graph_data_response.py +1 -1
  51. hindsight_client_api/models/http_validation_error.py +1 -1
  52. hindsight_client_api/models/include_options.py +1 -1
  53. hindsight_client_api/models/list_documents_response.py +1 -1
  54. hindsight_client_api/models/list_memory_units_response.py +1 -1
  55. hindsight_client_api/models/memory_item.py +1 -1
  56. hindsight_client_api/models/operation_response.py +109 -0
  57. hindsight_client_api/models/operations_list_response.py +97 -0
  58. hindsight_client_api/models/recall_request.py +1 -1
  59. hindsight_client_api/models/recall_response.py +1 -1
  60. hindsight_client_api/models/recall_result.py +1 -1
  61. hindsight_client_api/models/reflect_fact.py +1 -1
  62. hindsight_client_api/models/reflect_include_options.py +1 -1
  63. hindsight_client_api/models/reflect_request.py +1 -1
  64. hindsight_client_api/models/reflect_response.py +1 -1
  65. hindsight_client_api/models/retain_request.py +1 -1
  66. hindsight_client_api/models/retain_response.py +1 -1
  67. hindsight_client_api/models/update_disposition_request.py +1 -1
  68. hindsight_client_api/models/validation_error.py +1 -1
  69. hindsight_client_api/models/validation_error_loc_inner.py +1 -1
  70. hindsight_client_api/rest.py +1 -1
  71. hindsight_client-0.1.13.dist-info/RECORD +0 -131
  72. hindsight_client_api/api/default_api.py +0 -5976
  73. hindsight_client_api/docs/DefaultApi.md +0 -1568
  74. hindsight_client_api/test/__init__.py +0 -0
  75. hindsight_client_api/test/test_add_background_request.py +0 -53
  76. hindsight_client_api/test/test_background_response.py +0 -53
  77. hindsight_client_api/test/test_bank_list_item.py +0 -60
  78. hindsight_client_api/test/test_bank_list_response.py +0 -68
  79. hindsight_client_api/test/test_bank_profile_response.py +0 -58
  80. hindsight_client_api/test/test_budget.py +0 -33
  81. hindsight_client_api/test/test_chunk_data.py +0 -57
  82. hindsight_client_api/test/test_chunk_include_options.py +0 -51
  83. hindsight_client_api/test/test_chunk_response.py +0 -62
  84. hindsight_client_api/test/test_create_bank_request.py +0 -53
  85. hindsight_client_api/test/test_default_api.py +0 -178
  86. hindsight_client_api/test/test_delete_response.py +0 -52
  87. hindsight_client_api/test/test_disposition_traits.py +0 -56
  88. hindsight_client_api/test/test_document_response.py +0 -64
  89. hindsight_client_api/test/test_entity_detail_response.py +0 -71
  90. hindsight_client_api/test/test_entity_include_options.py +0 -51
  91. hindsight_client_api/test/test_entity_list_item.py +0 -61
  92. hindsight_client_api/test/test_entity_list_response.py +0 -56
  93. hindsight_client_api/test/test_entity_observation_response.py +0 -53
  94. hindsight_client_api/test/test_entity_state_response.py +0 -64
  95. hindsight_client_api/test/test_graph_data_response.py +0 -82
  96. hindsight_client_api/test/test_http_validation_error.py +0 -58
  97. hindsight_client_api/test/test_include_options.py +0 -54
  98. hindsight_client_api/test/test_list_documents_response.py +0 -66
  99. hindsight_client_api/test/test_list_memory_units_response.py +0 -66
  100. hindsight_client_api/test/test_memory_item.py +0 -58
  101. hindsight_client_api/test/test_monitoring_api.py +0 -45
  102. hindsight_client_api/test/test_recall_request.py +0 -64
  103. hindsight_client_api/test/test_recall_response.py +0 -76
  104. hindsight_client_api/test/test_recall_result.py +0 -67
  105. hindsight_client_api/test/test_reflect_fact.py +0 -57
  106. hindsight_client_api/test/test_reflect_include_options.py +0 -51
  107. hindsight_client_api/test/test_reflect_request.py +0 -56
  108. hindsight_client_api/test/test_reflect_response.py +0 -55
  109. hindsight_client_api/test/test_retain_request.py +0 -57
  110. hindsight_client_api/test/test_retain_response.py +0 -58
  111. hindsight_client_api/test/test_update_disposition_request.py +0 -52
  112. hindsight_client_api/test/test_validation_error.py +0 -60
  113. hindsight_client_api/test/test_validation_error_loc_inner.py +0 -50
  114. {hindsight_client-0.1.13.dist-info → hindsight_client-0.1.15.dist-info}/WHEEL +0 -0
@@ -1,64 +0,0 @@
1
- # coding: utf-8
2
-
3
- """
4
- Hindsight HTTP API
5
-
6
- HTTP API for Hindsight
7
-
8
- The version of the OpenAPI document: 1.0.0
9
- Generated by OpenAPI Generator (https://openapi-generator.tech)
10
-
11
- Do not edit the class manually.
12
- """ # noqa: E501
13
-
14
-
15
- import unittest
16
-
17
- from hindsight_client_api.models.entity_state_response import EntityStateResponse
18
-
19
- class TestEntityStateResponse(unittest.TestCase):
20
- """EntityStateResponse unit test stubs"""
21
-
22
- def setUp(self):
23
- pass
24
-
25
- def tearDown(self):
26
- pass
27
-
28
- def make_instance(self, include_optional) -> EntityStateResponse:
29
- """Test EntityStateResponse
30
- include_optional is a boolean, when False only required
31
- params are included, when True both required and
32
- optional params are included """
33
- # uncomment below to create an instance of `EntityStateResponse`
34
- """
35
- model = EntityStateResponse()
36
- if include_optional:
37
- return EntityStateResponse(
38
- entity_id = '',
39
- canonical_name = '',
40
- observations = [
41
- hindsight_client_api.models.entity_observation_response.EntityObservationResponse(
42
- text = '',
43
- mentioned_at = '', )
44
- ]
45
- )
46
- else:
47
- return EntityStateResponse(
48
- entity_id = '',
49
- canonical_name = '',
50
- observations = [
51
- hindsight_client_api.models.entity_observation_response.EntityObservationResponse(
52
- text = '',
53
- mentioned_at = '', )
54
- ],
55
- )
56
- """
57
-
58
- def testEntityStateResponse(self):
59
- """Test EntityStateResponse"""
60
- # inst_req_only = self.make_instance(include_optional=False)
61
- # inst_req_and_optional = self.make_instance(include_optional=True)
62
-
63
- if __name__ == '__main__':
64
- unittest.main()
@@ -1,82 +0,0 @@
1
- # coding: utf-8
2
-
3
- """
4
- Hindsight HTTP API
5
-
6
- HTTP API for Hindsight
7
-
8
- The version of the OpenAPI document: 1.0.0
9
- Generated by OpenAPI Generator (https://openapi-generator.tech)
10
-
11
- Do not edit the class manually.
12
- """ # noqa: E501
13
-
14
-
15
- import unittest
16
-
17
- from hindsight_client_api.models.graph_data_response import GraphDataResponse
18
-
19
- class TestGraphDataResponse(unittest.TestCase):
20
- """GraphDataResponse unit test stubs"""
21
-
22
- def setUp(self):
23
- pass
24
-
25
- def tearDown(self):
26
- pass
27
-
28
- def make_instance(self, include_optional) -> GraphDataResponse:
29
- """Test GraphDataResponse
30
- include_optional is a boolean, when False only required
31
- params are included, when True both required and
32
- optional params are included """
33
- # uncomment below to create an instance of `GraphDataResponse`
34
- """
35
- model = GraphDataResponse()
36
- if include_optional:
37
- return GraphDataResponse(
38
- nodes = [
39
- {
40
- 'key' : null
41
- }
42
- ],
43
- edges = [
44
- {
45
- 'key' : null
46
- }
47
- ],
48
- table_rows = [
49
- {
50
- 'key' : null
51
- }
52
- ],
53
- total_units = 56
54
- )
55
- else:
56
- return GraphDataResponse(
57
- nodes = [
58
- {
59
- 'key' : null
60
- }
61
- ],
62
- edges = [
63
- {
64
- 'key' : null
65
- }
66
- ],
67
- table_rows = [
68
- {
69
- 'key' : null
70
- }
71
- ],
72
- total_units = 56,
73
- )
74
- """
75
-
76
- def testGraphDataResponse(self):
77
- """Test GraphDataResponse"""
78
- # inst_req_only = self.make_instance(include_optional=False)
79
- # inst_req_and_optional = self.make_instance(include_optional=True)
80
-
81
- if __name__ == '__main__':
82
- unittest.main()
@@ -1,58 +0,0 @@
1
- # coding: utf-8
2
-
3
- """
4
- Hindsight HTTP API
5
-
6
- HTTP API for Hindsight
7
-
8
- The version of the OpenAPI document: 1.0.0
9
- Generated by OpenAPI Generator (https://openapi-generator.tech)
10
-
11
- Do not edit the class manually.
12
- """ # noqa: E501
13
-
14
-
15
- import unittest
16
-
17
- from hindsight_client_api.models.http_validation_error import HTTPValidationError
18
-
19
- class TestHTTPValidationError(unittest.TestCase):
20
- """HTTPValidationError unit test stubs"""
21
-
22
- def setUp(self):
23
- pass
24
-
25
- def tearDown(self):
26
- pass
27
-
28
- def make_instance(self, include_optional) -> HTTPValidationError:
29
- """Test HTTPValidationError
30
- include_optional is a boolean, when False only required
31
- params are included, when True both required and
32
- optional params are included """
33
- # uncomment below to create an instance of `HTTPValidationError`
34
- """
35
- model = HTTPValidationError()
36
- if include_optional:
37
- return HTTPValidationError(
38
- detail = [
39
- hindsight_client_api.models.validation_error.ValidationError(
40
- loc = [
41
- null
42
- ],
43
- msg = '',
44
- type = '', )
45
- ]
46
- )
47
- else:
48
- return HTTPValidationError(
49
- )
50
- """
51
-
52
- def testHTTPValidationError(self):
53
- """Test HTTPValidationError"""
54
- # inst_req_only = self.make_instance(include_optional=False)
55
- # inst_req_and_optional = self.make_instance(include_optional=True)
56
-
57
- if __name__ == '__main__':
58
- unittest.main()
@@ -1,54 +0,0 @@
1
- # coding: utf-8
2
-
3
- """
4
- Hindsight HTTP API
5
-
6
- HTTP API for Hindsight
7
-
8
- The version of the OpenAPI document: 1.0.0
9
- Generated by OpenAPI Generator (https://openapi-generator.tech)
10
-
11
- Do not edit the class manually.
12
- """ # noqa: E501
13
-
14
-
15
- import unittest
16
-
17
- from hindsight_client_api.models.include_options import IncludeOptions
18
-
19
- class TestIncludeOptions(unittest.TestCase):
20
- """IncludeOptions unit test stubs"""
21
-
22
- def setUp(self):
23
- pass
24
-
25
- def tearDown(self):
26
- pass
27
-
28
- def make_instance(self, include_optional) -> IncludeOptions:
29
- """Test IncludeOptions
30
- include_optional is a boolean, when False only required
31
- params are included, when True both required and
32
- optional params are included """
33
- # uncomment below to create an instance of `IncludeOptions`
34
- """
35
- model = IncludeOptions()
36
- if include_optional:
37
- return IncludeOptions(
38
- entities = hindsight_client_api.models.entity_include_options.EntityIncludeOptions(
39
- max_tokens = 56, ),
40
- chunks = hindsight_client_api.models.chunk_include_options.ChunkIncludeOptions(
41
- max_tokens = 56, )
42
- )
43
- else:
44
- return IncludeOptions(
45
- )
46
- """
47
-
48
- def testIncludeOptions(self):
49
- """Test IncludeOptions"""
50
- # inst_req_only = self.make_instance(include_optional=False)
51
- # inst_req_and_optional = self.make_instance(include_optional=True)
52
-
53
- if __name__ == '__main__':
54
- unittest.main()
@@ -1,66 +0,0 @@
1
- # coding: utf-8
2
-
3
- """
4
- Hindsight HTTP API
5
-
6
- HTTP API for Hindsight
7
-
8
- The version of the OpenAPI document: 1.0.0
9
- Generated by OpenAPI Generator (https://openapi-generator.tech)
10
-
11
- Do not edit the class manually.
12
- """ # noqa: E501
13
-
14
-
15
- import unittest
16
-
17
- from hindsight_client_api.models.list_documents_response import ListDocumentsResponse
18
-
19
- class TestListDocumentsResponse(unittest.TestCase):
20
- """ListDocumentsResponse unit test stubs"""
21
-
22
- def setUp(self):
23
- pass
24
-
25
- def tearDown(self):
26
- pass
27
-
28
- def make_instance(self, include_optional) -> ListDocumentsResponse:
29
- """Test ListDocumentsResponse
30
- include_optional is a boolean, when False only required
31
- params are included, when True both required and
32
- optional params are included """
33
- # uncomment below to create an instance of `ListDocumentsResponse`
34
- """
35
- model = ListDocumentsResponse()
36
- if include_optional:
37
- return ListDocumentsResponse(
38
- items = [
39
- {
40
- 'key' : null
41
- }
42
- ],
43
- total = 56,
44
- limit = 56,
45
- offset = 56
46
- )
47
- else:
48
- return ListDocumentsResponse(
49
- items = [
50
- {
51
- 'key' : null
52
- }
53
- ],
54
- total = 56,
55
- limit = 56,
56
- offset = 56,
57
- )
58
- """
59
-
60
- def testListDocumentsResponse(self):
61
- """Test ListDocumentsResponse"""
62
- # inst_req_only = self.make_instance(include_optional=False)
63
- # inst_req_and_optional = self.make_instance(include_optional=True)
64
-
65
- if __name__ == '__main__':
66
- unittest.main()
@@ -1,66 +0,0 @@
1
- # coding: utf-8
2
-
3
- """
4
- Hindsight HTTP API
5
-
6
- HTTP API for Hindsight
7
-
8
- The version of the OpenAPI document: 1.0.0
9
- Generated by OpenAPI Generator (https://openapi-generator.tech)
10
-
11
- Do not edit the class manually.
12
- """ # noqa: E501
13
-
14
-
15
- import unittest
16
-
17
- from hindsight_client_api.models.list_memory_units_response import ListMemoryUnitsResponse
18
-
19
- class TestListMemoryUnitsResponse(unittest.TestCase):
20
- """ListMemoryUnitsResponse unit test stubs"""
21
-
22
- def setUp(self):
23
- pass
24
-
25
- def tearDown(self):
26
- pass
27
-
28
- def make_instance(self, include_optional) -> ListMemoryUnitsResponse:
29
- """Test ListMemoryUnitsResponse
30
- include_optional is a boolean, when False only required
31
- params are included, when True both required and
32
- optional params are included """
33
- # uncomment below to create an instance of `ListMemoryUnitsResponse`
34
- """
35
- model = ListMemoryUnitsResponse()
36
- if include_optional:
37
- return ListMemoryUnitsResponse(
38
- items = [
39
- {
40
- 'key' : null
41
- }
42
- ],
43
- total = 56,
44
- limit = 56,
45
- offset = 56
46
- )
47
- else:
48
- return ListMemoryUnitsResponse(
49
- items = [
50
- {
51
- 'key' : null
52
- }
53
- ],
54
- total = 56,
55
- limit = 56,
56
- offset = 56,
57
- )
58
- """
59
-
60
- def testListMemoryUnitsResponse(self):
61
- """Test ListMemoryUnitsResponse"""
62
- # inst_req_only = self.make_instance(include_optional=False)
63
- # inst_req_and_optional = self.make_instance(include_optional=True)
64
-
65
- if __name__ == '__main__':
66
- unittest.main()
@@ -1,58 +0,0 @@
1
- # coding: utf-8
2
-
3
- """
4
- Hindsight HTTP API
5
-
6
- HTTP API for Hindsight
7
-
8
- The version of the OpenAPI document: 1.0.0
9
- Generated by OpenAPI Generator (https://openapi-generator.tech)
10
-
11
- Do not edit the class manually.
12
- """ # noqa: E501
13
-
14
-
15
- import unittest
16
-
17
- from hindsight_client_api.models.memory_item import MemoryItem
18
-
19
- class TestMemoryItem(unittest.TestCase):
20
- """MemoryItem unit test stubs"""
21
-
22
- def setUp(self):
23
- pass
24
-
25
- def tearDown(self):
26
- pass
27
-
28
- def make_instance(self, include_optional) -> MemoryItem:
29
- """Test MemoryItem
30
- include_optional is a boolean, when False only required
31
- params are included, when True both required and
32
- optional params are included """
33
- # uncomment below to create an instance of `MemoryItem`
34
- """
35
- model = MemoryItem()
36
- if include_optional:
37
- return MemoryItem(
38
- content = '',
39
- timestamp = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'),
40
- context = '',
41
- metadata = {
42
- 'key' : ''
43
- },
44
- document_id = ''
45
- )
46
- else:
47
- return MemoryItem(
48
- content = '',
49
- )
50
- """
51
-
52
- def testMemoryItem(self):
53
- """Test MemoryItem"""
54
- # inst_req_only = self.make_instance(include_optional=False)
55
- # inst_req_and_optional = self.make_instance(include_optional=True)
56
-
57
- if __name__ == '__main__':
58
- unittest.main()
@@ -1,45 +0,0 @@
1
- # coding: utf-8
2
-
3
- """
4
- Hindsight HTTP API
5
-
6
- HTTP API for Hindsight
7
-
8
- The version of the OpenAPI document: 1.0.0
9
- Generated by OpenAPI Generator (https://openapi-generator.tech)
10
-
11
- Do not edit the class manually.
12
- """ # noqa: E501
13
-
14
-
15
- import unittest
16
-
17
- from hindsight_client_api.api.monitoring_api import MonitoringApi
18
-
19
-
20
- class TestMonitoringApi(unittest.IsolatedAsyncioTestCase):
21
- """MonitoringApi unit test stubs"""
22
-
23
- async def asyncSetUp(self) -> None:
24
- self.api = MonitoringApi()
25
-
26
- async def asyncTearDown(self) -> None:
27
- await self.api.api_client.close()
28
-
29
- async def test_health_endpoint_health_get(self) -> None:
30
- """Test case for health_endpoint_health_get
31
-
32
- Health check endpoint
33
- """
34
- pass
35
-
36
- async def test_metrics_endpoint_metrics_get(self) -> None:
37
- """Test case for metrics_endpoint_metrics_get
38
-
39
- Prometheus metrics endpoint
40
- """
41
- pass
42
-
43
-
44
- if __name__ == '__main__':
45
- unittest.main()
@@ -1,64 +0,0 @@
1
- # coding: utf-8
2
-
3
- """
4
- Hindsight HTTP API
5
-
6
- HTTP API for Hindsight
7
-
8
- The version of the OpenAPI document: 1.0.0
9
- Generated by OpenAPI Generator (https://openapi-generator.tech)
10
-
11
- Do not edit the class manually.
12
- """ # noqa: E501
13
-
14
-
15
- import unittest
16
-
17
- from hindsight_client_api.models.recall_request import RecallRequest
18
-
19
- class TestRecallRequest(unittest.TestCase):
20
- """RecallRequest unit test stubs"""
21
-
22
- def setUp(self):
23
- pass
24
-
25
- def tearDown(self):
26
- pass
27
-
28
- def make_instance(self, include_optional) -> RecallRequest:
29
- """Test RecallRequest
30
- include_optional is a boolean, when False only required
31
- params are included, when True both required and
32
- optional params are included """
33
- # uncomment below to create an instance of `RecallRequest`
34
- """
35
- model = RecallRequest()
36
- if include_optional:
37
- return RecallRequest(
38
- query = '',
39
- types = [
40
- ''
41
- ],
42
- budget = 'low',
43
- max_tokens = 56,
44
- trace = True,
45
- query_timestamp = '',
46
- include = hindsight_client_api.models.include_options.IncludeOptions(
47
- entities = hindsight_client_api.models.entity_include_options.EntityIncludeOptions(
48
- max_tokens = 56, ),
49
- chunks = hindsight_client_api.models.chunk_include_options.ChunkIncludeOptions(
50
- max_tokens = 56, ), )
51
- )
52
- else:
53
- return RecallRequest(
54
- query = '',
55
- )
56
- """
57
-
58
- def testRecallRequest(self):
59
- """Test RecallRequest"""
60
- # inst_req_only = self.make_instance(include_optional=False)
61
- # inst_req_and_optional = self.make_instance(include_optional=True)
62
-
63
- if __name__ == '__main__':
64
- unittest.main()
@@ -1,76 +0,0 @@
1
- # coding: utf-8
2
-
3
- """
4
- Hindsight HTTP API
5
-
6
- HTTP API for Hindsight
7
-
8
- The version of the OpenAPI document: 1.0.0
9
- Generated by OpenAPI Generator (https://openapi-generator.tech)
10
-
11
- Do not edit the class manually.
12
- """ # noqa: E501
13
-
14
-
15
- import unittest
16
-
17
- from hindsight_client_api.models.recall_response import RecallResponse
18
-
19
- class TestRecallResponse(unittest.TestCase):
20
- """RecallResponse unit test stubs"""
21
-
22
- def setUp(self):
23
- pass
24
-
25
- def tearDown(self):
26
- pass
27
-
28
- def make_instance(self, include_optional) -> RecallResponse:
29
- """Test RecallResponse
30
- include_optional is a boolean, when False only required
31
- params are included, when True both required and
32
- optional params are included """
33
- # uncomment below to create an instance of `RecallResponse`
34
- """
35
- model = RecallResponse()
36
- if include_optional:
37
- return RecallResponse(
38
- results = [
39
- {chunk_id=456e7890-e12b-34d5-a678-901234567890, context=work info, document_id=session_abc123, entities=[Alice, Google], id=123e4567-e89b-12d3-a456-426614174000, mentioned_at=2024-01-15T10:30:00Z, metadata={source=slack}, occurred_end=2024-01-15T10:30:00Z, occurred_start=2024-01-15T10:30:00Z, text=Alice works at Google on the AI team, type=world}
40
- ],
41
- trace = {
42
- 'key' : null
43
- },
44
- entities = {
45
- 'key' : hindsight_client_api.models.entity_state_response.EntityStateResponse(
46
- entity_id = '',
47
- canonical_name = '',
48
- observations = [
49
- hindsight_client_api.models.entity_observation_response.EntityObservationResponse(
50
- text = '',
51
- mentioned_at = '', )
52
- ], )
53
- },
54
- chunks = {
55
- 'key' : hindsight_client_api.models.chunk_data.ChunkData(
56
- id = '',
57
- text = '',
58
- chunk_index = 56,
59
- truncated = True, )
60
- }
61
- )
62
- else:
63
- return RecallResponse(
64
- results = [
65
- {chunk_id=456e7890-e12b-34d5-a678-901234567890, context=work info, document_id=session_abc123, entities=[Alice, Google], id=123e4567-e89b-12d3-a456-426614174000, mentioned_at=2024-01-15T10:30:00Z, metadata={source=slack}, occurred_end=2024-01-15T10:30:00Z, occurred_start=2024-01-15T10:30:00Z, text=Alice works at Google on the AI team, type=world}
66
- ],
67
- )
68
- """
69
-
70
- def testRecallResponse(self):
71
- """Test RecallResponse"""
72
- # inst_req_only = self.make_instance(include_optional=False)
73
- # inst_req_and_optional = self.make_instance(include_optional=True)
74
-
75
- if __name__ == '__main__':
76
- unittest.main()