|
28 | 28 |
|
29 | 29 | from box_sdk_gen.schemas.ai_ask import AiAsk |
30 | 30 |
|
31 | | -from box_sdk_gen.box.errors import BoxSDKError |
32 | | - |
33 | 31 | from box_sdk_gen.schemas.ai_response import AiResponse |
34 | 32 |
|
35 | 33 | from box_sdk_gen.schemas.ai_text_gen import AiTextGen |
|
48 | 46 |
|
49 | 47 | from box_sdk_gen.schemas.ai_extract_structured import AiExtractStructured |
50 | 48 |
|
| 49 | +from box_sdk_gen.box.errors import BoxSDKError |
| 50 | + |
51 | 51 | from box_sdk_gen.networking.auth import Authentication |
52 | 52 |
|
53 | 53 | from box_sdk_gen.networking.network import NetworkSession |
54 | 54 |
|
| 55 | +from box_sdk_gen.networking.fetch_options import FetchOptions |
| 56 | + |
| 57 | +from box_sdk_gen.networking.fetch_response import FetchResponse |
| 58 | + |
55 | 59 | from box_sdk_gen.internal.utils import prepare_params |
56 | 60 |
|
57 | 61 | from box_sdk_gen.internal.utils import to_string |
58 | 62 |
|
59 | 63 | from box_sdk_gen.internal.utils import ByteStream |
60 | 64 |
|
61 | | -from box_sdk_gen.networking.fetch_options import FetchOptions |
62 | | - |
63 | | -from box_sdk_gen.networking.fetch_response import FetchResponse |
64 | | - |
65 | | -from box_sdk_gen.networking.fetch import fetch |
66 | | - |
67 | 65 | from box_sdk_gen.serialization.json import SerializedData |
68 | 66 |
|
69 | 67 | from box_sdk_gen.serialization.json import sd_to_json |
@@ -251,7 +249,7 @@ def create_ai_ask( |
251 | 249 | 'ai_agent': ai_agent, |
252 | 250 | } |
253 | 251 | headers_map: Dict[str, str] = prepare_params({**extra_headers}) |
254 | | - response: FetchResponse = fetch( |
| 252 | + response: FetchResponse = self.network_session.network_client.fetch( |
255 | 253 | FetchOptions( |
256 | 254 | url=''.join([self.network_session.base_urls.base_url, '/2.0/ai/ask']), |
257 | 255 | method='POST', |
@@ -298,7 +296,7 @@ def create_ai_text_gen( |
298 | 296 | 'ai_agent': ai_agent, |
299 | 297 | } |
300 | 298 | headers_map: Dict[str, str] = prepare_params({**extra_headers}) |
301 | | - response: FetchResponse = fetch( |
| 299 | + response: FetchResponse = self.network_session.network_client.fetch( |
302 | 300 | FetchOptions( |
303 | 301 | url=''.join( |
304 | 302 | [self.network_session.base_urls.base_url, '/2.0/ai/text_gen'] |
@@ -344,7 +342,7 @@ def get_ai_agent_default_config( |
344 | 342 | } |
345 | 343 | ) |
346 | 344 | headers_map: Dict[str, str] = prepare_params({**extra_headers}) |
347 | | - response: FetchResponse = fetch( |
| 345 | + response: FetchResponse = self.network_session.network_client.fetch( |
348 | 346 | FetchOptions( |
349 | 347 | url=''.join( |
350 | 348 | [self.network_session.base_urls.base_url, '/2.0/ai_agent_default'] |
@@ -389,7 +387,7 @@ def create_ai_extract( |
389 | 387 | extra_headers = {} |
390 | 388 | request_body: Dict = {'prompt': prompt, 'items': items, 'ai_agent': ai_agent} |
391 | 389 | headers_map: Dict[str, str] = prepare_params({**extra_headers}) |
392 | | - response: FetchResponse = fetch( |
| 390 | + response: FetchResponse = self.network_session.network_client.fetch( |
393 | 391 | FetchOptions( |
394 | 392 | url=''.join( |
395 | 393 | [self.network_session.base_urls.base_url, '/2.0/ai/extract'] |
@@ -448,7 +446,7 @@ def create_ai_extract_structured( |
448 | 446 | 'ai_agent': ai_agent, |
449 | 447 | } |
450 | 448 | headers_map: Dict[str, str] = prepare_params({**extra_headers}) |
451 | | - response: FetchResponse = fetch( |
| 449 | + response: FetchResponse = self.network_session.network_client.fetch( |
452 | 450 | FetchOptions( |
453 | 451 | url=''.join( |
454 | 452 | [ |
|
0 commit comments