Skip to content

Commit 8e3a2f4

Browse files
authored
Den 352 Fix extract retry not updated page (#63)
1 parent f23f203 commit 8e3a2f4

File tree

1 file changed

+11
-10
lines changed

1 file changed

+11
-10
lines changed

dendrite_sdk/async_api/_core/mixin/extract.py

Lines changed: 11 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -116,16 +116,6 @@ async def extract(
116116
init_start_time = time.time()
117117
page = await self._get_page()
118118

119-
page_information = await page.get_page_information()
120-
extract_dto = ExtractDTO(
121-
page_information=page_information,
122-
api_config=self._get_dendrite_browser().api_config,
123-
prompt=prompt,
124-
return_data_json_schema=json_schema,
125-
use_screenshot=True,
126-
use_cache=use_cache,
127-
)
128-
129119
delay = 1
130120
while True:
131121
elapsed_time = time.time() - init_start_time
@@ -135,6 +125,17 @@ async def extract(
135125
)
136126

137127
start_time = time.time()
128+
129+
page_information = await page.get_page_information()
130+
extract_dto = ExtractDTO(
131+
page_information=page_information,
132+
api_config=self._get_dendrite_browser().api_config,
133+
prompt=prompt,
134+
return_data_json_schema=json_schema,
135+
use_screenshot=True,
136+
use_cache=use_cache,
137+
)
138+
138139
res = await self._get_browser_api_client().extract(extract_dto)
139140
request_time = time.time() - start_time
140141

0 commit comments

Comments
 (0)