structifyai 1.158.0__py3-none-any.whl → 1.176.1__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 (364) hide show
  1. structify/_base_client.py +140 -11
  2. structify/_client.py +1116 -253
  3. structify/_models.py +16 -1
  4. structify/_streaming.py +12 -10
  5. structify/_types.py +12 -2
  6. structify/_version.py +1 -1
  7. structify/lib/__init__.py +5 -0
  8. structify/lib/cost_confirmation.py +63 -0
  9. structify/pagination.py +63 -1
  10. structify/resources/__init__.py +57 -27
  11. structify/resources/admin/__init__.py +26 -40
  12. structify/resources/admin/admin.py +39 -71
  13. structify/resources/admin/chat_templates.py +382 -0
  14. structify/resources/admin/jobs.py +76 -3
  15. structify/resources/admin/sandbox.py +184 -0
  16. structify/resources/admin/teams.py +88 -0
  17. structify/resources/admin/users.py +28 -238
  18. structify/resources/chat.py +232 -0
  19. structify/resources/connector_catalog/__init__.py +33 -0
  20. structify/resources/connector_catalog/admin.py +1510 -0
  21. structify/resources/connector_catalog/connector_catalog.py +394 -0
  22. structify/resources/connectors/__init__.py +33 -0
  23. structify/resources/{connectors.py → connectors/connectors.py} +1155 -325
  24. structify/resources/connectors/type_snippets.py +169 -0
  25. structify/resources/datasets/datasets.py +0 -4
  26. structify/resources/documents.py +1 -123
  27. structify/resources/external.py +35 -2
  28. structify/resources/jobs.py +19 -336
  29. structify/resources/nango.py +234 -0
  30. structify/resources/polars.py +225 -103
  31. structify/resources/sandbox.py +13 -110
  32. structify/resources/scrape.py +0 -16
  33. structify/resources/sessions.py +241 -153
  34. structify/resources/slack.py +1 -151
  35. structify/resources/structure.py +120 -48
  36. structify/resources/teams.py +96 -0
  37. structify/resources/user/__init__.py +14 -0
  38. structify/resources/user/api_keys.py +380 -0
  39. structify/resources/user/user.py +209 -2
  40. structify/resources/whitelabel.py +294 -187
  41. structify/resources/whitelabel_service.py +222 -0
  42. structify/resources/{report.py → wiki.py} +237 -210
  43. structify/resources/workflow_schedule.py +4 -5
  44. structify/types/__init__.py +67 -36
  45. structify/types/add_member_response.py +35 -4
  46. structify/types/admin/__init__.py +13 -88
  47. structify/types/admin/admin_dataset_return.py +6 -0
  48. structify/types/admin/{user_get_credits_response.py → admin_delete_jobs_response.py} +3 -3
  49. structify/types/admin/admin_list_jobs_response.py +35 -1
  50. structify/types/admin/admin_sandbox.py +38 -0
  51. structify/types/admin/chat_template_create_params.py +21 -0
  52. structify/types/admin/{human_llm_start_next_job_params.py → chat_template_list_params.py} +3 -3
  53. structify/types/admin/{human_llm_get_jobs_response.py → chat_template_list_response.py} +3 -3
  54. structify/types/admin/chat_template_update_params.py +22 -0
  55. structify/types/admin/{user_add_credits_response.py → expire_grants_response.py} +5 -3
  56. structify/types/admin/impersonate_response.py +17 -0
  57. structify/types/admin/{next_action_get_batched_training_data_params.py → job_delete_params.py} +2 -2
  58. structify/types/admin/{user_get_credits_params.py → sandbox_list_params.py} +4 -4
  59. structify/types/admin/sandbox_type.py +7 -0
  60. structify/types/admin/{user_set_credits_params.py → team_expire_grants_params.py} +4 -4
  61. structify/types/admin/user.py +16 -0
  62. structify/types/admin/user_create_params.py +1 -0
  63. structify/types/admin/{training_dataset_delete_params.py → user_impersonate_params.py} +3 -3
  64. structify/types/admin/user_list_response.py +14 -4
  65. structify/types/admin_grant_access_response.py +2 -0
  66. structify/types/{slack_user_mapping_response.py → admin_issue_found_response.py} +3 -5
  67. structify/types/chat_add_git_commit_response.py +2 -0
  68. structify/types/{admin/human_llm_get_next_step_params.py → chat_admin_issue_found_params.py} +4 -4
  69. structify/types/chat_create_session_params.py +7 -0
  70. structify/types/chat_dependency.py +17 -0
  71. structify/types/chat_event.py +51 -239
  72. structify/types/chat_get_git_commit_response.py +2 -0
  73. structify/types/chat_get_session_timeline_response.py +31 -1
  74. structify/types/{job_get_steps_response.py → chat_list_templates_response.py} +3 -3
  75. structify/types/chat_prompt.py +2 -0
  76. structify/types/chat_revert_to_commit_response.py +2 -0
  77. structify/types/chat_session.py +14 -0
  78. structify/types/chat_session_with_messages.py +33 -1
  79. structify/types/chat_template.py +31 -0
  80. structify/types/chat_update_session_params.py +2 -0
  81. structify/types/code_generate_code_params.py +7 -0
  82. structify/types/connector.py +16 -25
  83. structify/types/connector_auth_method.py +29 -0
  84. structify/types/connector_auth_method_with_fields.py +15 -0
  85. structify/types/connector_catalog/__init__.py +32 -0
  86. structify/types/connector_catalog/admin_batch_create_credential_fields_params.py +14 -0
  87. structify/types/connector_catalog/admin_batch_create_credential_fields_response.py +10 -0
  88. structify/types/connector_catalog/admin_batch_create_scopes_params.py +14 -0
  89. structify/types/connector_catalog/admin_create_auth_method_params.py +22 -0
  90. structify/types/connector_catalog/admin_create_catalog_params.py +22 -0
  91. structify/types/connector_catalog/admin_create_credential_field_params.py +28 -0
  92. structify/types/connector_catalog/admin_create_scope_params.py +19 -0
  93. structify/types/connector_catalog/admin_list_nango_pending_response.py +10 -0
  94. structify/types/{admin/training_dataset_get_datum_info_params.py → connector_catalog/admin_list_scopes_params.py} +3 -3
  95. structify/types/connector_catalog/admin_update_auth_method_params.py +16 -0
  96. structify/types/connector_catalog/admin_update_catalog_params.py +20 -0
  97. structify/types/connector_catalog/admin_update_credential_field_params.py +26 -0
  98. structify/types/connector_catalog/admin_update_scope_params.py +18 -0
  99. structify/types/connector_catalog/admin_upload_logo_params.py +13 -0
  100. structify/types/connector_catalog/batch_create_scopes_response.py +12 -0
  101. structify/types/connector_catalog/connector_auth_method_scope.py +25 -0
  102. structify/types/connector_catalog/connector_catalog.py +26 -0
  103. structify/types/connector_catalog/create_credential_field_request_param.py +28 -0
  104. structify/types/connector_catalog/create_scope_request_param.py +19 -0
  105. structify/types/connector_catalog/list_scopes_response.py +12 -0
  106. structify/types/connector_catalog/pending_nango_integration.py +17 -0
  107. structify/types/{admin/user_set_credits_response.py → connector_catalog/upload_logo_response.py} +3 -3
  108. structify/types/connector_catalog_list_params.py +20 -0
  109. structify/types/connector_catalog_list_response.py +22 -0
  110. structify/types/connector_catalog_with_methods.py +15 -0
  111. structify/types/connector_category.py +7 -0
  112. structify/types/connector_create_params.py +11 -25
  113. structify/types/connector_credential_field.py +34 -0
  114. structify/types/connector_delete_schema_object_params.py +35 -0
  115. structify/types/connector_explore_params.py +19 -0
  116. structify/types/connector_explorer_chat.py +9 -84
  117. structify/types/connector_get_clarification_requests_response.py +36 -0
  118. structify/types/connector_get_response.py +2 -0
  119. structify/types/connector_list_with_snippets_params.py +12 -0
  120. structify/types/connector_list_with_snippets_response.py +10 -0
  121. structify/types/connector_search_tables_params.py +15 -0
  122. structify/types/connector_search_tables_response.py +160 -0
  123. structify/types/connector_store_response.py +21 -0
  124. structify/types/connector_summaries_params.py +15 -0
  125. structify/types/connector_summaries_response.py +10 -0
  126. structify/types/connector_summary.py +16 -0
  127. structify/types/connector_table_info.py +15 -0
  128. structify/types/connector_update_column_params.py +12 -0
  129. structify/types/connector_update_params.py +4 -25
  130. structify/types/connector_update_table_params.py +14 -0
  131. structify/types/connector_with_secrets.py +2 -0
  132. structify/types/connector_with_snippets.py +11 -0
  133. structify/types/connectors/__init__.py +6 -0
  134. structify/types/connectors/snippet.py +20 -0
  135. structify/types/connectors/type_snippet_upsert_params.py +12 -0
  136. structify/types/create_chat_session_response.py +2 -0
  137. structify/types/dataset_create_params.py +0 -2
  138. structify/types/dataset_descriptor.py +6 -0
  139. structify/types/dataset_descriptor_param.py +6 -0
  140. structify/types/dataset_get_response.py +6 -0
  141. structify/types/dataset_list_response.py +6 -0
  142. structify/types/dataset_view_table_response.py +3 -1
  143. structify/types/delete_chat_session_response.py +2 -0
  144. structify/types/delete_schema_object_request_param.py +35 -0
  145. structify/types/delete_schema_object_response.py +13 -0
  146. structify/types/edge_spec_param.py +13 -0
  147. structify/types/entity_get_source_entities_response.py +0 -2
  148. structify/types/entity_list_jobs_response.py +3 -3
  149. structify/types/entity_view_response.py +0 -2
  150. structify/types/estimate_cost_response.py +9 -0
  151. structify/types/exploration_phase_id.py +79 -0
  152. structify/types/finalize_dag_response.py +11 -0
  153. structify/types/get_chat_session_response.py +14 -6
  154. structify/types/get_dependencies_response.py +15 -0
  155. structify/types/get_job_events_response.py +20 -0
  156. structify/types/get_node_response.py +10 -0
  157. structify/types/job_cancel_response.py +3 -3
  158. structify/types/job_event_body.py +108 -1
  159. structify/types/job_get_source_entities_response.py +0 -2
  160. structify/types/job_list_params.py +1 -1
  161. structify/types/job_list_response.py +35 -1
  162. structify/types/knowledge_graph.py +6 -0
  163. structify/types/knowledge_graph_param.py +6 -0
  164. structify/types/list_chat_sessions_response.py +2 -0
  165. structify/types/list_collaborators_response.py +4 -0
  166. structify/types/list_tables_response.py +12 -0
  167. structify/types/llm_information_store.py +62 -28
  168. structify/types/merge_config.py +2 -0
  169. structify/types/merge_config_param.py +2 -0
  170. structify/types/message.py +2 -0
  171. structify/types/nango_create_session_params.py +21 -0
  172. structify/types/nango_create_session_response.py +15 -0
  173. structify/types/nango_list_integrations_response.py +28 -0
  174. structify/types/{session_create_node_params.py → node_spec_param.py} +4 -4
  175. structify/types/refresh_session_response.py +15 -0
  176. structify/types/sandbox.py +8 -2
  177. structify/types/sandbox_get_params.py +2 -1
  178. structify/types/scrape_list_params.py +1 -22
  179. structify/types/scrape_list_response.py +2 -0
  180. structify/types/scrape_scrape_params.py +1 -14
  181. structify/types/scrape_scrape_response.py +2 -0
  182. structify/types/select_team_response.py +3 -1
  183. structify/types/{connector_approve_version_params.py → session_confirm_node_params.py} +3 -3
  184. structify/types/session_finalize_dag_params.py +20 -0
  185. structify/types/session_get_events_response.py +2 -0
  186. structify/types/session_request_confirmation_params.py +13 -0
  187. structify/types/session_upload_node_output_data_params.py +3 -0
  188. structify/types/slack_event_payload_param.py +18 -2
  189. structify/types/slack_events_params.py +18 -2
  190. structify/types/source_list_response.py +0 -2
  191. structify/types/structure_enhance_property_params.py +2 -17
  192. structify/types/structure_enhance_relationship_params.py +2 -17
  193. structify/types/structure_find_relationship_params.py +2 -18
  194. structify/types/structure_pdf_params.py +22 -0
  195. structify/types/structure_pdf_response.py +11 -0
  196. structify/types/structure_run_async_params.py +10 -17
  197. structify/types/table.py +2 -0
  198. structify/types/table_param.py +2 -0
  199. structify/types/team.py +14 -0
  200. structify/types/{admin/training_dataset_remove_datum_params.py → team_create_link_code_params.py} +3 -3
  201. structify/types/team_update_params.py +10 -0
  202. structify/types/team_wiki_page.py +28 -0
  203. structify/types/teams_link_code_response.py +13 -0
  204. structify/types/token_response.py +6 -2
  205. structify/types/tool_invocation.py +468 -0
  206. structify/types/tool_result.py +99 -0
  207. structify/types/update_table_response.py +46 -0
  208. structify/types/usage_group_key.py +14 -1
  209. structify/types/user/__init__.py +4 -0
  210. structify/types/{admin/training_dataset_size_params.py → user/api_key_create_params.py} +5 -5
  211. structify/types/user/api_key_info.py +24 -0
  212. structify/types/user/create_api_key_response.py +12 -0
  213. structify/types/user/list_api_keys_response.py +12 -0
  214. structify/types/user_enrich_params.py +11 -0
  215. structify/types/user_info.py +22 -0
  216. structify/types/user_refresh_params.py +13 -0
  217. structify/types/user_transactions_response.py +5 -5
  218. structify/types/user_update_params.py +32 -12
  219. structify/types/user_usage_response.py +0 -2
  220. structify/types/wiki_connector_reference.py +63 -0
  221. structify/types/wiki_create_params.py +16 -0
  222. structify/types/{admin/training_dataset_list_response.py → wiki_list_response.py} +4 -2
  223. structify/types/wiki_page_with_references.py +12 -0
  224. structify/types/wiki_update_params.py +16 -0
  225. structify/types/workflow_dag.py +2 -0
  226. structify/types/workflow_node_execution_status.py +3 -1
  227. structify/types/workflow_schedule_info.py +2 -2
  228. structify/types/workflow_session_node.py +10 -0
  229. {structifyai-1.158.0.dist-info → structifyai-1.176.1.dist-info}/METADATA +67 -4
  230. structifyai-1.176.1.dist-info/RECORD +496 -0
  231. {structifyai-1.158.0.dist-info → structifyai-1.176.1.dist-info}/licenses/LICENSE +1 -1
  232. structify/resources/admin/human_llm.py +0 -819
  233. structify/resources/admin/next_action.py +0 -656
  234. structify/resources/admin/training_datasets.py +0 -2028
  235. structify/resources/external/__init__.py +0 -61
  236. structify/resources/external/external.py +0 -166
  237. structify/resources/external/news.py +0 -519
  238. structify/resources/external/people.py +0 -998
  239. structify/resources/external/search_api.py +0 -1525
  240. structify/types/active_version_data.py +0 -21
  241. structify/types/active_version_response.py +0 -13
  242. structify/types/admin/action_training_data_entry.py +0 -219
  243. structify/types/admin/action_training_data_response.py +0 -12
  244. structify/types/admin/datum_status.py +0 -19
  245. structify/types/admin/delete_action_training_data_response.py +0 -9
  246. structify/types/admin/human_llm_add_search_for_job_params.py +0 -13
  247. structify/types/admin/human_llm_add_to_dataset_params.py +0 -186
  248. structify/types/admin/human_llm_finish_job_params.py +0 -13
  249. structify/types/admin/human_llm_get_jobs_params.py +0 -12
  250. structify/types/admin/human_llm_get_next_step_response.py +0 -196
  251. structify/types/admin/human_llm_job.py +0 -73
  252. structify/types/admin/human_llm_prelabel_step_response.py +0 -184
  253. structify/types/admin/human_llm_update_step_params.py +0 -186
  254. structify/types/admin/labeling_stats.py +0 -24
  255. structify/types/admin/next_action_add_training_datum_params.py +0 -203
  256. structify/types/admin/next_action_delete_training_data_params.py +0 -12
  257. structify/types/admin/next_action_get_training_data_params.py +0 -25
  258. structify/types/admin/next_action_get_training_datum_params.py +0 -12
  259. structify/types/admin/next_action_label_training_datum_params.py +0 -90
  260. structify/types/admin/step_choices.py +0 -32
  261. structify/types/admin/training_dataset_add_datum_params.py +0 -13
  262. structify/types/admin/training_dataset_add_params.py +0 -12
  263. structify/types/admin/training_dataset_download_datum_params.py +0 -13
  264. structify/types/admin/training_dataset_get_labeller_stats_params.py +0 -23
  265. structify/types/admin/training_dataset_get_labeller_stats_response.py +0 -10
  266. structify/types/admin/training_dataset_get_next_for_labeling_params.py +0 -15
  267. structify/types/admin/training_dataset_get_next_for_qa_params.py +0 -18
  268. structify/types/admin/training_dataset_get_next_suspicious_params.py +0 -23
  269. structify/types/admin/training_dataset_label_datum_params.py +0 -188
  270. structify/types/admin/training_dataset_list_datums_params.py +0 -17
  271. structify/types/admin/training_dataset_list_datums_response.py +0 -33
  272. structify/types/admin/training_dataset_mark_datum_suspicious_params.py +0 -19
  273. structify/types/admin/training_dataset_size_response.py +0 -37
  274. structify/types/admin/training_dataset_suspicious_count_params.py +0 -23
  275. structify/types/admin/training_dataset_suspicious_count_response.py +0 -9
  276. structify/types/admin/training_dataset_switch_dataset_params.py +0 -13
  277. structify/types/admin/training_dataset_update_datum_status_params.py +0 -18
  278. structify/types/admin/training_dataset_upload_labeled_step_params.py +0 -15
  279. structify/types/admin/training_dataset_verify_datum_params.py +0 -17
  280. structify/types/admin/training_datum_response.py +0 -405
  281. structify/types/admin/user_add_credits_params.py +0 -18
  282. structify/types/chat_prompt_param.py +0 -255
  283. structify/types/connector_column_descriptor.py +0 -21
  284. structify/types/connector_relational_database_descriptor.py +0 -13
  285. structify/types/connector_table_descriptor.py +0 -22
  286. structify/types/document_structure_params.py +0 -21
  287. structify/types/document_structure_response.py +0 -8
  288. structify/types/execution_step.py +0 -198
  289. structify/types/external/__init__.py +0 -96
  290. structify/types/external/companies_search_response.py +0 -28
  291. structify/types/external/company_search_result.py +0 -69
  292. structify/types/external/enriched_organization.py +0 -69
  293. structify/types/external/everything_response.py +0 -19
  294. structify/types/external/google_maps_result.py +0 -38
  295. structify/types/external/google_search_response.py +0 -19
  296. structify/types/external/google_search_result.py +0 -21
  297. structify/types/external/job_posting.py +0 -54
  298. structify/types/external/job_postings_response.py +0 -19
  299. structify/types/external/location_coordinates.py +0 -13
  300. structify/types/external/location_response.py +0 -16
  301. structify/types/external/location_result.py +0 -24
  302. structify/types/external/news_article.py +0 -33
  303. structify/types/external/news_everything_params.py +0 -45
  304. structify/types/external/news_source.py +0 -15
  305. structify/types/external/news_source_detail.py +0 -28
  306. structify/types/external/news_sources_params.py +0 -19
  307. structify/types/external/news_top_headlines_params.py +0 -28
  308. structify/types/external/organization_detail.py +0 -87
  309. structify/types/external/people_match_response.py +0 -18
  310. structify/types/external/people_search_response.py +0 -28
  311. structify/types/external/person_companies_search_params.py +0 -63
  312. structify/types/external/person_match.py +0 -60
  313. structify/types/external/person_organization_job_postings_params.py +0 -16
  314. structify/types/external/person_organizations_enrich_params.py +0 -16
  315. structify/types/external/person_people_match_params.py +0 -49
  316. structify/types/external/person_people_search_params.py +0 -57
  317. structify/types/external/person_search_result.py +0 -48
  318. structify/types/external/search_api_google_flights_calendar_params.py +0 -28
  319. structify/types/external/search_api_google_flights_calendar_response.py +0 -8
  320. structify/types/external/search_api_google_flights_location_search_params.py +0 -16
  321. structify/types/external/search_api_google_flights_location_search_response.py +0 -8
  322. structify/types/external/search_api_google_flights_search_params.py +0 -43
  323. structify/types/external/search_api_google_flights_search_response.py +0 -8
  324. structify/types/external/search_api_google_maps_place_details_params.py +0 -16
  325. structify/types/external/search_api_google_maps_place_details_response.py +0 -8
  326. structify/types/external/search_api_google_maps_place_photos_params.py +0 -16
  327. structify/types/external/search_api_google_maps_place_photos_response.py +0 -8
  328. structify/types/external/search_api_google_maps_place_reviews_params.py +0 -19
  329. structify/types/external/search_api_google_maps_place_reviews_response.py +0 -8
  330. structify/types/external/search_api_google_maps_search_params.py +0 -22
  331. structify/types/external/search_api_google_maps_search_response.py +0 -10
  332. structify/types/external/search_api_google_scholar_author_search_params.py +0 -22
  333. structify/types/external/search_api_google_scholar_author_search_response.py +0 -8
  334. structify/types/external/search_api_google_scholar_citations_params.py +0 -22
  335. structify/types/external/search_api_google_scholar_citations_response.py +0 -8
  336. structify/types/external/search_api_google_scholar_search_params.py +0 -28
  337. structify/types/external/search_api_google_scholar_search_response.py +0 -8
  338. structify/types/external/search_api_google_search_params.py +0 -25
  339. structify/types/external/search_api_location_search_params.py +0 -22
  340. structify/types/external/sources_response.py +0 -16
  341. structify/types/external/top_headlines_response.py +0 -19
  342. structify/types/job_delete_response.py +0 -7
  343. structify/types/job_get_response.py +0 -62
  344. structify/types/job_get_step_graph_response.py +0 -233
  345. structify/types/job_get_step_response.py +0 -196
  346. structify/types/message_param.py +0 -35
  347. structify/types/pending_version_response.py +0 -21
  348. structify/types/report_missing_params.py +0 -18
  349. structify/types/report_missing_response.py +0 -7
  350. structify/types/report_relationship_params.py +0 -18
  351. structify/types/report_relationship_response.py +0 -7
  352. structify/types/report_step_params.py +0 -15
  353. structify/types/report_step_response.py +0 -7
  354. structify/types/report_wrong_params.py +0 -18
  355. structify/types/report_wrong_response.py +0 -7
  356. structify/types/sandbox_create_params.py +0 -20
  357. structify/types/session_create_edge_params.py +0 -13
  358. structify/types/slack_connection_status.py +0 -15
  359. structify/types/slack_user_mapping_params.py +0 -18
  360. structify/types/tool_metadata_param.py +0 -17
  361. structify/types/user_team.py +0 -35
  362. structify/types/workflow_schedule_get_response.py +0 -10
  363. structifyai-1.158.0.dist-info/RECORD +0 -530
  364. {structifyai-1.158.0.dist-info → structifyai-1.176.1.dist-info}/WHEEL +0 -0
@@ -1,21 +0,0 @@
1
- # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
-
3
- from .._models import BaseModel
4
- from .llm_information_store import LlmInformationStore
5
-
6
- __all__ = ["ActiveVersionData"]
7
-
8
-
9
- class ActiveVersionData(BaseModel):
10
- store: LlmInformationStore
11
- """Information store for LLM context about a connector
12
-
13
- This enum represents different types of connector information that can be
14
- provided to LLMs for context. It's stored as JSON in the database.
15
-
16
- When deserializing from the database, we attempt to parse into the most specific
17
- variant first (RelationalDatabase), and fall back to Other if the structure
18
- doesn't match.
19
- """
20
-
21
- version_id: str
@@ -1,13 +0,0 @@
1
- # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
-
3
- from typing import Optional
4
-
5
- from .._models import BaseModel
6
- from .active_version_data import ActiveVersionData
7
-
8
- __all__ = ["ActiveVersionResponse"]
9
-
10
-
11
- class ActiveVersionResponse(BaseModel):
12
- active_version: Optional[ActiveVersionData] = None
13
- """Active version data"""
@@ -1,219 +0,0 @@
1
- # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
-
3
- from typing import Dict, List, Union, Optional
4
- from datetime import datetime
5
- from typing_extensions import Literal, TypeAlias
6
-
7
- from pydantic import Field as FieldInfo
8
-
9
- from ..._models import BaseModel
10
- from ..chat_prompt import ChatPrompt
11
- from ..knowledge_graph import KnowledgeGraph
12
- from ..save_requirement import SaveRequirement
13
- from ..dataset_descriptor import DatasetDescriptor
14
-
15
- __all__ = [
16
- "ActionTrainingDataEntry",
17
- "Input",
18
- "InputAllStep",
19
- "InputPreviousAction",
20
- "InputPreviousActionSelectedStep",
21
- "InputPreviousActionSelectedStepSelectedStep",
22
- "InputPreviousActionSelectedStepSelectedStepInfo",
23
- "InputPreviousActionSearchStep",
24
- "InputPreviousActionSearchStepSearchStep",
25
- "InputPreviousActionInvalidAction",
26
- "InputPreviousActionInvalidActionInvalidAction",
27
- "InputPreviousActionExit",
28
- "InputPreviousActionExitExit",
29
- "Output",
30
- "OutputOutput",
31
- "OutputOutputSelectedStep",
32
- "OutputOutputSelectedStepSelectedStep",
33
- "OutputOutputSelectedStepSelectedStepInfo",
34
- "OutputOutputSearchStep",
35
- "OutputOutputSearchStepSearchStep",
36
- "OutputOutputInvalidAction",
37
- "OutputOutputInvalidActionInvalidAction",
38
- "OutputOutputExit",
39
- "OutputOutputExitExit",
40
- ]
41
-
42
-
43
- class InputAllStep(BaseModel):
44
- id: str
45
-
46
- action_name: Optional[str] = None
47
-
48
- metadata: Optional[Dict[str, str]] = None
49
-
50
-
51
- class InputPreviousActionSelectedStepSelectedStepInfo(BaseModel):
52
- id: str
53
-
54
- action_name: Optional[str] = None
55
-
56
- metadata: Optional[Dict[str, str]] = None
57
-
58
-
59
- class InputPreviousActionSelectedStepSelectedStep(BaseModel):
60
- info: InputPreviousActionSelectedStepSelectedStepInfo
61
-
62
- llm_input: ChatPrompt
63
-
64
- llm_output: str
65
-
66
- step_id: str
67
-
68
-
69
- class InputPreviousActionSelectedStep(BaseModel):
70
- selected_step: InputPreviousActionSelectedStepSelectedStep = FieldInfo(alias="SelectedStep")
71
-
72
-
73
- class InputPreviousActionSearchStepSearchStep(BaseModel):
74
- llm_input: ChatPrompt
75
-
76
- llm_output: str
77
-
78
- search_query: str
79
-
80
-
81
- class InputPreviousActionSearchStep(BaseModel):
82
- search_step: InputPreviousActionSearchStepSearchStep = FieldInfo(alias="SearchStep")
83
-
84
-
85
- class InputPreviousActionInvalidActionInvalidAction(BaseModel):
86
- error: str
87
-
88
- llm_input: ChatPrompt
89
-
90
- llm_output: str
91
-
92
-
93
- class InputPreviousActionInvalidAction(BaseModel):
94
- invalid_action: InputPreviousActionInvalidActionInvalidAction = FieldInfo(alias="InvalidAction")
95
-
96
-
97
- class InputPreviousActionExitExit(BaseModel):
98
- llm_input: ChatPrompt
99
-
100
- llm_output: str
101
-
102
-
103
- class InputPreviousActionExit(BaseModel):
104
- exit: InputPreviousActionExitExit = FieldInfo(alias="Exit")
105
-
106
-
107
- InputPreviousAction: TypeAlias = Union[
108
- InputPreviousActionSelectedStep,
109
- InputPreviousActionSearchStep,
110
- InputPreviousActionInvalidAction,
111
- InputPreviousActionExit,
112
- ]
113
-
114
-
115
- class Input(BaseModel):
116
- all_steps: List[InputAllStep]
117
-
118
- descriptor: DatasetDescriptor
119
- """A dataset is where you put multiple referential schemas.
120
-
121
- A dataset is a complete namespace where all references between schemas are held
122
- within the dataset.
123
- """
124
-
125
- extraction_criteria: List[SaveRequirement]
126
-
127
- previous_actions: List[InputPreviousAction]
128
-
129
- seeded_kg: KnowledgeGraph
130
- """
131
- Knowledge graph info structured to deserialize and display in the same format
132
- that the LLM outputs. Also the first representation of an LLM output in the
133
- pipeline from raw tool output to being merged into a DB
134
- """
135
-
136
-
137
- class OutputOutputSelectedStepSelectedStepInfo(BaseModel):
138
- id: str
139
-
140
- action_name: Optional[str] = None
141
-
142
- metadata: Optional[Dict[str, str]] = None
143
-
144
-
145
- class OutputOutputSelectedStepSelectedStep(BaseModel):
146
- info: OutputOutputSelectedStepSelectedStepInfo
147
-
148
- llm_input: ChatPrompt
149
-
150
- llm_output: str
151
-
152
- step_id: str
153
-
154
-
155
- class OutputOutputSelectedStep(BaseModel):
156
- selected_step: OutputOutputSelectedStepSelectedStep = FieldInfo(alias="SelectedStep")
157
-
158
-
159
- class OutputOutputSearchStepSearchStep(BaseModel):
160
- llm_input: ChatPrompt
161
-
162
- llm_output: str
163
-
164
- search_query: str
165
-
166
-
167
- class OutputOutputSearchStep(BaseModel):
168
- search_step: OutputOutputSearchStepSearchStep = FieldInfo(alias="SearchStep")
169
-
170
-
171
- class OutputOutputInvalidActionInvalidAction(BaseModel):
172
- error: str
173
-
174
- llm_input: ChatPrompt
175
-
176
- llm_output: str
177
-
178
-
179
- class OutputOutputInvalidAction(BaseModel):
180
- invalid_action: OutputOutputInvalidActionInvalidAction = FieldInfo(alias="InvalidAction")
181
-
182
-
183
- class OutputOutputExitExit(BaseModel):
184
- llm_input: ChatPrompt
185
-
186
- llm_output: str
187
-
188
-
189
- class OutputOutputExit(BaseModel):
190
- exit: OutputOutputExitExit = FieldInfo(alias="Exit")
191
-
192
-
193
- OutputOutput: TypeAlias = Union[
194
- OutputOutputSelectedStep, OutputOutputSearchStep, OutputOutputInvalidAction, OutputOutputExit
195
- ]
196
-
197
-
198
- class Output(BaseModel):
199
- id: str
200
-
201
- created_at: datetime
202
-
203
- label: Literal["HumanLLMLabel", "LLMOutput", "Pending", "Reviewed", "Verified", "Others"]
204
-
205
- output: OutputOutput
206
-
207
-
208
- class ActionTrainingDataEntry(BaseModel):
209
- id: str
210
-
211
- author: str
212
-
213
- created_at: datetime
214
-
215
- input: Input
216
-
217
- input_prompt: str
218
-
219
- outputs: List[Output]
@@ -1,12 +0,0 @@
1
- # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
-
3
- from typing import List
4
-
5
- from ..._models import BaseModel
6
- from .action_training_data_entry import ActionTrainingDataEntry
7
-
8
- __all__ = ["ActionTrainingDataResponse"]
9
-
10
-
11
- class ActionTrainingDataResponse(BaseModel):
12
- data: List[ActionTrainingDataEntry]
@@ -1,19 +0,0 @@
1
- # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
-
3
- from typing_extensions import Literal, TypeAlias
4
-
5
- __all__ = ["DatumStatus"]
6
-
7
- DatumStatus: TypeAlias = Literal[
8
- "unlabeled",
9
- "nav_labeled",
10
- "save_labeled",
11
- "nav_verified",
12
- "save_verified",
13
- "pending",
14
- "skipped",
15
- "suspicious_nav",
16
- "suspicious_save",
17
- "potential_suspicious_nav",
18
- "potential_suspicious_save",
19
- ]
@@ -1,9 +0,0 @@
1
- # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
-
3
- from ..._models import BaseModel
4
-
5
- __all__ = ["DeleteActionTrainingDataResponse"]
6
-
7
-
8
- class DeleteActionTrainingDataResponse(BaseModel):
9
- deleted_count: int
@@ -1,13 +0,0 @@
1
- # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
-
3
- from __future__ import annotations
4
-
5
- from typing_extensions import Required, TypedDict
6
-
7
- __all__ = ["HumanLlmAddSearchForJobParams"]
8
-
9
-
10
- class HumanLlmAddSearchForJobParams(TypedDict, total=False):
11
- job_id: Required[str]
12
-
13
- search_query: Required[str]
@@ -1,186 +0,0 @@
1
- # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
-
3
- from __future__ import annotations
4
-
5
- from typing import Union, Iterable, Optional
6
- from typing_extensions import Literal, Required, Annotated, TypeAlias, TypedDict
7
-
8
- from ..._utils import PropertyInfo
9
- from ..knowledge_graph_param import KnowledgeGraphParam
10
-
11
- __all__ = [
12
- "HumanLlmAddToDatasetParams",
13
- "ToolCall",
14
- "ToolCallInput",
15
- "ToolCallInputSave",
16
- "ToolCallInputScroll",
17
- "ToolCallInputScrollScroll",
18
- "ToolCallInputScrollToBottom",
19
- "ToolCallInputScrollToBottomScrollToBottom",
20
- "ToolCallInputExit",
21
- "ToolCallInputExitExit",
22
- "ToolCallInputClick",
23
- "ToolCallInputClickClick",
24
- "ToolCallInputHover",
25
- "ToolCallInputHoverHover",
26
- "ToolCallInputWait",
27
- "ToolCallInputWaitWait",
28
- "ToolCallInputError",
29
- "ToolCallInputErrorError",
30
- "ToolCallInputGoogle",
31
- "ToolCallInputGoogleGoogle",
32
- "ToolCallInputType",
33
- "ToolCallInputTypeType",
34
- "ToolCallResult",
35
- "ToolCallResultToolQueued",
36
- "ToolCallResultToolFail",
37
- "ToolCallResultInputParseFail",
38
- "ToolCallResultSuccess",
39
- ]
40
-
41
-
42
- class HumanLlmAddToDatasetParams(TypedDict, total=False):
43
- extraction_criteria_met: Required[bool]
44
-
45
- job_id: Required[str]
46
-
47
- step_id: Required[str]
48
-
49
- tool_calls: Required[Iterable[ToolCall]]
50
-
51
-
52
- class ToolCallInputSave(TypedDict, total=False):
53
- save: Required[Annotated[KnowledgeGraphParam, PropertyInfo(alias="Save")]]
54
- """
55
- Knowledge graph info structured to deserialize and display in the same format
56
- that the LLM outputs. Also the first representation of an LLM output in the
57
- pipeline from raw tool output to being merged into a DB
58
- """
59
-
60
-
61
- class ToolCallInputScrollScroll(TypedDict, total=False):
62
- reason: Required[str]
63
- """Dummy argument"""
64
-
65
-
66
- class ToolCallInputScroll(TypedDict, total=False):
67
- scroll: Required[Annotated[ToolCallInputScrollScroll, PropertyInfo(alias="Scroll")]]
68
- """For tools with no inputs."""
69
-
70
-
71
- class ToolCallInputScrollToBottomScrollToBottom(TypedDict, total=False):
72
- reason: Required[str]
73
- """Dummy argument"""
74
-
75
-
76
- class ToolCallInputScrollToBottom(TypedDict, total=False):
77
- scroll_to_bottom: Required[
78
- Annotated[ToolCallInputScrollToBottomScrollToBottom, PropertyInfo(alias="ScrollToBottom")]
79
- ]
80
- """For tools with no inputs."""
81
-
82
-
83
- class ToolCallInputExitExit(TypedDict, total=False):
84
- reason: Required[str]
85
- """Dummy argument"""
86
-
87
-
88
- class ToolCallInputExit(TypedDict, total=False):
89
- exit: Required[Annotated[ToolCallInputExitExit, PropertyInfo(alias="Exit")]]
90
- """For tools with no inputs."""
91
-
92
-
93
- class ToolCallInputClickClick(TypedDict, total=False):
94
- flag: Required[int]
95
-
96
-
97
- class ToolCallInputClick(TypedDict, total=False):
98
- click: Required[Annotated[ToolCallInputClickClick, PropertyInfo(alias="Click")]]
99
-
100
-
101
- class ToolCallInputHoverHover(TypedDict, total=False):
102
- flag: Required[int]
103
-
104
-
105
- class ToolCallInputHover(TypedDict, total=False):
106
- hover: Required[Annotated[ToolCallInputHoverHover, PropertyInfo(alias="Hover")]]
107
-
108
-
109
- class ToolCallInputWaitWait(TypedDict, total=False):
110
- seconds: int
111
- """Time in seconds to wait"""
112
-
113
-
114
- class ToolCallInputWait(TypedDict, total=False):
115
- wait: Required[Annotated[ToolCallInputWaitWait, PropertyInfo(alias="Wait")]]
116
-
117
-
118
- class ToolCallInputErrorError(TypedDict, total=False):
119
- error: Required[str]
120
-
121
-
122
- class ToolCallInputError(TypedDict, total=False):
123
- error: Required[Annotated[ToolCallInputErrorError, PropertyInfo(alias="Error")]]
124
-
125
-
126
- class ToolCallInputGoogleGoogle(TypedDict, total=False):
127
- query: Required[str]
128
-
129
-
130
- class ToolCallInputGoogle(TypedDict, total=False):
131
- google: Required[Annotated[ToolCallInputGoogleGoogle, PropertyInfo(alias="Google")]]
132
-
133
-
134
- class ToolCallInputTypeType(TypedDict, total=False):
135
- flag: Required[int]
136
-
137
- input: Required[str]
138
-
139
-
140
- class ToolCallInputType(TypedDict, total=False):
141
- type: Required[Annotated[ToolCallInputTypeType, PropertyInfo(alias="Type")]]
142
-
143
-
144
- ToolCallInput: TypeAlias = Union[
145
- ToolCallInputSave,
146
- ToolCallInputScroll,
147
- ToolCallInputScrollToBottom,
148
- ToolCallInputExit,
149
- ToolCallInputClick,
150
- ToolCallInputHover,
151
- ToolCallInputWait,
152
- ToolCallInputError,
153
- ToolCallInputGoogle,
154
- ToolCallInputType,
155
- ]
156
-
157
-
158
- class ToolCallResultToolQueued(TypedDict, total=False):
159
- tool_queued: Required[Annotated[str, PropertyInfo(alias="ToolQueued")]]
160
-
161
-
162
- class ToolCallResultToolFail(TypedDict, total=False):
163
- tool_fail: Required[Annotated[str, PropertyInfo(alias="ToolFail")]]
164
-
165
-
166
- class ToolCallResultInputParseFail(TypedDict, total=False):
167
- input_parse_fail: Required[Annotated[str, PropertyInfo(alias="InputParseFail")]]
168
-
169
-
170
- class ToolCallResultSuccess(TypedDict, total=False):
171
- success: Required[Annotated[str, PropertyInfo(alias="Success")]]
172
-
173
-
174
- ToolCallResult: TypeAlias = Union[
175
- ToolCallResultToolQueued, ToolCallResultToolFail, ToolCallResultInputParseFail, ToolCallResultSuccess
176
- ]
177
-
178
-
179
- class ToolCall(TypedDict, total=False):
180
- input: Required[ToolCallInput]
181
-
182
- name: Required[
183
- Literal["Exit", "Save", "Wait", "Type", "Scroll", "ScrollToBottom", "Click", "Hover", "Error", "Google"]
184
- ]
185
-
186
- result: Optional[ToolCallResult]
@@ -1,13 +0,0 @@
1
- # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
-
3
- from __future__ import annotations
4
-
5
- from typing_extensions import Literal, Required, TypedDict
6
-
7
- __all__ = ["HumanLlmFinishJobParams"]
8
-
9
-
10
- class HumanLlmFinishJobParams(TypedDict, total=False):
11
- id: Required[str]
12
-
13
- status: Required[Literal["Queued", "Running", "Completed", "Failed"]]
@@ -1,12 +0,0 @@
1
- # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
-
3
- from __future__ import annotations
4
-
5
- from typing import Optional
6
- from typing_extensions import Literal, TypedDict
7
-
8
- __all__ = ["HumanLlmGetJobsParams"]
9
-
10
-
11
- class HumanLlmGetJobsParams(TypedDict, total=False):
12
- status: Optional[Literal["Queued", "Running", "Completed", "Failed"]]