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.
- mirascope/__init__.py +3 -59
- mirascope/graphs/__init__.py +22 -0
- mirascope/{experimental/graphs → graphs}/finite_state_machine.py +70 -159
- mirascope/llm/__init__.py +206 -16
- mirascope/llm/agents/__init__.py +15 -0
- mirascope/llm/agents/agent.py +97 -0
- mirascope/llm/agents/agent_template.py +45 -0
- mirascope/llm/agents/decorator.py +176 -0
- mirascope/llm/calls/__init__.py +16 -0
- mirascope/llm/calls/base_call.py +33 -0
- mirascope/llm/calls/calls.py +315 -0
- mirascope/llm/calls/decorator.py +255 -0
- mirascope/llm/clients/__init__.py +34 -0
- mirascope/llm/clients/anthropic/__init__.py +11 -0
- mirascope/llm/clients/anthropic/_utils/__init__.py +13 -0
- mirascope/llm/clients/anthropic/_utils/decode.py +244 -0
- mirascope/llm/clients/anthropic/_utils/encode.py +243 -0
- mirascope/llm/clients/anthropic/clients.py +819 -0
- mirascope/llm/clients/anthropic/model_ids.py +8 -0
- mirascope/llm/clients/base/__init__.py +15 -0
- mirascope/llm/clients/base/_utils.py +192 -0
- mirascope/llm/clients/base/client.py +1256 -0
- mirascope/llm/clients/base/kwargs.py +12 -0
- mirascope/llm/clients/base/params.py +93 -0
- mirascope/llm/clients/google/__init__.py +6 -0
- mirascope/llm/clients/google/_utils/__init__.py +13 -0
- mirascope/llm/clients/google/_utils/decode.py +231 -0
- mirascope/llm/clients/google/_utils/encode.py +279 -0
- mirascope/llm/clients/google/clients.py +853 -0
- mirascope/llm/clients/google/message.py +7 -0
- mirascope/llm/clients/google/model_ids.py +15 -0
- mirascope/llm/clients/openai/__init__.py +25 -0
- mirascope/llm/clients/openai/completions/__init__.py +9 -0
- mirascope/llm/clients/openai/completions/_utils/__init__.py +13 -0
- mirascope/llm/clients/openai/completions/_utils/decode.py +187 -0
- mirascope/llm/clients/openai/completions/_utils/encode.py +358 -0
- mirascope/llm/clients/openai/completions/_utils/model_features.py +81 -0
- mirascope/llm/clients/openai/completions/clients.py +833 -0
- mirascope/llm/clients/openai/completions/model_ids.py +8 -0
- mirascope/llm/clients/openai/responses/__init__.py +9 -0
- mirascope/llm/clients/openai/responses/_utils/__init__.py +13 -0
- mirascope/llm/clients/openai/responses/_utils/decode.py +194 -0
- mirascope/llm/clients/openai/responses/_utils/encode.py +333 -0
- mirascope/llm/clients/openai/responses/_utils/model_features.py +87 -0
- mirascope/llm/clients/openai/responses/clients.py +832 -0
- mirascope/llm/clients/openai/responses/model_ids.py +8 -0
- mirascope/llm/clients/openai/shared/__init__.py +7 -0
- mirascope/llm/clients/openai/shared/_utils.py +55 -0
- mirascope/llm/clients/providers.py +175 -0
- mirascope/llm/content/__init__.py +70 -0
- mirascope/llm/content/audio.py +173 -0
- mirascope/llm/content/document.py +94 -0
- mirascope/llm/content/image.py +206 -0
- mirascope/llm/content/text.py +47 -0
- mirascope/llm/content/thought.py +58 -0
- mirascope/llm/content/tool_call.py +63 -0
- mirascope/llm/content/tool_output.py +26 -0
- mirascope/llm/context/__init__.py +6 -0
- mirascope/llm/context/_utils.py +28 -0
- mirascope/llm/context/context.py +24 -0
- mirascope/llm/exceptions.py +105 -0
- mirascope/llm/formatting/__init__.py +22 -0
- mirascope/llm/formatting/_utils.py +74 -0
- mirascope/llm/formatting/format.py +104 -0
- mirascope/llm/formatting/from_call_args.py +30 -0
- mirascope/llm/formatting/partial.py +58 -0
- mirascope/llm/formatting/types.py +109 -0
- mirascope/llm/mcp/__init__.py +5 -0
- mirascope/llm/mcp/client.py +118 -0
- mirascope/llm/messages/__init__.py +32 -0
- mirascope/llm/messages/message.py +182 -0
- mirascope/llm/models/__init__.py +16 -0
- mirascope/llm/models/models.py +1243 -0
- mirascope/llm/prompts/__init__.py +33 -0
- mirascope/llm/prompts/_utils.py +60 -0
- mirascope/llm/prompts/decorator.py +286 -0
- mirascope/llm/prompts/protocols.py +99 -0
- mirascope/llm/responses/__init__.py +57 -0
- mirascope/llm/responses/_utils.py +56 -0
- mirascope/llm/responses/base_response.py +91 -0
- mirascope/llm/responses/base_stream_response.py +697 -0
- mirascope/llm/responses/finish_reason.py +27 -0
- mirascope/llm/responses/response.py +345 -0
- mirascope/llm/responses/root_response.py +177 -0
- mirascope/llm/responses/stream_response.py +572 -0
- mirascope/llm/responses/streams.py +363 -0
- mirascope/llm/tools/__init__.py +40 -0
- mirascope/llm/tools/_utils.py +25 -0
- mirascope/llm/tools/decorator.py +175 -0
- mirascope/llm/tools/protocols.py +96 -0
- mirascope/llm/tools/tool_schema.py +246 -0
- mirascope/llm/tools/toolkit.py +152 -0
- mirascope/llm/tools/tools.py +169 -0
- mirascope/llm/types/__init__.py +22 -0
- mirascope/llm/types/dataclass.py +9 -0
- mirascope/llm/types/jsonable.py +44 -0
- mirascope/llm/types/type_vars.py +19 -0
- mirascope-2.0.0a0.dist-info/METADATA +117 -0
- mirascope-2.0.0a0.dist-info/RECORD +101 -0
- mirascope/beta/__init__.py +0 -3
- mirascope/beta/openai/__init__.py +0 -17
- mirascope/beta/openai/realtime/__init__.py +0 -13
- mirascope/beta/openai/realtime/_utils/__init__.py +0 -3
- mirascope/beta/openai/realtime/_utils/_audio.py +0 -74
- mirascope/beta/openai/realtime/_utils/_protocols.py +0 -50
- mirascope/beta/openai/realtime/realtime.py +0 -500
- mirascope/beta/openai/realtime/recording.py +0 -98
- mirascope/beta/openai/realtime/tool.py +0 -113
- mirascope/beta/rag/__init__.py +0 -24
- mirascope/beta/rag/base/__init__.py +0 -22
- mirascope/beta/rag/base/chunkers/__init__.py +0 -2
- mirascope/beta/rag/base/chunkers/base_chunker.py +0 -37
- mirascope/beta/rag/base/chunkers/text_chunker.py +0 -33
- mirascope/beta/rag/base/config.py +0 -8
- mirascope/beta/rag/base/document.py +0 -11
- mirascope/beta/rag/base/embedders.py +0 -35
- mirascope/beta/rag/base/embedding_params.py +0 -18
- mirascope/beta/rag/base/embedding_response.py +0 -30
- mirascope/beta/rag/base/query_results.py +0 -7
- mirascope/beta/rag/base/vectorstore_params.py +0 -18
- mirascope/beta/rag/base/vectorstores.py +0 -37
- mirascope/beta/rag/chroma/__init__.py +0 -11
- mirascope/beta/rag/chroma/types.py +0 -62
- mirascope/beta/rag/chroma/vectorstores.py +0 -121
- mirascope/beta/rag/cohere/__init__.py +0 -11
- mirascope/beta/rag/cohere/embedders.py +0 -87
- mirascope/beta/rag/cohere/embedding_params.py +0 -29
- mirascope/beta/rag/cohere/embedding_response.py +0 -29
- mirascope/beta/rag/cohere/py.typed +0 -0
- mirascope/beta/rag/openai/__init__.py +0 -11
- mirascope/beta/rag/openai/embedders.py +0 -144
- mirascope/beta/rag/openai/embedding_params.py +0 -18
- mirascope/beta/rag/openai/embedding_response.py +0 -14
- mirascope/beta/rag/openai/py.typed +0 -0
- mirascope/beta/rag/pinecone/__init__.py +0 -19
- mirascope/beta/rag/pinecone/types.py +0 -143
- mirascope/beta/rag/pinecone/vectorstores.py +0 -148
- mirascope/beta/rag/weaviate/__init__.py +0 -6
- mirascope/beta/rag/weaviate/types.py +0 -92
- mirascope/beta/rag/weaviate/vectorstores.py +0 -103
- mirascope/core/__init__.py +0 -109
- mirascope/core/anthropic/__init__.py +0 -31
- mirascope/core/anthropic/_call.py +0 -67
- mirascope/core/anthropic/_call_kwargs.py +0 -13
- mirascope/core/anthropic/_thinking.py +0 -70
- mirascope/core/anthropic/_utils/__init__.py +0 -16
- mirascope/core/anthropic/_utils/_convert_common_call_params.py +0 -25
- mirascope/core/anthropic/_utils/_convert_finish_reason_to_common_finish_reasons.py +0 -21
- mirascope/core/anthropic/_utils/_convert_message_params.py +0 -102
- mirascope/core/anthropic/_utils/_get_json_output.py +0 -31
- mirascope/core/anthropic/_utils/_handle_stream.py +0 -113
- mirascope/core/anthropic/_utils/_message_param_converter.py +0 -154
- mirascope/core/anthropic/_utils/_setup_call.py +0 -146
- mirascope/core/anthropic/call_params.py +0 -44
- mirascope/core/anthropic/call_response.py +0 -226
- mirascope/core/anthropic/call_response_chunk.py +0 -152
- mirascope/core/anthropic/dynamic_config.py +0 -40
- mirascope/core/anthropic/py.typed +0 -0
- mirascope/core/anthropic/stream.py +0 -204
- mirascope/core/anthropic/tool.py +0 -101
- mirascope/core/azure/__init__.py +0 -31
- mirascope/core/azure/_call.py +0 -67
- mirascope/core/azure/_call_kwargs.py +0 -13
- mirascope/core/azure/_utils/__init__.py +0 -14
- mirascope/core/azure/_utils/_convert_common_call_params.py +0 -26
- mirascope/core/azure/_utils/_convert_finish_reason_to_common_finish_reasons.py +0 -21
- mirascope/core/azure/_utils/_convert_message_params.py +0 -121
- mirascope/core/azure/_utils/_get_credential.py +0 -33
- mirascope/core/azure/_utils/_get_json_output.py +0 -27
- mirascope/core/azure/_utils/_handle_stream.py +0 -130
- mirascope/core/azure/_utils/_message_param_converter.py +0 -117
- mirascope/core/azure/_utils/_setup_call.py +0 -183
- mirascope/core/azure/call_params.py +0 -59
- mirascope/core/azure/call_response.py +0 -215
- mirascope/core/azure/call_response_chunk.py +0 -105
- mirascope/core/azure/dynamic_config.py +0 -30
- mirascope/core/azure/py.typed +0 -0
- mirascope/core/azure/stream.py +0 -147
- mirascope/core/azure/tool.py +0 -93
- mirascope/core/base/__init__.py +0 -86
- mirascope/core/base/_call_factory.py +0 -256
- mirascope/core/base/_create.py +0 -253
- mirascope/core/base/_extract.py +0 -175
- mirascope/core/base/_extract_with_tools.py +0 -189
- mirascope/core/base/_partial.py +0 -95
- mirascope/core/base/_utils/__init__.py +0 -92
- mirascope/core/base/_utils/_base_message_param_converter.py +0 -22
- mirascope/core/base/_utils/_base_type.py +0 -26
- mirascope/core/base/_utils/_convert_base_model_to_base_tool.py +0 -48
- mirascope/core/base/_utils/_convert_base_type_to_base_tool.py +0 -24
- mirascope/core/base/_utils/_convert_function_to_base_tool.py +0 -139
- mirascope/core/base/_utils/_convert_messages_to_message_params.py +0 -178
- mirascope/core/base/_utils/_convert_provider_finish_reason_to_finish_reason.py +0 -20
- mirascope/core/base/_utils/_default_tool_docstring.py +0 -6
- mirascope/core/base/_utils/_extract_tool_return.py +0 -42
- mirascope/core/base/_utils/_fn_is_async.py +0 -24
- mirascope/core/base/_utils/_format_template.py +0 -32
- mirascope/core/base/_utils/_get_audio_type.py +0 -18
- mirascope/core/base/_utils/_get_common_usage.py +0 -20
- mirascope/core/base/_utils/_get_create_fn_or_async_create_fn.py +0 -137
- mirascope/core/base/_utils/_get_document_type.py +0 -7
- mirascope/core/base/_utils/_get_dynamic_configuration.py +0 -69
- mirascope/core/base/_utils/_get_fields_from_call_args.py +0 -34
- mirascope/core/base/_utils/_get_fn_args.py +0 -23
- mirascope/core/base/_utils/_get_image_dimensions.py +0 -39
- mirascope/core/base/_utils/_get_image_type.py +0 -26
- mirascope/core/base/_utils/_get_metadata.py +0 -17
- mirascope/core/base/_utils/_get_possible_user_message_param.py +0 -21
- mirascope/core/base/_utils/_get_prompt_template.py +0 -28
- mirascope/core/base/_utils/_get_template_values.py +0 -51
- mirascope/core/base/_utils/_get_template_variables.py +0 -38
- mirascope/core/base/_utils/_get_unsupported_tool_config_keys.py +0 -10
- mirascope/core/base/_utils/_is_prompt_template.py +0 -24
- mirascope/core/base/_utils/_json_mode_content.py +0 -17
- mirascope/core/base/_utils/_messages_decorator.py +0 -121
- mirascope/core/base/_utils/_parse_content_template.py +0 -323
- mirascope/core/base/_utils/_parse_prompt_messages.py +0 -63
- mirascope/core/base/_utils/_pil_image_to_bytes.py +0 -13
- mirascope/core/base/_utils/_protocols.py +0 -901
- mirascope/core/base/_utils/_setup_call.py +0 -79
- mirascope/core/base/_utils/_setup_extract_tool.py +0 -30
- mirascope/core/base/call_kwargs.py +0 -13
- mirascope/core/base/call_params.py +0 -36
- mirascope/core/base/call_response.py +0 -338
- mirascope/core/base/call_response_chunk.py +0 -130
- mirascope/core/base/dynamic_config.py +0 -82
- mirascope/core/base/from_call_args.py +0 -30
- mirascope/core/base/merge_decorators.py +0 -59
- mirascope/core/base/message_param.py +0 -175
- mirascope/core/base/messages.py +0 -116
- mirascope/core/base/metadata.py +0 -13
- mirascope/core/base/prompt.py +0 -497
- mirascope/core/base/response_model_config_dict.py +0 -9
- mirascope/core/base/stream.py +0 -479
- mirascope/core/base/stream_config.py +0 -11
- mirascope/core/base/structured_stream.py +0 -296
- mirascope/core/base/tool.py +0 -214
- mirascope/core/base/toolkit.py +0 -176
- mirascope/core/base/types.py +0 -344
- mirascope/core/bedrock/__init__.py +0 -34
- mirascope/core/bedrock/_call.py +0 -68
- mirascope/core/bedrock/_call_kwargs.py +0 -12
- mirascope/core/bedrock/_types.py +0 -104
- mirascope/core/bedrock/_utils/__init__.py +0 -14
- mirascope/core/bedrock/_utils/_convert_common_call_params.py +0 -39
- mirascope/core/bedrock/_utils/_convert_finish_reason_to_common_finish_reasons.py +0 -23
- mirascope/core/bedrock/_utils/_convert_message_params.py +0 -111
- mirascope/core/bedrock/_utils/_get_json_output.py +0 -30
- mirascope/core/bedrock/_utils/_handle_stream.py +0 -104
- mirascope/core/bedrock/_utils/_message_param_converter.py +0 -172
- mirascope/core/bedrock/_utils/_setup_call.py +0 -258
- mirascope/core/bedrock/call_params.py +0 -38
- mirascope/core/bedrock/call_response.py +0 -248
- mirascope/core/bedrock/call_response_chunk.py +0 -111
- mirascope/core/bedrock/dynamic_config.py +0 -37
- mirascope/core/bedrock/py.typed +0 -0
- mirascope/core/bedrock/stream.py +0 -154
- mirascope/core/bedrock/tool.py +0 -100
- mirascope/core/cohere/__init__.py +0 -30
- mirascope/core/cohere/_call.py +0 -67
- mirascope/core/cohere/_call_kwargs.py +0 -11
- mirascope/core/cohere/_types.py +0 -20
- mirascope/core/cohere/_utils/__init__.py +0 -14
- mirascope/core/cohere/_utils/_convert_common_call_params.py +0 -26
- mirascope/core/cohere/_utils/_convert_finish_reason_to_common_finish_reasons.py +0 -24
- mirascope/core/cohere/_utils/_convert_message_params.py +0 -32
- mirascope/core/cohere/_utils/_get_json_output.py +0 -30
- mirascope/core/cohere/_utils/_handle_stream.py +0 -35
- mirascope/core/cohere/_utils/_message_param_converter.py +0 -54
- mirascope/core/cohere/_utils/_setup_call.py +0 -150
- mirascope/core/cohere/call_params.py +0 -62
- mirascope/core/cohere/call_response.py +0 -205
- mirascope/core/cohere/call_response_chunk.py +0 -125
- mirascope/core/cohere/dynamic_config.py +0 -32
- mirascope/core/cohere/py.typed +0 -0
- mirascope/core/cohere/stream.py +0 -113
- mirascope/core/cohere/tool.py +0 -93
- mirascope/core/costs/__init__.py +0 -5
- mirascope/core/costs/_anthropic_calculate_cost.py +0 -219
- mirascope/core/costs/_azure_calculate_cost.py +0 -11
- mirascope/core/costs/_bedrock_calculate_cost.py +0 -15
- mirascope/core/costs/_cohere_calculate_cost.py +0 -44
- mirascope/core/costs/_gemini_calculate_cost.py +0 -67
- mirascope/core/costs/_google_calculate_cost.py +0 -427
- mirascope/core/costs/_groq_calculate_cost.py +0 -156
- mirascope/core/costs/_litellm_calculate_cost.py +0 -11
- mirascope/core/costs/_mistral_calculate_cost.py +0 -64
- mirascope/core/costs/_openai_calculate_cost.py +0 -416
- mirascope/core/costs/_vertex_calculate_cost.py +0 -67
- mirascope/core/costs/_xai_calculate_cost.py +0 -104
- mirascope/core/costs/calculate_cost.py +0 -86
- mirascope/core/gemini/__init__.py +0 -40
- mirascope/core/gemini/_call.py +0 -67
- mirascope/core/gemini/_call_kwargs.py +0 -12
- mirascope/core/gemini/_utils/__init__.py +0 -14
- mirascope/core/gemini/_utils/_convert_common_call_params.py +0 -39
- mirascope/core/gemini/_utils/_convert_finish_reason_to_common_finish_reasons.py +0 -23
- mirascope/core/gemini/_utils/_convert_message_params.py +0 -156
- mirascope/core/gemini/_utils/_get_json_output.py +0 -35
- mirascope/core/gemini/_utils/_handle_stream.py +0 -33
- mirascope/core/gemini/_utils/_message_param_converter.py +0 -209
- mirascope/core/gemini/_utils/_setup_call.py +0 -149
- mirascope/core/gemini/call_params.py +0 -52
- mirascope/core/gemini/call_response.py +0 -216
- mirascope/core/gemini/call_response_chunk.py +0 -100
- mirascope/core/gemini/dynamic_config.py +0 -26
- mirascope/core/gemini/stream.py +0 -120
- mirascope/core/gemini/tool.py +0 -104
- mirascope/core/google/__init__.py +0 -29
- mirascope/core/google/_call.py +0 -67
- mirascope/core/google/_call_kwargs.py +0 -13
- mirascope/core/google/_utils/__init__.py +0 -14
- mirascope/core/google/_utils/_convert_common_call_params.py +0 -38
- mirascope/core/google/_utils/_convert_finish_reason_to_common_finish_reasons.py +0 -27
- mirascope/core/google/_utils/_convert_message_params.py +0 -297
- mirascope/core/google/_utils/_get_json_output.py +0 -37
- mirascope/core/google/_utils/_handle_stream.py +0 -58
- mirascope/core/google/_utils/_message_param_converter.py +0 -200
- mirascope/core/google/_utils/_setup_call.py +0 -201
- mirascope/core/google/_utils/_validate_media_type.py +0 -58
- mirascope/core/google/call_params.py +0 -22
- mirascope/core/google/call_response.py +0 -255
- mirascope/core/google/call_response_chunk.py +0 -135
- mirascope/core/google/dynamic_config.py +0 -26
- mirascope/core/google/stream.py +0 -199
- mirascope/core/google/tool.py +0 -146
- mirascope/core/groq/__init__.py +0 -30
- mirascope/core/groq/_call.py +0 -67
- mirascope/core/groq/_call_kwargs.py +0 -13
- mirascope/core/groq/_utils/__init__.py +0 -14
- mirascope/core/groq/_utils/_convert_common_call_params.py +0 -26
- mirascope/core/groq/_utils/_convert_message_params.py +0 -112
- mirascope/core/groq/_utils/_get_json_output.py +0 -27
- mirascope/core/groq/_utils/_handle_stream.py +0 -123
- mirascope/core/groq/_utils/_message_param_converter.py +0 -89
- mirascope/core/groq/_utils/_setup_call.py +0 -132
- mirascope/core/groq/call_params.py +0 -52
- mirascope/core/groq/call_response.py +0 -213
- mirascope/core/groq/call_response_chunk.py +0 -104
- mirascope/core/groq/dynamic_config.py +0 -29
- mirascope/core/groq/py.typed +0 -0
- mirascope/core/groq/stream.py +0 -135
- mirascope/core/groq/tool.py +0 -80
- mirascope/core/litellm/__init__.py +0 -28
- mirascope/core/litellm/_call.py +0 -67
- mirascope/core/litellm/_utils/__init__.py +0 -5
- mirascope/core/litellm/_utils/_setup_call.py +0 -109
- mirascope/core/litellm/call_params.py +0 -10
- mirascope/core/litellm/call_response.py +0 -24
- mirascope/core/litellm/call_response_chunk.py +0 -14
- mirascope/core/litellm/dynamic_config.py +0 -8
- mirascope/core/litellm/py.typed +0 -0
- mirascope/core/litellm/stream.py +0 -86
- mirascope/core/litellm/tool.py +0 -13
- mirascope/core/mistral/__init__.py +0 -36
- mirascope/core/mistral/_call.py +0 -65
- mirascope/core/mistral/_call_kwargs.py +0 -19
- mirascope/core/mistral/_utils/__init__.py +0 -14
- mirascope/core/mistral/_utils/_convert_common_call_params.py +0 -24
- mirascope/core/mistral/_utils/_convert_finish_reason_to_common_finish_reasons.py +0 -22
- mirascope/core/mistral/_utils/_convert_message_params.py +0 -122
- mirascope/core/mistral/_utils/_get_json_output.py +0 -34
- mirascope/core/mistral/_utils/_handle_stream.py +0 -139
- mirascope/core/mistral/_utils/_message_param_converter.py +0 -176
- mirascope/core/mistral/_utils/_setup_call.py +0 -164
- mirascope/core/mistral/call_params.py +0 -36
- mirascope/core/mistral/call_response.py +0 -205
- mirascope/core/mistral/call_response_chunk.py +0 -105
- mirascope/core/mistral/dynamic_config.py +0 -33
- mirascope/core/mistral/py.typed +0 -0
- mirascope/core/mistral/stream.py +0 -120
- mirascope/core/mistral/tool.py +0 -81
- mirascope/core/openai/__init__.py +0 -31
- mirascope/core/openai/_call.py +0 -67
- mirascope/core/openai/_call_kwargs.py +0 -13
- mirascope/core/openai/_utils/__init__.py +0 -14
- mirascope/core/openai/_utils/_convert_common_call_params.py +0 -26
- mirascope/core/openai/_utils/_convert_message_params.py +0 -148
- mirascope/core/openai/_utils/_get_json_output.py +0 -31
- mirascope/core/openai/_utils/_handle_stream.py +0 -138
- mirascope/core/openai/_utils/_message_param_converter.py +0 -105
- mirascope/core/openai/_utils/_setup_call.py +0 -155
- mirascope/core/openai/call_params.py +0 -92
- mirascope/core/openai/call_response.py +0 -273
- mirascope/core/openai/call_response_chunk.py +0 -139
- mirascope/core/openai/dynamic_config.py +0 -34
- mirascope/core/openai/py.typed +0 -0
- mirascope/core/openai/stream.py +0 -185
- mirascope/core/openai/tool.py +0 -101
- mirascope/core/py.typed +0 -0
- mirascope/core/vertex/__init__.py +0 -45
- mirascope/core/vertex/_call.py +0 -62
- mirascope/core/vertex/_call_kwargs.py +0 -12
- mirascope/core/vertex/_utils/__init__.py +0 -14
- mirascope/core/vertex/_utils/_convert_common_call_params.py +0 -37
- mirascope/core/vertex/_utils/_convert_finish_reason_to_common_finish_reasons.py +0 -23
- mirascope/core/vertex/_utils/_convert_message_params.py +0 -171
- mirascope/core/vertex/_utils/_get_json_output.py +0 -36
- mirascope/core/vertex/_utils/_handle_stream.py +0 -33
- mirascope/core/vertex/_utils/_message_param_converter.py +0 -133
- mirascope/core/vertex/_utils/_setup_call.py +0 -160
- mirascope/core/vertex/call_params.py +0 -24
- mirascope/core/vertex/call_response.py +0 -206
- mirascope/core/vertex/call_response_chunk.py +0 -99
- mirascope/core/vertex/dynamic_config.py +0 -28
- mirascope/core/vertex/stream.py +0 -119
- mirascope/core/vertex/tool.py +0 -101
- mirascope/core/xai/__init__.py +0 -28
- mirascope/core/xai/_call.py +0 -67
- mirascope/core/xai/_utils/__init__.py +0 -5
- mirascope/core/xai/_utils/_setup_call.py +0 -113
- mirascope/core/xai/call_params.py +0 -10
- mirascope/core/xai/call_response.py +0 -16
- mirascope/core/xai/call_response_chunk.py +0 -14
- mirascope/core/xai/dynamic_config.py +0 -8
- mirascope/core/xai/py.typed +0 -0
- mirascope/core/xai/stream.py +0 -57
- mirascope/core/xai/tool.py +0 -13
- mirascope/experimental/graphs/__init__.py +0 -5
- mirascope/integrations/__init__.py +0 -16
- mirascope/integrations/_middleware_factory.py +0 -403
- mirascope/integrations/langfuse/__init__.py +0 -3
- mirascope/integrations/langfuse/_utils.py +0 -114
- mirascope/integrations/langfuse/_with_langfuse.py +0 -70
- mirascope/integrations/logfire/__init__.py +0 -3
- mirascope/integrations/logfire/_utils.py +0 -225
- mirascope/integrations/logfire/_with_logfire.py +0 -63
- mirascope/integrations/otel/__init__.py +0 -10
- mirascope/integrations/otel/_utils.py +0 -270
- mirascope/integrations/otel/_with_hyperdx.py +0 -60
- mirascope/integrations/otel/_with_otel.py +0 -59
- mirascope/integrations/tenacity.py +0 -14
- mirascope/llm/_call.py +0 -401
- mirascope/llm/_context.py +0 -384
- mirascope/llm/_override.py +0 -3639
- mirascope/llm/_protocols.py +0 -500
- mirascope/llm/_response_metaclass.py +0 -31
- mirascope/llm/call_response.py +0 -158
- mirascope/llm/call_response_chunk.py +0 -66
- mirascope/llm/stream.py +0 -162
- mirascope/llm/tool.py +0 -64
- mirascope/mcp/__init__.py +0 -7
- mirascope/mcp/_utils.py +0 -288
- mirascope/mcp/client.py +0 -167
- mirascope/mcp/server.py +0 -356
- mirascope/mcp/tools.py +0 -110
- mirascope/py.typed +0 -0
- mirascope/retries/__init__.py +0 -11
- mirascope/retries/fallback.py +0 -131
- mirascope/retries/tenacity.py +0 -50
- mirascope/tools/__init__.py +0 -37
- mirascope/tools/base.py +0 -98
- mirascope/tools/system/__init__.py +0 -0
- mirascope/tools/system/_docker_operation.py +0 -166
- mirascope/tools/system/_file_system.py +0 -267
- mirascope/tools/web/__init__.py +0 -0
- mirascope/tools/web/_duckduckgo.py +0 -111
- mirascope/tools/web/_httpx.py +0 -125
- mirascope/tools/web/_parse_url_content.py +0 -94
- mirascope/tools/web/_requests.py +0 -54
- mirascope/v0/__init__.py +0 -43
- mirascope/v0/anthropic.py +0 -54
- mirascope/v0/base/__init__.py +0 -12
- mirascope/v0/base/calls.py +0 -118
- mirascope/v0/base/extractors.py +0 -122
- mirascope/v0/base/ops_utils.py +0 -207
- mirascope/v0/base/prompts.py +0 -48
- mirascope/v0/base/types.py +0 -14
- mirascope/v0/base/utils.py +0 -21
- mirascope/v0/openai.py +0 -54
- mirascope-1.25.7.dist-info/METADATA +0 -169
- mirascope-1.25.7.dist-info/RECORD +0 -378
- {mirascope-1.25.7.dist-info → mirascope-2.0.0a0.dist-info}/WHEEL +0 -0
- {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]
|
mirascope/core/base/_create.py
DELETED
|
@@ -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
|
mirascope/core/base/_extract.py
DELETED
|
@@ -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
|