mirascope 2.0.0a2__py3-none-any.whl → 2.0.0a4__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 (252) hide show
  1. mirascope/__init__.py +2 -2
  2. mirascope/api/__init__.py +6 -0
  3. mirascope/api/_generated/README.md +207 -0
  4. mirascope/api/_generated/__init__.py +141 -0
  5. mirascope/api/_generated/client.py +163 -0
  6. mirascope/api/_generated/core/__init__.py +52 -0
  7. mirascope/api/_generated/core/api_error.py +23 -0
  8. mirascope/api/_generated/core/client_wrapper.py +58 -0
  9. mirascope/api/_generated/core/datetime_utils.py +30 -0
  10. mirascope/api/_generated/core/file.py +70 -0
  11. mirascope/api/_generated/core/force_multipart.py +16 -0
  12. mirascope/api/_generated/core/http_client.py +619 -0
  13. mirascope/api/_generated/core/http_response.py +55 -0
  14. mirascope/api/_generated/core/jsonable_encoder.py +102 -0
  15. mirascope/api/_generated/core/pydantic_utilities.py +310 -0
  16. mirascope/api/_generated/core/query_encoder.py +60 -0
  17. mirascope/api/_generated/core/remove_none_from_dict.py +11 -0
  18. mirascope/api/_generated/core/request_options.py +35 -0
  19. mirascope/api/_generated/core/serialization.py +282 -0
  20. mirascope/api/_generated/docs/__init__.py +4 -0
  21. mirascope/api/_generated/docs/client.py +95 -0
  22. mirascope/api/_generated/docs/raw_client.py +132 -0
  23. mirascope/api/_generated/environment.py +9 -0
  24. mirascope/api/_generated/errors/__init__.py +17 -0
  25. mirascope/api/_generated/errors/bad_request_error.py +15 -0
  26. mirascope/api/_generated/errors/conflict_error.py +15 -0
  27. mirascope/api/_generated/errors/forbidden_error.py +15 -0
  28. mirascope/api/_generated/errors/internal_server_error.py +15 -0
  29. mirascope/api/_generated/errors/not_found_error.py +15 -0
  30. mirascope/api/_generated/health/__init__.py +7 -0
  31. mirascope/api/_generated/health/client.py +96 -0
  32. mirascope/api/_generated/health/raw_client.py +129 -0
  33. mirascope/api/_generated/health/types/__init__.py +8 -0
  34. mirascope/api/_generated/health/types/health_check_response.py +24 -0
  35. mirascope/api/_generated/health/types/health_check_response_status.py +5 -0
  36. mirascope/api/_generated/organizations/__init__.py +25 -0
  37. mirascope/api/_generated/organizations/client.py +380 -0
  38. mirascope/api/_generated/organizations/raw_client.py +876 -0
  39. mirascope/api/_generated/organizations/types/__init__.py +23 -0
  40. mirascope/api/_generated/organizations/types/organizations_create_response.py +24 -0
  41. mirascope/api/_generated/organizations/types/organizations_create_response_role.py +7 -0
  42. mirascope/api/_generated/organizations/types/organizations_get_response.py +24 -0
  43. mirascope/api/_generated/organizations/types/organizations_get_response_role.py +7 -0
  44. mirascope/api/_generated/organizations/types/organizations_list_response_item.py +24 -0
  45. mirascope/api/_generated/organizations/types/organizations_list_response_item_role.py +7 -0
  46. mirascope/api/_generated/organizations/types/organizations_update_response.py +24 -0
  47. mirascope/api/_generated/organizations/types/organizations_update_response_role.py +7 -0
  48. mirascope/api/_generated/projects/__init__.py +17 -0
  49. mirascope/api/_generated/projects/client.py +458 -0
  50. mirascope/api/_generated/projects/raw_client.py +1016 -0
  51. mirascope/api/_generated/projects/types/__init__.py +15 -0
  52. mirascope/api/_generated/projects/types/projects_create_response.py +30 -0
  53. mirascope/api/_generated/projects/types/projects_get_response.py +30 -0
  54. mirascope/api/_generated/projects/types/projects_list_response_item.py +30 -0
  55. mirascope/api/_generated/projects/types/projects_update_response.py +30 -0
  56. mirascope/api/_generated/reference.md +753 -0
  57. mirascope/api/_generated/traces/__init__.py +55 -0
  58. mirascope/api/_generated/traces/client.py +162 -0
  59. mirascope/api/_generated/traces/raw_client.py +168 -0
  60. mirascope/api/_generated/traces/types/__init__.py +95 -0
  61. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item.py +36 -0
  62. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_resource.py +31 -0
  63. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_resource_attributes_item.py +25 -0
  64. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_resource_attributes_item_value.py +54 -0
  65. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_resource_attributes_item_value_array_value.py +23 -0
  66. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_resource_attributes_item_value_kvlist_value.py +28 -0
  67. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_resource_attributes_item_value_kvlist_value_values_item.py +24 -0
  68. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item.py +35 -0
  69. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item_scope.py +35 -0
  70. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item_scope_attributes_item.py +27 -0
  71. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item_scope_attributes_item_value.py +54 -0
  72. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item_scope_attributes_item_value_array_value.py +23 -0
  73. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item_scope_attributes_item_value_kvlist_value.py +28 -0
  74. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item_scope_attributes_item_value_kvlist_value_values_item.py +24 -0
  75. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item_spans_item.py +60 -0
  76. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item_spans_item_attributes_item.py +29 -0
  77. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item_spans_item_attributes_item_value.py +54 -0
  78. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item_spans_item_attributes_item_value_array_value.py +23 -0
  79. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item_spans_item_attributes_item_value_kvlist_value.py +28 -0
  80. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item_spans_item_attributes_item_value_kvlist_value_values_item.py +24 -0
  81. mirascope/api/_generated/traces/types/traces_create_request_resource_spans_item_scope_spans_item_spans_item_status.py +24 -0
  82. mirascope/api/_generated/traces/types/traces_create_response.py +27 -0
  83. mirascope/api/_generated/traces/types/traces_create_response_partial_success.py +28 -0
  84. mirascope/api/_generated/types/__init__.py +37 -0
  85. mirascope/api/_generated/types/already_exists_error.py +24 -0
  86. mirascope/api/_generated/types/already_exists_error_tag.py +5 -0
  87. mirascope/api/_generated/types/database_error.py +24 -0
  88. mirascope/api/_generated/types/database_error_tag.py +5 -0
  89. mirascope/api/_generated/types/http_api_decode_error.py +29 -0
  90. mirascope/api/_generated/types/http_api_decode_error_tag.py +5 -0
  91. mirascope/api/_generated/types/issue.py +40 -0
  92. mirascope/api/_generated/types/issue_tag.py +17 -0
  93. mirascope/api/_generated/types/not_found_error_body.py +24 -0
  94. mirascope/api/_generated/types/not_found_error_tag.py +5 -0
  95. mirascope/api/_generated/types/permission_denied_error.py +24 -0
  96. mirascope/api/_generated/types/permission_denied_error_tag.py +7 -0
  97. mirascope/api/_generated/types/property_key.py +7 -0
  98. mirascope/api/_generated/types/property_key_key.py +27 -0
  99. mirascope/api/_generated/types/property_key_key_tag.py +5 -0
  100. mirascope/api/client.py +255 -0
  101. mirascope/api/settings.py +81 -0
  102. mirascope/llm/__init__.py +45 -11
  103. mirascope/llm/calls/calls.py +81 -57
  104. mirascope/llm/calls/decorator.py +121 -115
  105. mirascope/llm/content/__init__.py +3 -2
  106. mirascope/llm/context/_utils.py +19 -6
  107. mirascope/llm/exceptions.py +30 -16
  108. mirascope/llm/formatting/_utils.py +9 -5
  109. mirascope/llm/formatting/format.py +2 -2
  110. mirascope/llm/formatting/from_call_args.py +2 -2
  111. mirascope/llm/messages/message.py +13 -5
  112. mirascope/llm/models/__init__.py +2 -2
  113. mirascope/llm/models/models.py +189 -81
  114. mirascope/llm/prompts/__init__.py +13 -12
  115. mirascope/llm/prompts/_utils.py +27 -24
  116. mirascope/llm/prompts/decorator.py +133 -204
  117. mirascope/llm/prompts/prompts.py +424 -0
  118. mirascope/llm/prompts/protocols.py +25 -59
  119. mirascope/llm/providers/__init__.py +44 -0
  120. mirascope/llm/{clients → providers}/_missing_import_stubs.py +8 -6
  121. mirascope/llm/providers/anthropic/__init__.py +29 -0
  122. mirascope/llm/providers/anthropic/_utils/__init__.py +23 -0
  123. mirascope/llm/providers/anthropic/_utils/beta_decode.py +271 -0
  124. mirascope/llm/providers/anthropic/_utils/beta_encode.py +216 -0
  125. mirascope/llm/{clients → providers}/anthropic/_utils/decode.py +44 -11
  126. mirascope/llm/providers/anthropic/_utils/encode.py +356 -0
  127. mirascope/llm/providers/anthropic/beta_provider.py +322 -0
  128. mirascope/llm/providers/anthropic/model_id.py +23 -0
  129. mirascope/llm/providers/anthropic/model_info.py +87 -0
  130. mirascope/llm/providers/anthropic/provider.py +416 -0
  131. mirascope/llm/{clients → providers}/base/__init__.py +3 -3
  132. mirascope/llm/{clients → providers}/base/_utils.py +25 -8
  133. mirascope/llm/{clients/base/client.py → providers/base/base_provider.py} +255 -126
  134. mirascope/llm/providers/google/__init__.py +21 -0
  135. mirascope/llm/{clients → providers}/google/_utils/decode.py +61 -7
  136. mirascope/llm/{clients → providers}/google/_utils/encode.py +44 -30
  137. mirascope/llm/providers/google/model_id.py +22 -0
  138. mirascope/llm/providers/google/model_info.py +62 -0
  139. mirascope/llm/providers/google/provider.py +442 -0
  140. mirascope/llm/providers/load_provider.py +54 -0
  141. mirascope/llm/providers/mlx/__init__.py +24 -0
  142. mirascope/llm/providers/mlx/_utils.py +129 -0
  143. mirascope/llm/providers/mlx/encoding/__init__.py +8 -0
  144. mirascope/llm/providers/mlx/encoding/base.py +69 -0
  145. mirascope/llm/providers/mlx/encoding/transformers.py +147 -0
  146. mirascope/llm/providers/mlx/mlx.py +237 -0
  147. mirascope/llm/providers/mlx/model_id.py +17 -0
  148. mirascope/llm/providers/mlx/provider.py +415 -0
  149. mirascope/llm/providers/model_id.py +16 -0
  150. mirascope/llm/providers/ollama/__init__.py +19 -0
  151. mirascope/llm/providers/ollama/provider.py +71 -0
  152. mirascope/llm/providers/openai/__init__.py +6 -0
  153. mirascope/llm/providers/openai/completions/__init__.py +25 -0
  154. mirascope/llm/{clients → providers}/openai/completions/_utils/__init__.py +2 -0
  155. mirascope/llm/{clients → providers}/openai/completions/_utils/decode.py +60 -6
  156. mirascope/llm/{clients → providers}/openai/completions/_utils/encode.py +37 -26
  157. mirascope/llm/providers/openai/completions/base_provider.py +513 -0
  158. mirascope/llm/providers/openai/completions/provider.py +22 -0
  159. mirascope/llm/providers/openai/model_id.py +31 -0
  160. mirascope/llm/providers/openai/model_info.py +303 -0
  161. mirascope/llm/providers/openai/provider.py +398 -0
  162. mirascope/llm/providers/openai/responses/__init__.py +21 -0
  163. mirascope/llm/{clients → providers}/openai/responses/_utils/decode.py +59 -6
  164. mirascope/llm/{clients → providers}/openai/responses/_utils/encode.py +34 -23
  165. mirascope/llm/providers/openai/responses/provider.py +469 -0
  166. mirascope/llm/providers/provider_id.py +23 -0
  167. mirascope/llm/providers/provider_registry.py +169 -0
  168. mirascope/llm/providers/together/__init__.py +19 -0
  169. mirascope/llm/providers/together/provider.py +40 -0
  170. mirascope/llm/responses/__init__.py +3 -0
  171. mirascope/llm/responses/base_response.py +14 -5
  172. mirascope/llm/responses/base_stream_response.py +35 -6
  173. mirascope/llm/responses/finish_reason.py +1 -0
  174. mirascope/llm/responses/response.py +33 -13
  175. mirascope/llm/responses/root_response.py +12 -13
  176. mirascope/llm/responses/stream_response.py +35 -23
  177. mirascope/llm/responses/usage.py +95 -0
  178. mirascope/llm/tools/__init__.py +9 -2
  179. mirascope/llm/tools/_utils.py +12 -3
  180. mirascope/llm/tools/protocols.py +4 -4
  181. mirascope/llm/tools/tool_schema.py +44 -9
  182. mirascope/llm/tools/tools.py +10 -9
  183. mirascope/ops/__init__.py +156 -0
  184. mirascope/ops/_internal/__init__.py +5 -0
  185. mirascope/ops/_internal/closure.py +1118 -0
  186. mirascope/ops/_internal/configuration.py +126 -0
  187. mirascope/ops/_internal/context.py +76 -0
  188. mirascope/ops/_internal/exporters/__init__.py +26 -0
  189. mirascope/ops/_internal/exporters/exporters.py +342 -0
  190. mirascope/ops/_internal/exporters/processors.py +104 -0
  191. mirascope/ops/_internal/exporters/types.py +165 -0
  192. mirascope/ops/_internal/exporters/utils.py +29 -0
  193. mirascope/ops/_internal/instrumentation/__init__.py +8 -0
  194. mirascope/ops/_internal/instrumentation/llm/__init__.py +8 -0
  195. mirascope/ops/_internal/instrumentation/llm/encode.py +238 -0
  196. mirascope/ops/_internal/instrumentation/llm/gen_ai_types/__init__.py +38 -0
  197. mirascope/ops/_internal/instrumentation/llm/gen_ai_types/gen_ai_input_messages.py +31 -0
  198. mirascope/ops/_internal/instrumentation/llm/gen_ai_types/gen_ai_output_messages.py +38 -0
  199. mirascope/ops/_internal/instrumentation/llm/gen_ai_types/gen_ai_system_instructions.py +18 -0
  200. mirascope/ops/_internal/instrumentation/llm/gen_ai_types/shared.py +100 -0
  201. mirascope/ops/_internal/instrumentation/llm/llm.py +1288 -0
  202. mirascope/ops/_internal/propagation.py +198 -0
  203. mirascope/ops/_internal/protocols.py +51 -0
  204. mirascope/ops/_internal/session.py +139 -0
  205. mirascope/ops/_internal/spans.py +232 -0
  206. mirascope/ops/_internal/traced_calls.py +371 -0
  207. mirascope/ops/_internal/traced_functions.py +394 -0
  208. mirascope/ops/_internal/tracing.py +276 -0
  209. mirascope/ops/_internal/types.py +13 -0
  210. mirascope/ops/_internal/utils.py +75 -0
  211. mirascope/ops/_internal/versioned_calls.py +512 -0
  212. mirascope/ops/_internal/versioned_functions.py +346 -0
  213. mirascope/ops/_internal/versioning.py +303 -0
  214. mirascope/ops/exceptions.py +21 -0
  215. {mirascope-2.0.0a2.dist-info → mirascope-2.0.0a4.dist-info}/METADATA +78 -3
  216. mirascope-2.0.0a4.dist-info/RECORD +247 -0
  217. {mirascope-2.0.0a2.dist-info → mirascope-2.0.0a4.dist-info}/WHEEL +1 -1
  218. mirascope/graphs/__init__.py +0 -22
  219. mirascope/graphs/finite_state_machine.py +0 -625
  220. mirascope/llm/agents/__init__.py +0 -15
  221. mirascope/llm/agents/agent.py +0 -97
  222. mirascope/llm/agents/agent_template.py +0 -45
  223. mirascope/llm/agents/decorator.py +0 -176
  224. mirascope/llm/calls/base_call.py +0 -33
  225. mirascope/llm/clients/__init__.py +0 -34
  226. mirascope/llm/clients/anthropic/__init__.py +0 -25
  227. mirascope/llm/clients/anthropic/_utils/encode.py +0 -243
  228. mirascope/llm/clients/anthropic/clients.py +0 -819
  229. mirascope/llm/clients/anthropic/model_ids.py +0 -8
  230. mirascope/llm/clients/google/__init__.py +0 -20
  231. mirascope/llm/clients/google/clients.py +0 -853
  232. mirascope/llm/clients/google/model_ids.py +0 -15
  233. mirascope/llm/clients/openai/__init__.py +0 -25
  234. mirascope/llm/clients/openai/completions/__init__.py +0 -28
  235. mirascope/llm/clients/openai/completions/_utils/model_features.py +0 -81
  236. mirascope/llm/clients/openai/completions/clients.py +0 -833
  237. mirascope/llm/clients/openai/completions/model_ids.py +0 -8
  238. mirascope/llm/clients/openai/responses/__init__.py +0 -26
  239. mirascope/llm/clients/openai/responses/_utils/__init__.py +0 -13
  240. mirascope/llm/clients/openai/responses/_utils/model_features.py +0 -87
  241. mirascope/llm/clients/openai/responses/clients.py +0 -832
  242. mirascope/llm/clients/openai/responses/model_ids.py +0 -8
  243. mirascope/llm/clients/openai/shared/__init__.py +0 -7
  244. mirascope/llm/clients/openai/shared/_utils.py +0 -55
  245. mirascope/llm/clients/providers.py +0 -175
  246. mirascope-2.0.0a2.dist-info/RECORD +0 -102
  247. /mirascope/llm/{clients → providers}/base/kwargs.py +0 -0
  248. /mirascope/llm/{clients → providers}/base/params.py +0 -0
  249. /mirascope/llm/{clients/anthropic → providers/google}/_utils/__init__.py +0 -0
  250. /mirascope/llm/{clients → providers}/google/message.py +0 -0
  251. /mirascope/llm/{clients/google → providers/openai/responses}/_utils/__init__.py +0 -0
  252. {mirascope-2.0.0a2.dist-info → mirascope-2.0.0a4.dist-info}/licenses/LICENSE +0 -0
mirascope/llm/providers/openai/completions/provider.py ADDED
@@ -0,0 +1,22 @@
1
+ """OpenAI Completions API provider implementation."""
2
+
3
+ from ..model_id import model_name
4
+ from .base_provider import BaseOpenAICompletionsProvider
5
+
6
+
7
+ class OpenAICompletionsProvider(BaseOpenAICompletionsProvider):
8
+ """Provider for OpenAI's ChatCompletions API."""
9
+
10
+ id = "openai:completions"
11
+ default_scope = "openai/"
12
+ default_base_url = None
13
+ api_key_env_var = "OPENAI_API_KEY"
14
+ api_key_required = False
15
+ provider_name = "OpenAI"
16
+
17
+ def _provider_model_name(self, model_id: str) -> str:
18
+ """Get the model name for tracking in Response.
19
+
20
+ Returns the model name with :completions suffix for tracking which API was used.
21
+ """
22
+ return model_name(model_id, "completions")
mirascope/llm/providers/openai/model_id.py ADDED
@@ -0,0 +1,31 @@
1
+ """OpenAI model ids, and related utilities."""
2
+
3
+ from typing import Literal, TypeAlias, get_args
4
+
5
+ from .model_info import OpenAIKnownModels
6
+
7
+ OpenAIModelId = OpenAIKnownModels | str
8
+ """Valid OpenAI model IDs including API-specific variants."""
9
+
10
+ OPENAI_KNOWN_MODELS: set[str] = set(get_args(OpenAIKnownModels))
11
+
12
+ ApiMode: TypeAlias = Literal["responses", "completions"]
13
+
14
+
15
+ def model_name(model_id: OpenAIModelId, api_mode: ApiMode | None) -> str:
16
+ """Extract the openai model name from the ModelId
17
+
18
+ Args:
19
+ model_id: Full model ID (e.g. "openai/gpt-4o")
20
+ api_mode: API mode to append as suffix ("responses" or "completions").
21
+ If None, no suffix will be added (just the base model name).
22
+
23
+ Returns:
24
+ Provider-specific model ID with API suffix (e.g. "gpt-4o:responses")
25
+ """
26
+ base_name = (
27
+ model_id.split("/")[1].removesuffix(":responses").removesuffix(":completions")
28
+ )
29
+ if api_mode is None:
30
+ return base_name
31
+ return f"{base_name}:{api_mode}"
mirascope/llm/providers/openai/model_info.py ADDED
@@ -0,0 +1,303 @@
1
+ """OpenAI model information.
2
+
3
+ This file is auto-generated by scripts/model_features/codegen_openai.py
4
+ Do not edit manually - run the codegen script to update."""
5
+
6
+ from typing import Literal
7
+
8
+ OpenAIKnownModels = Literal[
9
+ "openai/chatgpt-4o-latest",
10
+ "openai/chatgpt-4o-latest:completions",
11
+ "openai/chatgpt-4o-latest:responses",
12
+ "openai/codex-mini-latest",
13
+ "openai/codex-mini-latest:responses",
14
+ "openai/gpt-3.5-turbo",
15
+ "openai/gpt-3.5-turbo:completions",
16
+ "openai/gpt-3.5-turbo:responses",
17
+ "openai/gpt-3.5-turbo-0125",
18
+ "openai/gpt-3.5-turbo-0125:completions",
19
+ "openai/gpt-3.5-turbo-0125:responses",
20
+ "openai/gpt-3.5-turbo-1106",
21
+ "openai/gpt-3.5-turbo-1106:completions",
22
+ "openai/gpt-3.5-turbo-1106:responses",
23
+ "openai/gpt-3.5-turbo-16k",
24
+ "openai/gpt-3.5-turbo-16k:completions",
25
+ "openai/gpt-4",
26
+ "openai/gpt-4:completions",
27
+ "openai/gpt-4:responses",
28
+ "openai/gpt-4-0125-preview",
29
+ "openai/gpt-4-0125-preview:completions",
30
+ "openai/gpt-4-0125-preview:responses",
31
+ "openai/gpt-4-0613",
32
+ "openai/gpt-4-0613:completions",
33
+ "openai/gpt-4-0613:responses",
34
+ "openai/gpt-4-1106-preview",
35
+ "openai/gpt-4-1106-preview:completions",
36
+ "openai/gpt-4-1106-preview:responses",
37
+ "openai/gpt-4-turbo",
38
+ "openai/gpt-4-turbo:completions",
39
+ "openai/gpt-4-turbo:responses",
40
+ "openai/gpt-4-turbo-2024-04-09",
41
+ "openai/gpt-4-turbo-2024-04-09:completions",
42
+ "openai/gpt-4-turbo-2024-04-09:responses",
43
+ "openai/gpt-4-turbo-preview",
44
+ "openai/gpt-4-turbo-preview:completions",
45
+ "openai/gpt-4-turbo-preview:responses",
46
+ "openai/gpt-4.1",
47
+ "openai/gpt-4.1:completions",
48
+ "openai/gpt-4.1:responses",
49
+ "openai/gpt-4.1-2025-04-14",
50
+ "openai/gpt-4.1-2025-04-14:completions",
51
+ "openai/gpt-4.1-2025-04-14:responses",
52
+ "openai/gpt-4.1-mini",
53
+ "openai/gpt-4.1-mini:completions",
54
+ "openai/gpt-4.1-mini:responses",
55
+ "openai/gpt-4.1-mini-2025-04-14",
56
+ "openai/gpt-4.1-mini-2025-04-14:completions",
57
+ "openai/gpt-4.1-mini-2025-04-14:responses",
58
+ "openai/gpt-4.1-nano",
59
+ "openai/gpt-4.1-nano:completions",
60
+ "openai/gpt-4.1-nano:responses",
61
+ "openai/gpt-4.1-nano-2025-04-14",
62
+ "openai/gpt-4.1-nano-2025-04-14:completions",
63
+ "openai/gpt-4.1-nano-2025-04-14:responses",
64
+ "openai/gpt-4o",
65
+ "openai/gpt-4o:completions",
66
+ "openai/gpt-4o:responses",
67
+ "openai/gpt-4o-2024-05-13",
68
+ "openai/gpt-4o-2024-05-13:completions",
69
+ "openai/gpt-4o-2024-05-13:responses",
70
+ "openai/gpt-4o-2024-08-06",
71
+ "openai/gpt-4o-2024-08-06:completions",
72
+ "openai/gpt-4o-2024-08-06:responses",
73
+ "openai/gpt-4o-2024-11-20",
74
+ "openai/gpt-4o-2024-11-20:completions",
75
+ "openai/gpt-4o-2024-11-20:responses",
76
+ "openai/gpt-4o-mini",
77
+ "openai/gpt-4o-mini:completions",
78
+ "openai/gpt-4o-mini:responses",
79
+ "openai/gpt-4o-mini-2024-07-18",
80
+ "openai/gpt-4o-mini-2024-07-18:completions",
81
+ "openai/gpt-4o-mini-2024-07-18:responses",
82
+ "openai/gpt-4o-mini-search-preview",
83
+ "openai/gpt-4o-mini-search-preview:completions",
84
+ "openai/gpt-4o-mini-search-preview-2025-03-11",
85
+ "openai/gpt-4o-mini-search-preview-2025-03-11:completions",
86
+ "openai/gpt-4o-search-preview",
87
+ "openai/gpt-4o-search-preview:completions",
88
+ "openai/gpt-4o-search-preview-2025-03-11",
89
+ "openai/gpt-4o-search-preview-2025-03-11:completions",
90
+ "openai/gpt-5",
91
+ "openai/gpt-5:completions",
92
+ "openai/gpt-5:responses",
93
+ "openai/gpt-5-2025-08-07",
94
+ "openai/gpt-5-2025-08-07:completions",
95
+ "openai/gpt-5-2025-08-07:responses",
96
+ "openai/gpt-5-chat-latest",
97
+ "openai/gpt-5-chat-latest:completions",
98
+ "openai/gpt-5-chat-latest:responses",
99
+ "openai/gpt-5-codex",
100
+ "openai/gpt-5-codex:responses",
101
+ "openai/gpt-5-mini",
102
+ "openai/gpt-5-mini:completions",
103
+ "openai/gpt-5-mini:responses",
104
+ "openai/gpt-5-mini-2025-08-07",
105
+ "openai/gpt-5-mini-2025-08-07:completions",
106
+ "openai/gpt-5-mini-2025-08-07:responses",
107
+ "openai/gpt-5-nano",
108
+ "openai/gpt-5-nano:completions",
109
+ "openai/gpt-5-nano:responses",
110
+ "openai/gpt-5-nano-2025-08-07",
111
+ "openai/gpt-5-nano-2025-08-07:completions",
112
+ "openai/gpt-5-nano-2025-08-07:responses",
113
+ "openai/gpt-5-pro",
114
+ "openai/gpt-5-pro:responses",
115
+ "openai/gpt-5-pro-2025-10-06",
116
+ "openai/gpt-5-pro-2025-10-06:responses",
117
+ "openai/gpt-5-search-api",
118
+ "openai/gpt-5-search-api:completions",
119
+ "openai/gpt-5-search-api-2025-10-14",
120
+ "openai/gpt-5-search-api-2025-10-14:completions",
121
+ "openai/gpt-5.1",
122
+ "openai/gpt-5.1:responses",
123
+ "openai/gpt-5.1-2025-11-13",
124
+ "openai/gpt-5.1-2025-11-13:responses",
125
+ "openai/gpt-5.1-chat-latest",
126
+ "openai/gpt-5.1-chat-latest:completions",
127
+ "openai/gpt-5.1-chat-latest:responses",
128
+ "openai/gpt-5.1-codex",
129
+ "openai/gpt-5.1-codex:responses",
130
+ "openai/gpt-5.1-codex-max",
131
+ "openai/gpt-5.1-codex-max:responses",
132
+ "openai/gpt-5.1-codex-mini",
133
+ "openai/gpt-5.1-codex-mini:responses",
134
+ "openai/gpt-5.2",
135
+ "openai/gpt-5.2:completions",
136
+ "openai/gpt-5.2:responses",
137
+ "openai/gpt-5.2-2025-12-11",
138
+ "openai/gpt-5.2-2025-12-11:completions",
139
+ "openai/gpt-5.2-2025-12-11:responses",
140
+ "openai/gpt-5.2-chat-latest",
141
+ "openai/gpt-5.2-chat-latest:completions",
142
+ "openai/gpt-5.2-chat-latest:responses",
143
+ "openai/gpt-5.2-pro",
144
+ "openai/gpt-5.2-pro:responses",
145
+ "openai/gpt-5.2-pro-2025-12-11",
146
+ "openai/gpt-5.2-pro-2025-12-11:responses",
147
+ "openai/o1",
148
+ "openai/o1:completions",
149
+ "openai/o1:responses",
150
+ "openai/o1-2024-12-17",
151
+ "openai/o1-2024-12-17:completions",
152
+ "openai/o1-2024-12-17:responses",
153
+ "openai/o1-pro",
154
+ "openai/o1-pro:responses",
155
+ "openai/o1-pro-2025-03-19",
156
+ "openai/o1-pro-2025-03-19:responses",
157
+ "openai/o3",
158
+ "openai/o3:completions",
159
+ "openai/o3:responses",
160
+ "openai/o3-2025-04-16",
161
+ "openai/o3-2025-04-16:completions",
162
+ "openai/o3-2025-04-16:responses",
163
+ "openai/o3-mini",
164
+ "openai/o3-mini:completions",
165
+ "openai/o3-mini:responses",
166
+ "openai/o3-mini-2025-01-31",
167
+ "openai/o3-mini-2025-01-31:completions",
168
+ "openai/o3-mini-2025-01-31:responses",
169
+ "openai/o3-pro",
170
+ "openai/o3-pro:responses",
171
+ "openai/o3-pro-2025-06-10",
172
+ "openai/o3-pro-2025-06-10:responses",
173
+ "openai/o4-mini",
174
+ "openai/o4-mini:completions",
175
+ "openai/o4-mini:responses",
176
+ "openai/o4-mini-2025-04-16",
177
+ "openai/o4-mini-2025-04-16:completions",
178
+ "openai/o4-mini-2025-04-16:responses",
179
+ ]
180
+ """Valid OpenAI model IDs including API-specific variants."""
181
+
182
+
183
+ MODELS_WITHOUT_AUDIO_SUPPORT: set[str] = {
184
+ "chatgpt-4o-latest",
185
+ "gpt-3.5-turbo",
186
+ "gpt-3.5-turbo-0125",
187
+ "gpt-3.5-turbo-1106",
188
+ "gpt-3.5-turbo-16k",
189
+ "gpt-4",
190
+ "gpt-4-0125-preview",
191
+ "gpt-4-0613",
192
+ "gpt-4-1106-preview",
193
+ "gpt-4-turbo",
194
+ "gpt-4-turbo-2024-04-09",
195
+ "gpt-4-turbo-preview",
196
+ "gpt-4.1",
197
+ "gpt-4.1-2025-04-14",
198
+ "gpt-4.1-mini",
199
+ "gpt-4.1-mini-2025-04-14",
200
+ "gpt-4.1-nano",
201
+ "gpt-4.1-nano-2025-04-14",
202
+ "gpt-4o",
203
+ "gpt-4o-2024-05-13",
204
+ "gpt-4o-2024-08-06",
205
+ "gpt-4o-2024-11-20",
206
+ "gpt-4o-mini",
207
+ "gpt-4o-mini-2024-07-18",
208
+ "gpt-4o-mini-search-preview",
209
+ "gpt-4o-mini-search-preview-2025-03-11",
210
+ "gpt-4o-search-preview",
211
+ "gpt-4o-search-preview-2025-03-11",
212
+ "gpt-5",
213
+ "gpt-5-2025-08-07",
214
+ "gpt-5-chat-latest",
215
+ "gpt-5-mini",
216
+ "gpt-5-mini-2025-08-07",
217
+ "gpt-5-nano",
218
+ "gpt-5-nano-2025-08-07",
219
+ "gpt-5-search-api",
220
+ "gpt-5-search-api-2025-10-14",
221
+ "gpt-5.1-chat-latest",
222
+ "gpt-5.2",
223
+ "gpt-5.2-2025-12-11",
224
+ "gpt-5.2-chat-latest",
225
+ "o1",
226
+ "o1-2024-12-17",
227
+ "o3",
228
+ "o3-2025-04-16",
229
+ "o3-mini",
230
+ "o3-mini-2025-01-31",
231
+ "o4-mini",
232
+ "o4-mini-2025-04-16",
233
+ }
234
+ """Models that do not support audio inputs.
235
+
236
+ Models not in this set are assumed to support audio (optimistic default).
237
+ """
238
+
239
+ NON_REASONING_MODELS: set[str] = {
240
+ "chatgpt-4o-latest",
241
+ "gpt-3.5-turbo",
242
+ "gpt-3.5-turbo-0125",
243
+ "gpt-3.5-turbo-1106",
244
+ "gpt-4",
245
+ "gpt-4-0125-preview",
246
+ "gpt-4-0613",
247
+ "gpt-4-1106-preview",
248
+ "gpt-4-turbo",
249
+ "gpt-4-turbo-2024-04-09",
250
+ "gpt-4-turbo-preview",
251
+ "gpt-4.1",
252
+ "gpt-4.1-2025-04-14",
253
+ "gpt-4.1-mini",
254
+ "gpt-4.1-mini-2025-04-14",
255
+ "gpt-4.1-nano",
256
+ "gpt-4.1-nano-2025-04-14",
257
+ "gpt-4o",
258
+ "gpt-4o-2024-05-13",
259
+ "gpt-4o-2024-08-06",
260
+ "gpt-4o-2024-11-20",
261
+ "gpt-4o-mini",
262
+ "gpt-4o-mini-2024-07-18",
263
+ "gpt-5-chat-latest",
264
+ }
265
+ """Models that do not support the reasoning parameter.
266
+
267
+ Models not in this set are assumed to support reasoning (optimistic default).
268
+ """
269
+
270
+ MODELS_WITHOUT_JSON_SCHEMA_SUPPORT: set[str] = {
271
+ "chatgpt-4o-latest",
272
+ "gpt-3.5-turbo",
273
+ "gpt-3.5-turbo-0125",
274
+ "gpt-3.5-turbo-1106",
275
+ "gpt-3.5-turbo-16k",
276
+ "gpt-4",
277
+ "gpt-4-0125-preview",
278
+ "gpt-4-0613",
279
+ "gpt-4-1106-preview",
280
+ "gpt-4-turbo",
281
+ "gpt-4-turbo-2024-04-09",
282
+ "gpt-4-turbo-preview",
283
+ "gpt-4o-2024-05-13",
284
+ }
285
+ """Models that do not support JSON schema (structured outputs).
286
+
287
+ Models not in this set are assumed to support JSON schema (optimistic default).
288
+ """
289
+
290
+ MODELS_WITHOUT_JSON_OBJECT_SUPPORT: set[str] = {
291
+ "gpt-4",
292
+ "gpt-4-0613",
293
+ "gpt-4o-mini-search-preview",
294
+ "gpt-4o-mini-search-preview-2025-03-11",
295
+ "gpt-4o-search-preview",
296
+ "gpt-4o-search-preview-2025-03-11",
297
+ "gpt-5-search-api",
298
+ "gpt-5-search-api-2025-10-14",
299
+ }
300
+ """Models that do not support JSON object mode.
301
+
302
+ Models not in this set are assumed to support JSON object mode (optimistic default).
303
+ """