66
77import httpx
88
9- from ...types import (
10- tool_parse_pdf_params ,
11- tool_text_to_graph_params ,
12- tool_context_aware_splitting_params ,
13- )
9+ from ...types import tool_parse_pdf_params , tool_context_aware_splitting_params
1410from ..._types import NOT_GIVEN , Body , Query , Headers , NotGiven
1511from ..._utils import (
1612 maybe_transform ,
3430)
3531from ..._base_client import make_request_options
3632from ...types .tool_parse_pdf_response import ToolParsePdfResponse
37- from ...types .tool_text_to_graph_response import ToolTextToGraphResponse
3833from ...types .tool_context_aware_splitting_response import ToolContextAwareSplittingResponse
3934
4035__all__ = ["ToolsResource" , "AsyncToolsResource" ]
@@ -147,41 +142,6 @@ def parse_pdf(
147142 cast_to = ToolParsePdfResponse ,
148143 )
149144
150- def text_to_graph (
151- self ,
152- * ,
153- text : str ,
154- # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
155- # The extra values given here take precedence over values defined on the client or passed to this method.
156- extra_headers : Headers | None = None ,
157- extra_query : Query | None = None ,
158- extra_body : Body | None = None ,
159- timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
160- ) -> ToolTextToGraphResponse :
161- """
162- Performs name entity recognition on the supplied text accepting a maximum of
163- 35000 words.
164-
165- Args:
166- text: The text to be converted into a graph structure. Maximum of 35000 words.
167-
168- extra_headers: Send extra headers
169-
170- extra_query: Add additional query parameters to the request
171-
172- extra_body: Add additional JSON properties to the request
173-
174- timeout: Override the client-level default timeout for this request, in seconds
175- """
176- return self ._post (
177- "/v1/tools/text-to-graph" ,
178- body = maybe_transform ({"text" : text }, tool_text_to_graph_params .ToolTextToGraphParams ),
179- options = make_request_options (
180- extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
181- ),
182- cast_to = ToolTextToGraphResponse ,
183- )
184-
185145
186146class AsyncToolsResource (AsyncAPIResource ):
187147 @cached_property
@@ -290,41 +250,6 @@ async def parse_pdf(
290250 cast_to = ToolParsePdfResponse ,
291251 )
292252
293- async def text_to_graph (
294- self ,
295- * ,
296- text : str ,
297- # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
298- # The extra values given here take precedence over values defined on the client or passed to this method.
299- extra_headers : Headers | None = None ,
300- extra_query : Query | None = None ,
301- extra_body : Body | None = None ,
302- timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
303- ) -> ToolTextToGraphResponse :
304- """
305- Performs name entity recognition on the supplied text accepting a maximum of
306- 35000 words.
307-
308- Args:
309- text: The text to be converted into a graph structure. Maximum of 35000 words.
310-
311- extra_headers: Send extra headers
312-
313- extra_query: Add additional query parameters to the request
314-
315- extra_body: Add additional JSON properties to the request
316-
317- timeout: Override the client-level default timeout for this request, in seconds
318- """
319- return await self ._post (
320- "/v1/tools/text-to-graph" ,
321- body = await async_maybe_transform ({"text" : text }, tool_text_to_graph_params .ToolTextToGraphParams ),
322- options = make_request_options (
323- extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
324- ),
325- cast_to = ToolTextToGraphResponse ,
326- )
327-
328253
329254class ToolsResourceWithRawResponse :
330255 def __init__ (self , tools : ToolsResource ) -> None :
@@ -336,9 +261,6 @@ def __init__(self, tools: ToolsResource) -> None:
336261 self .parse_pdf = to_raw_response_wrapper (
337262 tools .parse_pdf ,
338263 )
339- self .text_to_graph = to_raw_response_wrapper (
340- tools .text_to_graph ,
341- )
342264
343265 @cached_property
344266 def comprehend (self ) -> ComprehendResourceWithRawResponse :
@@ -355,9 +277,6 @@ def __init__(self, tools: AsyncToolsResource) -> None:
355277 self .parse_pdf = async_to_raw_response_wrapper (
356278 tools .parse_pdf ,
357279 )
358- self .text_to_graph = async_to_raw_response_wrapper (
359- tools .text_to_graph ,
360- )
361280
362281 @cached_property
363282 def comprehend (self ) -> AsyncComprehendResourceWithRawResponse :
@@ -374,9 +293,6 @@ def __init__(self, tools: ToolsResource) -> None:
374293 self .parse_pdf = to_streamed_response_wrapper (
375294 tools .parse_pdf ,
376295 )
377- self .text_to_graph = to_streamed_response_wrapper (
378- tools .text_to_graph ,
379- )
380296
381297 @cached_property
382298 def comprehend (self ) -> ComprehendResourceWithStreamingResponse :
@@ -393,9 +309,6 @@ def __init__(self, tools: AsyncToolsResource) -> None:
393309 self .parse_pdf = async_to_streamed_response_wrapper (
394310 tools .parse_pdf ,
395311 )
396- self .text_to_graph = async_to_streamed_response_wrapper (
397- tools .text_to_graph ,
398- )
399312
400313 @cached_property
401314 def comprehend (self ) -> AsyncComprehendResourceWithStreamingResponse :
0 commit comments