casedev 0.1.0__py3-none-any.whl → 0.2.0__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 (135) hide show
  1. casedev/_base_client.py +140 -11
  2. casedev/_client.py +290 -119
  3. casedev/_models.py +16 -1
  4. casedev/_types.py +12 -2
  5. casedev/_version.py +1 -1
  6. casedev/resources/__init__.py +0 -70
  7. casedev/resources/compute/v1/__init__.py +0 -42
  8. casedev/resources/compute/v1/environments.py +16 -19
  9. casedev/resources/compute/v1/secrets.py +35 -41
  10. casedev/resources/compute/v1/v1.py +14 -276
  11. casedev/resources/format/v1/templates.py +11 -13
  12. casedev/resources/llm/llm.py +6 -7
  13. casedev/resources/llm/v1/v1.py +11 -13
  14. casedev/resources/ocr/v1.py +10 -13
  15. casedev/resources/search/v1.py +6 -7
  16. casedev/resources/vault/graphrag.py +11 -13
  17. casedev/resources/vault/objects.py +20 -25
  18. casedev/resources/vault/vault.py +64 -28
  19. casedev/resources/voice/streaming.py +6 -7
  20. casedev/resources/voice/transcription.py +82 -23
  21. casedev/resources/voice/v1/speak.py +1 -195
  22. casedev/resources/voice/v1/v1.py +6 -7
  23. casedev/types/__init__.py +2 -0
  24. casedev/types/compute/__init__.py +1 -2
  25. casedev/types/compute/v1/__init__.py +7 -5
  26. casedev/types/compute/v1/environment_list_response.py +40 -0
  27. casedev/types/compute/v1/environment_retrieve_response.py +36 -0
  28. casedev/types/compute/v1/environment_set_default_response.py +36 -0
  29. casedev/types/{convert/v1_webhook_response.py → compute/v1/secret_delete_group_response.py} +3 -3
  30. casedev/types/compute/v1/secret_list_response.py +31 -0
  31. casedev/types/compute/v1/secret_retrieve_group_response.py +38 -0
  32. casedev/types/compute/v1/secret_update_group_response.py +22 -0
  33. casedev/types/compute/v1_get_usage_response.py +51 -0
  34. casedev/types/format/v1/__init__.py +2 -0
  35. casedev/types/format/v1/template_list_response.py +40 -0
  36. casedev/types/format/v1/template_retrieve_response.py +33 -0
  37. casedev/types/llm/__init__.py +2 -0
  38. casedev/types/llm/v1_create_embedding_response.py +31 -0
  39. casedev/types/llm/v1_list_models_response.py +41 -0
  40. casedev/types/llm_get_config_response.py +33 -0
  41. casedev/types/ocr/__init__.py +2 -0
  42. casedev/types/ocr/v1_download_response.py +7 -0
  43. casedev/types/ocr/v1_process_params.py +2 -0
  44. casedev/types/ocr/v1_retrieve_response.py +32 -0
  45. casedev/types/search/__init__.py +1 -0
  46. casedev/types/search/v1_retrieve_research_response.py +74 -0
  47. casedev/types/vault/__init__.py +6 -0
  48. casedev/types/vault/graphrag_get_stats_response.py +31 -0
  49. casedev/types/{workflows/v1_delete_response.py → vault/graphrag_init_response.py} +6 -2
  50. casedev/types/vault/object_download_response.py +7 -0
  51. casedev/types/vault/object_get_text_response.py +35 -0
  52. casedev/types/vault/object_list_response.py +64 -0
  53. casedev/types/vault/object_retrieve_response.py +57 -0
  54. casedev/types/vault_create_params.py +16 -1
  55. casedev/types/vault_create_response.py +5 -2
  56. casedev/types/vault_ingest_response.py +15 -5
  57. casedev/types/vault_retrieve_response.py +76 -0
  58. casedev/types/vault_search_params.py +19 -5
  59. casedev/types/vault_search_response.py +25 -1
  60. casedev/types/vault_upload_params.py +7 -0
  61. casedev/types/vault_upload_response.py +6 -0
  62. casedev/types/voice/__init__.py +3 -0
  63. casedev/types/voice/streaming_get_url_response.py +44 -0
  64. casedev/types/voice/transcription_create_params.py +23 -3
  65. casedev/types/voice/transcription_create_response.py +22 -0
  66. casedev/types/voice/transcription_retrieve_response.py +19 -17
  67. casedev/types/voice/v1/__init__.py +0 -1
  68. casedev/types/voice/v1/speak_create_params.py +2 -0
  69. casedev/types/voice/v1_list_voices_response.py +40 -0
  70. casedev/types/webhooks/__init__.py +0 -3
  71. {casedev-0.1.0.dist-info → casedev-0.2.0.dist-info}/METADATA +10 -13
  72. casedev-0.2.0.dist-info/RECORD +149 -0
  73. {casedev-0.1.0.dist-info → casedev-0.2.0.dist-info}/licenses/LICENSE +1 -1
  74. casedev/resources/actions/__init__.py +0 -33
  75. casedev/resources/actions/actions.py +0 -102
  76. casedev/resources/actions/v1.py +0 -640
  77. casedev/resources/compute/v1/functions.py +0 -278
  78. casedev/resources/compute/v1/invoke.py +0 -216
  79. casedev/resources/compute/v1/runs.py +0 -290
  80. casedev/resources/convert/__init__.py +0 -33
  81. casedev/resources/convert/convert.py +0 -102
  82. casedev/resources/convert/v1/__init__.py +0 -33
  83. casedev/resources/convert/v1/jobs.py +0 -254
  84. casedev/resources/convert/v1/v1.py +0 -450
  85. casedev/resources/templates/__init__.py +0 -33
  86. casedev/resources/templates/templates.py +0 -102
  87. casedev/resources/templates/v1.py +0 -633
  88. casedev/resources/webhooks/__init__.py +0 -33
  89. casedev/resources/webhooks/v1.py +0 -447
  90. casedev/resources/webhooks/webhooks.py +0 -102
  91. casedev/resources/workflows/__init__.py +0 -33
  92. casedev/resources/workflows/v1.py +0 -1053
  93. casedev/resources/workflows/workflows.py +0 -102
  94. casedev/types/actions/__init__.py +0 -8
  95. casedev/types/actions/v1_create_params.py +0 -22
  96. casedev/types/actions/v1_create_response.py +0 -33
  97. casedev/types/actions/v1_execute_params.py +0 -16
  98. casedev/types/actions/v1_execute_response.py +0 -31
  99. casedev/types/compute/v1/function_get_logs_params.py +0 -12
  100. casedev/types/compute/v1/function_list_params.py +0 -12
  101. casedev/types/compute/v1/invoke_run_params.py +0 -21
  102. casedev/types/compute/v1/invoke_run_response.py +0 -39
  103. casedev/types/compute/v1/run_list_params.py +0 -18
  104. casedev/types/compute/v1_deploy_params.py +0 -114
  105. casedev/types/compute/v1_deploy_response.py +0 -30
  106. casedev/types/convert/__init__.py +0 -8
  107. casedev/types/convert/v1/__init__.py +0 -3
  108. casedev/types/convert/v1_process_params.py +0 -15
  109. casedev/types/convert/v1_process_response.py +0 -19
  110. casedev/types/convert/v1_webhook_params.py +0 -32
  111. casedev/types/templates/__init__.py +0 -8
  112. casedev/types/templates/v1_execute_params.py +0 -22
  113. casedev/types/templates/v1_execute_response.py +0 -31
  114. casedev/types/templates/v1_list_params.py +0 -32
  115. casedev/types/templates/v1_search_params.py +0 -18
  116. casedev/types/voice/v1/speak_stream_params.py +0 -58
  117. casedev/types/webhooks/v1_create_params.py +0 -20
  118. casedev/types/webhooks/v1_create_response.py +0 -33
  119. casedev/types/workflows/__init__.py +0 -19
  120. casedev/types/workflows/v1_create_params.py +0 -32
  121. casedev/types/workflows/v1_create_response.py +0 -29
  122. casedev/types/workflows/v1_deploy_response.py +0 -20
  123. casedev/types/workflows/v1_execute_params.py +0 -12
  124. casedev/types/workflows/v1_execute_response.py +0 -22
  125. casedev/types/workflows/v1_list_executions_params.py +0 -13
  126. casedev/types/workflows/v1_list_executions_response.py +0 -27
  127. casedev/types/workflows/v1_list_params.py +0 -18
  128. casedev/types/workflows/v1_list_response.py +0 -37
  129. casedev/types/workflows/v1_retrieve_execution_response.py +0 -31
  130. casedev/types/workflows/v1_retrieve_response.py +0 -35
  131. casedev/types/workflows/v1_undeploy_response.py +0 -13
  132. casedev/types/workflows/v1_update_params.py +0 -26
  133. casedev/types/workflows/v1_update_response.py +0 -17
  134. casedev-0.1.0.dist-info/RECORD +0 -185
  135. {casedev-0.1.0.dist-info → casedev-0.2.0.dist-info}/WHEEL +0 -0
@@ -15,7 +15,7 @@ from .chat import (
15
15
  ChatResourceWithStreamingResponse,
16
16
  AsyncChatResourceWithStreamingResponse,
17
17
  )
18
- from ...._types import Body, Omit, Query, Headers, NoneType, NotGiven, SequenceNotStr, omit, not_given
18
+ from ...._types import Body, Omit, Query, Headers, NotGiven, SequenceNotStr, omit, not_given
19
19
  from ...._utils import maybe_transform, async_maybe_transform
20
20
  from ...._compat import cached_property
21
21
  from ...._resource import SyncAPIResource, AsyncAPIResource
@@ -27,6 +27,8 @@ from ...._response import (
27
27
  )
28
28
  from ....types.llm import v1_create_embedding_params
29
29
  from ...._base_client import make_request_options
30
+ from ....types.llm.v1_list_models_response import V1ListModelsResponse
31
+ from ....types.llm.v1_create_embedding_response import V1CreateEmbeddingResponse
30
32
 
31
33
  __all__ = ["V1Resource", "AsyncV1Resource"]
32
34
 
@@ -69,7 +71,7 @@ class V1Resource(SyncAPIResource):
69
71
  extra_query: Query | None = None,
70
72
  extra_body: Body | None = None,
71
73
  timeout: float | httpx.Timeout | None | NotGiven = not_given,
72
- ) -> None:
74
+ ) -> V1CreateEmbeddingResponse:
73
75
  """Create vector embeddings from text using OpenAI-compatible models.
74
76
 
75
77
  Perfect for
@@ -95,7 +97,6 @@ class V1Resource(SyncAPIResource):
95
97
 
96
98
  timeout: Override the client-level default timeout for this request, in seconds
97
99
  """
98
- extra_headers = {"Accept": "*/*", **(extra_headers or {})}
99
100
  return self._post(
100
101
  "/llm/v1/embeddings",
101
102
  body=maybe_transform(
@@ -111,7 +112,7 @@ class V1Resource(SyncAPIResource):
111
112
  options=make_request_options(
112
113
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
113
114
  ),
114
- cast_to=NoneType,
115
+ cast_to=V1CreateEmbeddingResponse,
115
116
  )
116
117
 
117
118
  def list_models(
@@ -123,7 +124,7 @@ class V1Resource(SyncAPIResource):
123
124
  extra_query: Query | None = None,
124
125
  extra_body: Body | None = None,
125
126
  timeout: float | httpx.Timeout | None | NotGiven = not_given,
126
- ) -> None:
127
+ ) -> V1ListModelsResponse:
127
128
  """
128
129
  Retrieve a list of all available language models from 40+ providers including
129
130
  OpenAI, Anthropic, Google, and Case.dev's specialized legal models. Returns
@@ -132,13 +133,12 @@ class V1Resource(SyncAPIResource):
132
133
  This endpoint is compatible with OpenAI's models API format, making it easy to
133
134
  integrate with existing applications.
134
135
  """
135
- extra_headers = {"Accept": "*/*", **(extra_headers or {})}
136
136
  return self._get(
137
137
  "/llm/v1/models",
138
138
  options=make_request_options(
139
139
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
140
140
  ),
141
- cast_to=NoneType,
141
+ cast_to=V1ListModelsResponse,
142
142
  )
143
143
 
144
144
 
@@ -180,7 +180,7 @@ class AsyncV1Resource(AsyncAPIResource):
180
180
  extra_query: Query | None = None,
181
181
  extra_body: Body | None = None,
182
182
  timeout: float | httpx.Timeout | None | NotGiven = not_given,
183
- ) -> None:
183
+ ) -> V1CreateEmbeddingResponse:
184
184
  """Create vector embeddings from text using OpenAI-compatible models.
185
185
 
186
186
  Perfect for
@@ -206,7 +206,6 @@ class AsyncV1Resource(AsyncAPIResource):
206
206
 
207
207
  timeout: Override the client-level default timeout for this request, in seconds
208
208
  """
209
- extra_headers = {"Accept": "*/*", **(extra_headers or {})}
210
209
  return await self._post(
211
210
  "/llm/v1/embeddings",
212
211
  body=await async_maybe_transform(
@@ -222,7 +221,7 @@ class AsyncV1Resource(AsyncAPIResource):
222
221
  options=make_request_options(
223
222
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
224
223
  ),
225
- cast_to=NoneType,
224
+ cast_to=V1CreateEmbeddingResponse,
226
225
  )
227
226
 
228
227
  async def list_models(
@@ -234,7 +233,7 @@ class AsyncV1Resource(AsyncAPIResource):
234
233
  extra_query: Query | None = None,
235
234
  extra_body: Body | None = None,
236
235
  timeout: float | httpx.Timeout | None | NotGiven = not_given,
237
- ) -> None:
236
+ ) -> V1ListModelsResponse:
238
237
  """
239
238
  Retrieve a list of all available language models from 40+ providers including
240
239
  OpenAI, Anthropic, Google, and Case.dev's specialized legal models. Returns
@@ -243,13 +242,12 @@ class AsyncV1Resource(AsyncAPIResource):
243
242
  This endpoint is compatible with OpenAI's models API format, making it easy to
244
243
  integrate with existing applications.
245
244
  """
246
- extra_headers = {"Accept": "*/*", **(extra_headers or {})}
247
245
  return await self._get(
248
246
  "/llm/v1/models",
249
247
  options=make_request_options(
250
248
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
251
249
  ),
252
- cast_to=NoneType,
250
+ cast_to=V1ListModelsResponse,
253
251
  )
254
252
 
255
253
 
@@ -6,7 +6,7 @@ from typing_extensions import Literal
6
6
 
7
7
  import httpx
8
8
 
9
- from ..._types import Body, Omit, Query, Headers, NoneType, NotGiven, omit, not_given
9
+ from ..._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
10
10
  from ..._utils import maybe_transform, async_maybe_transform
11
11
  from ..._compat import cached_property
12
12
  from ..._resource import SyncAPIResource, AsyncAPIResource
@@ -19,6 +19,7 @@ from ..._response import (
19
19
  from ...types.ocr import v1_process_params
20
20
  from ..._base_client import make_request_options
21
21
  from ...types.ocr.v1_process_response import V1ProcessResponse
22
+ from ...types.ocr.v1_retrieve_response import V1RetrieveResponse
22
23
 
23
24
  __all__ = ["V1Resource", "AsyncV1Resource"]
24
25
 
@@ -53,7 +54,7 @@ class V1Resource(SyncAPIResource):
53
54
  extra_query: Query | None = None,
54
55
  extra_body: Body | None = None,
55
56
  timeout: float | httpx.Timeout | None | NotGiven = not_given,
56
- ) -> None:
57
+ ) -> V1RetrieveResponse:
57
58
  """Retrieve the status and results of an OCR job.
58
59
 
59
60
  Returns job progress, extracted
@@ -70,13 +71,12 @@ class V1Resource(SyncAPIResource):
70
71
  """
71
72
  if not id:
72
73
  raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
73
- extra_headers = {"Accept": "*/*", **(extra_headers or {})}
74
74
  return self._get(
75
75
  f"/ocr/v1/{id}",
76
76
  options=make_request_options(
77
77
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
78
78
  ),
79
- cast_to=NoneType,
79
+ cast_to=V1RetrieveResponse,
80
80
  )
81
81
 
82
82
  def download(
@@ -90,7 +90,7 @@ class V1Resource(SyncAPIResource):
90
90
  extra_query: Query | None = None,
91
91
  extra_body: Body | None = None,
92
92
  timeout: float | httpx.Timeout | None | NotGiven = not_given,
93
- ) -> None:
93
+ ) -> str:
94
94
  """Download OCR processing results in various formats.
95
95
 
96
96
  Returns the processed
@@ -110,13 +110,12 @@ class V1Resource(SyncAPIResource):
110
110
  raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
111
111
  if not type:
112
112
  raise ValueError(f"Expected a non-empty value for `type` but received {type!r}")
113
- extra_headers = {"Accept": "*/*", **(extra_headers or {})}
114
113
  return self._get(
115
114
  f"/ocr/v1/{id}/download/{type}",
116
115
  options=make_request_options(
117
116
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
118
117
  ),
119
- cast_to=NoneType,
118
+ cast_to=str,
120
119
  )
121
120
 
122
121
  def process(
@@ -215,7 +214,7 @@ class AsyncV1Resource(AsyncAPIResource):
215
214
  extra_query: Query | None = None,
216
215
  extra_body: Body | None = None,
217
216
  timeout: float | httpx.Timeout | None | NotGiven = not_given,
218
- ) -> None:
217
+ ) -> V1RetrieveResponse:
219
218
  """Retrieve the status and results of an OCR job.
220
219
 
221
220
  Returns job progress, extracted
@@ -232,13 +231,12 @@ class AsyncV1Resource(AsyncAPIResource):
232
231
  """
233
232
  if not id:
234
233
  raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
235
- extra_headers = {"Accept": "*/*", **(extra_headers or {})}
236
234
  return await self._get(
237
235
  f"/ocr/v1/{id}",
238
236
  options=make_request_options(
239
237
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
240
238
  ),
241
- cast_to=NoneType,
239
+ cast_to=V1RetrieveResponse,
242
240
  )
243
241
 
244
242
  async def download(
@@ -252,7 +250,7 @@ class AsyncV1Resource(AsyncAPIResource):
252
250
  extra_query: Query | None = None,
253
251
  extra_body: Body | None = None,
254
252
  timeout: float | httpx.Timeout | None | NotGiven = not_given,
255
- ) -> None:
253
+ ) -> str:
256
254
  """Download OCR processing results in various formats.
257
255
 
258
256
  Returns the processed
@@ -272,13 +270,12 @@ class AsyncV1Resource(AsyncAPIResource):
272
270
  raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
273
271
  if not type:
274
272
  raise ValueError(f"Expected a non-empty value for `type` but received {type!r}")
275
- extra_headers = {"Accept": "*/*", **(extra_headers or {})}
276
273
  return await self._get(
277
274
  f"/ocr/v1/{id}/download/{type}",
278
275
  options=make_request_options(
279
276
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
280
277
  ),
281
- cast_to=NoneType,
278
+ cast_to=str,
282
279
  )
283
280
 
284
281
  async def process(
@@ -8,7 +8,7 @@ from typing_extensions import Literal
8
8
 
9
9
  import httpx
10
10
 
11
- from ..._types import Body, Omit, Query, Headers, NoneType, NotGiven, SequenceNotStr, omit, not_given
11
+ from ..._types import Body, Omit, Query, Headers, NotGiven, SequenceNotStr, omit, not_given
12
12
  from ..._utils import maybe_transform, async_maybe_transform
13
13
  from ..._compat import cached_property
14
14
  from ..._resource import SyncAPIResource, AsyncAPIResource
@@ -32,6 +32,7 @@ from ...types.search.v1_search_response import V1SearchResponse
32
32
  from ...types.search.v1_similar_response import V1SimilarResponse
33
33
  from ...types.search.v1_contents_response import V1ContentsResponse
34
34
  from ...types.search.v1_research_response import V1ResearchResponse
35
+ from ...types.search.v1_retrieve_research_response import V1RetrieveResearchResponse
35
36
 
36
37
  __all__ = ["V1Resource", "AsyncV1Resource"]
37
38
 
@@ -282,7 +283,7 @@ class V1Resource(SyncAPIResource):
282
283
  extra_query: Query | None = None,
283
284
  extra_body: Body | None = None,
284
285
  timeout: float | httpx.Timeout | None | NotGiven = not_given,
285
- ) -> None:
286
+ ) -> V1RetrieveResearchResponse:
286
287
  """Retrieve the status and results of a deep research task by ID.
287
288
 
288
289
  Supports both
@@ -305,7 +306,6 @@ class V1Resource(SyncAPIResource):
305
306
  """
306
307
  if not id:
307
308
  raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
308
- extra_headers = {"Accept": "*/*", **(extra_headers or {})}
309
309
  return self._get(
310
310
  f"/search/v1/research/{id}",
311
311
  options=make_request_options(
@@ -321,7 +321,7 @@ class V1Resource(SyncAPIResource):
321
321
  v1_retrieve_research_params.V1RetrieveResearchParams,
322
322
  ),
323
323
  ),
324
- cast_to=NoneType,
324
+ cast_to=V1RetrieveResearchResponse,
325
325
  )
326
326
 
327
327
  def search(
@@ -742,7 +742,7 @@ class AsyncV1Resource(AsyncAPIResource):
742
742
  extra_query: Query | None = None,
743
743
  extra_body: Body | None = None,
744
744
  timeout: float | httpx.Timeout | None | NotGiven = not_given,
745
- ) -> None:
745
+ ) -> V1RetrieveResearchResponse:
746
746
  """Retrieve the status and results of a deep research task by ID.
747
747
 
748
748
  Supports both
@@ -765,7 +765,6 @@ class AsyncV1Resource(AsyncAPIResource):
765
765
  """
766
766
  if not id:
767
767
  raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
768
- extra_headers = {"Accept": "*/*", **(extra_headers or {})}
769
768
  return await self._get(
770
769
  f"/search/v1/research/{id}",
771
770
  options=make_request_options(
@@ -781,7 +780,7 @@ class AsyncV1Resource(AsyncAPIResource):
781
780
  v1_retrieve_research_params.V1RetrieveResearchParams,
782
781
  ),
783
782
  ),
784
- cast_to=NoneType,
783
+ cast_to=V1RetrieveResearchResponse,
785
784
  )
786
785
 
787
786
  async def search(
@@ -4,7 +4,7 @@ from __future__ import annotations
4
4
 
5
5
  import httpx
6
6
 
7
- from ..._types import Body, Query, Headers, NoneType, NotGiven, not_given
7
+ from ..._types import Body, Query, Headers, NotGiven, not_given
8
8
  from ..._compat import cached_property
9
9
  from ..._resource import SyncAPIResource, AsyncAPIResource
10
10
  from ..._response import (
@@ -14,6 +14,8 @@ from ..._response import (
14
14
  async_to_streamed_response_wrapper,
15
15
  )
16
16
  from ..._base_client import make_request_options
17
+ from ...types.vault.graphrag_init_response import GraphragInitResponse
18
+ from ...types.vault.graphrag_get_stats_response import GraphragGetStatsResponse
17
19
 
18
20
  __all__ = ["GraphragResource", "AsyncGraphragResource"]
19
21
 
@@ -48,7 +50,7 @@ class GraphragResource(SyncAPIResource):
48
50
  extra_query: Query | None = None,
49
51
  extra_body: Body | None = None,
50
52
  timeout: float | httpx.Timeout | None | NotGiven = not_given,
51
- ) -> None:
53
+ ) -> GraphragGetStatsResponse:
52
54
  """
53
55
  Retrieve GraphRAG (Graph Retrieval-Augmented Generation) statistics for a
54
56
  specific vault. This includes metrics about the knowledge graph structure,
@@ -66,13 +68,12 @@ class GraphragResource(SyncAPIResource):
66
68
  """
67
69
  if not id:
68
70
  raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
69
- extra_headers = {"Accept": "*/*", **(extra_headers or {})}
70
71
  return self._get(
71
72
  f"/vault/{id}/graphrag/stats",
72
73
  options=make_request_options(
73
74
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
74
75
  ),
75
- cast_to=NoneType,
76
+ cast_to=GraphragGetStatsResponse,
76
77
  )
77
78
 
78
79
  def init(
@@ -85,7 +86,7 @@ class GraphragResource(SyncAPIResource):
85
86
  extra_query: Query | None = None,
86
87
  extra_body: Body | None = None,
87
88
  timeout: float | httpx.Timeout | None | NotGiven = not_given,
88
- ) -> None:
89
+ ) -> GraphragInitResponse:
89
90
  """
90
91
  Initialize a GraphRAG workspace for a vault to enable advanced knowledge graph
91
92
  and retrieval-augmented generation capabilities. This creates the necessary
@@ -103,13 +104,12 @@ class GraphragResource(SyncAPIResource):
103
104
  """
104
105
  if not id:
105
106
  raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
106
- extra_headers = {"Accept": "*/*", **(extra_headers or {})}
107
107
  return self._post(
108
108
  f"/vault/{id}/graphrag/init",
109
109
  options=make_request_options(
110
110
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
111
111
  ),
112
- cast_to=NoneType,
112
+ cast_to=GraphragInitResponse,
113
113
  )
114
114
 
115
115
 
@@ -143,7 +143,7 @@ class AsyncGraphragResource(AsyncAPIResource):
143
143
  extra_query: Query | None = None,
144
144
  extra_body: Body | None = None,
145
145
  timeout: float | httpx.Timeout | None | NotGiven = not_given,
146
- ) -> None:
146
+ ) -> GraphragGetStatsResponse:
147
147
  """
148
148
  Retrieve GraphRAG (Graph Retrieval-Augmented Generation) statistics for a
149
149
  specific vault. This includes metrics about the knowledge graph structure,
@@ -161,13 +161,12 @@ class AsyncGraphragResource(AsyncAPIResource):
161
161
  """
162
162
  if not id:
163
163
  raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
164
- extra_headers = {"Accept": "*/*", **(extra_headers or {})}
165
164
  return await self._get(
166
165
  f"/vault/{id}/graphrag/stats",
167
166
  options=make_request_options(
168
167
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
169
168
  ),
170
- cast_to=NoneType,
169
+ cast_to=GraphragGetStatsResponse,
171
170
  )
172
171
 
173
172
  async def init(
@@ -180,7 +179,7 @@ class AsyncGraphragResource(AsyncAPIResource):
180
179
  extra_query: Query | None = None,
181
180
  extra_body: Body | None = None,
182
181
  timeout: float | httpx.Timeout | None | NotGiven = not_given,
183
- ) -> None:
182
+ ) -> GraphragInitResponse:
184
183
  """
185
184
  Initialize a GraphRAG workspace for a vault to enable advanced knowledge graph
186
185
  and retrieval-augmented generation capabilities. This creates the necessary
@@ -198,13 +197,12 @@ class AsyncGraphragResource(AsyncAPIResource):
198
197
  """
199
198
  if not id:
200
199
  raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
201
- extra_headers = {"Accept": "*/*", **(extra_headers or {})}
202
200
  return await self._post(
203
201
  f"/vault/{id}/graphrag/init",
204
202
  options=make_request_options(
205
203
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
206
204
  ),
207
- cast_to=NoneType,
205
+ cast_to=GraphragInitResponse,
208
206
  )
209
207
 
210
208
 
@@ -6,7 +6,7 @@ from typing_extensions import Literal
6
6
 
7
7
  import httpx
8
8
 
9
- from ..._types import Body, Omit, Query, Headers, NoneType, NotGiven, omit, not_given
9
+ from ..._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
10
10
  from ..._utils import maybe_transform, async_maybe_transform
11
11
  from ..._compat import cached_property
12
12
  from ..._resource import SyncAPIResource, AsyncAPIResource
@@ -18,6 +18,9 @@ from ..._response import (
18
18
  )
19
19
  from ...types.vault import object_create_presigned_url_params
20
20
  from ..._base_client import make_request_options
21
+ from ...types.vault.object_list_response import ObjectListResponse
22
+ from ...types.vault.object_get_text_response import ObjectGetTextResponse
23
+ from ...types.vault.object_retrieve_response import ObjectRetrieveResponse
21
24
  from ...types.vault.object_create_presigned_url_response import ObjectCreatePresignedURLResponse
22
25
 
23
26
  __all__ = ["ObjectsResource", "AsyncObjectsResource"]
@@ -54,7 +57,7 @@ class ObjectsResource(SyncAPIResource):
54
57
  extra_query: Query | None = None,
55
58
  extra_body: Body | None = None,
56
59
  timeout: float | httpx.Timeout | None | NotGiven = not_given,
57
- ) -> None:
60
+ ) -> ObjectRetrieveResponse:
58
61
  """
59
62
  Retrieves metadata for a specific document in a vault and generates a temporary
60
63
  download URL. The download URL expires after 1 hour for security. This endpoint
@@ -73,13 +76,12 @@ class ObjectsResource(SyncAPIResource):
73
76
  raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
74
77
  if not object_id:
75
78
  raise ValueError(f"Expected a non-empty value for `object_id` but received {object_id!r}")
76
- extra_headers = {"Accept": "*/*", **(extra_headers or {})}
77
79
  return self._get(
78
80
  f"/vault/{id}/objects/{object_id}",
79
81
  options=make_request_options(
80
82
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
81
83
  ),
82
- cast_to=NoneType,
84
+ cast_to=ObjectRetrieveResponse,
83
85
  )
84
86
 
85
87
  def list(
@@ -92,7 +94,7 @@ class ObjectsResource(SyncAPIResource):
92
94
  extra_query: Query | None = None,
93
95
  extra_body: Body | None = None,
94
96
  timeout: float | httpx.Timeout | None | NotGiven = not_given,
95
- ) -> None:
97
+ ) -> ObjectListResponse:
96
98
  """
97
99
  Retrieve all objects stored in a specific vault, including document metadata,
98
100
  ingestion status, and processing statistics.
@@ -108,13 +110,12 @@ class ObjectsResource(SyncAPIResource):
108
110
  """
109
111
  if not id:
110
112
  raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
111
- extra_headers = {"Accept": "*/*", **(extra_headers or {})}
112
113
  return self._get(
113
114
  f"/vault/{id}/objects",
114
115
  options=make_request_options(
115
116
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
116
117
  ),
117
- cast_to=NoneType,
118
+ cast_to=ObjectListResponse,
118
119
  )
119
120
 
120
121
  def create_presigned_url(
@@ -184,7 +185,7 @@ class ObjectsResource(SyncAPIResource):
184
185
  extra_query: Query | None = None,
185
186
  extra_body: Body | None = None,
186
187
  timeout: float | httpx.Timeout | None | NotGiven = not_given,
187
- ) -> None:
188
+ ) -> str:
188
189
  """Downloads a file from a vault.
189
190
 
190
191
  Returns the actual file content as a binary
@@ -205,13 +206,12 @@ class ObjectsResource(SyncAPIResource):
205
206
  raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
206
207
  if not object_id:
207
208
  raise ValueError(f"Expected a non-empty value for `object_id` but received {object_id!r}")
208
- extra_headers = {"Accept": "*/*", **(extra_headers or {})}
209
209
  return self._get(
210
210
  f"/vault/{id}/objects/{object_id}/download",
211
211
  options=make_request_options(
212
212
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
213
213
  ),
214
- cast_to=NoneType,
214
+ cast_to=str,
215
215
  )
216
216
 
217
217
  def get_text(
@@ -225,7 +225,7 @@ class ObjectsResource(SyncAPIResource):
225
225
  extra_query: Query | None = None,
226
226
  extra_body: Body | None = None,
227
227
  timeout: float | httpx.Timeout | None | NotGiven = not_given,
228
- ) -> None:
228
+ ) -> ObjectGetTextResponse:
229
229
  """Retrieves the full extracted text content from a processed vault object.
230
230
 
231
231
  Returns
@@ -245,13 +245,12 @@ class ObjectsResource(SyncAPIResource):
245
245
  raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
246
246
  if not object_id:
247
247
  raise ValueError(f"Expected a non-empty value for `object_id` but received {object_id!r}")
248
- extra_headers = {"Accept": "*/*", **(extra_headers or {})}
249
248
  return self._get(
250
249
  f"/vault/{id}/objects/{object_id}/text",
251
250
  options=make_request_options(
252
251
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
253
252
  ),
254
- cast_to=NoneType,
253
+ cast_to=ObjectGetTextResponse,
255
254
  )
256
255
 
257
256
 
@@ -286,7 +285,7 @@ class AsyncObjectsResource(AsyncAPIResource):
286
285
  extra_query: Query | None = None,
287
286
  extra_body: Body | None = None,
288
287
  timeout: float | httpx.Timeout | None | NotGiven = not_given,
289
- ) -> None:
288
+ ) -> ObjectRetrieveResponse:
290
289
  """
291
290
  Retrieves metadata for a specific document in a vault and generates a temporary
292
291
  download URL. The download URL expires after 1 hour for security. This endpoint
@@ -305,13 +304,12 @@ class AsyncObjectsResource(AsyncAPIResource):
305
304
  raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
306
305
  if not object_id:
307
306
  raise ValueError(f"Expected a non-empty value for `object_id` but received {object_id!r}")
308
- extra_headers = {"Accept": "*/*", **(extra_headers or {})}
309
307
  return await self._get(
310
308
  f"/vault/{id}/objects/{object_id}",
311
309
  options=make_request_options(
312
310
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
313
311
  ),
314
- cast_to=NoneType,
312
+ cast_to=ObjectRetrieveResponse,
315
313
  )
316
314
 
317
315
  async def list(
@@ -324,7 +322,7 @@ class AsyncObjectsResource(AsyncAPIResource):
324
322
  extra_query: Query | None = None,
325
323
  extra_body: Body | None = None,
326
324
  timeout: float | httpx.Timeout | None | NotGiven = not_given,
327
- ) -> None:
325
+ ) -> ObjectListResponse:
328
326
  """
329
327
  Retrieve all objects stored in a specific vault, including document metadata,
330
328
  ingestion status, and processing statistics.
@@ -340,13 +338,12 @@ class AsyncObjectsResource(AsyncAPIResource):
340
338
  """
341
339
  if not id:
342
340
  raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
343
- extra_headers = {"Accept": "*/*", **(extra_headers or {})}
344
341
  return await self._get(
345
342
  f"/vault/{id}/objects",
346
343
  options=make_request_options(
347
344
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
348
345
  ),
349
- cast_to=NoneType,
346
+ cast_to=ObjectListResponse,
350
347
  )
351
348
 
352
349
  async def create_presigned_url(
@@ -416,7 +413,7 @@ class AsyncObjectsResource(AsyncAPIResource):
416
413
  extra_query: Query | None = None,
417
414
  extra_body: Body | None = None,
418
415
  timeout: float | httpx.Timeout | None | NotGiven = not_given,
419
- ) -> None:
416
+ ) -> str:
420
417
  """Downloads a file from a vault.
421
418
 
422
419
  Returns the actual file content as a binary
@@ -437,13 +434,12 @@ class AsyncObjectsResource(AsyncAPIResource):
437
434
  raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
438
435
  if not object_id:
439
436
  raise ValueError(f"Expected a non-empty value for `object_id` but received {object_id!r}")
440
- extra_headers = {"Accept": "*/*", **(extra_headers or {})}
441
437
  return await self._get(
442
438
  f"/vault/{id}/objects/{object_id}/download",
443
439
  options=make_request_options(
444
440
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
445
441
  ),
446
- cast_to=NoneType,
442
+ cast_to=str,
447
443
  )
448
444
 
449
445
  async def get_text(
@@ -457,7 +453,7 @@ class AsyncObjectsResource(AsyncAPIResource):
457
453
  extra_query: Query | None = None,
458
454
  extra_body: Body | None = None,
459
455
  timeout: float | httpx.Timeout | None | NotGiven = not_given,
460
- ) -> None:
456
+ ) -> ObjectGetTextResponse:
461
457
  """Retrieves the full extracted text content from a processed vault object.
462
458
 
463
459
  Returns
@@ -477,13 +473,12 @@ class AsyncObjectsResource(AsyncAPIResource):
477
473
  raise ValueError(f"Expected a non-empty value for `id` but received {id!r}")
478
474
  if not object_id:
479
475
  raise ValueError(f"Expected a non-empty value for `object_id` but received {object_id!r}")
480
- extra_headers = {"Accept": "*/*", **(extra_headers or {})}
481
476
  return await self._get(
482
477
  f"/vault/{id}/objects/{object_id}/text",
483
478
  options=make_request_options(
484
479
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
485
480
  ),
486
- cast_to=NoneType,
481
+ cast_to=ObjectGetTextResponse,
487
482
  )
488
483
 
489
484