mirascope 2.0.0__py3-none-any.whl → 2.0.0a1__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 (443) hide show
  1. mirascope/__init__.py +2 -11
  2. mirascope/graphs/__init__.py +22 -0
  3. mirascope/graphs/finite_state_machine.py +625 -0
  4. mirascope/llm/__init__.py +15 -96
  5. mirascope/llm/agents/__init__.py +15 -0
  6. mirascope/llm/agents/agent.py +97 -0
  7. mirascope/llm/agents/agent_template.py +45 -0
  8. mirascope/llm/agents/decorator.py +176 -0
  9. mirascope/llm/calls/__init__.py +1 -2
  10. mirascope/llm/calls/base_call.py +33 -0
  11. mirascope/llm/calls/calls.py +58 -84
  12. mirascope/llm/calls/decorator.py +120 -140
  13. mirascope/llm/clients/__init__.py +34 -0
  14. mirascope/llm/clients/_missing_import_stubs.py +47 -0
  15. mirascope/llm/clients/anthropic/__init__.py +25 -0
  16. mirascope/llm/{providers/openai/completions → clients/anthropic}/_utils/__init__.py +0 -2
  17. mirascope/llm/{providers → clients}/anthropic/_utils/decode.py +22 -66
  18. mirascope/llm/clients/anthropic/_utils/encode.py +243 -0
  19. mirascope/llm/clients/anthropic/clients.py +819 -0
  20. mirascope/llm/clients/anthropic/model_ids.py +8 -0
  21. mirascope/llm/{providers → clients}/base/__init__.py +5 -4
  22. mirascope/llm/{providers → clients}/base/_utils.py +17 -78
  23. mirascope/llm/{providers/base/base_provider.py → clients/base/client.py} +145 -468
  24. mirascope/llm/{models → clients/base}/params.py +37 -16
  25. mirascope/llm/clients/google/__init__.py +20 -0
  26. mirascope/llm/{providers/openai/responses → clients/google}/_utils/__init__.py +0 -2
  27. mirascope/llm/{providers → clients}/google/_utils/decode.py +22 -98
  28. mirascope/llm/{providers → clients}/google/_utils/encode.py +46 -168
  29. mirascope/llm/clients/google/clients.py +853 -0
  30. mirascope/llm/clients/google/model_ids.py +15 -0
  31. mirascope/llm/clients/openai/__init__.py +25 -0
  32. mirascope/llm/clients/openai/completions/__init__.py +28 -0
  33. mirascope/llm/{providers/google → clients/openai/completions}/_utils/__init__.py +0 -4
  34. mirascope/llm/{providers → clients}/openai/completions/_utils/decode.py +9 -74
  35. mirascope/llm/{providers → clients}/openai/completions/_utils/encode.py +52 -70
  36. mirascope/llm/clients/openai/completions/_utils/model_features.py +81 -0
  37. mirascope/llm/clients/openai/completions/clients.py +833 -0
  38. mirascope/llm/clients/openai/completions/model_ids.py +8 -0
  39. mirascope/llm/clients/openai/responses/__init__.py +26 -0
  40. mirascope/llm/clients/openai/responses/_utils/__init__.py +13 -0
  41. mirascope/llm/{providers → clients}/openai/responses/_utils/decode.py +14 -80
  42. mirascope/llm/{providers → clients}/openai/responses/_utils/encode.py +41 -92
  43. mirascope/llm/clients/openai/responses/_utils/model_features.py +87 -0
  44. mirascope/llm/clients/openai/responses/clients.py +832 -0
  45. mirascope/llm/clients/openai/responses/model_ids.py +8 -0
  46. mirascope/llm/clients/openai/shared/__init__.py +7 -0
  47. mirascope/llm/clients/openai/shared/_utils.py +55 -0
  48. mirascope/llm/clients/providers.py +175 -0
  49. mirascope/llm/content/__init__.py +2 -3
  50. mirascope/llm/content/tool_call.py +0 -6
  51. mirascope/llm/content/tool_output.py +5 -22
  52. mirascope/llm/context/_utils.py +6 -19
  53. mirascope/llm/exceptions.py +43 -298
  54. mirascope/llm/formatting/__init__.py +2 -19
  55. mirascope/llm/formatting/_utils.py +74 -0
  56. mirascope/llm/formatting/format.py +30 -219
  57. mirascope/llm/formatting/from_call_args.py +2 -2
  58. mirascope/llm/formatting/partial.py +7 -80
  59. mirascope/llm/formatting/types.py +64 -21
  60. mirascope/llm/mcp/__init__.py +2 -2
  61. mirascope/llm/mcp/client.py +118 -0
  62. mirascope/llm/messages/__init__.py +0 -3
  63. mirascope/llm/messages/message.py +5 -13
  64. mirascope/llm/models/__init__.py +2 -7
  65. mirascope/llm/models/models.py +139 -315
  66. mirascope/llm/prompts/__init__.py +12 -13
  67. mirascope/llm/prompts/_utils.py +43 -14
  68. mirascope/llm/prompts/decorator.py +204 -144
  69. mirascope/llm/prompts/protocols.py +59 -25
  70. mirascope/llm/responses/__init__.py +1 -9
  71. mirascope/llm/responses/_utils.py +12 -102
  72. mirascope/llm/responses/base_response.py +6 -18
  73. mirascope/llm/responses/base_stream_response.py +50 -173
  74. mirascope/llm/responses/finish_reason.py +0 -1
  75. mirascope/llm/responses/response.py +13 -34
  76. mirascope/llm/responses/root_response.py +29 -100
  77. mirascope/llm/responses/stream_response.py +31 -40
  78. mirascope/llm/tools/__init__.py +2 -9
  79. mirascope/llm/tools/_utils.py +3 -12
  80. mirascope/llm/tools/decorator.py +16 -25
  81. mirascope/llm/tools/protocols.py +4 -4
  82. mirascope/llm/tools/tool_schema.py +19 -87
  83. mirascope/llm/tools/toolkit.py +27 -35
  84. mirascope/llm/tools/tools.py +41 -135
  85. {mirascope-2.0.0.dist-info → mirascope-2.0.0a1.dist-info}/METADATA +13 -90
  86. mirascope-2.0.0a1.dist-info/RECORD +102 -0
  87. {mirascope-2.0.0.dist-info → mirascope-2.0.0a1.dist-info}/WHEEL +1 -1
  88. {mirascope-2.0.0.dist-info → mirascope-2.0.0a1.dist-info}/licenses/LICENSE +1 -1
  89. mirascope/_stubs.py +0 -363
  90. mirascope/api/__init__.py +0 -14
  91. mirascope/api/_generated/README.md +0 -207
  92. mirascope/api/_generated/__init__.py +0 -440
  93. mirascope/api/_generated/annotations/__init__.py +0 -33
  94. mirascope/api/_generated/annotations/client.py +0 -506
  95. mirascope/api/_generated/annotations/raw_client.py +0 -1414
  96. mirascope/api/_generated/annotations/types/__init__.py +0 -31
  97. mirascope/api/_generated/annotations/types/annotations_create_request_label.py +0 -5
  98. mirascope/api/_generated/annotations/types/annotations_create_response.py +0 -48
  99. mirascope/api/_generated/annotations/types/annotations_create_response_label.py +0 -5
  100. mirascope/api/_generated/annotations/types/annotations_get_response.py +0 -48
  101. mirascope/api/_generated/annotations/types/annotations_get_response_label.py +0 -5
  102. mirascope/api/_generated/annotations/types/annotations_list_request_label.py +0 -5
  103. mirascope/api/_generated/annotations/types/annotations_list_response.py +0 -21
  104. mirascope/api/_generated/annotations/types/annotations_list_response_annotations_item.py +0 -50
  105. mirascope/api/_generated/annotations/types/annotations_list_response_annotations_item_label.py +0 -5
  106. mirascope/api/_generated/annotations/types/annotations_update_request_label.py +0 -5
  107. mirascope/api/_generated/annotations/types/annotations_update_response.py +0 -48
  108. mirascope/api/_generated/annotations/types/annotations_update_response_label.py +0 -5
  109. mirascope/api/_generated/api_keys/__init__.py +0 -17
  110. mirascope/api/_generated/api_keys/client.py +0 -530
  111. mirascope/api/_generated/api_keys/raw_client.py +0 -1236
  112. mirascope/api/_generated/api_keys/types/__init__.py +0 -15
  113. mirascope/api/_generated/api_keys/types/api_keys_create_response.py +0 -28
  114. mirascope/api/_generated/api_keys/types/api_keys_get_response.py +0 -27
  115. mirascope/api/_generated/api_keys/types/api_keys_list_all_for_org_response_item.py +0 -40
  116. mirascope/api/_generated/api_keys/types/api_keys_list_response_item.py +0 -27
  117. mirascope/api/_generated/client.py +0 -211
  118. mirascope/api/_generated/core/__init__.py +0 -52
  119. mirascope/api/_generated/core/api_error.py +0 -23
  120. mirascope/api/_generated/core/client_wrapper.py +0 -46
  121. mirascope/api/_generated/core/datetime_utils.py +0 -28
  122. mirascope/api/_generated/core/file.py +0 -67
  123. mirascope/api/_generated/core/force_multipart.py +0 -16
  124. mirascope/api/_generated/core/http_client.py +0 -543
  125. mirascope/api/_generated/core/http_response.py +0 -55
  126. mirascope/api/_generated/core/jsonable_encoder.py +0 -100
  127. mirascope/api/_generated/core/pydantic_utilities.py +0 -255
  128. mirascope/api/_generated/core/query_encoder.py +0 -58
  129. mirascope/api/_generated/core/remove_none_from_dict.py +0 -11
  130. mirascope/api/_generated/core/request_options.py +0 -35
  131. mirascope/api/_generated/core/serialization.py +0 -276
  132. mirascope/api/_generated/docs/__init__.py +0 -4
  133. mirascope/api/_generated/docs/client.py +0 -91
  134. mirascope/api/_generated/docs/raw_client.py +0 -178
  135. mirascope/api/_generated/environment.py +0 -9
  136. mirascope/api/_generated/environments/__init__.py +0 -23
  137. mirascope/api/_generated/environments/client.py +0 -649
  138. mirascope/api/_generated/environments/raw_client.py +0 -1567
  139. mirascope/api/_generated/environments/types/__init__.py +0 -25
  140. mirascope/api/_generated/environments/types/environments_create_response.py +0 -24
  141. mirascope/api/_generated/environments/types/environments_get_analytics_response.py +0 -60
  142. mirascope/api/_generated/environments/types/environments_get_analytics_response_top_functions_item.py +0 -24
  143. mirascope/api/_generated/environments/types/environments_get_analytics_response_top_models_item.py +0 -22
  144. mirascope/api/_generated/environments/types/environments_get_response.py +0 -24
  145. mirascope/api/_generated/environments/types/environments_list_response_item.py +0 -24
  146. mirascope/api/_generated/environments/types/environments_update_response.py +0 -24
  147. mirascope/api/_generated/errors/__init__.py +0 -25
  148. mirascope/api/_generated/errors/bad_request_error.py +0 -14
  149. mirascope/api/_generated/errors/conflict_error.py +0 -14
  150. mirascope/api/_generated/errors/forbidden_error.py +0 -11
  151. mirascope/api/_generated/errors/internal_server_error.py +0 -10
  152. mirascope/api/_generated/errors/not_found_error.py +0 -11
  153. mirascope/api/_generated/errors/payment_required_error.py +0 -15
  154. mirascope/api/_generated/errors/service_unavailable_error.py +0 -14
  155. mirascope/api/_generated/errors/too_many_requests_error.py +0 -15
  156. mirascope/api/_generated/errors/unauthorized_error.py +0 -11
  157. mirascope/api/_generated/functions/__init__.py +0 -39
  158. mirascope/api/_generated/functions/client.py +0 -647
  159. mirascope/api/_generated/functions/raw_client.py +0 -1890
  160. mirascope/api/_generated/functions/types/__init__.py +0 -53
  161. mirascope/api/_generated/functions/types/functions_create_request_dependencies_value.py +0 -20
  162. mirascope/api/_generated/functions/types/functions_create_response.py +0 -37
  163. mirascope/api/_generated/functions/types/functions_create_response_dependencies_value.py +0 -20
  164. mirascope/api/_generated/functions/types/functions_find_by_hash_response.py +0 -39
  165. mirascope/api/_generated/functions/types/functions_find_by_hash_response_dependencies_value.py +0 -20
  166. mirascope/api/_generated/functions/types/functions_get_by_env_response.py +0 -53
  167. mirascope/api/_generated/functions/types/functions_get_by_env_response_dependencies_value.py +0 -22
  168. mirascope/api/_generated/functions/types/functions_get_response.py +0 -37
  169. mirascope/api/_generated/functions/types/functions_get_response_dependencies_value.py +0 -20
  170. mirascope/api/_generated/functions/types/functions_list_by_env_response.py +0 -25
  171. mirascope/api/_generated/functions/types/functions_list_by_env_response_functions_item.py +0 -56
  172. mirascope/api/_generated/functions/types/functions_list_by_env_response_functions_item_dependencies_value.py +0 -22
  173. mirascope/api/_generated/functions/types/functions_list_response.py +0 -21
  174. mirascope/api/_generated/functions/types/functions_list_response_functions_item.py +0 -41
  175. mirascope/api/_generated/functions/types/functions_list_response_functions_item_dependencies_value.py +0 -20
  176. mirascope/api/_generated/health/__init__.py +0 -7
  177. mirascope/api/_generated/health/client.py +0 -92
  178. mirascope/api/_generated/health/raw_client.py +0 -175
  179. mirascope/api/_generated/health/types/__init__.py +0 -8
  180. mirascope/api/_generated/health/types/health_check_response.py +0 -22
  181. mirascope/api/_generated/health/types/health_check_response_status.py +0 -5
  182. mirascope/api/_generated/organization_invitations/__init__.py +0 -33
  183. mirascope/api/_generated/organization_invitations/client.py +0 -546
  184. mirascope/api/_generated/organization_invitations/raw_client.py +0 -1519
  185. mirascope/api/_generated/organization_invitations/types/__init__.py +0 -53
  186. mirascope/api/_generated/organization_invitations/types/organization_invitations_accept_response.py +0 -34
  187. mirascope/api/_generated/organization_invitations/types/organization_invitations_accept_response_role.py +0 -7
  188. mirascope/api/_generated/organization_invitations/types/organization_invitations_create_request_role.py +0 -7
  189. mirascope/api/_generated/organization_invitations/types/organization_invitations_create_response.py +0 -48
  190. mirascope/api/_generated/organization_invitations/types/organization_invitations_create_response_role.py +0 -7
  191. mirascope/api/_generated/organization_invitations/types/organization_invitations_create_response_status.py +0 -7
  192. mirascope/api/_generated/organization_invitations/types/organization_invitations_get_response.py +0 -48
  193. mirascope/api/_generated/organization_invitations/types/organization_invitations_get_response_role.py +0 -7
  194. mirascope/api/_generated/organization_invitations/types/organization_invitations_get_response_status.py +0 -7
  195. mirascope/api/_generated/organization_invitations/types/organization_invitations_list_response_item.py +0 -48
  196. mirascope/api/_generated/organization_invitations/types/organization_invitations_list_response_item_role.py +0 -7
  197. mirascope/api/_generated/organization_invitations/types/organization_invitations_list_response_item_status.py +0 -7
  198. mirascope/api/_generated/organization_memberships/__init__.py +0 -19
  199. mirascope/api/_generated/organization_memberships/client.py +0 -302
  200. mirascope/api/_generated/organization_memberships/raw_client.py +0 -736
  201. mirascope/api/_generated/organization_memberships/types/__init__.py +0 -27
  202. mirascope/api/_generated/organization_memberships/types/organization_memberships_list_response_item.py +0 -33
  203. mirascope/api/_generated/organization_memberships/types/organization_memberships_list_response_item_role.py +0 -7
  204. mirascope/api/_generated/organization_memberships/types/organization_memberships_update_request_role.py +0 -7
  205. mirascope/api/_generated/organization_memberships/types/organization_memberships_update_response.py +0 -31
  206. mirascope/api/_generated/organization_memberships/types/organization_memberships_update_response_role.py +0 -7
  207. mirascope/api/_generated/organizations/__init__.py +0 -51
  208. mirascope/api/_generated/organizations/client.py +0 -869
  209. mirascope/api/_generated/organizations/raw_client.py +0 -2593
  210. mirascope/api/_generated/organizations/types/__init__.py +0 -71
  211. mirascope/api/_generated/organizations/types/organizations_create_payment_intent_response.py +0 -24
  212. mirascope/api/_generated/organizations/types/organizations_create_response.py +0 -26
  213. mirascope/api/_generated/organizations/types/organizations_create_response_role.py +0 -5
  214. mirascope/api/_generated/organizations/types/organizations_get_response.py +0 -26
  215. mirascope/api/_generated/organizations/types/organizations_get_response_role.py +0 -5
  216. mirascope/api/_generated/organizations/types/organizations_list_response_item.py +0 -26
  217. mirascope/api/_generated/organizations/types/organizations_list_response_item_role.py +0 -5
  218. mirascope/api/_generated/organizations/types/organizations_preview_subscription_change_request_target_plan.py +0 -7
  219. mirascope/api/_generated/organizations/types/organizations_preview_subscription_change_response.py +0 -47
  220. mirascope/api/_generated/organizations/types/organizations_preview_subscription_change_response_validation_errors_item.py +0 -33
  221. mirascope/api/_generated/organizations/types/organizations_preview_subscription_change_response_validation_errors_item_resource.py +0 -7
  222. mirascope/api/_generated/organizations/types/organizations_router_balance_response.py +0 -24
  223. mirascope/api/_generated/organizations/types/organizations_subscription_response.py +0 -53
  224. mirascope/api/_generated/organizations/types/organizations_subscription_response_current_plan.py +0 -7
  225. mirascope/api/_generated/organizations/types/organizations_subscription_response_payment_method.py +0 -26
  226. mirascope/api/_generated/organizations/types/organizations_subscription_response_scheduled_change.py +0 -34
  227. mirascope/api/_generated/organizations/types/organizations_subscription_response_scheduled_change_target_plan.py +0 -7
  228. mirascope/api/_generated/organizations/types/organizations_update_response.py +0 -26
  229. mirascope/api/_generated/organizations/types/organizations_update_response_role.py +0 -5
  230. mirascope/api/_generated/organizations/types/organizations_update_subscription_request_target_plan.py +0 -7
  231. mirascope/api/_generated/organizations/types/organizations_update_subscription_response.py +0 -35
  232. mirascope/api/_generated/project_memberships/__init__.py +0 -25
  233. mirascope/api/_generated/project_memberships/client.py +0 -437
  234. mirascope/api/_generated/project_memberships/raw_client.py +0 -1039
  235. mirascope/api/_generated/project_memberships/types/__init__.py +0 -29
  236. mirascope/api/_generated/project_memberships/types/project_memberships_create_request_role.py +0 -7
  237. mirascope/api/_generated/project_memberships/types/project_memberships_create_response.py +0 -35
  238. mirascope/api/_generated/project_memberships/types/project_memberships_create_response_role.py +0 -7
  239. mirascope/api/_generated/project_memberships/types/project_memberships_list_response_item.py +0 -33
  240. mirascope/api/_generated/project_memberships/types/project_memberships_list_response_item_role.py +0 -7
  241. mirascope/api/_generated/project_memberships/types/project_memberships_update_request_role.py +0 -7
  242. mirascope/api/_generated/project_memberships/types/project_memberships_update_response.py +0 -35
  243. mirascope/api/_generated/project_memberships/types/project_memberships_update_response_role.py +0 -7
  244. mirascope/api/_generated/projects/__init__.py +0 -7
  245. mirascope/api/_generated/projects/client.py +0 -428
  246. mirascope/api/_generated/projects/raw_client.py +0 -1302
  247. mirascope/api/_generated/projects/types/__init__.py +0 -10
  248. mirascope/api/_generated/projects/types/projects_create_response.py +0 -25
  249. mirascope/api/_generated/projects/types/projects_get_response.py +0 -25
  250. mirascope/api/_generated/projects/types/projects_list_response_item.py +0 -25
  251. mirascope/api/_generated/projects/types/projects_update_response.py +0 -25
  252. mirascope/api/_generated/reference.md +0 -4915
  253. mirascope/api/_generated/tags/__init__.py +0 -19
  254. mirascope/api/_generated/tags/client.py +0 -504
  255. mirascope/api/_generated/tags/raw_client.py +0 -1288
  256. mirascope/api/_generated/tags/types/__init__.py +0 -17
  257. mirascope/api/_generated/tags/types/tags_create_response.py +0 -41
  258. mirascope/api/_generated/tags/types/tags_get_response.py +0 -41
  259. mirascope/api/_generated/tags/types/tags_list_response.py +0 -23
  260. mirascope/api/_generated/tags/types/tags_list_response_tags_item.py +0 -41
  261. mirascope/api/_generated/tags/types/tags_update_response.py +0 -41
  262. mirascope/api/_generated/token_cost/__init__.py +0 -7
  263. mirascope/api/_generated/token_cost/client.py +0 -160
  264. mirascope/api/_generated/token_cost/raw_client.py +0 -264
  265. mirascope/api/_generated/token_cost/types/__init__.py +0 -8
  266. mirascope/api/_generated/token_cost/types/token_cost_calculate_request_usage.py +0 -54
  267. mirascope/api/_generated/token_cost/types/token_cost_calculate_response.py +0 -52
  268. mirascope/api/_generated/traces/__init__.py +0 -97
  269. mirascope/api/_generated/traces/client.py +0 -1103
  270. mirascope/api/_generated/traces/raw_client.py +0 -2322
  271. mirascope/api/_generated/traces/types/__init__.py +0 -155
  272. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item.py +0 -29
  273. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_resource.py +0 -27
  274. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_resource_attributes_item.py +0 -23
  275. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_resource_attributes_item_value.py +0 -38
  276. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_resource_attributes_item_value_array_value.py +0 -19
  277. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_resource_attributes_item_value_kvlist_value.py +0 -22
  278. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_resource_attributes_item_value_kvlist_value_values_item.py +0 -20
  279. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item.py +0 -29
  280. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item_scope.py +0 -31
  281. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item_scope_attributes_item.py +0 -23
  282. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item_scope_attributes_item_value.py +0 -38
  283. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item_scope_attributes_item_value_array_value.py +0 -19
  284. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item_scope_attributes_item_value_kvlist_value.py +0 -22
  285. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item_scope_attributes_item_value_kvlist_value_values_item.py +0 -22
  286. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item_spans_item.py +0 -48
  287. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item_spans_item_attributes_item.py +0 -23
  288. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item_spans_item_attributes_item_value.py +0 -38
  289. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item_spans_item_attributes_item_value_array_value.py +0 -19
  290. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item_spans_item_attributes_item_value_kvlist_value.py +0 -24
  291. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item_spans_item_attributes_item_value_kvlist_value_values_item.py +0 -22
  292. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item_spans_item_status.py +0 -20
  293. mirascope/api/_generated/traces/types/traces_create_response.py +0 -24
  294. mirascope/api/_generated/traces/types/traces_create_response_partial_success.py +0 -22
  295. mirascope/api/_generated/traces/types/traces_get_analytics_summary_response.py +0 -60
  296. mirascope/api/_generated/traces/types/traces_get_analytics_summary_response_top_functions_item.py +0 -24
  297. mirascope/api/_generated/traces/types/traces_get_analytics_summary_response_top_models_item.py +0 -22
  298. mirascope/api/_generated/traces/types/traces_get_trace_detail_by_env_response.py +0 -33
  299. mirascope/api/_generated/traces/types/traces_get_trace_detail_by_env_response_spans_item.py +0 -88
  300. mirascope/api/_generated/traces/types/traces_get_trace_detail_response.py +0 -33
  301. mirascope/api/_generated/traces/types/traces_get_trace_detail_response_spans_item.py +0 -88
  302. mirascope/api/_generated/traces/types/traces_list_by_function_hash_response.py +0 -25
  303. mirascope/api/_generated/traces/types/traces_list_by_function_hash_response_traces_item.py +0 -44
  304. mirascope/api/_generated/traces/types/traces_search_by_env_request_attribute_filters_item.py +0 -26
  305. mirascope/api/_generated/traces/types/traces_search_by_env_request_attribute_filters_item_operator.py +0 -7
  306. mirascope/api/_generated/traces/types/traces_search_by_env_request_sort_by.py +0 -7
  307. mirascope/api/_generated/traces/types/traces_search_by_env_request_sort_order.py +0 -7
  308. mirascope/api/_generated/traces/types/traces_search_by_env_response.py +0 -26
  309. mirascope/api/_generated/traces/types/traces_search_by_env_response_spans_item.py +0 -50
  310. mirascope/api/_generated/traces/types/traces_search_request_attribute_filters_item.py +0 -26
  311. mirascope/api/_generated/traces/types/traces_search_request_attribute_filters_item_operator.py +0 -7
  312. mirascope/api/_generated/traces/types/traces_search_request_sort_by.py +0 -7
  313. mirascope/api/_generated/traces/types/traces_search_request_sort_order.py +0 -5
  314. mirascope/api/_generated/traces/types/traces_search_response.py +0 -26
  315. mirascope/api/_generated/traces/types/traces_search_response_spans_item.py +0 -50
  316. mirascope/api/_generated/types/__init__.py +0 -85
  317. mirascope/api/_generated/types/already_exists_error.py +0 -22
  318. mirascope/api/_generated/types/already_exists_error_tag.py +0 -5
  319. mirascope/api/_generated/types/bad_request_error_body.py +0 -50
  320. mirascope/api/_generated/types/click_house_error.py +0 -22
  321. mirascope/api/_generated/types/database_error.py +0 -22
  322. mirascope/api/_generated/types/database_error_tag.py +0 -5
  323. mirascope/api/_generated/types/date.py +0 -3
  324. mirascope/api/_generated/types/http_api_decode_error.py +0 -27
  325. mirascope/api/_generated/types/http_api_decode_error_tag.py +0 -5
  326. mirascope/api/_generated/types/immutable_resource_error.py +0 -22
  327. mirascope/api/_generated/types/internal_server_error_body.py +0 -49
  328. mirascope/api/_generated/types/issue.py +0 -38
  329. mirascope/api/_generated/types/issue_tag.py +0 -10
  330. mirascope/api/_generated/types/not_found_error_body.py +0 -22
  331. mirascope/api/_generated/types/not_found_error_tag.py +0 -5
  332. mirascope/api/_generated/types/number_from_string.py +0 -3
  333. mirascope/api/_generated/types/permission_denied_error.py +0 -22
  334. mirascope/api/_generated/types/permission_denied_error_tag.py +0 -5
  335. mirascope/api/_generated/types/plan_limit_exceeded_error.py +0 -32
  336. mirascope/api/_generated/types/plan_limit_exceeded_error_tag.py +0 -7
  337. mirascope/api/_generated/types/pricing_unavailable_error.py +0 -23
  338. mirascope/api/_generated/types/property_key.py +0 -7
  339. mirascope/api/_generated/types/property_key_key.py +0 -25
  340. mirascope/api/_generated/types/property_key_key_tag.py +0 -5
  341. mirascope/api/_generated/types/rate_limit_error.py +0 -31
  342. mirascope/api/_generated/types/rate_limit_error_tag.py +0 -5
  343. mirascope/api/_generated/types/service_unavailable_error_body.py +0 -24
  344. mirascope/api/_generated/types/service_unavailable_error_tag.py +0 -7
  345. mirascope/api/_generated/types/stripe_error.py +0 -20
  346. mirascope/api/_generated/types/subscription_past_due_error.py +0 -31
  347. mirascope/api/_generated/types/subscription_past_due_error_tag.py +0 -7
  348. mirascope/api/_generated/types/unauthorized_error_body.py +0 -21
  349. mirascope/api/_generated/types/unauthorized_error_tag.py +0 -5
  350. mirascope/api/client.py +0 -255
  351. mirascope/api/settings.py +0 -99
  352. mirascope/llm/formatting/output_parser.py +0 -178
  353. mirascope/llm/formatting/primitives.py +0 -192
  354. mirascope/llm/mcp/mcp_client.py +0 -130
  355. mirascope/llm/messages/_utils.py +0 -34
  356. mirascope/llm/models/thinking_config.py +0 -61
  357. mirascope/llm/prompts/prompts.py +0 -487
  358. mirascope/llm/providers/__init__.py +0 -62
  359. mirascope/llm/providers/anthropic/__init__.py +0 -11
  360. mirascope/llm/providers/anthropic/_utils/__init__.py +0 -27
  361. mirascope/llm/providers/anthropic/_utils/beta_decode.py +0 -282
  362. mirascope/llm/providers/anthropic/_utils/beta_encode.py +0 -266
  363. mirascope/llm/providers/anthropic/_utils/encode.py +0 -418
  364. mirascope/llm/providers/anthropic/_utils/errors.py +0 -46
  365. mirascope/llm/providers/anthropic/beta_provider.py +0 -374
  366. mirascope/llm/providers/anthropic/model_id.py +0 -23
  367. mirascope/llm/providers/anthropic/model_info.py +0 -87
  368. mirascope/llm/providers/anthropic/provider.py +0 -479
  369. mirascope/llm/providers/google/__init__.py +0 -6
  370. mirascope/llm/providers/google/_utils/errors.py +0 -50
  371. mirascope/llm/providers/google/model_id.py +0 -22
  372. mirascope/llm/providers/google/model_info.py +0 -63
  373. mirascope/llm/providers/google/provider.py +0 -492
  374. mirascope/llm/providers/mirascope/__init__.py +0 -5
  375. mirascope/llm/providers/mirascope/_utils.py +0 -73
  376. mirascope/llm/providers/mirascope/provider.py +0 -349
  377. mirascope/llm/providers/mlx/__init__.py +0 -9
  378. mirascope/llm/providers/mlx/_utils.py +0 -141
  379. mirascope/llm/providers/mlx/encoding/__init__.py +0 -8
  380. mirascope/llm/providers/mlx/encoding/base.py +0 -72
  381. mirascope/llm/providers/mlx/encoding/transformers.py +0 -150
  382. mirascope/llm/providers/mlx/mlx.py +0 -254
  383. mirascope/llm/providers/mlx/model_id.py +0 -17
  384. mirascope/llm/providers/mlx/provider.py +0 -452
  385. mirascope/llm/providers/model_id.py +0 -16
  386. mirascope/llm/providers/ollama/__init__.py +0 -7
  387. mirascope/llm/providers/ollama/provider.py +0 -71
  388. mirascope/llm/providers/openai/__init__.py +0 -15
  389. mirascope/llm/providers/openai/_utils/__init__.py +0 -5
  390. mirascope/llm/providers/openai/_utils/errors.py +0 -46
  391. mirascope/llm/providers/openai/completions/__init__.py +0 -7
  392. mirascope/llm/providers/openai/completions/base_provider.py +0 -542
  393. mirascope/llm/providers/openai/completions/provider.py +0 -22
  394. mirascope/llm/providers/openai/model_id.py +0 -31
  395. mirascope/llm/providers/openai/model_info.py +0 -303
  396. mirascope/llm/providers/openai/provider.py +0 -441
  397. mirascope/llm/providers/openai/responses/__init__.py +0 -5
  398. mirascope/llm/providers/openai/responses/provider.py +0 -513
  399. mirascope/llm/providers/provider_id.py +0 -24
  400. mirascope/llm/providers/provider_registry.py +0 -299
  401. mirascope/llm/providers/together/__init__.py +0 -7
  402. mirascope/llm/providers/together/provider.py +0 -40
  403. mirascope/llm/responses/usage.py +0 -95
  404. mirascope/ops/__init__.py +0 -111
  405. mirascope/ops/_internal/__init__.py +0 -5
  406. mirascope/ops/_internal/closure.py +0 -1169
  407. mirascope/ops/_internal/configuration.py +0 -177
  408. mirascope/ops/_internal/context.py +0 -76
  409. mirascope/ops/_internal/exporters/__init__.py +0 -26
  410. mirascope/ops/_internal/exporters/exporters.py +0 -395
  411. mirascope/ops/_internal/exporters/processors.py +0 -104
  412. mirascope/ops/_internal/exporters/types.py +0 -165
  413. mirascope/ops/_internal/exporters/utils.py +0 -29
  414. mirascope/ops/_internal/instrumentation/__init__.py +0 -8
  415. mirascope/ops/_internal/instrumentation/llm/__init__.py +0 -8
  416. mirascope/ops/_internal/instrumentation/llm/common.py +0 -530
  417. mirascope/ops/_internal/instrumentation/llm/cost.py +0 -190
  418. mirascope/ops/_internal/instrumentation/llm/encode.py +0 -238
  419. mirascope/ops/_internal/instrumentation/llm/gen_ai_types/__init__.py +0 -38
  420. mirascope/ops/_internal/instrumentation/llm/gen_ai_types/gen_ai_input_messages.py +0 -31
  421. mirascope/ops/_internal/instrumentation/llm/gen_ai_types/gen_ai_output_messages.py +0 -38
  422. mirascope/ops/_internal/instrumentation/llm/gen_ai_types/gen_ai_system_instructions.py +0 -18
  423. mirascope/ops/_internal/instrumentation/llm/gen_ai_types/shared.py +0 -100
  424. mirascope/ops/_internal/instrumentation/llm/llm.py +0 -161
  425. mirascope/ops/_internal/instrumentation/llm/model.py +0 -1798
  426. mirascope/ops/_internal/instrumentation/llm/response.py +0 -521
  427. mirascope/ops/_internal/instrumentation/llm/serialize.py +0 -300
  428. mirascope/ops/_internal/propagation.py +0 -198
  429. mirascope/ops/_internal/protocols.py +0 -133
  430. mirascope/ops/_internal/session.py +0 -139
  431. mirascope/ops/_internal/spans.py +0 -232
  432. mirascope/ops/_internal/traced_calls.py +0 -375
  433. mirascope/ops/_internal/traced_functions.py +0 -523
  434. mirascope/ops/_internal/tracing.py +0 -353
  435. mirascope/ops/_internal/types.py +0 -13
  436. mirascope/ops/_internal/utils.py +0 -123
  437. mirascope/ops/_internal/versioned_calls.py +0 -512
  438. mirascope/ops/_internal/versioned_functions.py +0 -357
  439. mirascope/ops/_internal/versioning.py +0 -303
  440. mirascope/ops/exceptions.py +0 -21
  441. mirascope-2.0.0.dist-info/RECORD +0 -423
  442. /mirascope/llm/{providers → clients}/base/kwargs.py +0 -0
  443. /mirascope/llm/{providers → clients}/google/message.py +0 -0
@@ -1,15 +0,0 @@
1
- # This file was auto-generated by Fern from our API Definition.
2
-
3
- # isort: skip_file
4
-
5
- from .api_keys_create_response import ApiKeysCreateResponse
6
- from .api_keys_get_response import ApiKeysGetResponse
7
- from .api_keys_list_all_for_org_response_item import ApiKeysListAllForOrgResponseItem
8
- from .api_keys_list_response_item import ApiKeysListResponseItem
9
-
10
- __all__ = [
11
- "ApiKeysCreateResponse",
12
- "ApiKeysGetResponse",
13
- "ApiKeysListAllForOrgResponseItem",
14
- "ApiKeysListResponseItem",
15
- ]
@@ -1,28 +0,0 @@
1
- # This file was auto-generated by Fern from our API Definition.
2
-
3
- import typing
4
-
5
- import pydantic
6
- import typing_extensions
7
- from ...core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
8
- from ...core.serialization import FieldMetadata
9
-
10
-
11
- class ApiKeysCreateResponse(UniversalBaseModel):
12
- id: str
13
- name: str
14
- key_prefix: typing_extensions.Annotated[str, FieldMetadata(alias="keyPrefix")]
15
- environment_id: typing_extensions.Annotated[str, FieldMetadata(alias="environmentId")]
16
- owner_id: typing_extensions.Annotated[str, FieldMetadata(alias="ownerId")]
17
- created_at: typing_extensions.Annotated[typing.Optional[str], FieldMetadata(alias="createdAt")] = None
18
- last_used_at: typing_extensions.Annotated[typing.Optional[str], FieldMetadata(alias="lastUsedAt")] = None
19
- key: str
20
-
21
- if IS_PYDANTIC_V2:
22
- model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True) # type: ignore # Pydantic v2
23
- else:
24
-
25
- class Config:
26
- frozen = True
27
- smart_union = True
28
- extra = pydantic.Extra.allow
@@ -1,27 +0,0 @@
1
- # This file was auto-generated by Fern from our API Definition.
2
-
3
- import typing
4
-
5
- import pydantic
6
- import typing_extensions
7
- from ...core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
8
- from ...core.serialization import FieldMetadata
9
-
10
-
11
- class ApiKeysGetResponse(UniversalBaseModel):
12
- id: str
13
- name: str
14
- key_prefix: typing_extensions.Annotated[str, FieldMetadata(alias="keyPrefix")]
15
- environment_id: typing_extensions.Annotated[str, FieldMetadata(alias="environmentId")]
16
- owner_id: typing_extensions.Annotated[str, FieldMetadata(alias="ownerId")]
17
- created_at: typing_extensions.Annotated[typing.Optional[str], FieldMetadata(alias="createdAt")] = None
18
- last_used_at: typing_extensions.Annotated[typing.Optional[str], FieldMetadata(alias="lastUsedAt")] = None
19
-
20
- if IS_PYDANTIC_V2:
21
- model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True) # type: ignore # Pydantic v2
22
- else:
23
-
24
- class Config:
25
- frozen = True
26
- smart_union = True
27
- extra = pydantic.Extra.allow
@@ -1,40 +0,0 @@
1
- # This file was auto-generated by Fern from our API Definition.
2
-
3
- import typing
4
-
5
- import pydantic
6
- import typing_extensions
7
- from ...core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
8
- from ...core.serialization import FieldMetadata
9
-
10
-
11
- class ApiKeysListAllForOrgResponseItem(UniversalBaseModel):
12
- id: str
13
- name: str
14
- key_prefix: typing_extensions.Annotated[str, FieldMetadata(alias="keyPrefix")]
15
- environment_id: typing_extensions.Annotated[
16
- str, FieldMetadata(alias="environmentId")
17
- ]
18
- owner_id: typing_extensions.Annotated[str, FieldMetadata(alias="ownerId")]
19
- created_at: typing_extensions.Annotated[
20
- typing.Optional[str], FieldMetadata(alias="createdAt")
21
- ] = None
22
- last_used_at: typing_extensions.Annotated[
23
- typing.Optional[str], FieldMetadata(alias="lastUsedAt")
24
- ] = None
25
- project_id: typing_extensions.Annotated[str, FieldMetadata(alias="projectId")]
26
- project_name: typing_extensions.Annotated[str, FieldMetadata(alias="projectName")]
27
- environment_name: typing_extensions.Annotated[
28
- str, FieldMetadata(alias="environmentName")
29
- ]
30
-
31
- if IS_PYDANTIC_V2:
32
- model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
33
- extra="allow", frozen=True
34
- ) # type: ignore # Pydantic v2
35
- else:
36
-
37
- class Config:
38
- frozen = True
39
- smart_union = True
40
- extra = pydantic.Extra.allow
@@ -1,27 +0,0 @@
1
- # This file was auto-generated by Fern from our API Definition.
2
-
3
- import typing
4
-
5
- import pydantic
6
- import typing_extensions
7
- from ...core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
8
- from ...core.serialization import FieldMetadata
9
-
10
-
11
- class ApiKeysListResponseItem(UniversalBaseModel):
12
- id: str
13
- name: str
14
- key_prefix: typing_extensions.Annotated[str, FieldMetadata(alias="keyPrefix")]
15
- environment_id: typing_extensions.Annotated[str, FieldMetadata(alias="environmentId")]
16
- owner_id: typing_extensions.Annotated[str, FieldMetadata(alias="ownerId")]
17
- created_at: typing_extensions.Annotated[typing.Optional[str], FieldMetadata(alias="createdAt")] = None
18
- last_used_at: typing_extensions.Annotated[typing.Optional[str], FieldMetadata(alias="lastUsedAt")] = None
19
-
20
- if IS_PYDANTIC_V2:
21
- model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True) # type: ignore # Pydantic v2
22
- else:
23
-
24
- class Config:
25
- frozen = True
26
- smart_union = True
27
- extra = pydantic.Extra.allow
@@ -1,211 +0,0 @@
1
- # This file was auto-generated by Fern from our API Definition.
2
-
3
- import typing
4
-
5
- import httpx
6
- from .annotations.client import AnnotationsClient, AsyncAnnotationsClient
7
- from .api_keys.client import ApiKeysClient, AsyncApiKeysClient
8
- from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
9
- from .docs.client import AsyncDocsClient, DocsClient
10
- from .environment import MirascopeEnvironment
11
- from .environments.client import AsyncEnvironmentsClient, EnvironmentsClient
12
- from .functions.client import AsyncFunctionsClient, FunctionsClient
13
- from .health.client import AsyncHealthClient, HealthClient
14
- from .organization_invitations.client import (
15
- AsyncOrganizationInvitationsClient,
16
- OrganizationInvitationsClient,
17
- )
18
- from .organization_memberships.client import (
19
- AsyncOrganizationMembershipsClient,
20
- OrganizationMembershipsClient,
21
- )
22
- from .organizations.client import AsyncOrganizationsClient, OrganizationsClient
23
- from .project_memberships.client import (
24
- AsyncProjectMembershipsClient,
25
- ProjectMembershipsClient,
26
- )
27
- from .projects.client import AsyncProjectsClient, ProjectsClient
28
- from .tags.client import AsyncTagsClient, TagsClient
29
- from .token_cost.client import AsyncTokenCostClient, TokenCostClient
30
- from .traces.client import AsyncTracesClient, TracesClient
31
-
32
-
33
- class Mirascope:
34
- """
35
- Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propagate to these functions.
36
-
37
- Parameters
38
- ----------
39
- base_url : typing.Optional[str]
40
- The base url to use for requests from the client.
41
-
42
- environment : MirascopeEnvironment
43
- The environment to use for requests from the client. from .environment import MirascopeEnvironment
44
-
45
-
46
-
47
- Defaults to MirascopeEnvironment.PRODUCTION
48
-
49
-
50
-
51
- timeout : typing.Optional[float]
52
- The timeout to be used, in seconds, for requests. By default the timeout is 180 seconds, unless a custom httpx client is used, in which case this default is not enforced.
53
-
54
- follow_redirects : typing.Optional[bool]
55
- Whether the default httpx client follows redirects or not, this is irrelevant if a custom httpx client is passed in.
56
-
57
- httpx_client : typing.Optional[httpx.Client]
58
- The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration.
59
-
60
- Examples
61
- --------
62
- from mirascope.api._generated import Mirascope
63
-
64
- client = Mirascope()
65
- """
66
-
67
- def __init__(
68
- self,
69
- *,
70
- base_url: typing.Optional[str] = None,
71
- environment: MirascopeEnvironment = MirascopeEnvironment.PRODUCTION,
72
- timeout: typing.Optional[float] = None,
73
- follow_redirects: typing.Optional[bool] = True,
74
- httpx_client: typing.Optional[httpx.Client] = None,
75
- ):
76
- _defaulted_timeout = (
77
- timeout
78
- if timeout is not None
79
- else 180
80
- if httpx_client is None
81
- else httpx_client.timeout.read
82
- )
83
- self._client_wrapper = SyncClientWrapper(
84
- base_url=_get_base_url(base_url=base_url, environment=environment),
85
- httpx_client=httpx_client
86
- if httpx_client is not None
87
- else httpx.Client(
88
- timeout=_defaulted_timeout, follow_redirects=follow_redirects
89
- )
90
- if follow_redirects is not None
91
- else httpx.Client(timeout=_defaulted_timeout),
92
- timeout=_defaulted_timeout,
93
- )
94
- self.health = HealthClient(client_wrapper=self._client_wrapper)
95
- self.traces = TracesClient(client_wrapper=self._client_wrapper)
96
- self.docs = DocsClient(client_wrapper=self._client_wrapper)
97
- self.organizations = OrganizationsClient(client_wrapper=self._client_wrapper)
98
- self.organization_invitations = OrganizationInvitationsClient(
99
- client_wrapper=self._client_wrapper
100
- )
101
- self.organization_memberships = OrganizationMembershipsClient(
102
- client_wrapper=self._client_wrapper
103
- )
104
- self.projects = ProjectsClient(client_wrapper=self._client_wrapper)
105
- self.project_memberships = ProjectMembershipsClient(
106
- client_wrapper=self._client_wrapper
107
- )
108
- self.environments = EnvironmentsClient(client_wrapper=self._client_wrapper)
109
- self.api_keys = ApiKeysClient(client_wrapper=self._client_wrapper)
110
- self.functions = FunctionsClient(client_wrapper=self._client_wrapper)
111
- self.annotations = AnnotationsClient(client_wrapper=self._client_wrapper)
112
- self.tags = TagsClient(client_wrapper=self._client_wrapper)
113
- self.token_cost = TokenCostClient(client_wrapper=self._client_wrapper)
114
-
115
-
116
- class AsyncMirascope:
117
- """
118
- Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propagate to these functions.
119
-
120
- Parameters
121
- ----------
122
- base_url : typing.Optional[str]
123
- The base url to use for requests from the client.
124
-
125
- environment : MirascopeEnvironment
126
- The environment to use for requests from the client. from .environment import MirascopeEnvironment
127
-
128
-
129
-
130
- Defaults to MirascopeEnvironment.PRODUCTION
131
-
132
-
133
-
134
- timeout : typing.Optional[float]
135
- The timeout to be used, in seconds, for requests. By default the timeout is 180 seconds, unless a custom httpx client is used, in which case this default is not enforced.
136
-
137
- follow_redirects : typing.Optional[bool]
138
- Whether the default httpx client follows redirects or not, this is irrelevant if a custom httpx client is passed in.
139
-
140
- httpx_client : typing.Optional[httpx.AsyncClient]
141
- The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration.
142
-
143
- Examples
144
- --------
145
- from mirascope.api._generated import AsyncMirascope
146
-
147
- client = AsyncMirascope()
148
- """
149
-
150
- def __init__(
151
- self,
152
- *,
153
- base_url: typing.Optional[str] = None,
154
- environment: MirascopeEnvironment = MirascopeEnvironment.PRODUCTION,
155
- timeout: typing.Optional[float] = None,
156
- follow_redirects: typing.Optional[bool] = True,
157
- httpx_client: typing.Optional[httpx.AsyncClient] = None,
158
- ):
159
- _defaulted_timeout = (
160
- timeout
161
- if timeout is not None
162
- else 180
163
- if httpx_client is None
164
- else httpx_client.timeout.read
165
- )
166
- self._client_wrapper = AsyncClientWrapper(
167
- base_url=_get_base_url(base_url=base_url, environment=environment),
168
- httpx_client=httpx_client
169
- if httpx_client is not None
170
- else httpx.AsyncClient(
171
- timeout=_defaulted_timeout, follow_redirects=follow_redirects
172
- )
173
- if follow_redirects is not None
174
- else httpx.AsyncClient(timeout=_defaulted_timeout),
175
- timeout=_defaulted_timeout,
176
- )
177
- self.health = AsyncHealthClient(client_wrapper=self._client_wrapper)
178
- self.traces = AsyncTracesClient(client_wrapper=self._client_wrapper)
179
- self.docs = AsyncDocsClient(client_wrapper=self._client_wrapper)
180
- self.organizations = AsyncOrganizationsClient(
181
- client_wrapper=self._client_wrapper
182
- )
183
- self.organization_invitations = AsyncOrganizationInvitationsClient(
184
- client_wrapper=self._client_wrapper
185
- )
186
- self.organization_memberships = AsyncOrganizationMembershipsClient(
187
- client_wrapper=self._client_wrapper
188
- )
189
- self.projects = AsyncProjectsClient(client_wrapper=self._client_wrapper)
190
- self.project_memberships = AsyncProjectMembershipsClient(
191
- client_wrapper=self._client_wrapper
192
- )
193
- self.environments = AsyncEnvironmentsClient(client_wrapper=self._client_wrapper)
194
- self.api_keys = AsyncApiKeysClient(client_wrapper=self._client_wrapper)
195
- self.functions = AsyncFunctionsClient(client_wrapper=self._client_wrapper)
196
- self.annotations = AsyncAnnotationsClient(client_wrapper=self._client_wrapper)
197
- self.tags = AsyncTagsClient(client_wrapper=self._client_wrapper)
198
- self.token_cost = AsyncTokenCostClient(client_wrapper=self._client_wrapper)
199
-
200
-
201
- def _get_base_url(
202
- *, base_url: typing.Optional[str] = None, environment: MirascopeEnvironment
203
- ) -> str:
204
- if base_url is not None:
205
- return base_url
206
- elif environment is not None:
207
- return environment.value
208
- else:
209
- raise Exception(
210
- "Please pass in either base_url or environment to construct the client"
211
- )
@@ -1,52 +0,0 @@
1
- # This file was auto-generated by Fern from our API Definition.
2
-
3
- # isort: skip_file
4
-
5
- from .api_error import ApiError
6
- from .client_wrapper import AsyncClientWrapper, BaseClientWrapper, SyncClientWrapper
7
- from .datetime_utils import serialize_datetime
8
- from .file import File, convert_file_dict_to_httpx_tuples, with_content_type
9
- from .http_client import AsyncHttpClient, HttpClient
10
- from .http_response import AsyncHttpResponse, HttpResponse
11
- from .jsonable_encoder import jsonable_encoder
12
- from .pydantic_utilities import (
13
- IS_PYDANTIC_V2,
14
- UniversalBaseModel,
15
- UniversalRootModel,
16
- parse_obj_as,
17
- universal_field_validator,
18
- universal_root_validator,
19
- update_forward_refs,
20
- )
21
- from .query_encoder import encode_query
22
- from .remove_none_from_dict import remove_none_from_dict
23
- from .request_options import RequestOptions
24
- from .serialization import FieldMetadata, convert_and_respect_annotation_metadata
25
-
26
- __all__ = [
27
- "ApiError",
28
- "AsyncClientWrapper",
29
- "AsyncHttpClient",
30
- "AsyncHttpResponse",
31
- "BaseClientWrapper",
32
- "FieldMetadata",
33
- "File",
34
- "HttpClient",
35
- "HttpResponse",
36
- "IS_PYDANTIC_V2",
37
- "RequestOptions",
38
- "SyncClientWrapper",
39
- "UniversalBaseModel",
40
- "UniversalRootModel",
41
- "convert_and_respect_annotation_metadata",
42
- "convert_file_dict_to_httpx_tuples",
43
- "encode_query",
44
- "jsonable_encoder",
45
- "parse_obj_as",
46
- "remove_none_from_dict",
47
- "serialize_datetime",
48
- "universal_field_validator",
49
- "universal_root_validator",
50
- "update_forward_refs",
51
- "with_content_type",
52
- ]
@@ -1,23 +0,0 @@
1
- # This file was auto-generated by Fern from our API Definition.
2
-
3
- from typing import Any, Dict, Optional
4
-
5
-
6
- class ApiError(Exception):
7
- headers: Optional[Dict[str, str]]
8
- status_code: Optional[int]
9
- body: Any
10
-
11
- def __init__(
12
- self,
13
- *,
14
- headers: Optional[Dict[str, str]] = None,
15
- status_code: Optional[int] = None,
16
- body: Any = None,
17
- ) -> None:
18
- self.headers = headers
19
- self.status_code = status_code
20
- self.body = body
21
-
22
- def __str__(self) -> str:
23
- return f"headers: {self.headers}, status_code: {self.status_code}, body: {self.body}"
@@ -1,46 +0,0 @@
1
- # This file was auto-generated by Fern from our API Definition.
2
-
3
- import typing
4
-
5
- import httpx
6
- from .http_client import AsyncHttpClient, HttpClient
7
-
8
-
9
- class BaseClientWrapper:
10
- def __init__(self, *, base_url: str, timeout: typing.Optional[float] = None):
11
- self._base_url = base_url
12
- self._timeout = timeout
13
-
14
- def get_headers(self) -> typing.Dict[str, str]:
15
- headers: typing.Dict[str, str] = {
16
- "X-Fern-Language": "Python",
17
- }
18
- return headers
19
-
20
- def get_base_url(self) -> str:
21
- return self._base_url
22
-
23
- def get_timeout(self) -> typing.Optional[float]:
24
- return self._timeout
25
-
26
-
27
- class SyncClientWrapper(BaseClientWrapper):
28
- def __init__(self, *, base_url: str, timeout: typing.Optional[float] = None, httpx_client: httpx.Client):
29
- super().__init__(base_url=base_url, timeout=timeout)
30
- self.httpx_client = HttpClient(
31
- httpx_client=httpx_client,
32
- base_headers=self.get_headers,
33
- base_timeout=self.get_timeout,
34
- base_url=self.get_base_url,
35
- )
36
-
37
-
38
- class AsyncClientWrapper(BaseClientWrapper):
39
- def __init__(self, *, base_url: str, timeout: typing.Optional[float] = None, httpx_client: httpx.AsyncClient):
40
- super().__init__(base_url=base_url, timeout=timeout)
41
- self.httpx_client = AsyncHttpClient(
42
- httpx_client=httpx_client,
43
- base_headers=self.get_headers,
44
- base_timeout=self.get_timeout,
45
- base_url=self.get_base_url,
46
- )
@@ -1,28 +0,0 @@
1
- # This file was auto-generated by Fern from our API Definition.
2
-
3
- import datetime as dt
4
-
5
-
6
- def serialize_datetime(v: dt.datetime) -> str:
7
- """
8
- Serialize a datetime including timezone info.
9
-
10
- Uses the timezone info provided if present, otherwise uses the current runtime's timezone info.
11
-
12
- UTC datetimes end in "Z" while all other timezones are represented as offset from UTC, e.g. +05:00.
13
- """
14
-
15
- def _serialize_zoned_datetime(v: dt.datetime) -> str:
16
- if v.tzinfo is not None and v.tzinfo.tzname(None) == dt.timezone.utc.tzname(None):
17
- # UTC is a special case where we use "Z" at the end instead of "+00:00"
18
- return v.isoformat().replace("+00:00", "Z")
19
- else:
20
- # Delegate to the typical +/- offset format
21
- return v.isoformat()
22
-
23
- if v.tzinfo is not None:
24
- return _serialize_zoned_datetime(v)
25
- else:
26
- local_tz = dt.datetime.now().astimezone().tzinfo
27
- localized_dt = v.replace(tzinfo=local_tz)
28
- return _serialize_zoned_datetime(localized_dt)
@@ -1,67 +0,0 @@
1
- # This file was auto-generated by Fern from our API Definition.
2
-
3
- from typing import IO, Dict, List, Mapping, Optional, Tuple, Union, cast
4
-
5
- # File typing inspired by the flexibility of types within the httpx library
6
- # https://github.com/encode/httpx/blob/master/httpx/_types.py
7
- FileContent = Union[IO[bytes], bytes, str]
8
- File = Union[
9
- # file (or bytes)
10
- FileContent,
11
- # (filename, file (or bytes))
12
- Tuple[Optional[str], FileContent],
13
- # (filename, file (or bytes), content_type)
14
- Tuple[Optional[str], FileContent, Optional[str]],
15
- # (filename, file (or bytes), content_type, headers)
16
- Tuple[
17
- Optional[str],
18
- FileContent,
19
- Optional[str],
20
- Mapping[str, str],
21
- ],
22
- ]
23
-
24
-
25
- def convert_file_dict_to_httpx_tuples(
26
- d: Dict[str, Union[File, List[File]]],
27
- ) -> List[Tuple[str, File]]:
28
- """
29
- The format we use is a list of tuples, where the first element is the
30
- name of the file and the second is the file object. Typically HTTPX wants
31
- a dict, but to be able to send lists of files, you have to use the list
32
- approach (which also works for non-lists)
33
- https://github.com/encode/httpx/pull/1032
34
- """
35
-
36
- httpx_tuples = []
37
- for key, file_like in d.items():
38
- if isinstance(file_like, list):
39
- for file_like_item in file_like:
40
- httpx_tuples.append((key, file_like_item))
41
- else:
42
- httpx_tuples.append((key, file_like))
43
- return httpx_tuples
44
-
45
-
46
- def with_content_type(*, file: File, default_content_type: str) -> File:
47
- """
48
- This function resolves to the file's content type, if provided, and defaults
49
- to the default_content_type value if not.
50
- """
51
- if isinstance(file, tuple):
52
- if len(file) == 2:
53
- filename, content = cast(Tuple[Optional[str], FileContent], file) # type: ignore
54
- return (filename, content, default_content_type)
55
- elif len(file) == 3:
56
- filename, content, file_content_type = cast(Tuple[Optional[str], FileContent, Optional[str]], file) # type: ignore
57
- out_content_type = file_content_type or default_content_type
58
- return (filename, content, out_content_type)
59
- elif len(file) == 4:
60
- filename, content, file_content_type, headers = cast( # type: ignore
61
- Tuple[Optional[str], FileContent, Optional[str], Mapping[str, str]], file
62
- )
63
- out_content_type = file_content_type or default_content_type
64
- return (filename, content, out_content_type, headers)
65
- else:
66
- raise ValueError(f"Unexpected tuple length: {len(file)}")
67
- return (None, file, default_content_type)
@@ -1,16 +0,0 @@
1
- # This file was auto-generated by Fern from our API Definition.
2
-
3
-
4
- class ForceMultipartDict(dict):
5
- """
6
- A dictionary subclass that always evaluates to True in boolean contexts.
7
-
8
- This is used to force multipart/form-data encoding in HTTP requests even when
9
- the dictionary is empty, which would normally evaluate to False.
10
- """
11
-
12
- def __bool__(self):
13
- return True
14
-
15
-
16
- FORCE_MULTIPART = ForceMultipartDict()