mirascope 1.25.7__py3-none-any.whl → 2.0.0a0__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 (474) hide show
  1. mirascope/__init__.py +3 -59
  2. mirascope/graphs/__init__.py +22 -0
  3. mirascope/{experimental/graphs → graphs}/finite_state_machine.py +70 -159
  4. mirascope/llm/__init__.py +206 -16
  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 +16 -0
  10. mirascope/llm/calls/base_call.py +33 -0
  11. mirascope/llm/calls/calls.py +315 -0
  12. mirascope/llm/calls/decorator.py +255 -0
  13. mirascope/llm/clients/__init__.py +34 -0
  14. mirascope/llm/clients/anthropic/__init__.py +11 -0
  15. mirascope/llm/clients/anthropic/_utils/__init__.py +13 -0
  16. mirascope/llm/clients/anthropic/_utils/decode.py +244 -0
  17. mirascope/llm/clients/anthropic/_utils/encode.py +243 -0
  18. mirascope/llm/clients/anthropic/clients.py +819 -0
  19. mirascope/llm/clients/anthropic/model_ids.py +8 -0
  20. mirascope/llm/clients/base/__init__.py +15 -0
  21. mirascope/llm/clients/base/_utils.py +192 -0
  22. mirascope/llm/clients/base/client.py +1256 -0
  23. mirascope/llm/clients/base/kwargs.py +12 -0
  24. mirascope/llm/clients/base/params.py +93 -0
  25. mirascope/llm/clients/google/__init__.py +6 -0
  26. mirascope/llm/clients/google/_utils/__init__.py +13 -0
  27. mirascope/llm/clients/google/_utils/decode.py +231 -0
  28. mirascope/llm/clients/google/_utils/encode.py +279 -0
  29. mirascope/llm/clients/google/clients.py +853 -0
  30. mirascope/llm/clients/google/message.py +7 -0
  31. mirascope/llm/clients/google/model_ids.py +15 -0
  32. mirascope/llm/clients/openai/__init__.py +25 -0
  33. mirascope/llm/clients/openai/completions/__init__.py +9 -0
  34. mirascope/llm/clients/openai/completions/_utils/__init__.py +13 -0
  35. mirascope/llm/clients/openai/completions/_utils/decode.py +187 -0
  36. mirascope/llm/clients/openai/completions/_utils/encode.py +358 -0
  37. mirascope/llm/clients/openai/completions/_utils/model_features.py +81 -0
  38. mirascope/llm/clients/openai/completions/clients.py +833 -0
  39. mirascope/llm/clients/openai/completions/model_ids.py +8 -0
  40. mirascope/llm/clients/openai/responses/__init__.py +9 -0
  41. mirascope/llm/clients/openai/responses/_utils/__init__.py +13 -0
  42. mirascope/llm/clients/openai/responses/_utils/decode.py +194 -0
  43. mirascope/llm/clients/openai/responses/_utils/encode.py +333 -0
  44. mirascope/llm/clients/openai/responses/_utils/model_features.py +87 -0
  45. mirascope/llm/clients/openai/responses/clients.py +832 -0
  46. mirascope/llm/clients/openai/responses/model_ids.py +8 -0
  47. mirascope/llm/clients/openai/shared/__init__.py +7 -0
  48. mirascope/llm/clients/openai/shared/_utils.py +55 -0
  49. mirascope/llm/clients/providers.py +175 -0
  50. mirascope/llm/content/__init__.py +70 -0
  51. mirascope/llm/content/audio.py +173 -0
  52. mirascope/llm/content/document.py +94 -0
  53. mirascope/llm/content/image.py +206 -0
  54. mirascope/llm/content/text.py +47 -0
  55. mirascope/llm/content/thought.py +58 -0
  56. mirascope/llm/content/tool_call.py +63 -0
  57. mirascope/llm/content/tool_output.py +26 -0
  58. mirascope/llm/context/__init__.py +6 -0
  59. mirascope/llm/context/_utils.py +28 -0
  60. mirascope/llm/context/context.py +24 -0
  61. mirascope/llm/exceptions.py +105 -0
  62. mirascope/llm/formatting/__init__.py +22 -0
  63. mirascope/llm/formatting/_utils.py +74 -0
  64. mirascope/llm/formatting/format.py +104 -0
  65. mirascope/llm/formatting/from_call_args.py +30 -0
  66. mirascope/llm/formatting/partial.py +58 -0
  67. mirascope/llm/formatting/types.py +109 -0
  68. mirascope/llm/mcp/__init__.py +5 -0
  69. mirascope/llm/mcp/client.py +118 -0
  70. mirascope/llm/messages/__init__.py +32 -0
  71. mirascope/llm/messages/message.py +182 -0
  72. mirascope/llm/models/__init__.py +16 -0
  73. mirascope/llm/models/models.py +1243 -0
  74. mirascope/llm/prompts/__init__.py +33 -0
  75. mirascope/llm/prompts/_utils.py +60 -0
  76. mirascope/llm/prompts/decorator.py +286 -0
  77. mirascope/llm/prompts/protocols.py +99 -0
  78. mirascope/llm/responses/__init__.py +57 -0
  79. mirascope/llm/responses/_utils.py +56 -0
  80. mirascope/llm/responses/base_response.py +91 -0
  81. mirascope/llm/responses/base_stream_response.py +697 -0
  82. mirascope/llm/responses/finish_reason.py +27 -0
  83. mirascope/llm/responses/response.py +345 -0
  84. mirascope/llm/responses/root_response.py +177 -0
  85. mirascope/llm/responses/stream_response.py +572 -0
  86. mirascope/llm/responses/streams.py +363 -0
  87. mirascope/llm/tools/__init__.py +40 -0
  88. mirascope/llm/tools/_utils.py +25 -0
  89. mirascope/llm/tools/decorator.py +175 -0
  90. mirascope/llm/tools/protocols.py +96 -0
  91. mirascope/llm/tools/tool_schema.py +246 -0
  92. mirascope/llm/tools/toolkit.py +152 -0
  93. mirascope/llm/tools/tools.py +169 -0
  94. mirascope/llm/types/__init__.py +22 -0
  95. mirascope/llm/types/dataclass.py +9 -0
  96. mirascope/llm/types/jsonable.py +44 -0
  97. mirascope/llm/types/type_vars.py +19 -0
  98. mirascope-2.0.0a0.dist-info/METADATA +117 -0
  99. mirascope-2.0.0a0.dist-info/RECORD +101 -0
  100. mirascope/beta/__init__.py +0 -3
  101. mirascope/beta/openai/__init__.py +0 -17
  102. mirascope/beta/openai/realtime/__init__.py +0 -13
  103. mirascope/beta/openai/realtime/_utils/__init__.py +0 -3
  104. mirascope/beta/openai/realtime/_utils/_audio.py +0 -74
  105. mirascope/beta/openai/realtime/_utils/_protocols.py +0 -50
  106. mirascope/beta/openai/realtime/realtime.py +0 -500
  107. mirascope/beta/openai/realtime/recording.py +0 -98
  108. mirascope/beta/openai/realtime/tool.py +0 -113
  109. mirascope/beta/rag/__init__.py +0 -24
  110. mirascope/beta/rag/base/__init__.py +0 -22
  111. mirascope/beta/rag/base/chunkers/__init__.py +0 -2
  112. mirascope/beta/rag/base/chunkers/base_chunker.py +0 -37
  113. mirascope/beta/rag/base/chunkers/text_chunker.py +0 -33
  114. mirascope/beta/rag/base/config.py +0 -8
  115. mirascope/beta/rag/base/document.py +0 -11
  116. mirascope/beta/rag/base/embedders.py +0 -35
  117. mirascope/beta/rag/base/embedding_params.py +0 -18
  118. mirascope/beta/rag/base/embedding_response.py +0 -30
  119. mirascope/beta/rag/base/query_results.py +0 -7
  120. mirascope/beta/rag/base/vectorstore_params.py +0 -18
  121. mirascope/beta/rag/base/vectorstores.py +0 -37
  122. mirascope/beta/rag/chroma/__init__.py +0 -11
  123. mirascope/beta/rag/chroma/types.py +0 -62
  124. mirascope/beta/rag/chroma/vectorstores.py +0 -121
  125. mirascope/beta/rag/cohere/__init__.py +0 -11
  126. mirascope/beta/rag/cohere/embedders.py +0 -87
  127. mirascope/beta/rag/cohere/embedding_params.py +0 -29
  128. mirascope/beta/rag/cohere/embedding_response.py +0 -29
  129. mirascope/beta/rag/cohere/py.typed +0 -0
  130. mirascope/beta/rag/openai/__init__.py +0 -11
  131. mirascope/beta/rag/openai/embedders.py +0 -144
  132. mirascope/beta/rag/openai/embedding_params.py +0 -18
  133. mirascope/beta/rag/openai/embedding_response.py +0 -14
  134. mirascope/beta/rag/openai/py.typed +0 -0
  135. mirascope/beta/rag/pinecone/__init__.py +0 -19
  136. mirascope/beta/rag/pinecone/types.py +0 -143
  137. mirascope/beta/rag/pinecone/vectorstores.py +0 -148
  138. mirascope/beta/rag/weaviate/__init__.py +0 -6
  139. mirascope/beta/rag/weaviate/types.py +0 -92
  140. mirascope/beta/rag/weaviate/vectorstores.py +0 -103
  141. mirascope/core/__init__.py +0 -109
  142. mirascope/core/anthropic/__init__.py +0 -31
  143. mirascope/core/anthropic/_call.py +0 -67
  144. mirascope/core/anthropic/_call_kwargs.py +0 -13
  145. mirascope/core/anthropic/_thinking.py +0 -70
  146. mirascope/core/anthropic/_utils/__init__.py +0 -16
  147. mirascope/core/anthropic/_utils/_convert_common_call_params.py +0 -25
  148. mirascope/core/anthropic/_utils/_convert_finish_reason_to_common_finish_reasons.py +0 -21
  149. mirascope/core/anthropic/_utils/_convert_message_params.py +0 -102
  150. mirascope/core/anthropic/_utils/_get_json_output.py +0 -31
  151. mirascope/core/anthropic/_utils/_handle_stream.py +0 -113
  152. mirascope/core/anthropic/_utils/_message_param_converter.py +0 -154
  153. mirascope/core/anthropic/_utils/_setup_call.py +0 -146
  154. mirascope/core/anthropic/call_params.py +0 -44
  155. mirascope/core/anthropic/call_response.py +0 -226
  156. mirascope/core/anthropic/call_response_chunk.py +0 -152
  157. mirascope/core/anthropic/dynamic_config.py +0 -40
  158. mirascope/core/anthropic/py.typed +0 -0
  159. mirascope/core/anthropic/stream.py +0 -204
  160. mirascope/core/anthropic/tool.py +0 -101
  161. mirascope/core/azure/__init__.py +0 -31
  162. mirascope/core/azure/_call.py +0 -67
  163. mirascope/core/azure/_call_kwargs.py +0 -13
  164. mirascope/core/azure/_utils/__init__.py +0 -14
  165. mirascope/core/azure/_utils/_convert_common_call_params.py +0 -26
  166. mirascope/core/azure/_utils/_convert_finish_reason_to_common_finish_reasons.py +0 -21
  167. mirascope/core/azure/_utils/_convert_message_params.py +0 -121
  168. mirascope/core/azure/_utils/_get_credential.py +0 -33
  169. mirascope/core/azure/_utils/_get_json_output.py +0 -27
  170. mirascope/core/azure/_utils/_handle_stream.py +0 -130
  171. mirascope/core/azure/_utils/_message_param_converter.py +0 -117
  172. mirascope/core/azure/_utils/_setup_call.py +0 -183
  173. mirascope/core/azure/call_params.py +0 -59
  174. mirascope/core/azure/call_response.py +0 -215
  175. mirascope/core/azure/call_response_chunk.py +0 -105
  176. mirascope/core/azure/dynamic_config.py +0 -30
  177. mirascope/core/azure/py.typed +0 -0
  178. mirascope/core/azure/stream.py +0 -147
  179. mirascope/core/azure/tool.py +0 -93
  180. mirascope/core/base/__init__.py +0 -86
  181. mirascope/core/base/_call_factory.py +0 -256
  182. mirascope/core/base/_create.py +0 -253
  183. mirascope/core/base/_extract.py +0 -175
  184. mirascope/core/base/_extract_with_tools.py +0 -189
  185. mirascope/core/base/_partial.py +0 -95
  186. mirascope/core/base/_utils/__init__.py +0 -92
  187. mirascope/core/base/_utils/_base_message_param_converter.py +0 -22
  188. mirascope/core/base/_utils/_base_type.py +0 -26
  189. mirascope/core/base/_utils/_convert_base_model_to_base_tool.py +0 -48
  190. mirascope/core/base/_utils/_convert_base_type_to_base_tool.py +0 -24
  191. mirascope/core/base/_utils/_convert_function_to_base_tool.py +0 -139
  192. mirascope/core/base/_utils/_convert_messages_to_message_params.py +0 -178
  193. mirascope/core/base/_utils/_convert_provider_finish_reason_to_finish_reason.py +0 -20
  194. mirascope/core/base/_utils/_default_tool_docstring.py +0 -6
  195. mirascope/core/base/_utils/_extract_tool_return.py +0 -42
  196. mirascope/core/base/_utils/_fn_is_async.py +0 -24
  197. mirascope/core/base/_utils/_format_template.py +0 -32
  198. mirascope/core/base/_utils/_get_audio_type.py +0 -18
  199. mirascope/core/base/_utils/_get_common_usage.py +0 -20
  200. mirascope/core/base/_utils/_get_create_fn_or_async_create_fn.py +0 -137
  201. mirascope/core/base/_utils/_get_document_type.py +0 -7
  202. mirascope/core/base/_utils/_get_dynamic_configuration.py +0 -69
  203. mirascope/core/base/_utils/_get_fields_from_call_args.py +0 -34
  204. mirascope/core/base/_utils/_get_fn_args.py +0 -23
  205. mirascope/core/base/_utils/_get_image_dimensions.py +0 -39
  206. mirascope/core/base/_utils/_get_image_type.py +0 -26
  207. mirascope/core/base/_utils/_get_metadata.py +0 -17
  208. mirascope/core/base/_utils/_get_possible_user_message_param.py +0 -21
  209. mirascope/core/base/_utils/_get_prompt_template.py +0 -28
  210. mirascope/core/base/_utils/_get_template_values.py +0 -51
  211. mirascope/core/base/_utils/_get_template_variables.py +0 -38
  212. mirascope/core/base/_utils/_get_unsupported_tool_config_keys.py +0 -10
  213. mirascope/core/base/_utils/_is_prompt_template.py +0 -24
  214. mirascope/core/base/_utils/_json_mode_content.py +0 -17
  215. mirascope/core/base/_utils/_messages_decorator.py +0 -121
  216. mirascope/core/base/_utils/_parse_content_template.py +0 -323
  217. mirascope/core/base/_utils/_parse_prompt_messages.py +0 -63
  218. mirascope/core/base/_utils/_pil_image_to_bytes.py +0 -13
  219. mirascope/core/base/_utils/_protocols.py +0 -901
  220. mirascope/core/base/_utils/_setup_call.py +0 -79
  221. mirascope/core/base/_utils/_setup_extract_tool.py +0 -30
  222. mirascope/core/base/call_kwargs.py +0 -13
  223. mirascope/core/base/call_params.py +0 -36
  224. mirascope/core/base/call_response.py +0 -338
  225. mirascope/core/base/call_response_chunk.py +0 -130
  226. mirascope/core/base/dynamic_config.py +0 -82
  227. mirascope/core/base/from_call_args.py +0 -30
  228. mirascope/core/base/merge_decorators.py +0 -59
  229. mirascope/core/base/message_param.py +0 -175
  230. mirascope/core/base/messages.py +0 -116
  231. mirascope/core/base/metadata.py +0 -13
  232. mirascope/core/base/prompt.py +0 -497
  233. mirascope/core/base/response_model_config_dict.py +0 -9
  234. mirascope/core/base/stream.py +0 -479
  235. mirascope/core/base/stream_config.py +0 -11
  236. mirascope/core/base/structured_stream.py +0 -296
  237. mirascope/core/base/tool.py +0 -214
  238. mirascope/core/base/toolkit.py +0 -176
  239. mirascope/core/base/types.py +0 -344
  240. mirascope/core/bedrock/__init__.py +0 -34
  241. mirascope/core/bedrock/_call.py +0 -68
  242. mirascope/core/bedrock/_call_kwargs.py +0 -12
  243. mirascope/core/bedrock/_types.py +0 -104
  244. mirascope/core/bedrock/_utils/__init__.py +0 -14
  245. mirascope/core/bedrock/_utils/_convert_common_call_params.py +0 -39
  246. mirascope/core/bedrock/_utils/_convert_finish_reason_to_common_finish_reasons.py +0 -23
  247. mirascope/core/bedrock/_utils/_convert_message_params.py +0 -111
  248. mirascope/core/bedrock/_utils/_get_json_output.py +0 -30
  249. mirascope/core/bedrock/_utils/_handle_stream.py +0 -104
  250. mirascope/core/bedrock/_utils/_message_param_converter.py +0 -172
  251. mirascope/core/bedrock/_utils/_setup_call.py +0 -258
  252. mirascope/core/bedrock/call_params.py +0 -38
  253. mirascope/core/bedrock/call_response.py +0 -248
  254. mirascope/core/bedrock/call_response_chunk.py +0 -111
  255. mirascope/core/bedrock/dynamic_config.py +0 -37
  256. mirascope/core/bedrock/py.typed +0 -0
  257. mirascope/core/bedrock/stream.py +0 -154
  258. mirascope/core/bedrock/tool.py +0 -100
  259. mirascope/core/cohere/__init__.py +0 -30
  260. mirascope/core/cohere/_call.py +0 -67
  261. mirascope/core/cohere/_call_kwargs.py +0 -11
  262. mirascope/core/cohere/_types.py +0 -20
  263. mirascope/core/cohere/_utils/__init__.py +0 -14
  264. mirascope/core/cohere/_utils/_convert_common_call_params.py +0 -26
  265. mirascope/core/cohere/_utils/_convert_finish_reason_to_common_finish_reasons.py +0 -24
  266. mirascope/core/cohere/_utils/_convert_message_params.py +0 -32
  267. mirascope/core/cohere/_utils/_get_json_output.py +0 -30
  268. mirascope/core/cohere/_utils/_handle_stream.py +0 -35
  269. mirascope/core/cohere/_utils/_message_param_converter.py +0 -54
  270. mirascope/core/cohere/_utils/_setup_call.py +0 -150
  271. mirascope/core/cohere/call_params.py +0 -62
  272. mirascope/core/cohere/call_response.py +0 -205
  273. mirascope/core/cohere/call_response_chunk.py +0 -125
  274. mirascope/core/cohere/dynamic_config.py +0 -32
  275. mirascope/core/cohere/py.typed +0 -0
  276. mirascope/core/cohere/stream.py +0 -113
  277. mirascope/core/cohere/tool.py +0 -93
  278. mirascope/core/costs/__init__.py +0 -5
  279. mirascope/core/costs/_anthropic_calculate_cost.py +0 -219
  280. mirascope/core/costs/_azure_calculate_cost.py +0 -11
  281. mirascope/core/costs/_bedrock_calculate_cost.py +0 -15
  282. mirascope/core/costs/_cohere_calculate_cost.py +0 -44
  283. mirascope/core/costs/_gemini_calculate_cost.py +0 -67
  284. mirascope/core/costs/_google_calculate_cost.py +0 -427
  285. mirascope/core/costs/_groq_calculate_cost.py +0 -156
  286. mirascope/core/costs/_litellm_calculate_cost.py +0 -11
  287. mirascope/core/costs/_mistral_calculate_cost.py +0 -64
  288. mirascope/core/costs/_openai_calculate_cost.py +0 -416
  289. mirascope/core/costs/_vertex_calculate_cost.py +0 -67
  290. mirascope/core/costs/_xai_calculate_cost.py +0 -104
  291. mirascope/core/costs/calculate_cost.py +0 -86
  292. mirascope/core/gemini/__init__.py +0 -40
  293. mirascope/core/gemini/_call.py +0 -67
  294. mirascope/core/gemini/_call_kwargs.py +0 -12
  295. mirascope/core/gemini/_utils/__init__.py +0 -14
  296. mirascope/core/gemini/_utils/_convert_common_call_params.py +0 -39
  297. mirascope/core/gemini/_utils/_convert_finish_reason_to_common_finish_reasons.py +0 -23
  298. mirascope/core/gemini/_utils/_convert_message_params.py +0 -156
  299. mirascope/core/gemini/_utils/_get_json_output.py +0 -35
  300. mirascope/core/gemini/_utils/_handle_stream.py +0 -33
  301. mirascope/core/gemini/_utils/_message_param_converter.py +0 -209
  302. mirascope/core/gemini/_utils/_setup_call.py +0 -149
  303. mirascope/core/gemini/call_params.py +0 -52
  304. mirascope/core/gemini/call_response.py +0 -216
  305. mirascope/core/gemini/call_response_chunk.py +0 -100
  306. mirascope/core/gemini/dynamic_config.py +0 -26
  307. mirascope/core/gemini/stream.py +0 -120
  308. mirascope/core/gemini/tool.py +0 -104
  309. mirascope/core/google/__init__.py +0 -29
  310. mirascope/core/google/_call.py +0 -67
  311. mirascope/core/google/_call_kwargs.py +0 -13
  312. mirascope/core/google/_utils/__init__.py +0 -14
  313. mirascope/core/google/_utils/_convert_common_call_params.py +0 -38
  314. mirascope/core/google/_utils/_convert_finish_reason_to_common_finish_reasons.py +0 -27
  315. mirascope/core/google/_utils/_convert_message_params.py +0 -297
  316. mirascope/core/google/_utils/_get_json_output.py +0 -37
  317. mirascope/core/google/_utils/_handle_stream.py +0 -58
  318. mirascope/core/google/_utils/_message_param_converter.py +0 -200
  319. mirascope/core/google/_utils/_setup_call.py +0 -201
  320. mirascope/core/google/_utils/_validate_media_type.py +0 -58
  321. mirascope/core/google/call_params.py +0 -22
  322. mirascope/core/google/call_response.py +0 -255
  323. mirascope/core/google/call_response_chunk.py +0 -135
  324. mirascope/core/google/dynamic_config.py +0 -26
  325. mirascope/core/google/stream.py +0 -199
  326. mirascope/core/google/tool.py +0 -146
  327. mirascope/core/groq/__init__.py +0 -30
  328. mirascope/core/groq/_call.py +0 -67
  329. mirascope/core/groq/_call_kwargs.py +0 -13
  330. mirascope/core/groq/_utils/__init__.py +0 -14
  331. mirascope/core/groq/_utils/_convert_common_call_params.py +0 -26
  332. mirascope/core/groq/_utils/_convert_message_params.py +0 -112
  333. mirascope/core/groq/_utils/_get_json_output.py +0 -27
  334. mirascope/core/groq/_utils/_handle_stream.py +0 -123
  335. mirascope/core/groq/_utils/_message_param_converter.py +0 -89
  336. mirascope/core/groq/_utils/_setup_call.py +0 -132
  337. mirascope/core/groq/call_params.py +0 -52
  338. mirascope/core/groq/call_response.py +0 -213
  339. mirascope/core/groq/call_response_chunk.py +0 -104
  340. mirascope/core/groq/dynamic_config.py +0 -29
  341. mirascope/core/groq/py.typed +0 -0
  342. mirascope/core/groq/stream.py +0 -135
  343. mirascope/core/groq/tool.py +0 -80
  344. mirascope/core/litellm/__init__.py +0 -28
  345. mirascope/core/litellm/_call.py +0 -67
  346. mirascope/core/litellm/_utils/__init__.py +0 -5
  347. mirascope/core/litellm/_utils/_setup_call.py +0 -109
  348. mirascope/core/litellm/call_params.py +0 -10
  349. mirascope/core/litellm/call_response.py +0 -24
  350. mirascope/core/litellm/call_response_chunk.py +0 -14
  351. mirascope/core/litellm/dynamic_config.py +0 -8
  352. mirascope/core/litellm/py.typed +0 -0
  353. mirascope/core/litellm/stream.py +0 -86
  354. mirascope/core/litellm/tool.py +0 -13
  355. mirascope/core/mistral/__init__.py +0 -36
  356. mirascope/core/mistral/_call.py +0 -65
  357. mirascope/core/mistral/_call_kwargs.py +0 -19
  358. mirascope/core/mistral/_utils/__init__.py +0 -14
  359. mirascope/core/mistral/_utils/_convert_common_call_params.py +0 -24
  360. mirascope/core/mistral/_utils/_convert_finish_reason_to_common_finish_reasons.py +0 -22
  361. mirascope/core/mistral/_utils/_convert_message_params.py +0 -122
  362. mirascope/core/mistral/_utils/_get_json_output.py +0 -34
  363. mirascope/core/mistral/_utils/_handle_stream.py +0 -139
  364. mirascope/core/mistral/_utils/_message_param_converter.py +0 -176
  365. mirascope/core/mistral/_utils/_setup_call.py +0 -164
  366. mirascope/core/mistral/call_params.py +0 -36
  367. mirascope/core/mistral/call_response.py +0 -205
  368. mirascope/core/mistral/call_response_chunk.py +0 -105
  369. mirascope/core/mistral/dynamic_config.py +0 -33
  370. mirascope/core/mistral/py.typed +0 -0
  371. mirascope/core/mistral/stream.py +0 -120
  372. mirascope/core/mistral/tool.py +0 -81
  373. mirascope/core/openai/__init__.py +0 -31
  374. mirascope/core/openai/_call.py +0 -67
  375. mirascope/core/openai/_call_kwargs.py +0 -13
  376. mirascope/core/openai/_utils/__init__.py +0 -14
  377. mirascope/core/openai/_utils/_convert_common_call_params.py +0 -26
  378. mirascope/core/openai/_utils/_convert_message_params.py +0 -148
  379. mirascope/core/openai/_utils/_get_json_output.py +0 -31
  380. mirascope/core/openai/_utils/_handle_stream.py +0 -138
  381. mirascope/core/openai/_utils/_message_param_converter.py +0 -105
  382. mirascope/core/openai/_utils/_setup_call.py +0 -155
  383. mirascope/core/openai/call_params.py +0 -92
  384. mirascope/core/openai/call_response.py +0 -273
  385. mirascope/core/openai/call_response_chunk.py +0 -139
  386. mirascope/core/openai/dynamic_config.py +0 -34
  387. mirascope/core/openai/py.typed +0 -0
  388. mirascope/core/openai/stream.py +0 -185
  389. mirascope/core/openai/tool.py +0 -101
  390. mirascope/core/py.typed +0 -0
  391. mirascope/core/vertex/__init__.py +0 -45
  392. mirascope/core/vertex/_call.py +0 -62
  393. mirascope/core/vertex/_call_kwargs.py +0 -12
  394. mirascope/core/vertex/_utils/__init__.py +0 -14
  395. mirascope/core/vertex/_utils/_convert_common_call_params.py +0 -37
  396. mirascope/core/vertex/_utils/_convert_finish_reason_to_common_finish_reasons.py +0 -23
  397. mirascope/core/vertex/_utils/_convert_message_params.py +0 -171
  398. mirascope/core/vertex/_utils/_get_json_output.py +0 -36
  399. mirascope/core/vertex/_utils/_handle_stream.py +0 -33
  400. mirascope/core/vertex/_utils/_message_param_converter.py +0 -133
  401. mirascope/core/vertex/_utils/_setup_call.py +0 -160
  402. mirascope/core/vertex/call_params.py +0 -24
  403. mirascope/core/vertex/call_response.py +0 -206
  404. mirascope/core/vertex/call_response_chunk.py +0 -99
  405. mirascope/core/vertex/dynamic_config.py +0 -28
  406. mirascope/core/vertex/stream.py +0 -119
  407. mirascope/core/vertex/tool.py +0 -101
  408. mirascope/core/xai/__init__.py +0 -28
  409. mirascope/core/xai/_call.py +0 -67
  410. mirascope/core/xai/_utils/__init__.py +0 -5
  411. mirascope/core/xai/_utils/_setup_call.py +0 -113
  412. mirascope/core/xai/call_params.py +0 -10
  413. mirascope/core/xai/call_response.py +0 -16
  414. mirascope/core/xai/call_response_chunk.py +0 -14
  415. mirascope/core/xai/dynamic_config.py +0 -8
  416. mirascope/core/xai/py.typed +0 -0
  417. mirascope/core/xai/stream.py +0 -57
  418. mirascope/core/xai/tool.py +0 -13
  419. mirascope/experimental/graphs/__init__.py +0 -5
  420. mirascope/integrations/__init__.py +0 -16
  421. mirascope/integrations/_middleware_factory.py +0 -403
  422. mirascope/integrations/langfuse/__init__.py +0 -3
  423. mirascope/integrations/langfuse/_utils.py +0 -114
  424. mirascope/integrations/langfuse/_with_langfuse.py +0 -70
  425. mirascope/integrations/logfire/__init__.py +0 -3
  426. mirascope/integrations/logfire/_utils.py +0 -225
  427. mirascope/integrations/logfire/_with_logfire.py +0 -63
  428. mirascope/integrations/otel/__init__.py +0 -10
  429. mirascope/integrations/otel/_utils.py +0 -270
  430. mirascope/integrations/otel/_with_hyperdx.py +0 -60
  431. mirascope/integrations/otel/_with_otel.py +0 -59
  432. mirascope/integrations/tenacity.py +0 -14
  433. mirascope/llm/_call.py +0 -401
  434. mirascope/llm/_context.py +0 -384
  435. mirascope/llm/_override.py +0 -3639
  436. mirascope/llm/_protocols.py +0 -500
  437. mirascope/llm/_response_metaclass.py +0 -31
  438. mirascope/llm/call_response.py +0 -158
  439. mirascope/llm/call_response_chunk.py +0 -66
  440. mirascope/llm/stream.py +0 -162
  441. mirascope/llm/tool.py +0 -64
  442. mirascope/mcp/__init__.py +0 -7
  443. mirascope/mcp/_utils.py +0 -288
  444. mirascope/mcp/client.py +0 -167
  445. mirascope/mcp/server.py +0 -356
  446. mirascope/mcp/tools.py +0 -110
  447. mirascope/py.typed +0 -0
  448. mirascope/retries/__init__.py +0 -11
  449. mirascope/retries/fallback.py +0 -131
  450. mirascope/retries/tenacity.py +0 -50
  451. mirascope/tools/__init__.py +0 -37
  452. mirascope/tools/base.py +0 -98
  453. mirascope/tools/system/__init__.py +0 -0
  454. mirascope/tools/system/_docker_operation.py +0 -166
  455. mirascope/tools/system/_file_system.py +0 -267
  456. mirascope/tools/web/__init__.py +0 -0
  457. mirascope/tools/web/_duckduckgo.py +0 -111
  458. mirascope/tools/web/_httpx.py +0 -125
  459. mirascope/tools/web/_parse_url_content.py +0 -94
  460. mirascope/tools/web/_requests.py +0 -54
  461. mirascope/v0/__init__.py +0 -43
  462. mirascope/v0/anthropic.py +0 -54
  463. mirascope/v0/base/__init__.py +0 -12
  464. mirascope/v0/base/calls.py +0 -118
  465. mirascope/v0/base/extractors.py +0 -122
  466. mirascope/v0/base/ops_utils.py +0 -207
  467. mirascope/v0/base/prompts.py +0 -48
  468. mirascope/v0/base/types.py +0 -14
  469. mirascope/v0/base/utils.py +0 -21
  470. mirascope/v0/openai.py +0 -54
  471. mirascope-1.25.7.dist-info/METADATA +0 -169
  472. mirascope-1.25.7.dist-info/RECORD +0 -378
  473. {mirascope-1.25.7.dist-info → mirascope-2.0.0a0.dist-info}/WHEEL +0 -0
  474. {mirascope-1.25.7.dist-info → mirascope-2.0.0a0.dist-info}/licenses/LICENSE +0 -0
@@ -1,256 +0,0 @@
1
- """The `call_factory` method for generating provider specific call decorators."""
2
-
3
- from collections.abc import AsyncIterable, Callable, Iterable
4
- from enum import Enum
5
- from functools import partial
6
- from typing import TypeVar
7
-
8
- from pydantic import BaseModel
9
-
10
- from ._create import create_factory
11
- from ._extract import extract_factory
12
- from ._extract_with_tools import extract_with_tools_factory
13
- from ._utils import (
14
- BaseType,
15
- GetJsonOutput,
16
- HandleStream,
17
- HandleStreamAsync,
18
- LLMFunctionDecorator,
19
- SameSyncAndAsyncClientSetupCall,
20
- SetupCall,
21
- )
22
- from ._utils._protocols import (
23
- AsyncLLMFunctionDecorator,
24
- CallDecorator,
25
- SyncLLMFunctionDecorator,
26
- )
27
- from .call_params import BaseCallParams
28
- from .call_response import BaseCallResponse
29
- from .call_response_chunk import BaseCallResponseChunk
30
- from .dynamic_config import BaseDynamicConfig
31
- from .stream import BaseStream, stream_factory
32
- from .stream_config import StreamConfig
33
- from .structured_stream import structured_stream_factory
34
- from .tool import BaseTool
35
-
36
- _BaseCallResponseT = TypeVar("_BaseCallResponseT", bound=BaseCallResponse)
37
- _BaseCallResponseChunkT = TypeVar(
38
- "_BaseCallResponseChunkT", bound=BaseCallResponseChunk
39
- )
40
- _ResponseModelT = TypeVar("_ResponseModelT", bound=BaseModel | BaseType | Enum)
41
- _ParsedOutputT = TypeVar("_ParsedOutputT")
42
- _BaseCallParamsT = TypeVar("_BaseCallParamsT", bound=BaseCallParams)
43
- _BaseDynamicConfigT = TypeVar("_BaseDynamicConfigT", bound=BaseDynamicConfig)
44
- _AsyncBaseDynamicConfigT = TypeVar("_AsyncBaseDynamicConfigT", bound=BaseDynamicConfig)
45
- _BaseStreamT = TypeVar("_BaseStreamT", bound=BaseStream)
46
- _SameSyncAndAsyncClientT = TypeVar("_SameSyncAndAsyncClientT", contravariant=True)
47
- _SyncBaseClientT = TypeVar("_SyncBaseClientT", contravariant=True)
48
- _AsyncBaseClientT = TypeVar("_AsyncBaseClientT", contravariant=True)
49
- _BaseToolT = TypeVar("_BaseToolT", bound=BaseTool)
50
- _ResponseT = TypeVar("_ResponseT")
51
- _ResponseChunkT = TypeVar("_ResponseChunkT")
52
- _AsyncResponseT = TypeVar("_AsyncResponseT")
53
- _AsyncResponseChunkT = TypeVar("_AsyncResponseChunkT")
54
-
55
-
56
- def call_factory( # noqa: ANN202
57
- *,
58
- TCallResponse: type[_BaseCallResponseT],
59
- TCallResponseChunk: type[_BaseCallResponseChunkT],
60
- TToolType: type[_BaseToolT],
61
- TStream: type[_BaseStreamT],
62
- default_call_params: BaseCallParams,
63
- setup_call: SameSyncAndAsyncClientSetupCall[
64
- _SameSyncAndAsyncClientT,
65
- _BaseDynamicConfigT,
66
- _AsyncBaseDynamicConfigT,
67
- _BaseCallParamsT,
68
- _ResponseT,
69
- _ResponseChunkT,
70
- _AsyncResponseT,
71
- _AsyncResponseChunkT,
72
- _BaseToolT,
73
- ]
74
- | SetupCall[
75
- _SyncBaseClientT,
76
- _AsyncBaseClientT,
77
- _BaseDynamicConfigT,
78
- _AsyncBaseDynamicConfigT,
79
- _BaseCallParamsT,
80
- _ResponseT,
81
- _ResponseChunkT,
82
- _AsyncResponseT,
83
- _AsyncResponseChunkT,
84
- _BaseToolT,
85
- ],
86
- get_json_output: GetJsonOutput[_BaseCallResponseT | _BaseCallResponseChunkT],
87
- handle_stream: HandleStream[_ResponseChunkT, _BaseCallResponseChunkT, _BaseToolT],
88
- handle_stream_async: HandleStreamAsync[
89
- _AsyncResponseChunkT, _BaseCallResponseChunkT, _BaseToolT
90
- ],
91
- ) -> CallDecorator[
92
- _BaseCallResponseT,
93
- _BaseCallResponseChunkT,
94
- _BaseDynamicConfigT,
95
- _AsyncBaseDynamicConfigT,
96
- _BaseCallParamsT,
97
- _BaseStreamT,
98
- _SyncBaseClientT,
99
- _AsyncBaseClientT,
100
- _SameSyncAndAsyncClientT,
101
- ]:
102
- """A factory method for creating provider-specific call decorators.
103
-
104
- Args:
105
- TCallResponse: The provider-specific `BaseCallResponse` type.
106
- TCallResponseChunk: The provider-specific `BaseCallResponseChunk` type.
107
- TToolType: The provider-specific `BaseTool` type.
108
- TStream: The provider-specific `BaseStream` type.
109
- default_call_params: The default call parameters to use, which must match the
110
- `TCallParams` type if provided.
111
- setup_call: The helper method for setting up a call, which returns the
112
- configured create function, the prompt template, the list of
113
- provider-specific messages, the list of provider-specific tool types, and
114
- the finalized `call_kwargs` with which to make the API call with the create
115
- function.
116
- get_json_output: The helper method for getting JSON output from a call response.
117
- handle_stream: The helper method for converting a provider's original stream
118
- generator into a generator that returns tuples of `(chunk, tool)` where
119
- `chunk` and `tool` are provider-specific `BaseCallResponseChunk` and
120
- `BaseTool` instances, respectively.
121
- handle_stream_async: The same helper method as `handle_stream` except for
122
- handling asynchronous streaming.
123
- """
124
-
125
- def base_call(
126
- model: str,
127
- *,
128
- stream: bool | StreamConfig = False,
129
- tools: list[type[BaseTool] | Callable] | None = None,
130
- response_model: type[_ResponseModelT] | None = None,
131
- output_parser: Callable[[_BaseCallResponseT], _ParsedOutputT]
132
- | Callable[[_BaseCallResponseChunkT], _ParsedOutputT]
133
- | Callable[[_ResponseModelT], _ParsedOutputT]
134
- | None = None,
135
- json_mode: bool = False,
136
- client: _SameSyncAndAsyncClientT
137
- | _AsyncBaseClientT
138
- | _SyncBaseClientT
139
- | None = None,
140
- call_params: BaseCallParams | None = None,
141
- ) -> (
142
- AsyncLLMFunctionDecorator[
143
- _AsyncBaseDynamicConfigT,
144
- _BaseCallResponseT
145
- | _ParsedOutputT
146
- | BaseStream
147
- | _ResponseModelT
148
- | AsyncIterable[_ResponseModelT],
149
- ]
150
- | SyncLLMFunctionDecorator[
151
- _BaseDynamicConfigT,
152
- _BaseCallResponseT
153
- | _ParsedOutputT
154
- | BaseStream
155
- | _ResponseModelT
156
- | Iterable[_ResponseModelT],
157
- ]
158
- | LLMFunctionDecorator[
159
- _BaseDynamicConfigT,
160
- _AsyncBaseDynamicConfigT,
161
- _BaseCallResponseT
162
- | _ParsedOutputT
163
- | BaseStream
164
- | _ResponseModelT
165
- | Iterable[_ResponseModelT],
166
- _BaseCallResponseT
167
- | _ParsedOutputT
168
- | BaseStream
169
- | _ResponseModelT
170
- | AsyncIterable[_ResponseModelT],
171
- ]
172
- ):
173
- if stream and output_parser:
174
- raise ValueError("Cannot use `output_parser` with `stream=True`.")
175
-
176
- if call_params is None:
177
- call_params = default_call_params
178
-
179
- if response_model:
180
- if stream:
181
- return partial(
182
- structured_stream_factory(
183
- TCallResponse=TCallResponse,
184
- TCallResponseChunk=TCallResponseChunk,
185
- TStream=TStream,
186
- TToolType=TToolType,
187
- setup_call=setup_call,
188
- get_json_output=get_json_output,
189
- ),
190
- model=model,
191
- response_model=response_model,
192
- json_mode=json_mode,
193
- client=client,
194
- call_params=call_params,
195
- ) # pyright: ignore [reportReturnType, reportCallIssue]
196
- elif tools:
197
- return partial(
198
- extract_with_tools_factory(
199
- TCallResponse=TCallResponse,
200
- setup_call=setup_call,
201
- get_json_output=get_json_output,
202
- ),
203
- model=model,
204
- tools=tools,
205
- response_model=response_model,
206
- output_parser=output_parser,
207
- client=client,
208
- call_params=call_params,
209
- ) # pyright: ignore [reportReturnType, reportCallIssue]
210
- else:
211
- return partial(
212
- extract_factory(
213
- TCallResponse=TCallResponse,
214
- TToolType=TToolType,
215
- setup_call=setup_call,
216
- get_json_output=get_json_output,
217
- ),
218
- model=model,
219
- response_model=response_model,
220
- output_parser=output_parser,
221
- json_mode=json_mode,
222
- client=client,
223
- call_params=call_params,
224
- ) # pyright: ignore [reportCallIssue]
225
-
226
- if stream:
227
- return partial(
228
- stream_factory(
229
- TCallResponse=TCallResponse,
230
- TStream=TStream,
231
- setup_call=setup_call,
232
- handle_stream=handle_stream,
233
- handle_stream_async=handle_stream_async,
234
- ),
235
- model=model,
236
- tools=tools,
237
- json_mode=json_mode,
238
- client=client,
239
- call_params=call_params,
240
- partial_tools=isinstance(stream, dict) and stream.get("partial_tools"),
241
- ) # pyright: ignore [reportReturnType, reportCallIssue]
242
- return partial(
243
- create_factory(
244
- TCallResponse=TCallResponse,
245
- setup_call=setup_call,
246
- ),
247
- model=model,
248
- tools=tools,
249
- response_model=None,
250
- output_parser=output_parser,
251
- json_mode=json_mode,
252
- client=client,
253
- call_params=call_params,
254
- ) # pyright: ignore [reportReturnType, reportCallIssue]
255
-
256
- return base_call # pyright: ignore [reportReturnType]
@@ -1,253 +0,0 @@
1
- """The `create_factory` method for generating provider specific create decorators."""
2
-
3
- import datetime
4
- from collections.abc import Awaitable, Callable, Coroutine
5
- from functools import wraps
6
- from typing import Any, ParamSpec, TypeVar, cast, overload
7
-
8
- from pydantic import BaseModel
9
-
10
- from ._utils import (
11
- SameSyncAndAsyncClientSetupCall,
12
- SetupCall,
13
- fn_is_async,
14
- get_dynamic_configuration,
15
- get_fn_args,
16
- get_metadata,
17
- get_possible_user_message_param,
18
- is_prompt_template,
19
- )
20
- from .call_params import BaseCallParams
21
- from .call_response import BaseCallResponse
22
- from .dynamic_config import BaseDynamicConfig
23
- from .messages import Messages
24
- from .prompt import prompt_template
25
- from .tool import BaseTool
26
-
27
- _BaseCallResponseT = TypeVar("_BaseCallResponseT", bound=BaseCallResponse)
28
- _SameSyncAndAsyncClientT = TypeVar("_SameSyncAndAsyncClientT", contravariant=True)
29
- _SyncBaseClientT = TypeVar("_SyncBaseClientT", contravariant=True)
30
- _AsyncBaseClientT = TypeVar("_AsyncBaseClientT", contravariant=True)
31
- _BaseDynamicConfigT = TypeVar("_BaseDynamicConfigT", bound=BaseDynamicConfig)
32
- _AsyncBaseDynamicConfigT = TypeVar("_AsyncBaseDynamicConfigT", bound=BaseDynamicConfig)
33
- _ParsedOutputT = TypeVar("_ParsedOutputT")
34
- _BaseCallParamsT = TypeVar("_BaseCallParamsT", bound=BaseCallParams)
35
- _ResponseT = TypeVar("_ResponseT")
36
- _ResponseChunkT = TypeVar("_ResponseChunkT")
37
- _AsyncResponseT = TypeVar("_AsyncResponseT")
38
- _AsyncResponseChunkT = TypeVar("_AsyncResponseChunkT")
39
- _BaseToolT = TypeVar("_BaseToolT", bound=BaseTool)
40
- _P = ParamSpec("_P")
41
-
42
-
43
- def create_factory( # noqa: ANN202
44
- *,
45
- TCallResponse: type[_BaseCallResponseT],
46
- setup_call: SameSyncAndAsyncClientSetupCall[
47
- _SameSyncAndAsyncClientT,
48
- _BaseDynamicConfigT,
49
- _AsyncBaseDynamicConfigT,
50
- _BaseCallParamsT,
51
- _ResponseT,
52
- _ResponseChunkT,
53
- _AsyncResponseT,
54
- _AsyncResponseChunkT,
55
- _BaseToolT,
56
- ]
57
- | SetupCall[
58
- _SyncBaseClientT,
59
- _AsyncBaseClientT,
60
- _BaseDynamicConfigT,
61
- _AsyncBaseDynamicConfigT,
62
- _BaseCallParamsT,
63
- _ResponseT,
64
- _ResponseChunkT,
65
- _AsyncResponseT,
66
- _AsyncResponseChunkT,
67
- _BaseToolT,
68
- ],
69
- ):
70
- """Returns the wrapped function with the provider specific interfaces."""
71
-
72
- @overload
73
- def decorator(
74
- fn: Callable[_P, _BaseDynamicConfigT],
75
- model: str,
76
- tools: list[type[BaseTool] | Callable] | None,
77
- response_model: type[BaseModel] | None,
78
- output_parser: Callable[[_BaseCallResponseT], _ParsedOutputT] | None,
79
- json_mode: bool,
80
- client: _SameSyncAndAsyncClientT | _SyncBaseClientT | None,
81
- call_params: _BaseCallParamsT,
82
- ) -> Callable[_P, _BaseCallResponseT | _ParsedOutputT]: ...
83
-
84
- @overload
85
- def decorator(
86
- fn: Callable[_P, Messages.Type],
87
- model: str,
88
- tools: list[type[BaseTool] | Callable] | None,
89
- response_model: type[BaseModel] | None,
90
- output_parser: Callable[[_BaseCallResponseT], _ParsedOutputT] | None,
91
- json_mode: bool,
92
- client: _SameSyncAndAsyncClientT | _SyncBaseClientT | None,
93
- call_params: _BaseCallParamsT,
94
- ) -> Callable[_P, _BaseCallResponseT | _ParsedOutputT]: ...
95
-
96
- @overload
97
- def decorator(
98
- fn: Callable[
99
- _P,
100
- Awaitable[_AsyncBaseDynamicConfigT]
101
- | Coroutine[Any, Any, _AsyncBaseDynamicConfigT],
102
- ],
103
- model: str,
104
- tools: list[type[BaseTool] | Callable] | None,
105
- response_model: type[BaseModel] | None,
106
- output_parser: Callable[[_BaseCallResponseT], _ParsedOutputT] | None,
107
- json_mode: bool,
108
- client: _SameSyncAndAsyncClientT | _AsyncBaseClientT | None,
109
- call_params: _BaseCallParamsT,
110
- ) -> Callable[
111
- _P,
112
- Awaitable[_BaseCallResponseT | _ParsedOutputT],
113
- ]: ...
114
-
115
- @overload
116
- def decorator(
117
- fn: Callable[_P, Awaitable[Messages.Type] | Coroutine[Any, Any, Messages.Type]],
118
- model: str,
119
- tools: list[type[BaseTool] | Callable] | None,
120
- response_model: type[BaseModel] | None,
121
- output_parser: Callable[[_BaseCallResponseT], _ParsedOutputT] | None,
122
- json_mode: bool,
123
- client: _SameSyncAndAsyncClientT | _AsyncBaseClientT | None,
124
- call_params: _BaseCallParamsT,
125
- ) -> Callable[
126
- _P,
127
- Awaitable[_BaseCallResponseT | _ParsedOutputT],
128
- ]: ...
129
-
130
- def decorator(
131
- fn: Callable[_P, _BaseDynamicConfigT]
132
- | Callable[_P, Messages.Type]
133
- | Callable[
134
- _P,
135
- Awaitable[_AsyncBaseDynamicConfigT]
136
- | Coroutine[Any, Any, _AsyncBaseDynamicConfigT],
137
- ]
138
- | Callable[_P, Awaitable[Messages.Type] | Coroutine[Any, Any, Messages.Type]],
139
- model: str,
140
- tools: list[type[BaseTool] | Callable] | None,
141
- response_model: type[BaseModel] | None,
142
- output_parser: Callable[[_BaseCallResponseT], _ParsedOutputT] | None,
143
- json_mode: bool,
144
- client: _SameSyncAndAsyncClientT | _AsyncBaseClientT | _SyncBaseClientT | None,
145
- call_params: _BaseCallParamsT,
146
- ) -> Callable[
147
- _P,
148
- _BaseCallResponseT
149
- | _ParsedOutputT
150
- | Awaitable[_BaseCallResponseT | _ParsedOutputT],
151
- ]:
152
- if not is_prompt_template(fn):
153
- fn = cast(
154
- Callable[_P, Messages.Type] | Callable[_P, Awaitable[Messages.Type]], fn
155
- )
156
- fn = prompt_template()(fn)
157
- fn = cast(
158
- Callable[_P, _BaseDynamicConfigT]
159
- | Callable[_P, Awaitable[_AsyncBaseDynamicConfigT]],
160
- fn,
161
- )
162
- fn._model = model # pyright: ignore [reportFunctionMemberAccess]
163
- fn.__mirascope_call__ = True # pyright: ignore [reportFunctionMemberAccess]
164
- if fn_is_async(fn):
165
-
166
- @wraps(fn)
167
- async def inner_async(
168
- *args: _P.args, **kwargs: _P.kwargs
169
- ) -> TCallResponse | _ParsedOutputT: # pyright: ignore [reportInvalidTypeForm]
170
- fn_args = get_fn_args(fn, args, kwargs)
171
- dynamic_config = await get_dynamic_configuration(fn, args, kwargs)
172
- nonlocal client
173
- if dynamic_config is not None:
174
- client = dynamic_config.get("client", None) or client
175
- create, prompt_template, messages, tool_types, call_kwargs = setup_call( # pyright: ignore [reportCallIssue]
176
- model=model,
177
- client=client, # pyright: ignore [reportArgumentType]
178
- fn=fn,
179
- fn_args=fn_args,
180
- dynamic_config=dynamic_config,
181
- tools=tools,
182
- json_mode=json_mode,
183
- call_params=call_params,
184
- response_model=response_model,
185
- stream=False,
186
- )
187
- start_time = datetime.datetime.now().timestamp() * 1000
188
- response = await create(stream=False, **call_kwargs)
189
- end_time = datetime.datetime.now().timestamp() * 1000
190
- output = TCallResponse( # pyright: ignore [reportCallIssue]
191
- metadata=get_metadata(fn, dynamic_config),
192
- response=response,
193
- tool_types=tool_types, # pyright: ignore [reportArgumentType]
194
- prompt_template=prompt_template,
195
- fn_args=fn_args,
196
- dynamic_config=dynamic_config,
197
- messages=messages,
198
- call_params=call_params,
199
- call_kwargs=call_kwargs,
200
- user_message_param=get_possible_user_message_param(messages),
201
- start_time=start_time,
202
- end_time=end_time,
203
- )
204
- output._model = model
205
- return output if not output_parser else output_parser(output)
206
-
207
- return inner_async
208
- else:
209
-
210
- @wraps(fn)
211
- def inner(
212
- *args: _P.args, **kwargs: _P.kwargs
213
- ) -> TCallResponse | _ParsedOutputT:
214
- fn_args = get_fn_args(fn, args, kwargs)
215
- dynamic_config = get_dynamic_configuration(fn, args, kwargs)
216
- nonlocal client
217
- if dynamic_config is not None:
218
- client = dynamic_config.get("client", None) or client
219
- create, prompt_template, messages, tool_types, call_kwargs = setup_call( # pyright: ignore [reportCallIssue]
220
- model=model,
221
- client=client, # pyright: ignore [reportArgumentType]
222
- fn=fn,
223
- fn_args=fn_args,
224
- dynamic_config=dynamic_config,
225
- tools=tools,
226
- json_mode=json_mode,
227
- call_params=call_params,
228
- response_model=response_model,
229
- stream=False,
230
- )
231
- start_time = datetime.datetime.now().timestamp() * 1000
232
- response = create(stream=False, **call_kwargs)
233
- end_time = datetime.datetime.now().timestamp() * 1000
234
- output = TCallResponse( # pyright: ignore [reportCallIssue]
235
- metadata=get_metadata(fn, dynamic_config),
236
- response=response,
237
- tool_types=tool_types, # pyright: ignore [reportArgumentType]
238
- prompt_template=prompt_template,
239
- fn_args=fn_args,
240
- dynamic_config=dynamic_config,
241
- messages=messages,
242
- call_params=call_params,
243
- call_kwargs=call_kwargs,
244
- user_message_param=get_possible_user_message_param(messages),
245
- start_time=start_time,
246
- end_time=end_time,
247
- )
248
- output._model = model
249
- return output if not output_parser else output_parser(output)
250
-
251
- return inner
252
-
253
- return decorator
@@ -1,175 +0,0 @@
1
- """The `extract_factory` method for generating provider specific create decorators."""
2
-
3
- from collections.abc import Awaitable, Callable
4
- from functools import wraps
5
- from typing import ParamSpec, TypeVar, overload
6
-
7
- from pydantic import BaseModel
8
-
9
- from ._create import create_factory
10
- from ._utils import (
11
- BaseType,
12
- GetJsonOutput,
13
- SameSyncAndAsyncClientSetupCall,
14
- SetupCall,
15
- extract_tool_return,
16
- fn_is_async,
17
- setup_extract_tool,
18
- )
19
- from ._utils._get_fields_from_call_args import get_fields_from_call_args
20
- from .call_params import BaseCallParams
21
- from .call_response import BaseCallResponse
22
- from .dynamic_config import BaseDynamicConfig
23
- from .tool import BaseTool
24
-
25
- _BaseCallResponseT = TypeVar("_BaseCallResponseT", bound=BaseCallResponse)
26
- _SameSyncAndAsyncClientT = TypeVar("_SameSyncAndAsyncClientT", contravariant=True)
27
- _SyncBaseClientT = TypeVar("_SyncBaseClientT", contravariant=True)
28
- _AsyncBaseClientT = TypeVar("_AsyncBaseClientT", contravariant=True)
29
- _BaseDynamicConfigT = TypeVar("_BaseDynamicConfigT", bound=BaseDynamicConfig)
30
- _AsyncBaseDynamicConfigT = TypeVar("_AsyncBaseDynamicConfigT", bound=BaseDynamicConfig)
31
- _ParsedOutputT = TypeVar("_ParsedOutputT")
32
- _BaseCallParamsT = TypeVar("_BaseCallParamsT", bound=BaseCallParams)
33
- _ResponseT = TypeVar("_ResponseT")
34
- _ResponseChunkT = TypeVar("_ResponseChunkT")
35
- _AsyncResponseT = TypeVar("_AsyncResponseT")
36
- _AsyncResponseChunkT = TypeVar("_AsyncResponseChunkT")
37
- _BaseToolT = TypeVar("_BaseToolT", bound=BaseTool)
38
- _ResponseModelT = TypeVar("_ResponseModelT", bound=BaseModel | BaseType)
39
- _P = ParamSpec("_P")
40
-
41
-
42
- def extract_factory( # noqa: ANN202
43
- *,
44
- TCallResponse: type[_BaseCallResponseT],
45
- TToolType: type[BaseTool],
46
- setup_call: SameSyncAndAsyncClientSetupCall[
47
- _SameSyncAndAsyncClientT,
48
- _BaseDynamicConfigT,
49
- _AsyncBaseDynamicConfigT,
50
- _BaseCallParamsT,
51
- _ResponseT,
52
- _ResponseChunkT,
53
- _AsyncResponseT,
54
- _AsyncResponseChunkT,
55
- _BaseToolT,
56
- ]
57
- | SetupCall[
58
- _SyncBaseClientT,
59
- _AsyncBaseClientT,
60
- _BaseDynamicConfigT,
61
- _AsyncBaseDynamicConfigT,
62
- _BaseCallParamsT,
63
- _ResponseT,
64
- _ResponseChunkT,
65
- _AsyncResponseT,
66
- _AsyncResponseChunkT,
67
- _BaseToolT,
68
- ],
69
- get_json_output: GetJsonOutput[_BaseCallResponseT],
70
- ):
71
- """Returns the wrapped function with the provider specific interfaces."""
72
- create_decorator = create_factory(
73
- TCallResponse=TCallResponse,
74
- setup_call=setup_call,
75
- )
76
-
77
- @overload
78
- def decorator(
79
- fn: Callable[_P, _BaseDynamicConfigT],
80
- model: str,
81
- response_model: type[_ResponseModelT],
82
- output_parser: Callable[[_ResponseModelT], _ParsedOutputT] | None,
83
- json_mode: bool,
84
- client: _SameSyncAndAsyncClientT | _SyncBaseClientT | None,
85
- call_params: _BaseCallParamsT,
86
- ) -> Callable[_P, _ResponseModelT | _ParsedOutputT]: ...
87
-
88
- @overload
89
- def decorator(
90
- fn: Callable[_P, Awaitable[_AsyncBaseDynamicConfigT]],
91
- model: str,
92
- response_model: type[_ResponseModelT],
93
- output_parser: Callable[[_ResponseModelT], _ParsedOutputT] | None,
94
- json_mode: bool,
95
- client: _SameSyncAndAsyncClientT | _AsyncBaseClientT | None,
96
- call_params: _BaseCallParamsT,
97
- ) -> Callable[_P, Awaitable[_ResponseModelT | _ParsedOutputT]]: ...
98
-
99
- def decorator(
100
- fn: Callable[_P, _BaseDynamicConfigT]
101
- | Callable[_P, Awaitable[_AsyncBaseDynamicConfigT]],
102
- model: str,
103
- response_model: type[_ResponseModelT],
104
- output_parser: Callable[[_ResponseModelT], _ParsedOutputT] | None,
105
- json_mode: bool,
106
- client: _SameSyncAndAsyncClientT | _SyncBaseClientT | None,
107
- call_params: _BaseCallParamsT,
108
- ) -> Callable[
109
- _P,
110
- _ResponseModelT | _ParsedOutputT | Awaitable[_ResponseModelT | _ParsedOutputT],
111
- ]:
112
- fn._model = model # pyright: ignore [reportFunctionMemberAccess]
113
- fn.__mirascope_call__ = True # pyright: ignore [reportFunctionMemberAccess]
114
- create_decorator_kwargs = {
115
- "model": model,
116
- "tools": [setup_extract_tool(response_model, TToolType)]
117
- if not json_mode
118
- else None,
119
- "response_model": response_model,
120
- "output_parser": None,
121
- "json_mode": json_mode,
122
- "client": client,
123
- "call_params": call_params,
124
- }
125
-
126
- if fn_is_async(fn):
127
-
128
- @wraps(fn)
129
- async def inner_async(
130
- *args: _P.args, **kwargs: _P.kwargs
131
- ) -> _ResponseModelT:
132
- fields_from_call_args = get_fields_from_call_args(
133
- response_model, fn, args, kwargs
134
- )
135
- call_response = await create_decorator(
136
- fn=fn, **create_decorator_kwargs
137
- )(*args, **kwargs)
138
- try:
139
- json_output = get_json_output(call_response, json_mode)
140
- output = extract_tool_return(
141
- response_model, json_output, False, fields_from_call_args
142
- )
143
- except Exception as e:
144
- e._response = call_response # pyright: ignore [reportAttributeAccessIssue]
145
- raise e
146
- if isinstance(output, BaseModel):
147
- output._response = call_response # pyright: ignore [reportAttributeAccessIssue]
148
- return output if not output_parser else output_parser(output) # pyright: ignore [reportArgumentType, reportReturnType]
149
-
150
- return inner_async
151
- else:
152
-
153
- @wraps(fn)
154
- def inner(*args: _P.args, **kwargs: _P.kwargs) -> _ResponseModelT:
155
- fields_from_call_args = get_fields_from_call_args(
156
- response_model, fn, args, kwargs
157
- )
158
- call_response = create_decorator(fn=fn, **create_decorator_kwargs)(
159
- *args, **kwargs
160
- )
161
- try:
162
- json_output = get_json_output(call_response, json_mode)
163
- output = extract_tool_return(
164
- response_model, json_output, False, fields_from_call_args
165
- )
166
- except Exception as e:
167
- e._response = call_response # pyright: ignore [reportAttributeAccessIssue]
168
- raise e
169
- if isinstance(output, BaseModel):
170
- output._response = call_response # pyright: ignore [reportAttributeAccessIssue]
171
- return output if not output_parser else output_parser(output) # pyright: ignore [reportReturnType, reportArgumentType]
172
-
173
- return inner
174
-
175
- return decorator