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.
- casedev/_base_client.py +140 -11
- casedev/_client.py +290 -119
- casedev/_models.py +16 -1
- casedev/_types.py +12 -2
- casedev/_version.py +1 -1
- casedev/resources/__init__.py +0 -70
- casedev/resources/compute/v1/__init__.py +0 -42
- casedev/resources/compute/v1/environments.py +16 -19
- casedev/resources/compute/v1/secrets.py +35 -41
- casedev/resources/compute/v1/v1.py +14 -276
- casedev/resources/format/v1/templates.py +11 -13
- casedev/resources/llm/llm.py +6 -7
- casedev/resources/llm/v1/v1.py +11 -13
- casedev/resources/ocr/v1.py +10 -13
- casedev/resources/search/v1.py +6 -7
- casedev/resources/vault/graphrag.py +11 -13
- casedev/resources/vault/objects.py +20 -25
- casedev/resources/vault/vault.py +64 -28
- casedev/resources/voice/streaming.py +6 -7
- casedev/resources/voice/transcription.py +82 -23
- casedev/resources/voice/v1/speak.py +1 -195
- casedev/resources/voice/v1/v1.py +6 -7
- casedev/types/__init__.py +2 -0
- casedev/types/compute/__init__.py +1 -2
- casedev/types/compute/v1/__init__.py +7 -5
- casedev/types/compute/v1/environment_list_response.py +40 -0
- casedev/types/compute/v1/environment_retrieve_response.py +36 -0
- casedev/types/compute/v1/environment_set_default_response.py +36 -0
- casedev/types/{convert/v1_webhook_response.py → compute/v1/secret_delete_group_response.py} +3 -3
- casedev/types/compute/v1/secret_list_response.py +31 -0
- casedev/types/compute/v1/secret_retrieve_group_response.py +38 -0
- casedev/types/compute/v1/secret_update_group_response.py +22 -0
- casedev/types/compute/v1_get_usage_response.py +51 -0
- casedev/types/format/v1/__init__.py +2 -0
- casedev/types/format/v1/template_list_response.py +40 -0
- casedev/types/format/v1/template_retrieve_response.py +33 -0
- casedev/types/llm/__init__.py +2 -0
- casedev/types/llm/v1_create_embedding_response.py +31 -0
- casedev/types/llm/v1_list_models_response.py +41 -0
- casedev/types/llm_get_config_response.py +33 -0
- casedev/types/ocr/__init__.py +2 -0
- casedev/types/ocr/v1_download_response.py +7 -0
- casedev/types/ocr/v1_process_params.py +2 -0
- casedev/types/ocr/v1_retrieve_response.py +32 -0
- casedev/types/search/__init__.py +1 -0
- casedev/types/search/v1_retrieve_research_response.py +74 -0
- casedev/types/vault/__init__.py +6 -0
- casedev/types/vault/graphrag_get_stats_response.py +31 -0
- casedev/types/{workflows/v1_delete_response.py → vault/graphrag_init_response.py} +6 -2
- casedev/types/vault/object_download_response.py +7 -0
- casedev/types/vault/object_get_text_response.py +35 -0
- casedev/types/vault/object_list_response.py +64 -0
- casedev/types/vault/object_retrieve_response.py +57 -0
- casedev/types/vault_create_params.py +16 -1
- casedev/types/vault_create_response.py +5 -2
- casedev/types/vault_ingest_response.py +15 -5
- casedev/types/vault_retrieve_response.py +76 -0
- casedev/types/vault_search_params.py +19 -5
- casedev/types/vault_search_response.py +25 -1
- casedev/types/vault_upload_params.py +7 -0
- casedev/types/vault_upload_response.py +6 -0
- casedev/types/voice/__init__.py +3 -0
- casedev/types/voice/streaming_get_url_response.py +44 -0
- casedev/types/voice/transcription_create_params.py +23 -3
- casedev/types/voice/transcription_create_response.py +22 -0
- casedev/types/voice/transcription_retrieve_response.py +19 -17
- casedev/types/voice/v1/__init__.py +0 -1
- casedev/types/voice/v1/speak_create_params.py +2 -0
- casedev/types/voice/v1_list_voices_response.py +40 -0
- casedev/types/webhooks/__init__.py +0 -3
- {casedev-0.1.0.dist-info → casedev-0.2.0.dist-info}/METADATA +10 -13
- casedev-0.2.0.dist-info/RECORD +149 -0
- {casedev-0.1.0.dist-info → casedev-0.2.0.dist-info}/licenses/LICENSE +1 -1
- casedev/resources/actions/__init__.py +0 -33
- casedev/resources/actions/actions.py +0 -102
- casedev/resources/actions/v1.py +0 -640
- casedev/resources/compute/v1/functions.py +0 -278
- casedev/resources/compute/v1/invoke.py +0 -216
- casedev/resources/compute/v1/runs.py +0 -290
- casedev/resources/convert/__init__.py +0 -33
- casedev/resources/convert/convert.py +0 -102
- casedev/resources/convert/v1/__init__.py +0 -33
- casedev/resources/convert/v1/jobs.py +0 -254
- casedev/resources/convert/v1/v1.py +0 -450
- casedev/resources/templates/__init__.py +0 -33
- casedev/resources/templates/templates.py +0 -102
- casedev/resources/templates/v1.py +0 -633
- casedev/resources/webhooks/__init__.py +0 -33
- casedev/resources/webhooks/v1.py +0 -447
- casedev/resources/webhooks/webhooks.py +0 -102
- casedev/resources/workflows/__init__.py +0 -33
- casedev/resources/workflows/v1.py +0 -1053
- casedev/resources/workflows/workflows.py +0 -102
- casedev/types/actions/__init__.py +0 -8
- casedev/types/actions/v1_create_params.py +0 -22
- casedev/types/actions/v1_create_response.py +0 -33
- casedev/types/actions/v1_execute_params.py +0 -16
- casedev/types/actions/v1_execute_response.py +0 -31
- casedev/types/compute/v1/function_get_logs_params.py +0 -12
- casedev/types/compute/v1/function_list_params.py +0 -12
- casedev/types/compute/v1/invoke_run_params.py +0 -21
- casedev/types/compute/v1/invoke_run_response.py +0 -39
- casedev/types/compute/v1/run_list_params.py +0 -18
- casedev/types/compute/v1_deploy_params.py +0 -114
- casedev/types/compute/v1_deploy_response.py +0 -30
- casedev/types/convert/__init__.py +0 -8
- casedev/types/convert/v1/__init__.py +0 -3
- casedev/types/convert/v1_process_params.py +0 -15
- casedev/types/convert/v1_process_response.py +0 -19
- casedev/types/convert/v1_webhook_params.py +0 -32
- casedev/types/templates/__init__.py +0 -8
- casedev/types/templates/v1_execute_params.py +0 -22
- casedev/types/templates/v1_execute_response.py +0 -31
- casedev/types/templates/v1_list_params.py +0 -32
- casedev/types/templates/v1_search_params.py +0 -18
- casedev/types/voice/v1/speak_stream_params.py +0 -58
- casedev/types/webhooks/v1_create_params.py +0 -20
- casedev/types/webhooks/v1_create_response.py +0 -33
- casedev/types/workflows/__init__.py +0 -19
- casedev/types/workflows/v1_create_params.py +0 -32
- casedev/types/workflows/v1_create_response.py +0 -29
- casedev/types/workflows/v1_deploy_response.py +0 -20
- casedev/types/workflows/v1_execute_params.py +0 -12
- casedev/types/workflows/v1_execute_response.py +0 -22
- casedev/types/workflows/v1_list_executions_params.py +0 -13
- casedev/types/workflows/v1_list_executions_response.py +0 -27
- casedev/types/workflows/v1_list_params.py +0 -18
- casedev/types/workflows/v1_list_response.py +0 -37
- casedev/types/workflows/v1_retrieve_execution_response.py +0 -31
- casedev/types/workflows/v1_retrieve_response.py +0 -35
- casedev/types/workflows/v1_undeploy_response.py +0 -13
- casedev/types/workflows/v1_update_params.py +0 -26
- casedev/types/workflows/v1_update_response.py +0 -17
- casedev-0.1.0.dist-info/RECORD +0 -185
- {casedev-0.1.0.dist-info → casedev-0.2.0.dist-info}/WHEEL +0 -0
casedev/resources/llm/v1/v1.py
CHANGED
|
@@ -15,7 +15,7 @@ from .chat import (
|
|
|
15
15
|
ChatResourceWithStreamingResponse,
|
|
16
16
|
AsyncChatResourceWithStreamingResponse,
|
|
17
17
|
)
|
|
18
|
-
from ...._types import Body, Omit, Query, Headers,
|
|
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
|
-
) ->
|
|
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=
|
|
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
|
-
) ->
|
|
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=
|
|
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
|
-
) ->
|
|
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=
|
|
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
|
-
) ->
|
|
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=
|
|
250
|
+
cast_to=V1ListModelsResponse,
|
|
253
251
|
)
|
|
254
252
|
|
|
255
253
|
|
casedev/resources/ocr/v1.py
CHANGED
|
@@ -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,
|
|
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
|
-
) ->
|
|
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=
|
|
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
|
-
) ->
|
|
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=
|
|
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
|
-
) ->
|
|
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=
|
|
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
|
-
) ->
|
|
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=
|
|
278
|
+
cast_to=str,
|
|
282
279
|
)
|
|
283
280
|
|
|
284
281
|
async def process(
|
casedev/resources/search/v1.py
CHANGED
|
@@ -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,
|
|
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
|
-
) ->
|
|
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=
|
|
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
|
-
) ->
|
|
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=
|
|
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,
|
|
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
|
-
) ->
|
|
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=
|
|
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
|
-
) ->
|
|
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=
|
|
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
|
-
) ->
|
|
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=
|
|
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
|
-
) ->
|
|
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=
|
|
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,
|
|
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
|
-
) ->
|
|
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=
|
|
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
|
-
) ->
|
|
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=
|
|
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
|
-
) ->
|
|
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=
|
|
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
|
-
) ->
|
|
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=
|
|
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
|
-
) ->
|
|
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=
|
|
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
|
-
) ->
|
|
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=
|
|
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
|
-
) ->
|
|
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=
|
|
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
|
-
) ->
|
|
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=
|
|
481
|
+
cast_to=ObjectGetTextResponse,
|
|
487
482
|
)
|
|
488
483
|
|
|
489
484
|
|