opik 1.6.4__py3-none-any.whl → 1.9.71__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 (1021) hide show
  1. opik/__init__.py +33 -2
  2. opik/anonymizer/__init__.py +5 -0
  3. opik/anonymizer/anonymizer.py +12 -0
  4. opik/anonymizer/factory.py +80 -0
  5. opik/anonymizer/recursive_anonymizer.py +64 -0
  6. opik/anonymizer/rules.py +56 -0
  7. opik/anonymizer/rules_anonymizer.py +35 -0
  8. opik/api_objects/attachment/__init__.py +5 -0
  9. opik/api_objects/attachment/attachment.py +20 -0
  10. opik/api_objects/attachment/attachment_context.py +36 -0
  11. opik/api_objects/attachment/attachments_extractor.py +153 -0
  12. opik/api_objects/attachment/client.py +220 -0
  13. opik/api_objects/attachment/converters.py +51 -0
  14. opik/api_objects/attachment/decoder.py +18 -0
  15. opik/api_objects/attachment/decoder_base64.py +83 -0
  16. opik/api_objects/attachment/decoder_helpers.py +137 -0
  17. opik/api_objects/conversation/__init__.py +0 -0
  18. opik/api_objects/conversation/conversation_factory.py +43 -0
  19. opik/api_objects/conversation/conversation_thread.py +49 -0
  20. opik/api_objects/data_helpers.py +79 -0
  21. opik/api_objects/dataset/dataset.py +107 -45
  22. opik/api_objects/dataset/rest_operations.py +12 -3
  23. opik/api_objects/experiment/experiment.py +81 -45
  24. opik/api_objects/experiment/experiment_item.py +2 -1
  25. opik/api_objects/experiment/experiments_client.py +64 -0
  26. opik/api_objects/experiment/helpers.py +35 -11
  27. opik/api_objects/experiment/rest_operations.py +88 -19
  28. opik/api_objects/helpers.py +104 -7
  29. opik/api_objects/local_recording.py +81 -0
  30. opik/api_objects/opik_client.py +872 -174
  31. opik/api_objects/opik_query_language.py +136 -18
  32. opik/api_objects/optimization/__init__.py +3 -0
  33. opik/api_objects/optimization/optimization.py +39 -0
  34. opik/api_objects/prompt/__init__.py +13 -1
  35. opik/api_objects/prompt/base_prompt.py +69 -0
  36. opik/api_objects/prompt/base_prompt_template.py +29 -0
  37. opik/api_objects/prompt/chat/__init__.py +1 -0
  38. opik/api_objects/prompt/chat/chat_prompt.py +210 -0
  39. opik/api_objects/prompt/chat/chat_prompt_template.py +350 -0
  40. opik/api_objects/prompt/chat/content_renderer_registry.py +203 -0
  41. opik/api_objects/prompt/client.py +193 -41
  42. opik/api_objects/prompt/text/__init__.py +1 -0
  43. opik/api_objects/prompt/text/prompt.py +174 -0
  44. opik/api_objects/prompt/text/prompt_template.py +55 -0
  45. opik/api_objects/prompt/types.py +29 -0
  46. opik/api_objects/rest_stream_parser.py +98 -0
  47. opik/api_objects/search_helpers.py +89 -0
  48. opik/api_objects/span/span_client.py +165 -45
  49. opik/api_objects/span/span_data.py +136 -25
  50. opik/api_objects/threads/__init__.py +0 -0
  51. opik/api_objects/threads/threads_client.py +185 -0
  52. opik/api_objects/trace/trace_client.py +72 -36
  53. opik/api_objects/trace/trace_data.py +112 -26
  54. opik/api_objects/validation_helpers.py +3 -3
  55. opik/cli/__init__.py +5 -0
  56. opik/cli/__main__.py +6 -0
  57. opik/cli/configure.py +66 -0
  58. opik/cli/exports/__init__.py +131 -0
  59. opik/cli/exports/dataset.py +278 -0
  60. opik/cli/exports/experiment.py +784 -0
  61. opik/cli/exports/project.py +685 -0
  62. opik/cli/exports/prompt.py +578 -0
  63. opik/cli/exports/utils.py +406 -0
  64. opik/cli/harbor.py +39 -0
  65. opik/cli/healthcheck.py +21 -0
  66. opik/cli/imports/__init__.py +439 -0
  67. opik/cli/imports/dataset.py +143 -0
  68. opik/cli/imports/experiment.py +1192 -0
  69. opik/cli/imports/project.py +262 -0
  70. opik/cli/imports/prompt.py +177 -0
  71. opik/cli/imports/utils.py +280 -0
  72. opik/cli/main.py +49 -0
  73. opik/cli/proxy.py +93 -0
  74. opik/cli/usage_report/__init__.py +16 -0
  75. opik/cli/usage_report/charts.py +783 -0
  76. opik/cli/usage_report/cli.py +274 -0
  77. opik/cli/usage_report/constants.py +9 -0
  78. opik/cli/usage_report/extraction.py +749 -0
  79. opik/cli/usage_report/pdf.py +244 -0
  80. opik/cli/usage_report/statistics.py +78 -0
  81. opik/cli/usage_report/utils.py +235 -0
  82. opik/config.py +62 -4
  83. opik/configurator/configure.py +45 -6
  84. opik/configurator/opik_rest_helpers.py +4 -1
  85. opik/context_storage.py +164 -65
  86. opik/datetime_helpers.py +12 -0
  87. opik/decorator/arguments_helpers.py +9 -1
  88. opik/decorator/base_track_decorator.py +298 -146
  89. opik/decorator/context_manager/__init__.py +0 -0
  90. opik/decorator/context_manager/span_context_manager.py +123 -0
  91. opik/decorator/context_manager/trace_context_manager.py +84 -0
  92. opik/decorator/generator_wrappers.py +3 -2
  93. opik/decorator/inspect_helpers.py +11 -0
  94. opik/decorator/opik_args/__init__.py +13 -0
  95. opik/decorator/opik_args/api_classes.py +71 -0
  96. opik/decorator/opik_args/helpers.py +120 -0
  97. opik/decorator/span_creation_handler.py +49 -21
  98. opik/decorator/tracker.py +9 -1
  99. opik/dict_utils.py +3 -3
  100. opik/environment.py +13 -1
  101. opik/error_tracking/api.py +1 -1
  102. opik/error_tracking/before_send.py +6 -5
  103. opik/error_tracking/environment_details.py +29 -7
  104. opik/error_tracking/error_filtering/filter_by_response_status_code.py +42 -0
  105. opik/error_tracking/error_filtering/filter_chain_builder.py +14 -3
  106. opik/evaluation/__init__.py +14 -2
  107. opik/evaluation/engine/engine.py +280 -82
  108. opik/evaluation/engine/evaluation_tasks_executor.py +15 -10
  109. opik/evaluation/engine/helpers.py +34 -9
  110. opik/evaluation/engine/metrics_evaluator.py +237 -0
  111. opik/evaluation/engine/types.py +5 -4
  112. opik/evaluation/evaluation_result.py +169 -2
  113. opik/evaluation/evaluator.py +659 -58
  114. opik/evaluation/metrics/__init__.py +121 -6
  115. opik/evaluation/metrics/aggregated_metric.py +92 -0
  116. opik/evaluation/metrics/arguments_helpers.py +15 -21
  117. opik/evaluation/metrics/arguments_validator.py +38 -0
  118. opik/evaluation/metrics/base_metric.py +20 -10
  119. opik/evaluation/metrics/conversation/__init__.py +48 -0
  120. opik/evaluation/metrics/conversation/conversation_thread_metric.py +79 -0
  121. opik/evaluation/metrics/conversation/conversation_turns_factory.py +39 -0
  122. opik/evaluation/metrics/conversation/g_eval_wrappers.py +19 -0
  123. opik/evaluation/metrics/conversation/helpers.py +84 -0
  124. opik/evaluation/metrics/conversation/heuristics/__init__.py +14 -0
  125. opik/evaluation/metrics/conversation/heuristics/degeneration/__init__.py +3 -0
  126. opik/evaluation/metrics/conversation/heuristics/degeneration/metric.py +189 -0
  127. opik/evaluation/metrics/conversation/heuristics/degeneration/phrases.py +12 -0
  128. opik/evaluation/metrics/conversation/heuristics/knowledge_retention/__init__.py +3 -0
  129. opik/evaluation/metrics/conversation/heuristics/knowledge_retention/metric.py +172 -0
  130. opik/evaluation/metrics/conversation/llm_judges/__init__.py +32 -0
  131. opik/evaluation/metrics/conversation/llm_judges/conversational_coherence/__init__.py +0 -0
  132. opik/evaluation/metrics/conversation/llm_judges/conversational_coherence/metric.py +274 -0
  133. opik/evaluation/metrics/conversation/llm_judges/conversational_coherence/schema.py +16 -0
  134. opik/evaluation/metrics/conversation/llm_judges/conversational_coherence/templates.py +95 -0
  135. opik/evaluation/metrics/conversation/llm_judges/g_eval_wrappers.py +442 -0
  136. opik/evaluation/metrics/conversation/llm_judges/session_completeness/__init__.py +0 -0
  137. opik/evaluation/metrics/conversation/llm_judges/session_completeness/metric.py +295 -0
  138. opik/evaluation/metrics/conversation/llm_judges/session_completeness/schema.py +22 -0
  139. opik/evaluation/metrics/conversation/llm_judges/session_completeness/templates.py +139 -0
  140. opik/evaluation/metrics/conversation/llm_judges/user_frustration/__init__.py +0 -0
  141. opik/evaluation/metrics/conversation/llm_judges/user_frustration/metric.py +277 -0
  142. opik/evaluation/metrics/conversation/llm_judges/user_frustration/schema.py +16 -0
  143. opik/evaluation/metrics/conversation/llm_judges/user_frustration/templates.py +135 -0
  144. opik/evaluation/metrics/conversation/types.py +34 -0
  145. opik/evaluation/metrics/conversation_types.py +9 -0
  146. opik/evaluation/metrics/heuristics/bertscore.py +107 -0
  147. opik/evaluation/metrics/heuristics/bleu.py +43 -16
  148. opik/evaluation/metrics/heuristics/chrf.py +127 -0
  149. opik/evaluation/metrics/heuristics/contains.py +50 -11
  150. opik/evaluation/metrics/heuristics/distribution_metrics.py +331 -0
  151. opik/evaluation/metrics/heuristics/equals.py +4 -1
  152. opik/evaluation/metrics/heuristics/gleu.py +113 -0
  153. opik/evaluation/metrics/heuristics/is_json.py +9 -3
  154. opik/evaluation/metrics/heuristics/language_adherence.py +123 -0
  155. opik/evaluation/metrics/heuristics/levenshtein_ratio.py +6 -5
  156. opik/evaluation/metrics/heuristics/meteor.py +119 -0
  157. opik/evaluation/metrics/heuristics/prompt_injection.py +150 -0
  158. opik/evaluation/metrics/heuristics/readability.py +129 -0
  159. opik/evaluation/metrics/heuristics/regex_match.py +4 -1
  160. opik/evaluation/metrics/heuristics/rouge.py +148 -0
  161. opik/evaluation/metrics/heuristics/sentiment.py +98 -0
  162. opik/evaluation/metrics/heuristics/spearman.py +88 -0
  163. opik/evaluation/metrics/heuristics/tone.py +155 -0
  164. opik/evaluation/metrics/heuristics/vader_sentiment.py +77 -0
  165. opik/evaluation/metrics/llm_judges/answer_relevance/metric.py +27 -30
  166. opik/evaluation/metrics/llm_judges/answer_relevance/parser.py +27 -0
  167. opik/evaluation/metrics/llm_judges/answer_relevance/templates.py +10 -10
  168. opik/evaluation/metrics/llm_judges/context_precision/metric.py +28 -31
  169. opik/evaluation/metrics/llm_judges/context_precision/parser.py +27 -0
  170. opik/evaluation/metrics/llm_judges/context_precision/template.py +7 -7
  171. opik/evaluation/metrics/llm_judges/context_recall/metric.py +27 -31
  172. opik/evaluation/metrics/llm_judges/context_recall/parser.py +27 -0
  173. opik/evaluation/metrics/llm_judges/context_recall/template.py +7 -7
  174. opik/evaluation/metrics/llm_judges/factuality/metric.py +7 -26
  175. opik/evaluation/metrics/llm_judges/factuality/parser.py +35 -0
  176. opik/evaluation/metrics/llm_judges/factuality/template.py +1 -1
  177. opik/evaluation/metrics/llm_judges/g_eval/__init__.py +5 -0
  178. opik/evaluation/metrics/llm_judges/g_eval/metric.py +244 -113
  179. opik/evaluation/metrics/llm_judges/g_eval/parser.py +161 -0
  180. opik/evaluation/metrics/llm_judges/g_eval/presets.py +209 -0
  181. opik/evaluation/metrics/llm_judges/g_eval_presets/__init__.py +36 -0
  182. opik/evaluation/metrics/llm_judges/g_eval_presets/agent_assessment.py +77 -0
  183. opik/evaluation/metrics/llm_judges/g_eval_presets/bias_classifier.py +181 -0
  184. opik/evaluation/metrics/llm_judges/g_eval_presets/compliance_risk.py +41 -0
  185. opik/evaluation/metrics/llm_judges/g_eval_presets/prompt_uncertainty.py +41 -0
  186. opik/evaluation/metrics/llm_judges/g_eval_presets/qa_suite.py +146 -0
  187. opik/evaluation/metrics/llm_judges/hallucination/metric.py +23 -27
  188. opik/evaluation/metrics/llm_judges/hallucination/parser.py +29 -0
  189. opik/evaluation/metrics/llm_judges/hallucination/template.py +2 -4
  190. opik/evaluation/metrics/llm_judges/llm_juries/__init__.py +3 -0
  191. opik/evaluation/metrics/llm_judges/llm_juries/metric.py +76 -0
  192. opik/evaluation/metrics/llm_judges/moderation/metric.py +23 -28
  193. opik/evaluation/metrics/llm_judges/moderation/parser.py +27 -0
  194. opik/evaluation/metrics/llm_judges/moderation/template.py +2 -2
  195. opik/evaluation/metrics/llm_judges/parsing_helpers.py +26 -0
  196. opik/evaluation/metrics/llm_judges/structure_output_compliance/__init__.py +0 -0
  197. opik/evaluation/metrics/llm_judges/structure_output_compliance/metric.py +144 -0
  198. opik/evaluation/metrics/llm_judges/structure_output_compliance/parser.py +79 -0
  199. opik/evaluation/metrics/llm_judges/structure_output_compliance/schema.py +15 -0
  200. opik/evaluation/metrics/llm_judges/structure_output_compliance/template.py +50 -0
  201. opik/evaluation/metrics/llm_judges/syc_eval/__init__.py +0 -0
  202. opik/evaluation/metrics/llm_judges/syc_eval/metric.py +252 -0
  203. opik/evaluation/metrics/llm_judges/syc_eval/parser.py +82 -0
  204. opik/evaluation/metrics/llm_judges/syc_eval/template.py +155 -0
  205. opik/evaluation/metrics/llm_judges/trajectory_accuracy/__init__.py +3 -0
  206. opik/evaluation/metrics/llm_judges/trajectory_accuracy/metric.py +171 -0
  207. opik/evaluation/metrics/llm_judges/trajectory_accuracy/parser.py +38 -0
  208. opik/evaluation/metrics/llm_judges/trajectory_accuracy/templates.py +65 -0
  209. opik/evaluation/metrics/llm_judges/usefulness/metric.py +23 -32
  210. opik/evaluation/metrics/llm_judges/usefulness/parser.py +28 -0
  211. opik/evaluation/metrics/ragas_metric.py +112 -0
  212. opik/evaluation/models/__init__.py +10 -0
  213. opik/evaluation/models/base_model.py +140 -18
  214. opik/evaluation/models/langchain/__init__.py +3 -0
  215. opik/evaluation/models/langchain/langchain_chat_model.py +166 -0
  216. opik/evaluation/models/langchain/message_converters.py +106 -0
  217. opik/evaluation/models/langchain/opik_monitoring.py +23 -0
  218. opik/evaluation/models/litellm/litellm_chat_model.py +186 -40
  219. opik/evaluation/models/litellm/opik_monitor.py +24 -21
  220. opik/evaluation/models/litellm/util.py +125 -0
  221. opik/evaluation/models/litellm/warning_filters.py +16 -4
  222. opik/evaluation/models/model_capabilities.py +187 -0
  223. opik/evaluation/models/models_factory.py +25 -3
  224. opik/evaluation/preprocessing.py +92 -0
  225. opik/evaluation/report.py +70 -12
  226. opik/evaluation/rest_operations.py +49 -45
  227. opik/evaluation/samplers/__init__.py +4 -0
  228. opik/evaluation/samplers/base_dataset_sampler.py +40 -0
  229. opik/evaluation/samplers/random_dataset_sampler.py +48 -0
  230. opik/evaluation/score_statistics.py +66 -0
  231. opik/evaluation/scorers/__init__.py +4 -0
  232. opik/evaluation/scorers/scorer_function.py +55 -0
  233. opik/evaluation/scorers/scorer_wrapper_metric.py +130 -0
  234. opik/evaluation/test_case.py +3 -2
  235. opik/evaluation/test_result.py +1 -0
  236. opik/evaluation/threads/__init__.py +0 -0
  237. opik/evaluation/threads/context_helper.py +32 -0
  238. opik/evaluation/threads/evaluation_engine.py +181 -0
  239. opik/evaluation/threads/evaluation_result.py +18 -0
  240. opik/evaluation/threads/evaluator.py +120 -0
  241. opik/evaluation/threads/helpers.py +51 -0
  242. opik/evaluation/types.py +9 -1
  243. opik/exceptions.py +116 -3
  244. opik/file_upload/__init__.py +0 -0
  245. opik/file_upload/base_upload_manager.py +39 -0
  246. opik/file_upload/file_upload_monitor.py +14 -0
  247. opik/file_upload/file_uploader.py +141 -0
  248. opik/file_upload/mime_type.py +9 -0
  249. opik/file_upload/s3_multipart_upload/__init__.py +0 -0
  250. opik/file_upload/s3_multipart_upload/file_parts_strategy.py +89 -0
  251. opik/file_upload/s3_multipart_upload/s3_file_uploader.py +86 -0
  252. opik/file_upload/s3_multipart_upload/s3_upload_error.py +29 -0
  253. opik/file_upload/thread_pool.py +17 -0
  254. opik/file_upload/upload_client.py +114 -0
  255. opik/file_upload/upload_manager.py +255 -0
  256. opik/file_upload/upload_options.py +37 -0
  257. opik/format_helpers.py +17 -0
  258. opik/guardrails/__init__.py +4 -0
  259. opik/guardrails/guardrail.py +157 -0
  260. opik/guardrails/guards/__init__.py +5 -0
  261. opik/guardrails/guards/guard.py +17 -0
  262. opik/guardrails/guards/pii.py +47 -0
  263. opik/guardrails/guards/topic.py +76 -0
  264. opik/guardrails/rest_api_client.py +34 -0
  265. opik/guardrails/schemas.py +24 -0
  266. opik/guardrails/tracing.py +61 -0
  267. opik/healthcheck/__init__.py +2 -1
  268. opik/healthcheck/checks.py +2 -2
  269. opik/healthcheck/rich_representation.py +1 -1
  270. opik/hooks/__init__.py +23 -0
  271. opik/hooks/anonymizer_hook.py +36 -0
  272. opik/hooks/httpx_client_hook.py +112 -0
  273. opik/httpx_client.py +75 -4
  274. opik/id_helpers.py +18 -0
  275. opik/integrations/adk/__init__.py +14 -0
  276. opik/integrations/adk/callback_context_info_extractors.py +32 -0
  277. opik/integrations/adk/graph/__init__.py +0 -0
  278. opik/integrations/adk/graph/mermaid_graph_builder.py +128 -0
  279. opik/integrations/adk/graph/nodes.py +101 -0
  280. opik/integrations/adk/graph/subgraph_edges_builders.py +41 -0
  281. opik/integrations/adk/helpers.py +48 -0
  282. opik/integrations/adk/legacy_opik_tracer.py +381 -0
  283. opik/integrations/adk/opik_tracer.py +370 -0
  284. opik/integrations/adk/patchers/__init__.py +4 -0
  285. opik/integrations/adk/patchers/adk_otel_tracer/__init__.py +0 -0
  286. opik/integrations/adk/patchers/adk_otel_tracer/llm_span_helpers.py +30 -0
  287. opik/integrations/adk/patchers/adk_otel_tracer/opik_adk_otel_tracer.py +201 -0
  288. opik/integrations/adk/patchers/litellm_wrappers.py +91 -0
  289. opik/integrations/adk/patchers/llm_response_wrapper.py +105 -0
  290. opik/integrations/adk/patchers/patchers.py +64 -0
  291. opik/integrations/adk/recursive_callback_injector.py +126 -0
  292. opik/integrations/aisuite/aisuite_decorator.py +8 -3
  293. opik/integrations/aisuite/opik_tracker.py +1 -0
  294. opik/integrations/anthropic/messages_create_decorator.py +8 -3
  295. opik/integrations/anthropic/opik_tracker.py +0 -1
  296. opik/integrations/bedrock/converse/__init__.py +0 -0
  297. opik/integrations/bedrock/converse/chunks_aggregator.py +188 -0
  298. opik/integrations/bedrock/{converse_decorator.py → converse/converse_decorator.py} +18 -8
  299. opik/integrations/bedrock/invoke_agent_decorator.py +12 -7
  300. opik/integrations/bedrock/invoke_model/__init__.py +0 -0
  301. opik/integrations/bedrock/invoke_model/chunks_aggregator/__init__.py +78 -0
  302. opik/integrations/bedrock/invoke_model/chunks_aggregator/api.py +45 -0
  303. opik/integrations/bedrock/invoke_model/chunks_aggregator/base.py +23 -0
  304. opik/integrations/bedrock/invoke_model/chunks_aggregator/claude.py +121 -0
  305. opik/integrations/bedrock/invoke_model/chunks_aggregator/format_detector.py +107 -0
  306. opik/integrations/bedrock/invoke_model/chunks_aggregator/llama.py +108 -0
  307. opik/integrations/bedrock/invoke_model/chunks_aggregator/mistral.py +118 -0
  308. opik/integrations/bedrock/invoke_model/chunks_aggregator/nova.py +99 -0
  309. opik/integrations/bedrock/invoke_model/invoke_model_decorator.py +178 -0
  310. opik/integrations/bedrock/invoke_model/response_types.py +34 -0
  311. opik/integrations/bedrock/invoke_model/stream_wrappers.py +122 -0
  312. opik/integrations/bedrock/invoke_model/usage_converters.py +87 -0
  313. opik/integrations/bedrock/invoke_model/usage_extraction.py +108 -0
  314. opik/integrations/bedrock/opik_tracker.py +43 -4
  315. opik/integrations/bedrock/types.py +19 -0
  316. opik/integrations/crewai/crewai_decorator.py +34 -56
  317. opik/integrations/crewai/opik_tracker.py +31 -10
  318. opik/integrations/crewai/patchers/__init__.py +5 -0
  319. opik/integrations/crewai/patchers/flow.py +118 -0
  320. opik/integrations/crewai/patchers/litellm_completion.py +30 -0
  321. opik/integrations/crewai/patchers/llm_client.py +207 -0
  322. opik/integrations/dspy/callback.py +246 -84
  323. opik/integrations/dspy/graph.py +88 -0
  324. opik/integrations/dspy/parsers.py +168 -0
  325. opik/integrations/genai/encoder_extension.py +2 -6
  326. opik/integrations/genai/generate_content_decorator.py +20 -13
  327. opik/integrations/guardrails/guardrails_decorator.py +4 -0
  328. opik/integrations/harbor/__init__.py +17 -0
  329. opik/integrations/harbor/experiment_service.py +269 -0
  330. opik/integrations/harbor/opik_tracker.py +528 -0
  331. opik/integrations/haystack/constants.py +35 -0
  332. opik/integrations/haystack/converters.py +1 -2
  333. opik/integrations/haystack/opik_connector.py +28 -6
  334. opik/integrations/haystack/opik_span_bridge.py +284 -0
  335. opik/integrations/haystack/opik_tracer.py +124 -222
  336. opik/integrations/langchain/__init__.py +3 -1
  337. opik/integrations/langchain/helpers.py +96 -0
  338. opik/integrations/langchain/langgraph_async_context_bridge.py +131 -0
  339. opik/integrations/langchain/langgraph_tracer_injector.py +88 -0
  340. opik/integrations/langchain/opik_encoder_extension.py +2 -2
  341. opik/integrations/langchain/opik_tracer.py +641 -206
  342. opik/integrations/langchain/provider_usage_extractors/__init__.py +5 -0
  343. opik/integrations/langchain/provider_usage_extractors/anthropic_usage_extractor.py +101 -0
  344. opik/integrations/langchain/provider_usage_extractors/anthropic_vertexai_usage_extractor.py +67 -0
  345. opik/integrations/langchain/provider_usage_extractors/bedrock_usage_extractor.py +94 -0
  346. opik/integrations/langchain/provider_usage_extractors/google_generative_ai_usage_extractor.py +109 -0
  347. opik/integrations/langchain/provider_usage_extractors/groq_usage_extractor.py +92 -0
  348. opik/integrations/langchain/provider_usage_extractors/langchain_run_helpers/__init__.py +15 -0
  349. opik/integrations/langchain/provider_usage_extractors/langchain_run_helpers/helpers.py +134 -0
  350. opik/integrations/langchain/provider_usage_extractors/langchain_run_helpers/langchain_usage.py +163 -0
  351. opik/integrations/langchain/provider_usage_extractors/openai_usage_extractor.py +124 -0
  352. opik/integrations/langchain/provider_usage_extractors/provider_usage_extractor_protocol.py +29 -0
  353. opik/integrations/langchain/provider_usage_extractors/usage_extractor.py +48 -0
  354. opik/integrations/langchain/provider_usage_extractors/vertexai_usage_extractor.py +109 -0
  355. opik/integrations/litellm/__init__.py +5 -0
  356. opik/integrations/litellm/completion_chunks_aggregator.py +115 -0
  357. opik/integrations/litellm/litellm_completion_decorator.py +242 -0
  358. opik/integrations/litellm/opik_tracker.py +43 -0
  359. opik/integrations/litellm/stream_patchers.py +151 -0
  360. opik/integrations/llama_index/callback.py +179 -78
  361. opik/integrations/llama_index/event_parsing_utils.py +29 -9
  362. opik/integrations/openai/agents/opik_tracing_processor.py +204 -32
  363. opik/integrations/openai/agents/span_data_parsers.py +15 -6
  364. opik/integrations/openai/chat_completion_chunks_aggregator.py +1 -1
  365. opik/integrations/openai/{openai_decorator.py → openai_chat_completions_decorator.py} +45 -35
  366. opik/integrations/openai/openai_responses_decorator.py +158 -0
  367. opik/integrations/openai/opik_tracker.py +94 -13
  368. opik/integrations/openai/response_events_aggregator.py +36 -0
  369. opik/integrations/openai/stream_patchers.py +125 -15
  370. opik/integrations/sagemaker/auth.py +5 -1
  371. opik/jsonable_encoder.py +29 -1
  372. opik/llm_usage/base_original_provider_usage.py +15 -8
  373. opik/llm_usage/bedrock_usage.py +8 -2
  374. opik/llm_usage/google_usage.py +6 -1
  375. opik/llm_usage/llm_usage_info.py +6 -0
  376. opik/llm_usage/{openai_usage.py → openai_chat_completions_usage.py} +2 -12
  377. opik/llm_usage/{openai_agent_usage.py → openai_responses_usage.py} +7 -15
  378. opik/llm_usage/opik_usage.py +36 -10
  379. opik/llm_usage/opik_usage_factory.py +35 -19
  380. opik/logging_messages.py +19 -7
  381. opik/message_processing/arguments_utils.py +22 -0
  382. opik/message_processing/batching/base_batcher.py +45 -17
  383. opik/message_processing/batching/batch_manager.py +22 -10
  384. opik/message_processing/batching/batch_manager_constuctors.py +36 -11
  385. opik/message_processing/batching/batchers.py +167 -44
  386. opik/message_processing/batching/flushing_thread.py +0 -3
  387. opik/message_processing/batching/sequence_splitter.py +50 -5
  388. opik/message_processing/emulation/__init__.py +0 -0
  389. opik/message_processing/emulation/emulator_message_processor.py +578 -0
  390. opik/message_processing/emulation/local_emulator_message_processor.py +140 -0
  391. opik/message_processing/emulation/models.py +162 -0
  392. opik/message_processing/encoder_helpers.py +79 -0
  393. opik/message_processing/message_queue.py +79 -0
  394. opik/message_processing/messages.py +154 -12
  395. opik/message_processing/preprocessing/__init__.py +0 -0
  396. opik/message_processing/preprocessing/attachments_preprocessor.py +70 -0
  397. opik/message_processing/preprocessing/batching_preprocessor.py +53 -0
  398. opik/message_processing/preprocessing/constants.py +1 -0
  399. opik/message_processing/preprocessing/file_upload_preprocessor.py +38 -0
  400. opik/message_processing/preprocessing/preprocessor.py +36 -0
  401. opik/message_processing/processors/__init__.py +0 -0
  402. opik/message_processing/processors/attachments_extraction_processor.py +146 -0
  403. opik/message_processing/processors/message_processors.py +92 -0
  404. opik/message_processing/processors/message_processors_chain.py +96 -0
  405. opik/message_processing/processors/online_message_processor.py +324 -0
  406. opik/message_processing/queue_consumer.py +61 -13
  407. opik/message_processing/streamer.py +102 -31
  408. opik/message_processing/streamer_constructors.py +67 -12
  409. opik/opik_context.py +103 -11
  410. opik/plugins/pytest/decorator.py +2 -2
  411. opik/plugins/pytest/experiment_runner.py +3 -2
  412. opik/plugins/pytest/hooks.py +6 -4
  413. opik/rate_limit/__init__.py +0 -0
  414. opik/rate_limit/rate_limit.py +25 -0
  415. opik/rest_api/__init__.py +643 -11
  416. opik/rest_api/alerts/__init__.py +7 -0
  417. opik/rest_api/alerts/client.py +667 -0
  418. opik/rest_api/alerts/raw_client.py +1015 -0
  419. opik/rest_api/alerts/types/__init__.py +7 -0
  420. opik/rest_api/alerts/types/get_webhook_examples_request_alert_type.py +5 -0
  421. opik/rest_api/annotation_queues/__init__.py +4 -0
  422. opik/rest_api/annotation_queues/client.py +668 -0
  423. opik/rest_api/annotation_queues/raw_client.py +1019 -0
  424. opik/rest_api/attachments/__init__.py +17 -0
  425. opik/rest_api/attachments/client.py +752 -0
  426. opik/rest_api/attachments/raw_client.py +1125 -0
  427. opik/rest_api/attachments/types/__init__.py +15 -0
  428. opik/rest_api/attachments/types/attachment_list_request_entity_type.py +5 -0
  429. opik/rest_api/attachments/types/download_attachment_request_entity_type.py +5 -0
  430. opik/rest_api/attachments/types/start_multipart_upload_request_entity_type.py +5 -0
  431. opik/rest_api/attachments/types/upload_attachment_request_entity_type.py +5 -0
  432. opik/rest_api/automation_rule_evaluators/__init__.py +2 -0
  433. opik/rest_api/automation_rule_evaluators/client.py +182 -1162
  434. opik/rest_api/automation_rule_evaluators/raw_client.py +598 -0
  435. opik/rest_api/chat_completions/__init__.py +2 -0
  436. opik/rest_api/chat_completions/client.py +115 -149
  437. opik/rest_api/chat_completions/raw_client.py +339 -0
  438. opik/rest_api/check/__init__.py +2 -0
  439. opik/rest_api/check/client.py +88 -106
  440. opik/rest_api/check/raw_client.py +258 -0
  441. opik/rest_api/client.py +112 -212
  442. opik/rest_api/core/__init__.py +5 -0
  443. opik/rest_api/core/api_error.py +12 -6
  444. opik/rest_api/core/client_wrapper.py +4 -14
  445. opik/rest_api/core/datetime_utils.py +1 -3
  446. opik/rest_api/core/file.py +2 -5
  447. opik/rest_api/core/http_client.py +42 -120
  448. opik/rest_api/core/http_response.py +55 -0
  449. opik/rest_api/core/jsonable_encoder.py +1 -4
  450. opik/rest_api/core/pydantic_utilities.py +79 -147
  451. opik/rest_api/core/query_encoder.py +1 -3
  452. opik/rest_api/core/serialization.py +10 -10
  453. opik/rest_api/dashboards/__init__.py +4 -0
  454. opik/rest_api/dashboards/client.py +462 -0
  455. opik/rest_api/dashboards/raw_client.py +648 -0
  456. opik/rest_api/datasets/__init__.py +5 -0
  457. opik/rest_api/datasets/client.py +1638 -1091
  458. opik/rest_api/datasets/raw_client.py +3389 -0
  459. opik/rest_api/datasets/types/__init__.py +8 -0
  460. opik/rest_api/datasets/types/dataset_update_visibility.py +5 -0
  461. opik/rest_api/datasets/types/dataset_write_visibility.py +5 -0
  462. opik/rest_api/errors/__init__.py +2 -0
  463. opik/rest_api/errors/bad_request_error.py +4 -3
  464. opik/rest_api/errors/conflict_error.py +4 -3
  465. opik/rest_api/errors/forbidden_error.py +4 -2
  466. opik/rest_api/errors/not_found_error.py +4 -3
  467. opik/rest_api/errors/not_implemented_error.py +4 -3
  468. opik/rest_api/errors/unauthorized_error.py +4 -3
  469. opik/rest_api/errors/unprocessable_entity_error.py +4 -3
  470. opik/rest_api/experiments/__init__.py +5 -0
  471. opik/rest_api/experiments/client.py +676 -752
  472. opik/rest_api/experiments/raw_client.py +1872 -0
  473. opik/rest_api/experiments/types/__init__.py +10 -0
  474. opik/rest_api/experiments/types/experiment_update_status.py +5 -0
  475. opik/rest_api/experiments/types/experiment_update_type.py +5 -0
  476. opik/rest_api/experiments/types/experiment_write_status.py +5 -0
  477. opik/rest_api/experiments/types/experiment_write_type.py +5 -0
  478. opik/rest_api/feedback_definitions/__init__.py +2 -0
  479. opik/rest_api/feedback_definitions/client.py +96 -370
  480. opik/rest_api/feedback_definitions/raw_client.py +541 -0
  481. opik/rest_api/feedback_definitions/types/__init__.py +2 -0
  482. opik/rest_api/feedback_definitions/types/find_feedback_definitions_request_type.py +1 -3
  483. opik/rest_api/guardrails/__init__.py +4 -0
  484. opik/rest_api/guardrails/client.py +104 -0
  485. opik/rest_api/guardrails/raw_client.py +102 -0
  486. opik/rest_api/llm_provider_key/__init__.py +2 -0
  487. opik/rest_api/llm_provider_key/client.py +166 -440
  488. opik/rest_api/llm_provider_key/raw_client.py +643 -0
  489. opik/rest_api/llm_provider_key/types/__init__.py +2 -0
  490. opik/rest_api/llm_provider_key/types/provider_api_key_write_provider.py +1 -1
  491. opik/rest_api/manual_evaluation/__init__.py +4 -0
  492. opik/rest_api/manual_evaluation/client.py +347 -0
  493. opik/rest_api/manual_evaluation/raw_client.py +543 -0
  494. opik/rest_api/open_telemetry_ingestion/__init__.py +2 -0
  495. opik/rest_api/open_telemetry_ingestion/client.py +38 -63
  496. opik/rest_api/open_telemetry_ingestion/raw_client.py +88 -0
  497. opik/rest_api/optimizations/__init__.py +7 -0
  498. opik/rest_api/optimizations/client.py +704 -0
  499. opik/rest_api/optimizations/raw_client.py +920 -0
  500. opik/rest_api/optimizations/types/__init__.py +7 -0
  501. opik/rest_api/optimizations/types/optimization_update_status.py +7 -0
  502. opik/rest_api/projects/__init__.py +10 -1
  503. opik/rest_api/projects/client.py +180 -855
  504. opik/rest_api/projects/raw_client.py +1216 -0
  505. opik/rest_api/projects/types/__init__.py +11 -4
  506. opik/rest_api/projects/types/project_metric_request_public_interval.py +1 -3
  507. opik/rest_api/projects/types/project_metric_request_public_metric_type.py +11 -1
  508. opik/rest_api/projects/types/project_update_visibility.py +5 -0
  509. opik/rest_api/projects/types/project_write_visibility.py +5 -0
  510. opik/rest_api/prompts/__init__.py +4 -2
  511. opik/rest_api/prompts/client.py +381 -970
  512. opik/rest_api/prompts/raw_client.py +1634 -0
  513. opik/rest_api/prompts/types/__init__.py +5 -1
  514. opik/rest_api/prompts/types/create_prompt_version_detail_template_structure.py +5 -0
  515. opik/rest_api/prompts/types/prompt_write_template_structure.py +5 -0
  516. opik/rest_api/raw_client.py +156 -0
  517. opik/rest_api/redirect/__init__.py +4 -0
  518. opik/rest_api/redirect/client.py +375 -0
  519. opik/rest_api/redirect/raw_client.py +566 -0
  520. opik/rest_api/service_toggles/__init__.py +4 -0
  521. opik/rest_api/service_toggles/client.py +91 -0
  522. opik/rest_api/service_toggles/raw_client.py +93 -0
  523. opik/rest_api/spans/__init__.py +2 -0
  524. opik/rest_api/spans/client.py +659 -1354
  525. opik/rest_api/spans/raw_client.py +2383 -0
  526. opik/rest_api/spans/types/__init__.py +2 -0
  527. opik/rest_api/spans/types/find_feedback_score_names_1_request_type.py +1 -3
  528. opik/rest_api/spans/types/get_span_stats_request_type.py +1 -3
  529. opik/rest_api/spans/types/get_spans_by_project_request_type.py +1 -3
  530. opik/rest_api/spans/types/span_search_stream_request_public_type.py +1 -3
  531. opik/rest_api/system_usage/__init__.py +2 -0
  532. opik/rest_api/system_usage/client.py +157 -216
  533. opik/rest_api/system_usage/raw_client.py +455 -0
  534. opik/rest_api/traces/__init__.py +2 -0
  535. opik/rest_api/traces/client.py +2102 -1625
  536. opik/rest_api/traces/raw_client.py +4144 -0
  537. opik/rest_api/types/__init__.py +629 -24
  538. opik/rest_api/types/aggregation_data.py +27 -0
  539. opik/rest_api/types/alert.py +33 -0
  540. opik/rest_api/types/alert_alert_type.py +5 -0
  541. opik/rest_api/types/alert_page_public.py +24 -0
  542. opik/rest_api/types/alert_public.py +33 -0
  543. opik/rest_api/types/alert_public_alert_type.py +5 -0
  544. opik/rest_api/types/alert_trigger.py +27 -0
  545. opik/rest_api/types/alert_trigger_config.py +28 -0
  546. opik/rest_api/types/alert_trigger_config_public.py +28 -0
  547. opik/rest_api/types/alert_trigger_config_public_type.py +10 -0
  548. opik/rest_api/types/alert_trigger_config_type.py +10 -0
  549. opik/rest_api/types/alert_trigger_config_write.py +22 -0
  550. opik/rest_api/types/alert_trigger_config_write_type.py +10 -0
  551. opik/rest_api/types/alert_trigger_event_type.py +19 -0
  552. opik/rest_api/types/alert_trigger_public.py +27 -0
  553. opik/rest_api/types/alert_trigger_public_event_type.py +19 -0
  554. opik/rest_api/types/alert_trigger_write.py +23 -0
  555. opik/rest_api/types/alert_trigger_write_event_type.py +19 -0
  556. opik/rest_api/types/alert_write.py +28 -0
  557. opik/rest_api/types/alert_write_alert_type.py +5 -0
  558. opik/rest_api/types/annotation_queue.py +42 -0
  559. opik/rest_api/types/annotation_queue_batch.py +27 -0
  560. opik/rest_api/types/{json_schema_element.py → annotation_queue_item_ids.py} +5 -7
  561. opik/rest_api/types/annotation_queue_page_public.py +28 -0
  562. opik/rest_api/types/annotation_queue_public.py +38 -0
  563. opik/rest_api/types/annotation_queue_public_scope.py +5 -0
  564. opik/rest_api/types/{workspace_metadata.py → annotation_queue_reviewer.py} +6 -7
  565. opik/rest_api/types/annotation_queue_reviewer_public.py +20 -0
  566. opik/rest_api/types/annotation_queue_scope.py +5 -0
  567. opik/rest_api/types/annotation_queue_write.py +31 -0
  568. opik/rest_api/types/annotation_queue_write_scope.py +5 -0
  569. opik/rest_api/types/assistant_message.py +7 -8
  570. opik/rest_api/types/assistant_message_role.py +1 -3
  571. opik/rest_api/types/attachment.py +22 -0
  572. opik/rest_api/types/attachment_page.py +28 -0
  573. opik/rest_api/types/audio_url.py +19 -0
  574. opik/rest_api/types/audio_url_public.py +19 -0
  575. opik/rest_api/types/audio_url_write.py +19 -0
  576. opik/rest_api/types/automation_rule_evaluator.py +160 -0
  577. opik/rest_api/types/automation_rule_evaluator_llm_as_judge.py +6 -6
  578. opik/rest_api/types/automation_rule_evaluator_llm_as_judge_public.py +6 -6
  579. opik/rest_api/types/automation_rule_evaluator_llm_as_judge_write.py +6 -6
  580. opik/rest_api/types/automation_rule_evaluator_object_object_public.py +155 -0
  581. opik/rest_api/types/automation_rule_evaluator_page_public.py +6 -6
  582. opik/rest_api/types/automation_rule_evaluator_public.py +155 -0
  583. opik/rest_api/types/automation_rule_evaluator_span_llm_as_judge.py +22 -0
  584. opik/rest_api/types/automation_rule_evaluator_span_llm_as_judge_public.py +22 -0
  585. opik/rest_api/types/automation_rule_evaluator_span_llm_as_judge_write.py +22 -0
  586. opik/rest_api/types/automation_rule_evaluator_span_user_defined_metric_python.py +22 -0
  587. opik/rest_api/types/automation_rule_evaluator_span_user_defined_metric_python_public.py +22 -0
  588. opik/rest_api/types/automation_rule_evaluator_span_user_defined_metric_python_write.py +22 -0
  589. opik/rest_api/types/automation_rule_evaluator_trace_thread_llm_as_judge.py +22 -0
  590. opik/rest_api/types/automation_rule_evaluator_trace_thread_llm_as_judge_public.py +22 -0
  591. opik/rest_api/types/automation_rule_evaluator_trace_thread_llm_as_judge_write.py +22 -0
  592. opik/rest_api/types/automation_rule_evaluator_trace_thread_user_defined_metric_python.py +22 -0
  593. opik/rest_api/types/automation_rule_evaluator_trace_thread_user_defined_metric_python_public.py +22 -0
  594. opik/rest_api/types/automation_rule_evaluator_trace_thread_user_defined_metric_python_write.py +22 -0
  595. opik/rest_api/types/automation_rule_evaluator_update.py +143 -0
  596. opik/rest_api/types/automation_rule_evaluator_update_llm_as_judge.py +6 -6
  597. opik/rest_api/types/automation_rule_evaluator_update_span_llm_as_judge.py +22 -0
  598. opik/rest_api/types/automation_rule_evaluator_update_span_user_defined_metric_python.py +22 -0
  599. opik/rest_api/types/automation_rule_evaluator_update_trace_thread_llm_as_judge.py +22 -0
  600. opik/rest_api/types/automation_rule_evaluator_update_trace_thread_user_defined_metric_python.py +22 -0
  601. opik/rest_api/types/automation_rule_evaluator_update_user_defined_metric_python.py +6 -6
  602. opik/rest_api/types/automation_rule_evaluator_user_defined_metric_python.py +6 -6
  603. opik/rest_api/types/automation_rule_evaluator_user_defined_metric_python_public.py +6 -6
  604. opik/rest_api/types/automation_rule_evaluator_user_defined_metric_python_write.py +6 -6
  605. opik/rest_api/types/automation_rule_evaluator_write.py +143 -0
  606. opik/rest_api/types/avg_value_stat_public.py +3 -5
  607. opik/rest_api/types/batch_delete.py +3 -5
  608. opik/rest_api/types/batch_delete_by_project.py +20 -0
  609. opik/rest_api/types/bi_information.py +3 -5
  610. opik/rest_api/types/bi_information_response.py +4 -6
  611. opik/rest_api/types/boolean_feedback_definition.py +25 -0
  612. opik/rest_api/types/boolean_feedback_definition_create.py +20 -0
  613. opik/rest_api/types/boolean_feedback_definition_public.py +25 -0
  614. opik/rest_api/types/boolean_feedback_definition_update.py +20 -0
  615. opik/rest_api/types/boolean_feedback_detail.py +29 -0
  616. opik/rest_api/types/boolean_feedback_detail_create.py +29 -0
  617. opik/rest_api/types/boolean_feedback_detail_public.py +29 -0
  618. opik/rest_api/types/boolean_feedback_detail_update.py +29 -0
  619. opik/rest_api/types/categorical_feedback_definition.py +5 -7
  620. opik/rest_api/types/categorical_feedback_definition_create.py +4 -6
  621. opik/rest_api/types/categorical_feedback_definition_public.py +5 -7
  622. opik/rest_api/types/categorical_feedback_definition_update.py +4 -6
  623. opik/rest_api/types/categorical_feedback_detail.py +3 -5
  624. opik/rest_api/types/categorical_feedback_detail_create.py +3 -5
  625. opik/rest_api/types/categorical_feedback_detail_public.py +3 -5
  626. opik/rest_api/types/categorical_feedback_detail_update.py +3 -5
  627. opik/rest_api/types/chat_completion_choice.py +4 -6
  628. opik/rest_api/types/chat_completion_response.py +5 -6
  629. opik/rest_api/types/check.py +22 -0
  630. opik/rest_api/types/{json_node_compare.py → check_name.py} +1 -1
  631. opik/rest_api/types/check_public.py +22 -0
  632. opik/rest_api/types/check_public_name.py +5 -0
  633. opik/rest_api/types/check_public_result.py +5 -0
  634. opik/rest_api/types/check_result.py +5 -0
  635. opik/rest_api/types/chunked_output_json_node.py +4 -6
  636. opik/rest_api/types/chunked_output_json_node_public.py +4 -6
  637. opik/rest_api/types/chunked_output_json_node_public_type.py +6 -10
  638. opik/rest_api/types/chunked_output_json_node_type.py +6 -10
  639. opik/rest_api/types/column.py +8 -10
  640. opik/rest_api/types/column_compare.py +8 -10
  641. opik/rest_api/types/column_public.py +8 -10
  642. opik/rest_api/types/column_types_item.py +1 -3
  643. opik/rest_api/types/comment.py +4 -6
  644. opik/rest_api/types/comment_compare.py +4 -6
  645. opik/rest_api/types/comment_public.py +4 -6
  646. opik/rest_api/types/complete_multipart_upload_request.py +33 -0
  647. opik/rest_api/types/complete_multipart_upload_request_entity_type.py +5 -0
  648. opik/rest_api/types/completion_tokens_details.py +3 -5
  649. opik/rest_api/types/count_value_stat_public.py +3 -5
  650. opik/rest_api/types/dashboard_page_public.py +24 -0
  651. opik/rest_api/types/dashboard_public.py +30 -0
  652. opik/rest_api/types/data_point_double.py +21 -0
  653. opik/rest_api/types/data_point_number_public.py +3 -5
  654. opik/rest_api/types/dataset.py +14 -6
  655. opik/rest_api/types/dataset_expansion.py +42 -0
  656. opik/rest_api/types/dataset_expansion_response.py +39 -0
  657. opik/rest_api/types/dataset_item.py +9 -8
  658. opik/rest_api/types/dataset_item_batch.py +3 -5
  659. opik/rest_api/types/dataset_item_changes_public.py +5 -0
  660. opik/rest_api/types/dataset_item_compare.py +9 -8
  661. opik/rest_api/types/dataset_item_compare_source.py +1 -3
  662. opik/rest_api/types/dataset_item_filter.py +27 -0
  663. opik/rest_api/types/dataset_item_filter_operator.py +21 -0
  664. opik/rest_api/types/dataset_item_page_compare.py +10 -7
  665. opik/rest_api/types/dataset_item_page_public.py +10 -7
  666. opik/rest_api/types/dataset_item_public.py +9 -8
  667. opik/rest_api/types/dataset_item_public_source.py +1 -3
  668. opik/rest_api/types/dataset_item_source.py +1 -3
  669. opik/rest_api/types/dataset_item_update.py +39 -0
  670. opik/rest_api/types/dataset_item_write.py +5 -6
  671. opik/rest_api/types/dataset_item_write_source.py +1 -3
  672. opik/rest_api/types/dataset_page_public.py +9 -6
  673. opik/rest_api/types/dataset_public.py +14 -6
  674. opik/rest_api/types/dataset_public_status.py +5 -0
  675. opik/rest_api/types/dataset_public_visibility.py +5 -0
  676. opik/rest_api/types/dataset_status.py +5 -0
  677. opik/rest_api/types/dataset_version_diff.py +22 -0
  678. opik/rest_api/types/dataset_version_diff_stats.py +24 -0
  679. opik/rest_api/types/dataset_version_page_public.py +23 -0
  680. opik/rest_api/types/dataset_version_public.py +59 -0
  681. opik/rest_api/types/dataset_version_summary.py +46 -0
  682. opik/rest_api/types/dataset_version_summary_public.py +46 -0
  683. opik/rest_api/types/dataset_visibility.py +5 -0
  684. opik/rest_api/types/delete_attachments_request.py +23 -0
  685. opik/rest_api/types/delete_attachments_request_entity_type.py +5 -0
  686. opik/rest_api/types/delete_feedback_score.py +4 -5
  687. opik/rest_api/types/delete_ids_holder.py +19 -0
  688. opik/rest_api/types/delta.py +7 -9
  689. opik/rest_api/types/error_count_with_deviation.py +21 -0
  690. opik/rest_api/types/error_count_with_deviation_detailed.py +21 -0
  691. opik/rest_api/types/error_info.py +3 -5
  692. opik/rest_api/types/error_info_experiment_item_bulk_write_view.py +21 -0
  693. opik/rest_api/types/error_info_public.py +3 -5
  694. opik/rest_api/types/error_info_write.py +3 -5
  695. opik/rest_api/types/error_message.py +3 -5
  696. opik/rest_api/types/error_message_detail.py +3 -5
  697. opik/rest_api/types/error_message_detailed.py +3 -5
  698. opik/rest_api/types/error_message_public.py +3 -5
  699. opik/rest_api/types/experiment.py +21 -10
  700. opik/rest_api/types/experiment_group_aggregations_response.py +20 -0
  701. opik/rest_api/types/experiment_group_response.py +22 -0
  702. opik/rest_api/types/experiment_item.py +14 -11
  703. opik/rest_api/types/experiment_item_bulk_record.py +27 -0
  704. opik/rest_api/types/experiment_item_bulk_record_experiment_item_bulk_write_view.py +27 -0
  705. opik/rest_api/types/experiment_item_bulk_upload.py +27 -0
  706. opik/rest_api/types/experiment_item_compare.py +14 -11
  707. opik/rest_api/types/experiment_item_compare_trace_visibility_mode.py +5 -0
  708. opik/rest_api/types/experiment_item_public.py +6 -6
  709. opik/rest_api/types/experiment_item_public_trace_visibility_mode.py +5 -0
  710. opik/rest_api/types/experiment_item_trace_visibility_mode.py +5 -0
  711. opik/rest_api/types/experiment_page_public.py +9 -6
  712. opik/rest_api/types/experiment_public.py +21 -10
  713. opik/rest_api/types/experiment_public_status.py +5 -0
  714. opik/rest_api/types/experiment_public_type.py +5 -0
  715. opik/rest_api/types/experiment_score.py +20 -0
  716. opik/rest_api/types/experiment_score_public.py +20 -0
  717. opik/rest_api/types/experiment_score_write.py +20 -0
  718. opik/rest_api/types/experiment_status.py +5 -0
  719. opik/rest_api/types/experiment_type.py +5 -0
  720. opik/rest_api/types/export_trace_service_request.py +5 -0
  721. opik/rest_api/types/feedback.py +40 -27
  722. opik/rest_api/types/feedback_create.py +27 -13
  723. opik/rest_api/types/feedback_definition_page_public.py +4 -6
  724. opik/rest_api/types/feedback_object_public.py +40 -27
  725. opik/rest_api/types/feedback_public.py +40 -27
  726. opik/rest_api/types/feedback_score.py +7 -7
  727. opik/rest_api/types/feedback_score_average.py +3 -5
  728. opik/rest_api/types/feedback_score_average_detailed.py +3 -5
  729. opik/rest_api/types/feedback_score_average_public.py +3 -5
  730. opik/rest_api/types/feedback_score_batch.py +4 -6
  731. opik/rest_api/types/feedback_score_batch_item.py +6 -6
  732. opik/rest_api/types/feedback_score_batch_item_source.py +1 -3
  733. opik/rest_api/types/feedback_score_batch_item_thread.py +32 -0
  734. opik/rest_api/types/feedback_score_batch_item_thread_source.py +5 -0
  735. opik/rest_api/types/feedback_score_compare.py +7 -7
  736. opik/rest_api/types/feedback_score_compare_source.py +1 -3
  737. opik/rest_api/types/feedback_score_experiment_item_bulk_write_view.py +31 -0
  738. opik/rest_api/types/feedback_score_experiment_item_bulk_write_view_source.py +5 -0
  739. opik/rest_api/types/feedback_score_names.py +4 -6
  740. opik/rest_api/types/feedback_score_public.py +11 -7
  741. opik/rest_api/types/feedback_score_public_source.py +1 -3
  742. opik/rest_api/types/feedback_score_source.py +1 -3
  743. opik/rest_api/types/feedback_update.py +27 -13
  744. opik/rest_api/types/function.py +4 -7
  745. opik/rest_api/types/function_call.py +3 -5
  746. opik/rest_api/types/group_content.py +19 -0
  747. opik/rest_api/types/group_content_with_aggregations.py +21 -0
  748. opik/rest_api/types/group_detail.py +19 -0
  749. opik/rest_api/types/group_details.py +20 -0
  750. opik/rest_api/types/guardrail.py +34 -0
  751. opik/rest_api/types/guardrail_batch.py +20 -0
  752. opik/rest_api/types/guardrail_name.py +5 -0
  753. opik/rest_api/types/guardrail_result.py +5 -0
  754. opik/rest_api/types/guardrail_write.py +33 -0
  755. opik/rest_api/types/guardrail_write_name.py +5 -0
  756. opik/rest_api/types/guardrail_write_result.py +5 -0
  757. opik/rest_api/types/guardrails_validation.py +21 -0
  758. opik/rest_api/types/guardrails_validation_public.py +21 -0
  759. opik/rest_api/types/ids_holder.py +19 -0
  760. opik/rest_api/types/image_url.py +20 -0
  761. opik/rest_api/types/image_url_public.py +20 -0
  762. opik/rest_api/types/image_url_write.py +20 -0
  763. opik/rest_api/types/json_list_string.py +7 -0
  764. opik/rest_api/types/json_list_string_compare.py +7 -0
  765. opik/rest_api/types/json_list_string_experiment_item_bulk_write_view.py +7 -0
  766. opik/rest_api/types/json_list_string_public.py +7 -0
  767. opik/rest_api/types/json_list_string_write.py +7 -0
  768. opik/rest_api/types/json_schema.py +5 -8
  769. opik/rest_api/types/llm_as_judge_code.py +8 -12
  770. opik/rest_api/types/llm_as_judge_code_public.py +8 -12
  771. opik/rest_api/types/llm_as_judge_code_write.py +8 -12
  772. opik/rest_api/types/llm_as_judge_message.py +9 -7
  773. opik/rest_api/types/llm_as_judge_message_content.py +26 -0
  774. opik/rest_api/types/llm_as_judge_message_content_public.py +26 -0
  775. opik/rest_api/types/llm_as_judge_message_content_write.py +26 -0
  776. opik/rest_api/types/llm_as_judge_message_public.py +9 -7
  777. opik/rest_api/types/llm_as_judge_message_public_role.py +1 -1
  778. opik/rest_api/types/llm_as_judge_message_role.py +1 -1
  779. opik/rest_api/types/llm_as_judge_message_write.py +9 -7
  780. opik/rest_api/types/llm_as_judge_message_write_role.py +1 -1
  781. opik/rest_api/types/llm_as_judge_model_parameters.py +6 -5
  782. opik/rest_api/types/llm_as_judge_model_parameters_public.py +6 -5
  783. opik/rest_api/types/llm_as_judge_model_parameters_write.py +6 -5
  784. opik/rest_api/types/llm_as_judge_output_schema.py +4 -6
  785. opik/rest_api/types/llm_as_judge_output_schema_public.py +4 -6
  786. opik/rest_api/types/llm_as_judge_output_schema_public_type.py +1 -3
  787. opik/rest_api/types/llm_as_judge_output_schema_type.py +1 -3
  788. opik/rest_api/types/llm_as_judge_output_schema_write.py +4 -6
  789. opik/rest_api/types/llm_as_judge_output_schema_write_type.py +1 -3
  790. opik/rest_api/types/log_item.py +5 -7
  791. opik/rest_api/types/log_item_level.py +1 -3
  792. opik/rest_api/types/log_page.py +4 -6
  793. opik/rest_api/types/manual_evaluation_request.py +38 -0
  794. opik/rest_api/types/manual_evaluation_request_entity_type.py +5 -0
  795. opik/rest_api/types/manual_evaluation_response.py +27 -0
  796. opik/rest_api/types/multipart_upload_part.py +20 -0
  797. opik/rest_api/types/numerical_feedback_definition.py +5 -7
  798. opik/rest_api/types/numerical_feedback_definition_create.py +4 -6
  799. opik/rest_api/types/numerical_feedback_definition_public.py +5 -7
  800. opik/rest_api/types/numerical_feedback_definition_update.py +4 -6
  801. opik/rest_api/types/numerical_feedback_detail.py +3 -5
  802. opik/rest_api/types/numerical_feedback_detail_create.py +3 -5
  803. opik/rest_api/types/numerical_feedback_detail_public.py +3 -5
  804. opik/rest_api/types/numerical_feedback_detail_update.py +3 -5
  805. opik/rest_api/types/optimization.py +37 -0
  806. opik/rest_api/types/optimization_page_public.py +28 -0
  807. opik/rest_api/types/optimization_public.py +37 -0
  808. opik/rest_api/types/optimization_public_status.py +7 -0
  809. opik/rest_api/types/optimization_status.py +7 -0
  810. opik/rest_api/types/optimization_studio_config.py +27 -0
  811. opik/rest_api/types/optimization_studio_config_public.py +27 -0
  812. opik/rest_api/types/optimization_studio_config_write.py +27 -0
  813. opik/rest_api/types/optimization_studio_log.py +22 -0
  814. opik/rest_api/types/optimization_write.py +30 -0
  815. opik/rest_api/types/optimization_write_status.py +7 -0
  816. opik/rest_api/types/page_columns.py +4 -6
  817. opik/rest_api/types/percentage_value_stat_public.py +4 -6
  818. opik/rest_api/types/percentage_values.py +8 -16
  819. opik/rest_api/types/percentage_values_detailed.py +8 -16
  820. opik/rest_api/types/percentage_values_public.py +8 -16
  821. opik/rest_api/types/project.py +12 -7
  822. opik/rest_api/types/project_detailed.py +12 -7
  823. opik/rest_api/types/project_detailed_visibility.py +5 -0
  824. opik/rest_api/types/project_metric_response_public.py +5 -9
  825. opik/rest_api/types/project_metric_response_public_interval.py +1 -3
  826. opik/rest_api/types/project_metric_response_public_metric_type.py +11 -1
  827. opik/rest_api/types/project_page_public.py +8 -10
  828. opik/rest_api/types/project_public.py +6 -6
  829. opik/rest_api/types/project_public_visibility.py +5 -0
  830. opik/rest_api/types/project_reference.py +31 -0
  831. opik/rest_api/types/project_reference_public.py +31 -0
  832. opik/rest_api/types/project_stat_item_object_public.py +8 -17
  833. opik/rest_api/types/project_stats_public.py +4 -6
  834. opik/rest_api/types/project_stats_summary.py +4 -6
  835. opik/rest_api/types/project_stats_summary_item.py +9 -6
  836. opik/rest_api/types/project_visibility.py +5 -0
  837. opik/rest_api/types/prompt.py +12 -7
  838. opik/rest_api/types/prompt_detail.py +12 -7
  839. opik/rest_api/types/prompt_detail_template_structure.py +5 -0
  840. opik/rest_api/types/prompt_page_public.py +9 -6
  841. opik/rest_api/types/prompt_public.py +11 -6
  842. opik/rest_api/types/prompt_public_template_structure.py +5 -0
  843. opik/rest_api/types/prompt_template_structure.py +5 -0
  844. opik/rest_api/types/prompt_tokens_details.py +19 -0
  845. opik/rest_api/types/prompt_version.py +7 -6
  846. opik/rest_api/types/prompt_version_detail.py +7 -6
  847. opik/rest_api/types/prompt_version_detail_template_structure.py +5 -0
  848. opik/rest_api/types/prompt_version_link.py +4 -5
  849. opik/rest_api/types/prompt_version_link_public.py +4 -5
  850. opik/rest_api/types/prompt_version_link_write.py +3 -5
  851. opik/rest_api/types/prompt_version_page_public.py +9 -6
  852. opik/rest_api/types/prompt_version_public.py +7 -6
  853. opik/rest_api/types/prompt_version_public_template_structure.py +5 -0
  854. opik/rest_api/types/prompt_version_template_structure.py +5 -0
  855. opik/rest_api/types/prompt_version_update.py +33 -0
  856. opik/rest_api/types/provider_api_key.py +18 -8
  857. opik/rest_api/types/provider_api_key_page_public.py +27 -0
  858. opik/rest_api/types/provider_api_key_provider.py +1 -1
  859. opik/rest_api/types/provider_api_key_public.py +18 -8
  860. opik/rest_api/types/provider_api_key_public_provider.py +1 -1
  861. opik/rest_api/types/response_format.py +5 -7
  862. opik/rest_api/types/response_format_type.py +1 -3
  863. opik/rest_api/types/result.py +21 -0
  864. opik/rest_api/types/results_number_public.py +4 -6
  865. opik/rest_api/types/score_name.py +4 -5
  866. opik/rest_api/types/service_toggles_config.py +44 -0
  867. opik/rest_api/types/span.py +13 -15
  868. opik/rest_api/types/span_batch.py +4 -6
  869. opik/rest_api/types/span_enrichment_options.py +31 -0
  870. opik/rest_api/types/span_experiment_item_bulk_write_view.py +39 -0
  871. opik/rest_api/types/span_experiment_item_bulk_write_view_type.py +5 -0
  872. opik/rest_api/types/span_filter.py +23 -0
  873. opik/rest_api/types/span_filter_operator.py +21 -0
  874. opik/rest_api/types/span_filter_public.py +4 -6
  875. opik/rest_api/types/span_filter_public_operator.py +2 -0
  876. opik/rest_api/types/span_filter_write.py +23 -0
  877. opik/rest_api/types/span_filter_write_operator.py +21 -0
  878. opik/rest_api/types/span_llm_as_judge_code.py +27 -0
  879. opik/rest_api/types/span_llm_as_judge_code_public.py +27 -0
  880. opik/rest_api/types/span_llm_as_judge_code_write.py +27 -0
  881. opik/rest_api/types/span_page_public.py +9 -6
  882. opik/rest_api/types/span_public.py +19 -16
  883. opik/rest_api/types/span_public_type.py +1 -1
  884. opik/rest_api/types/span_type.py +1 -1
  885. opik/rest_api/types/span_update.py +46 -0
  886. opik/rest_api/types/span_update_type.py +5 -0
  887. opik/rest_api/types/span_user_defined_metric_python_code.py +20 -0
  888. opik/rest_api/types/span_user_defined_metric_python_code_public.py +20 -0
  889. opik/rest_api/types/span_user_defined_metric_python_code_write.py +20 -0
  890. opik/rest_api/types/span_write.py +13 -14
  891. opik/rest_api/types/span_write_type.py +1 -1
  892. opik/rest_api/types/spans_count_response.py +20 -0
  893. opik/rest_api/types/start_multipart_upload_response.py +20 -0
  894. opik/rest_api/types/stream_options.py +3 -5
  895. opik/rest_api/types/studio_evaluation.py +20 -0
  896. opik/rest_api/types/studio_evaluation_public.py +20 -0
  897. opik/rest_api/types/studio_evaluation_write.py +20 -0
  898. opik/rest_api/types/studio_llm_model.py +21 -0
  899. opik/rest_api/types/studio_llm_model_public.py +21 -0
  900. opik/rest_api/types/studio_llm_model_write.py +21 -0
  901. opik/rest_api/types/studio_message.py +20 -0
  902. opik/rest_api/types/studio_message_public.py +20 -0
  903. opik/rest_api/types/studio_message_write.py +20 -0
  904. opik/rest_api/types/studio_metric.py +21 -0
  905. opik/rest_api/types/studio_metric_public.py +21 -0
  906. opik/rest_api/types/studio_metric_write.py +21 -0
  907. opik/rest_api/types/studio_optimizer.py +21 -0
  908. opik/rest_api/types/studio_optimizer_public.py +21 -0
  909. opik/rest_api/types/studio_optimizer_write.py +21 -0
  910. opik/rest_api/types/studio_prompt.py +20 -0
  911. opik/rest_api/types/studio_prompt_public.py +20 -0
  912. opik/rest_api/types/studio_prompt_write.py +20 -0
  913. opik/rest_api/types/tool.py +4 -6
  914. opik/rest_api/types/tool_call.py +4 -6
  915. opik/rest_api/types/trace.py +26 -12
  916. opik/rest_api/types/trace_batch.py +4 -6
  917. opik/rest_api/types/trace_count_response.py +4 -6
  918. opik/rest_api/types/trace_enrichment_options.py +32 -0
  919. opik/rest_api/types/trace_experiment_item_bulk_write_view.py +41 -0
  920. opik/rest_api/types/trace_filter.py +23 -0
  921. opik/rest_api/types/trace_filter_operator.py +21 -0
  922. opik/rest_api/types/trace_filter_public.py +23 -0
  923. opik/rest_api/types/trace_filter_public_operator.py +21 -0
  924. opik/rest_api/types/trace_filter_write.py +23 -0
  925. opik/rest_api/types/trace_filter_write_operator.py +21 -0
  926. opik/rest_api/types/trace_page_public.py +8 -10
  927. opik/rest_api/types/trace_public.py +27 -13
  928. opik/rest_api/types/trace_public_visibility_mode.py +5 -0
  929. opik/rest_api/types/trace_thread.py +18 -9
  930. opik/rest_api/types/trace_thread_filter.py +23 -0
  931. opik/rest_api/types/trace_thread_filter_operator.py +21 -0
  932. opik/rest_api/types/trace_thread_filter_public.py +23 -0
  933. opik/rest_api/types/trace_thread_filter_public_operator.py +21 -0
  934. opik/rest_api/types/trace_thread_filter_write.py +23 -0
  935. opik/rest_api/types/trace_thread_filter_write_operator.py +21 -0
  936. opik/rest_api/types/trace_thread_identifier.py +22 -0
  937. opik/rest_api/types/trace_thread_llm_as_judge_code.py +26 -0
  938. opik/rest_api/types/trace_thread_llm_as_judge_code_public.py +26 -0
  939. opik/rest_api/types/trace_thread_llm_as_judge_code_write.py +26 -0
  940. opik/rest_api/types/trace_thread_page.py +9 -6
  941. opik/rest_api/types/trace_thread_status.py +5 -0
  942. opik/rest_api/types/trace_thread_update.py +19 -0
  943. opik/rest_api/types/trace_thread_user_defined_metric_python_code.py +19 -0
  944. opik/rest_api/types/trace_thread_user_defined_metric_python_code_public.py +19 -0
  945. opik/rest_api/types/trace_thread_user_defined_metric_python_code_write.py +19 -0
  946. opik/rest_api/types/trace_update.py +39 -0
  947. opik/rest_api/types/trace_visibility_mode.py +5 -0
  948. opik/rest_api/types/trace_write.py +10 -11
  949. opik/rest_api/types/usage.py +6 -6
  950. opik/rest_api/types/user_defined_metric_python_code.py +3 -5
  951. opik/rest_api/types/user_defined_metric_python_code_public.py +3 -5
  952. opik/rest_api/types/user_defined_metric_python_code_write.py +3 -5
  953. opik/rest_api/types/value_entry.py +27 -0
  954. opik/rest_api/types/value_entry_compare.py +27 -0
  955. opik/rest_api/types/value_entry_compare_source.py +5 -0
  956. opik/rest_api/types/value_entry_experiment_item_bulk_write_view.py +27 -0
  957. opik/rest_api/types/value_entry_experiment_item_bulk_write_view_source.py +5 -0
  958. opik/rest_api/types/value_entry_public.py +27 -0
  959. opik/rest_api/types/value_entry_public_source.py +5 -0
  960. opik/rest_api/types/value_entry_source.py +5 -0
  961. opik/rest_api/types/video_url.py +19 -0
  962. opik/rest_api/types/video_url_public.py +19 -0
  963. opik/rest_api/types/video_url_write.py +19 -0
  964. opik/rest_api/types/webhook.py +28 -0
  965. opik/rest_api/types/webhook_examples.py +19 -0
  966. opik/rest_api/types/webhook_public.py +28 -0
  967. opik/rest_api/types/webhook_test_result.py +23 -0
  968. opik/rest_api/types/webhook_test_result_status.py +5 -0
  969. opik/rest_api/types/webhook_write.py +23 -0
  970. opik/rest_api/types/welcome_wizard_tracking.py +22 -0
  971. opik/rest_api/types/workspace_configuration.py +27 -0
  972. opik/rest_api/types/workspace_metric_request.py +24 -0
  973. opik/rest_api/types/workspace_metric_response.py +20 -0
  974. opik/rest_api/types/workspace_metrics_summary_request.py +23 -0
  975. opik/rest_api/types/workspace_metrics_summary_response.py +20 -0
  976. opik/rest_api/types/workspace_name_holder.py +19 -0
  977. opik/rest_api/types/workspace_spans_count.py +20 -0
  978. opik/rest_api/types/workspace_trace_count.py +3 -5
  979. opik/rest_api/welcome_wizard/__init__.py +4 -0
  980. opik/rest_api/welcome_wizard/client.py +195 -0
  981. opik/rest_api/welcome_wizard/raw_client.py +208 -0
  982. opik/rest_api/workspaces/__init__.py +2 -0
  983. opik/rest_api/workspaces/client.py +550 -77
  984. opik/rest_api/workspaces/raw_client.py +923 -0
  985. opik/rest_client_configurator/api.py +1 -0
  986. opik/rest_client_configurator/retry_decorator.py +1 -0
  987. opik/s3_httpx_client.py +67 -0
  988. opik/simulation/__init__.py +6 -0
  989. opik/simulation/simulated_user.py +99 -0
  990. opik/simulation/simulator.py +108 -0
  991. opik/synchronization.py +11 -24
  992. opik/tracing_runtime_config.py +48 -0
  993. opik/types.py +48 -2
  994. opik/url_helpers.py +13 -3
  995. opik/validation/chat_prompt_messages.py +241 -0
  996. opik/validation/feedback_score.py +4 -5
  997. opik/validation/parameter.py +122 -0
  998. opik/validation/parameters_validator.py +175 -0
  999. opik/validation/validator.py +30 -2
  1000. opik/validation/validator_helpers.py +147 -0
  1001. opik-1.9.71.dist-info/METADATA +370 -0
  1002. opik-1.9.71.dist-info/RECORD +1110 -0
  1003. {opik-1.6.4.dist-info → opik-1.9.71.dist-info}/WHEEL +1 -1
  1004. opik-1.9.71.dist-info/licenses/LICENSE +203 -0
  1005. opik/api_objects/prompt/prompt.py +0 -107
  1006. opik/api_objects/prompt/prompt_template.py +0 -35
  1007. opik/cli.py +0 -193
  1008. opik/evaluation/metrics/models.py +0 -8
  1009. opik/hooks.py +0 -13
  1010. opik/integrations/bedrock/chunks_aggregator.py +0 -55
  1011. opik/integrations/bedrock/helpers.py +0 -8
  1012. opik/integrations/langchain/google_run_helpers.py +0 -75
  1013. opik/integrations/langchain/openai_run_helpers.py +0 -122
  1014. opik/message_processing/message_processors.py +0 -203
  1015. opik/rest_api/types/delta_role.py +0 -7
  1016. opik/rest_api/types/json_object_schema.py +0 -34
  1017. opik-1.6.4.dist-info/METADATA +0 -270
  1018. opik-1.6.4.dist-info/RECORD +0 -507
  1019. /opik/integrations/bedrock/{stream_wrappers.py → converse/stream_wrappers.py} +0 -0
  1020. {opik-1.6.4.dist-info → opik-1.9.71.dist-info}/entry_points.txt +0 -0
  1021. {opik-1.6.4.dist-info → opik-1.9.71.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,2383 @@
1
+ # This file was auto-generated by Fern from our API Definition.
2
+
3
+ import contextlib
4
+ import datetime as dt
5
+ import typing
6
+ from json.decoder import JSONDecodeError
7
+
8
+ from ..core.api_error import ApiError
9
+ from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
10
+ from ..core.datetime_utils import serialize_datetime
11
+ from ..core.http_response import AsyncHttpResponse, HttpResponse
12
+ from ..core.jsonable_encoder import jsonable_encoder
13
+ from ..core.pydantic_utilities import parse_obj_as
14
+ from ..core.request_options import RequestOptions
15
+ from ..core.serialization import convert_and_respect_annotation_metadata
16
+ from ..errors.bad_request_error import BadRequestError
17
+ from ..errors.conflict_error import ConflictError
18
+ from ..errors.not_found_error import NotFoundError
19
+ from ..errors.not_implemented_error import NotImplementedError
20
+ from ..types.comment import Comment
21
+ from ..types.error_info import ErrorInfo
22
+ from ..types.error_info_write import ErrorInfoWrite
23
+ from ..types.feedback_score_batch_item import FeedbackScoreBatchItem
24
+ from ..types.feedback_score_source import FeedbackScoreSource
25
+ from ..types.json_list_string import JsonListString
26
+ from ..types.json_list_string_write import JsonListStringWrite
27
+ from ..types.project_stats_public import ProjectStatsPublic
28
+ from ..types.span_filter_public import SpanFilterPublic
29
+ from ..types.span_page_public import SpanPagePublic
30
+ from ..types.span_public import SpanPublic
31
+ from ..types.span_update import SpanUpdate
32
+ from ..types.span_update_type import SpanUpdateType
33
+ from ..types.span_write import SpanWrite
34
+ from ..types.span_write_type import SpanWriteType
35
+ from ..types.value_entry import ValueEntry
36
+ from .types.find_feedback_score_names_1_request_type import FindFeedbackScoreNames1RequestType
37
+ from .types.get_span_stats_request_type import GetSpanStatsRequestType
38
+ from .types.get_spans_by_project_request_type import GetSpansByProjectRequestType
39
+ from .types.span_search_stream_request_public_type import SpanSearchStreamRequestPublicType
40
+
41
+ # this is used as the default value for optional parameters
42
+ OMIT = typing.cast(typing.Any, ...)
43
+
44
+
45
+ class RawSpansClient:
46
+ def __init__(self, *, client_wrapper: SyncClientWrapper):
47
+ self._client_wrapper = client_wrapper
48
+
49
+ def add_span_comment(
50
+ self,
51
+ id_: str,
52
+ *,
53
+ text: str,
54
+ id: typing.Optional[str] = OMIT,
55
+ created_at: typing.Optional[dt.datetime] = OMIT,
56
+ last_updated_at: typing.Optional[dt.datetime] = OMIT,
57
+ created_by: typing.Optional[str] = OMIT,
58
+ last_updated_by: typing.Optional[str] = OMIT,
59
+ request_options: typing.Optional[RequestOptions] = None,
60
+ ) -> HttpResponse[None]:
61
+ """
62
+ Add span comment
63
+
64
+ Parameters
65
+ ----------
66
+ id_ : str
67
+
68
+ text : str
69
+
70
+ id : typing.Optional[str]
71
+
72
+ created_at : typing.Optional[dt.datetime]
73
+
74
+ last_updated_at : typing.Optional[dt.datetime]
75
+
76
+ created_by : typing.Optional[str]
77
+
78
+ last_updated_by : typing.Optional[str]
79
+
80
+ request_options : typing.Optional[RequestOptions]
81
+ Request-specific configuration.
82
+
83
+ Returns
84
+ -------
85
+ HttpResponse[None]
86
+ """
87
+ _response = self._client_wrapper.httpx_client.request(
88
+ f"v1/private/spans/{jsonable_encoder(id_)}/comments",
89
+ method="POST",
90
+ json={
91
+ "id": id,
92
+ "text": text,
93
+ "created_at": created_at,
94
+ "last_updated_at": last_updated_at,
95
+ "created_by": created_by,
96
+ "last_updated_by": last_updated_by,
97
+ },
98
+ headers={
99
+ "content-type": "application/json",
100
+ },
101
+ request_options=request_options,
102
+ omit=OMIT,
103
+ )
104
+ try:
105
+ if 200 <= _response.status_code < 300:
106
+ return HttpResponse(response=_response, data=None)
107
+ _response_json = _response.json()
108
+ except JSONDecodeError:
109
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
110
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
111
+
112
+ def add_span_feedback_score(
113
+ self,
114
+ id: str,
115
+ *,
116
+ name: str,
117
+ value: float,
118
+ source: FeedbackScoreSource,
119
+ category_name: typing.Optional[str] = OMIT,
120
+ reason: typing.Optional[str] = OMIT,
121
+ created_at: typing.Optional[dt.datetime] = OMIT,
122
+ last_updated_at: typing.Optional[dt.datetime] = OMIT,
123
+ created_by: typing.Optional[str] = OMIT,
124
+ last_updated_by: typing.Optional[str] = OMIT,
125
+ value_by_author: typing.Optional[typing.Dict[str, ValueEntry]] = OMIT,
126
+ request_options: typing.Optional[RequestOptions] = None,
127
+ ) -> HttpResponse[None]:
128
+ """
129
+ Add span feedback score
130
+
131
+ Parameters
132
+ ----------
133
+ id : str
134
+
135
+ name : str
136
+
137
+ value : float
138
+
139
+ source : FeedbackScoreSource
140
+
141
+ category_name : typing.Optional[str]
142
+
143
+ reason : typing.Optional[str]
144
+
145
+ created_at : typing.Optional[dt.datetime]
146
+
147
+ last_updated_at : typing.Optional[dt.datetime]
148
+
149
+ created_by : typing.Optional[str]
150
+
151
+ last_updated_by : typing.Optional[str]
152
+
153
+ value_by_author : typing.Optional[typing.Dict[str, ValueEntry]]
154
+
155
+ request_options : typing.Optional[RequestOptions]
156
+ Request-specific configuration.
157
+
158
+ Returns
159
+ -------
160
+ HttpResponse[None]
161
+ """
162
+ _response = self._client_wrapper.httpx_client.request(
163
+ f"v1/private/spans/{jsonable_encoder(id)}/feedback-scores",
164
+ method="PUT",
165
+ json={
166
+ "name": name,
167
+ "category_name": category_name,
168
+ "value": value,
169
+ "reason": reason,
170
+ "source": source,
171
+ "created_at": created_at,
172
+ "last_updated_at": last_updated_at,
173
+ "created_by": created_by,
174
+ "last_updated_by": last_updated_by,
175
+ "value_by_author": convert_and_respect_annotation_metadata(
176
+ object_=value_by_author, annotation=typing.Dict[str, ValueEntry], direction="write"
177
+ ),
178
+ },
179
+ headers={
180
+ "content-type": "application/json",
181
+ },
182
+ request_options=request_options,
183
+ omit=OMIT,
184
+ )
185
+ try:
186
+ if 200 <= _response.status_code < 300:
187
+ return HttpResponse(response=_response, data=None)
188
+ _response_json = _response.json()
189
+ except JSONDecodeError:
190
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
191
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
192
+
193
+ def create_spans(
194
+ self, *, spans: typing.Sequence[SpanWrite], request_options: typing.Optional[RequestOptions] = None
195
+ ) -> HttpResponse[None]:
196
+ """
197
+ Create spans
198
+
199
+ Parameters
200
+ ----------
201
+ spans : typing.Sequence[SpanWrite]
202
+
203
+ request_options : typing.Optional[RequestOptions]
204
+ Request-specific configuration.
205
+
206
+ Returns
207
+ -------
208
+ HttpResponse[None]
209
+ """
210
+ _response = self._client_wrapper.httpx_client.request(
211
+ "v1/private/spans/batch",
212
+ method="POST",
213
+ json={
214
+ "spans": convert_and_respect_annotation_metadata(
215
+ object_=spans, annotation=typing.Sequence[SpanWrite], direction="write"
216
+ ),
217
+ },
218
+ headers={
219
+ "content-type": "application/json",
220
+ },
221
+ request_options=request_options,
222
+ omit=OMIT,
223
+ )
224
+ try:
225
+ if 200 <= _response.status_code < 300:
226
+ return HttpResponse(response=_response, data=None)
227
+ _response_json = _response.json()
228
+ except JSONDecodeError:
229
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
230
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
231
+
232
+ def batch_update_spans(
233
+ self,
234
+ *,
235
+ ids: typing.Sequence[str],
236
+ update: SpanUpdate,
237
+ merge_tags: typing.Optional[bool] = OMIT,
238
+ request_options: typing.Optional[RequestOptions] = None,
239
+ ) -> HttpResponse[None]:
240
+ """
241
+ Update multiple spans
242
+
243
+ Parameters
244
+ ----------
245
+ ids : typing.Sequence[str]
246
+ List of span IDs to update (max 1000)
247
+
248
+ update : SpanUpdate
249
+
250
+ merge_tags : typing.Optional[bool]
251
+ If true, merge tags with existing tags instead of replacing them. Default: false
252
+
253
+ request_options : typing.Optional[RequestOptions]
254
+ Request-specific configuration.
255
+
256
+ Returns
257
+ -------
258
+ HttpResponse[None]
259
+ """
260
+ _response = self._client_wrapper.httpx_client.request(
261
+ "v1/private/spans/batch",
262
+ method="PATCH",
263
+ json={
264
+ "ids": ids,
265
+ "update": convert_and_respect_annotation_metadata(
266
+ object_=update, annotation=SpanUpdate, direction="write"
267
+ ),
268
+ "merge_tags": merge_tags,
269
+ },
270
+ headers={
271
+ "content-type": "application/json",
272
+ },
273
+ request_options=request_options,
274
+ omit=OMIT,
275
+ )
276
+ try:
277
+ if 200 <= _response.status_code < 300:
278
+ return HttpResponse(response=_response, data=None)
279
+ if _response.status_code == 400:
280
+ raise BadRequestError(
281
+ headers=dict(_response.headers),
282
+ body=typing.cast(
283
+ typing.Optional[typing.Any],
284
+ parse_obj_as(
285
+ type_=typing.Optional[typing.Any], # type: ignore
286
+ object_=_response.json(),
287
+ ),
288
+ ),
289
+ )
290
+ _response_json = _response.json()
291
+ except JSONDecodeError:
292
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
293
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
294
+
295
+ def get_spans_by_project(
296
+ self,
297
+ *,
298
+ page: typing.Optional[int] = None,
299
+ size: typing.Optional[int] = None,
300
+ project_name: typing.Optional[str] = None,
301
+ project_id: typing.Optional[str] = None,
302
+ trace_id: typing.Optional[str] = None,
303
+ type: typing.Optional[GetSpansByProjectRequestType] = None,
304
+ filters: typing.Optional[str] = None,
305
+ truncate: typing.Optional[bool] = None,
306
+ strip_attachments: typing.Optional[bool] = None,
307
+ sorting: typing.Optional[str] = None,
308
+ exclude: typing.Optional[str] = None,
309
+ from_time: typing.Optional[dt.datetime] = None,
310
+ to_time: typing.Optional[dt.datetime] = None,
311
+ request_options: typing.Optional[RequestOptions] = None,
312
+ ) -> HttpResponse[SpanPagePublic]:
313
+ """
314
+ Get spans by project_name or project_id and optionally by trace_id and/or type
315
+
316
+ Parameters
317
+ ----------
318
+ page : typing.Optional[int]
319
+
320
+ size : typing.Optional[int]
321
+
322
+ project_name : typing.Optional[str]
323
+
324
+ project_id : typing.Optional[str]
325
+
326
+ trace_id : typing.Optional[str]
327
+
328
+ type : typing.Optional[GetSpansByProjectRequestType]
329
+
330
+ filters : typing.Optional[str]
331
+
332
+ truncate : typing.Optional[bool]
333
+
334
+ strip_attachments : typing.Optional[bool]
335
+
336
+ sorting : typing.Optional[str]
337
+
338
+ exclude : typing.Optional[str]
339
+
340
+ from_time : typing.Optional[dt.datetime]
341
+
342
+ to_time : typing.Optional[dt.datetime]
343
+
344
+ request_options : typing.Optional[RequestOptions]
345
+ Request-specific configuration.
346
+
347
+ Returns
348
+ -------
349
+ HttpResponse[SpanPagePublic]
350
+ Spans resource
351
+ """
352
+ _response = self._client_wrapper.httpx_client.request(
353
+ "v1/private/spans",
354
+ method="GET",
355
+ params={
356
+ "page": page,
357
+ "size": size,
358
+ "project_name": project_name,
359
+ "project_id": project_id,
360
+ "trace_id": trace_id,
361
+ "type": type,
362
+ "filters": filters,
363
+ "truncate": truncate,
364
+ "strip_attachments": strip_attachments,
365
+ "sorting": sorting,
366
+ "exclude": exclude,
367
+ "from_time": serialize_datetime(from_time) if from_time is not None else None,
368
+ "to_time": serialize_datetime(to_time) if to_time is not None else None,
369
+ },
370
+ request_options=request_options,
371
+ )
372
+ try:
373
+ if 200 <= _response.status_code < 300:
374
+ _data = typing.cast(
375
+ SpanPagePublic,
376
+ parse_obj_as(
377
+ type_=SpanPagePublic, # type: ignore
378
+ object_=_response.json(),
379
+ ),
380
+ )
381
+ return HttpResponse(response=_response, data=_data)
382
+ _response_json = _response.json()
383
+ except JSONDecodeError:
384
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
385
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
386
+
387
+ def create_span(
388
+ self,
389
+ *,
390
+ start_time: dt.datetime,
391
+ id: typing.Optional[str] = OMIT,
392
+ project_name: typing.Optional[str] = OMIT,
393
+ trace_id: typing.Optional[str] = OMIT,
394
+ parent_span_id: typing.Optional[str] = OMIT,
395
+ name: typing.Optional[str] = OMIT,
396
+ type: typing.Optional[SpanWriteType] = OMIT,
397
+ end_time: typing.Optional[dt.datetime] = OMIT,
398
+ input: typing.Optional[JsonListStringWrite] = OMIT,
399
+ output: typing.Optional[JsonListStringWrite] = OMIT,
400
+ metadata: typing.Optional[JsonListStringWrite] = OMIT,
401
+ model: typing.Optional[str] = OMIT,
402
+ provider: typing.Optional[str] = OMIT,
403
+ tags: typing.Optional[typing.Sequence[str]] = OMIT,
404
+ usage: typing.Optional[typing.Dict[str, int]] = OMIT,
405
+ error_info: typing.Optional[ErrorInfoWrite] = OMIT,
406
+ last_updated_at: typing.Optional[dt.datetime] = OMIT,
407
+ total_estimated_cost: typing.Optional[float] = OMIT,
408
+ total_estimated_cost_version: typing.Optional[str] = OMIT,
409
+ request_options: typing.Optional[RequestOptions] = None,
410
+ ) -> HttpResponse[None]:
411
+ """
412
+ Create span
413
+
414
+ Parameters
415
+ ----------
416
+ start_time : dt.datetime
417
+
418
+ id : typing.Optional[str]
419
+
420
+ project_name : typing.Optional[str]
421
+ If null, the default project is used
422
+
423
+ trace_id : typing.Optional[str]
424
+
425
+ parent_span_id : typing.Optional[str]
426
+
427
+ name : typing.Optional[str]
428
+
429
+ type : typing.Optional[SpanWriteType]
430
+
431
+ end_time : typing.Optional[dt.datetime]
432
+
433
+ input : typing.Optional[JsonListStringWrite]
434
+
435
+ output : typing.Optional[JsonListStringWrite]
436
+
437
+ metadata : typing.Optional[JsonListStringWrite]
438
+
439
+ model : typing.Optional[str]
440
+
441
+ provider : typing.Optional[str]
442
+
443
+ tags : typing.Optional[typing.Sequence[str]]
444
+
445
+ usage : typing.Optional[typing.Dict[str, int]]
446
+
447
+ error_info : typing.Optional[ErrorInfoWrite]
448
+
449
+ last_updated_at : typing.Optional[dt.datetime]
450
+
451
+ total_estimated_cost : typing.Optional[float]
452
+
453
+ total_estimated_cost_version : typing.Optional[str]
454
+
455
+ request_options : typing.Optional[RequestOptions]
456
+ Request-specific configuration.
457
+
458
+ Returns
459
+ -------
460
+ HttpResponse[None]
461
+ """
462
+ _response = self._client_wrapper.httpx_client.request(
463
+ "v1/private/spans",
464
+ method="POST",
465
+ json={
466
+ "id": id,
467
+ "project_name": project_name,
468
+ "trace_id": trace_id,
469
+ "parent_span_id": parent_span_id,
470
+ "name": name,
471
+ "type": type,
472
+ "start_time": start_time,
473
+ "end_time": end_time,
474
+ "input": convert_and_respect_annotation_metadata(
475
+ object_=input, annotation=JsonListStringWrite, direction="write"
476
+ ),
477
+ "output": convert_and_respect_annotation_metadata(
478
+ object_=output, annotation=JsonListStringWrite, direction="write"
479
+ ),
480
+ "metadata": convert_and_respect_annotation_metadata(
481
+ object_=metadata, annotation=JsonListStringWrite, direction="write"
482
+ ),
483
+ "model": model,
484
+ "provider": provider,
485
+ "tags": tags,
486
+ "usage": usage,
487
+ "error_info": convert_and_respect_annotation_metadata(
488
+ object_=error_info, annotation=ErrorInfoWrite, direction="write"
489
+ ),
490
+ "last_updated_at": last_updated_at,
491
+ "total_estimated_cost": total_estimated_cost,
492
+ "total_estimated_cost_version": total_estimated_cost_version,
493
+ },
494
+ headers={
495
+ "content-type": "application/json",
496
+ },
497
+ request_options=request_options,
498
+ omit=OMIT,
499
+ )
500
+ try:
501
+ if 200 <= _response.status_code < 300:
502
+ return HttpResponse(response=_response, data=None)
503
+ if _response.status_code == 409:
504
+ raise ConflictError(
505
+ headers=dict(_response.headers),
506
+ body=typing.cast(
507
+ typing.Optional[typing.Any],
508
+ parse_obj_as(
509
+ type_=typing.Optional[typing.Any], # type: ignore
510
+ object_=_response.json(),
511
+ ),
512
+ ),
513
+ )
514
+ _response_json = _response.json()
515
+ except JSONDecodeError:
516
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
517
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
518
+
519
+ def get_span_by_id(
520
+ self,
521
+ id: str,
522
+ *,
523
+ strip_attachments: typing.Optional[bool] = None,
524
+ request_options: typing.Optional[RequestOptions] = None,
525
+ ) -> HttpResponse[SpanPublic]:
526
+ """
527
+ Get span by id
528
+
529
+ Parameters
530
+ ----------
531
+ id : str
532
+
533
+ strip_attachments : typing.Optional[bool]
534
+
535
+ request_options : typing.Optional[RequestOptions]
536
+ Request-specific configuration.
537
+
538
+ Returns
539
+ -------
540
+ HttpResponse[SpanPublic]
541
+ Span resource
542
+ """
543
+ _response = self._client_wrapper.httpx_client.request(
544
+ f"v1/private/spans/{jsonable_encoder(id)}",
545
+ method="GET",
546
+ params={
547
+ "strip_attachments": strip_attachments,
548
+ },
549
+ request_options=request_options,
550
+ )
551
+ try:
552
+ if 200 <= _response.status_code < 300:
553
+ _data = typing.cast(
554
+ SpanPublic,
555
+ parse_obj_as(
556
+ type_=SpanPublic, # type: ignore
557
+ object_=_response.json(),
558
+ ),
559
+ )
560
+ return HttpResponse(response=_response, data=_data)
561
+ if _response.status_code == 404:
562
+ raise NotFoundError(
563
+ headers=dict(_response.headers),
564
+ body=typing.cast(
565
+ typing.Optional[typing.Any],
566
+ parse_obj_as(
567
+ type_=typing.Optional[typing.Any], # type: ignore
568
+ object_=_response.json(),
569
+ ),
570
+ ),
571
+ )
572
+ _response_json = _response.json()
573
+ except JSONDecodeError:
574
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
575
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
576
+
577
+ def delete_span_by_id(
578
+ self, id: str, *, request_options: typing.Optional[RequestOptions] = None
579
+ ) -> HttpResponse[None]:
580
+ """
581
+ Delete span by id
582
+
583
+ Parameters
584
+ ----------
585
+ id : str
586
+
587
+ request_options : typing.Optional[RequestOptions]
588
+ Request-specific configuration.
589
+
590
+ Returns
591
+ -------
592
+ HttpResponse[None]
593
+ """
594
+ _response = self._client_wrapper.httpx_client.request(
595
+ f"v1/private/spans/{jsonable_encoder(id)}",
596
+ method="DELETE",
597
+ request_options=request_options,
598
+ )
599
+ try:
600
+ if 200 <= _response.status_code < 300:
601
+ return HttpResponse(response=_response, data=None)
602
+ if _response.status_code == 501:
603
+ raise NotImplementedError(
604
+ headers=dict(_response.headers),
605
+ body=typing.cast(
606
+ typing.Optional[typing.Any],
607
+ parse_obj_as(
608
+ type_=typing.Optional[typing.Any], # type: ignore
609
+ object_=_response.json(),
610
+ ),
611
+ ),
612
+ )
613
+ _response_json = _response.json()
614
+ except JSONDecodeError:
615
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
616
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
617
+
618
+ def update_span(
619
+ self,
620
+ id: str,
621
+ *,
622
+ trace_id: str,
623
+ project_name: typing.Optional[str] = OMIT,
624
+ project_id: typing.Optional[str] = OMIT,
625
+ parent_span_id: typing.Optional[str] = OMIT,
626
+ name: typing.Optional[str] = OMIT,
627
+ type: typing.Optional[SpanUpdateType] = OMIT,
628
+ end_time: typing.Optional[dt.datetime] = OMIT,
629
+ input: typing.Optional[JsonListString] = OMIT,
630
+ output: typing.Optional[JsonListString] = OMIT,
631
+ metadata: typing.Optional[JsonListString] = OMIT,
632
+ model: typing.Optional[str] = OMIT,
633
+ provider: typing.Optional[str] = OMIT,
634
+ tags: typing.Optional[typing.Sequence[str]] = OMIT,
635
+ usage: typing.Optional[typing.Dict[str, int]] = OMIT,
636
+ total_estimated_cost: typing.Optional[float] = OMIT,
637
+ error_info: typing.Optional[ErrorInfo] = OMIT,
638
+ request_options: typing.Optional[RequestOptions] = None,
639
+ ) -> HttpResponse[None]:
640
+ """
641
+ Update span by id
642
+
643
+ Parameters
644
+ ----------
645
+ id : str
646
+
647
+ trace_id : str
648
+
649
+ project_name : typing.Optional[str]
650
+ If null and project_id not specified, Default Project is assumed
651
+
652
+ project_id : typing.Optional[str]
653
+ If null and project_name not specified, Default Project is assumed
654
+
655
+ parent_span_id : typing.Optional[str]
656
+
657
+ name : typing.Optional[str]
658
+
659
+ type : typing.Optional[SpanUpdateType]
660
+
661
+ end_time : typing.Optional[dt.datetime]
662
+
663
+ input : typing.Optional[JsonListString]
664
+
665
+ output : typing.Optional[JsonListString]
666
+
667
+ metadata : typing.Optional[JsonListString]
668
+
669
+ model : typing.Optional[str]
670
+
671
+ provider : typing.Optional[str]
672
+
673
+ tags : typing.Optional[typing.Sequence[str]]
674
+
675
+ usage : typing.Optional[typing.Dict[str, int]]
676
+
677
+ total_estimated_cost : typing.Optional[float]
678
+
679
+ error_info : typing.Optional[ErrorInfo]
680
+
681
+ request_options : typing.Optional[RequestOptions]
682
+ Request-specific configuration.
683
+
684
+ Returns
685
+ -------
686
+ HttpResponse[None]
687
+ """
688
+ _response = self._client_wrapper.httpx_client.request(
689
+ f"v1/private/spans/{jsonable_encoder(id)}",
690
+ method="PATCH",
691
+ json={
692
+ "project_name": project_name,
693
+ "project_id": project_id,
694
+ "trace_id": trace_id,
695
+ "parent_span_id": parent_span_id,
696
+ "name": name,
697
+ "type": type,
698
+ "end_time": end_time,
699
+ "input": convert_and_respect_annotation_metadata(
700
+ object_=input, annotation=JsonListString, direction="write"
701
+ ),
702
+ "output": convert_and_respect_annotation_metadata(
703
+ object_=output, annotation=JsonListString, direction="write"
704
+ ),
705
+ "metadata": convert_and_respect_annotation_metadata(
706
+ object_=metadata, annotation=JsonListString, direction="write"
707
+ ),
708
+ "model": model,
709
+ "provider": provider,
710
+ "tags": tags,
711
+ "usage": usage,
712
+ "total_estimated_cost": total_estimated_cost,
713
+ "error_info": convert_and_respect_annotation_metadata(
714
+ object_=error_info, annotation=ErrorInfo, direction="write"
715
+ ),
716
+ },
717
+ headers={
718
+ "content-type": "application/json",
719
+ },
720
+ request_options=request_options,
721
+ omit=OMIT,
722
+ )
723
+ try:
724
+ if 200 <= _response.status_code < 300:
725
+ return HttpResponse(response=_response, data=None)
726
+ if _response.status_code == 404:
727
+ raise NotFoundError(
728
+ headers=dict(_response.headers),
729
+ body=typing.cast(
730
+ typing.Optional[typing.Any],
731
+ parse_obj_as(
732
+ type_=typing.Optional[typing.Any], # type: ignore
733
+ object_=_response.json(),
734
+ ),
735
+ ),
736
+ )
737
+ _response_json = _response.json()
738
+ except JSONDecodeError:
739
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
740
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
741
+
742
+ def delete_span_comments(
743
+ self, *, ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
744
+ ) -> HttpResponse[None]:
745
+ """
746
+ Delete span comments
747
+
748
+ Parameters
749
+ ----------
750
+ ids : typing.Sequence[str]
751
+
752
+ request_options : typing.Optional[RequestOptions]
753
+ Request-specific configuration.
754
+
755
+ Returns
756
+ -------
757
+ HttpResponse[None]
758
+ """
759
+ _response = self._client_wrapper.httpx_client.request(
760
+ "v1/private/spans/comments/delete",
761
+ method="POST",
762
+ json={
763
+ "ids": ids,
764
+ },
765
+ headers={
766
+ "content-type": "application/json",
767
+ },
768
+ request_options=request_options,
769
+ omit=OMIT,
770
+ )
771
+ try:
772
+ if 200 <= _response.status_code < 300:
773
+ return HttpResponse(response=_response, data=None)
774
+ _response_json = _response.json()
775
+ except JSONDecodeError:
776
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
777
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
778
+
779
+ def delete_span_feedback_score(
780
+ self,
781
+ id: str,
782
+ *,
783
+ name: str,
784
+ author: typing.Optional[str] = OMIT,
785
+ request_options: typing.Optional[RequestOptions] = None,
786
+ ) -> HttpResponse[None]:
787
+ """
788
+ Delete span feedback score
789
+
790
+ Parameters
791
+ ----------
792
+ id : str
793
+
794
+ name : str
795
+
796
+ author : typing.Optional[str]
797
+
798
+ request_options : typing.Optional[RequestOptions]
799
+ Request-specific configuration.
800
+
801
+ Returns
802
+ -------
803
+ HttpResponse[None]
804
+ """
805
+ _response = self._client_wrapper.httpx_client.request(
806
+ f"v1/private/spans/{jsonable_encoder(id)}/feedback-scores/delete",
807
+ method="POST",
808
+ json={
809
+ "name": name,
810
+ "author": author,
811
+ },
812
+ headers={
813
+ "content-type": "application/json",
814
+ },
815
+ request_options=request_options,
816
+ omit=OMIT,
817
+ )
818
+ try:
819
+ if 200 <= _response.status_code < 300:
820
+ return HttpResponse(response=_response, data=None)
821
+ _response_json = _response.json()
822
+ except JSONDecodeError:
823
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
824
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
825
+
826
+ def find_feedback_score_names_1(
827
+ self,
828
+ *,
829
+ project_id: typing.Optional[str] = None,
830
+ type: typing.Optional[FindFeedbackScoreNames1RequestType] = None,
831
+ request_options: typing.Optional[RequestOptions] = None,
832
+ ) -> HttpResponse[typing.List[str]]:
833
+ """
834
+ Find Feedback Score names
835
+
836
+ Parameters
837
+ ----------
838
+ project_id : typing.Optional[str]
839
+
840
+ type : typing.Optional[FindFeedbackScoreNames1RequestType]
841
+
842
+ request_options : typing.Optional[RequestOptions]
843
+ Request-specific configuration.
844
+
845
+ Returns
846
+ -------
847
+ HttpResponse[typing.List[str]]
848
+ Feedback Scores resource
849
+ """
850
+ _response = self._client_wrapper.httpx_client.request(
851
+ "v1/private/spans/feedback-scores/names",
852
+ method="GET",
853
+ params={
854
+ "project_id": project_id,
855
+ "type": type,
856
+ },
857
+ request_options=request_options,
858
+ )
859
+ try:
860
+ if 200 <= _response.status_code < 300:
861
+ _data = typing.cast(
862
+ typing.List[str],
863
+ parse_obj_as(
864
+ type_=typing.List[str], # type: ignore
865
+ object_=_response.json(),
866
+ ),
867
+ )
868
+ return HttpResponse(response=_response, data=_data)
869
+ _response_json = _response.json()
870
+ except JSONDecodeError:
871
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
872
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
873
+
874
+ def get_span_comment(
875
+ self, comment_id: str, span_id: str, *, request_options: typing.Optional[RequestOptions] = None
876
+ ) -> HttpResponse[Comment]:
877
+ """
878
+ Get span comment
879
+
880
+ Parameters
881
+ ----------
882
+ comment_id : str
883
+
884
+ span_id : str
885
+
886
+ request_options : typing.Optional[RequestOptions]
887
+ Request-specific configuration.
888
+
889
+ Returns
890
+ -------
891
+ HttpResponse[Comment]
892
+ Comment resource
893
+ """
894
+ _response = self._client_wrapper.httpx_client.request(
895
+ f"v1/private/spans/{jsonable_encoder(span_id)}/comments/{jsonable_encoder(comment_id)}",
896
+ method="GET",
897
+ request_options=request_options,
898
+ )
899
+ try:
900
+ if 200 <= _response.status_code < 300:
901
+ _data = typing.cast(
902
+ Comment,
903
+ parse_obj_as(
904
+ type_=Comment, # type: ignore
905
+ object_=_response.json(),
906
+ ),
907
+ )
908
+ return HttpResponse(response=_response, data=_data)
909
+ if _response.status_code == 404:
910
+ raise NotFoundError(
911
+ headers=dict(_response.headers),
912
+ body=typing.cast(
913
+ typing.Optional[typing.Any],
914
+ parse_obj_as(
915
+ type_=typing.Optional[typing.Any], # type: ignore
916
+ object_=_response.json(),
917
+ ),
918
+ ),
919
+ )
920
+ _response_json = _response.json()
921
+ except JSONDecodeError:
922
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
923
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
924
+
925
+ def get_span_stats(
926
+ self,
927
+ *,
928
+ project_id: typing.Optional[str] = None,
929
+ project_name: typing.Optional[str] = None,
930
+ trace_id: typing.Optional[str] = None,
931
+ type: typing.Optional[GetSpanStatsRequestType] = None,
932
+ filters: typing.Optional[str] = None,
933
+ from_time: typing.Optional[dt.datetime] = None,
934
+ to_time: typing.Optional[dt.datetime] = None,
935
+ request_options: typing.Optional[RequestOptions] = None,
936
+ ) -> HttpResponse[ProjectStatsPublic]:
937
+ """
938
+ Get span stats
939
+
940
+ Parameters
941
+ ----------
942
+ project_id : typing.Optional[str]
943
+
944
+ project_name : typing.Optional[str]
945
+
946
+ trace_id : typing.Optional[str]
947
+
948
+ type : typing.Optional[GetSpanStatsRequestType]
949
+
950
+ filters : typing.Optional[str]
951
+
952
+ from_time : typing.Optional[dt.datetime]
953
+
954
+ to_time : typing.Optional[dt.datetime]
955
+
956
+ request_options : typing.Optional[RequestOptions]
957
+ Request-specific configuration.
958
+
959
+ Returns
960
+ -------
961
+ HttpResponse[ProjectStatsPublic]
962
+ Span stats resource
963
+ """
964
+ _response = self._client_wrapper.httpx_client.request(
965
+ "v1/private/spans/stats",
966
+ method="GET",
967
+ params={
968
+ "project_id": project_id,
969
+ "project_name": project_name,
970
+ "trace_id": trace_id,
971
+ "type": type,
972
+ "filters": filters,
973
+ "from_time": serialize_datetime(from_time) if from_time is not None else None,
974
+ "to_time": serialize_datetime(to_time) if to_time is not None else None,
975
+ },
976
+ request_options=request_options,
977
+ )
978
+ try:
979
+ if 200 <= _response.status_code < 300:
980
+ _data = typing.cast(
981
+ ProjectStatsPublic,
982
+ parse_obj_as(
983
+ type_=ProjectStatsPublic, # type: ignore
984
+ object_=_response.json(),
985
+ ),
986
+ )
987
+ return HttpResponse(response=_response, data=_data)
988
+ _response_json = _response.json()
989
+ except JSONDecodeError:
990
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
991
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
992
+
993
+ def score_batch_of_spans(
994
+ self,
995
+ *,
996
+ scores: typing.Sequence[FeedbackScoreBatchItem],
997
+ request_options: typing.Optional[RequestOptions] = None,
998
+ ) -> HttpResponse[None]:
999
+ """
1000
+ Batch feedback scoring for spans
1001
+
1002
+ Parameters
1003
+ ----------
1004
+ scores : typing.Sequence[FeedbackScoreBatchItem]
1005
+
1006
+ request_options : typing.Optional[RequestOptions]
1007
+ Request-specific configuration.
1008
+
1009
+ Returns
1010
+ -------
1011
+ HttpResponse[None]
1012
+ """
1013
+ _response = self._client_wrapper.httpx_client.request(
1014
+ "v1/private/spans/feedback-scores",
1015
+ method="PUT",
1016
+ json={
1017
+ "scores": convert_and_respect_annotation_metadata(
1018
+ object_=scores, annotation=typing.Sequence[FeedbackScoreBatchItem], direction="write"
1019
+ ),
1020
+ },
1021
+ headers={
1022
+ "content-type": "application/json",
1023
+ },
1024
+ request_options=request_options,
1025
+ omit=OMIT,
1026
+ )
1027
+ try:
1028
+ if 200 <= _response.status_code < 300:
1029
+ return HttpResponse(response=_response, data=None)
1030
+ _response_json = _response.json()
1031
+ except JSONDecodeError:
1032
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
1033
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
1034
+
1035
+ @contextlib.contextmanager
1036
+ def search_spans(
1037
+ self,
1038
+ *,
1039
+ trace_id: typing.Optional[str] = OMIT,
1040
+ project_name: typing.Optional[str] = OMIT,
1041
+ project_id: typing.Optional[str] = OMIT,
1042
+ type: typing.Optional[SpanSearchStreamRequestPublicType] = OMIT,
1043
+ filters: typing.Optional[typing.Sequence[SpanFilterPublic]] = OMIT,
1044
+ limit: typing.Optional[int] = OMIT,
1045
+ last_retrieved_id: typing.Optional[str] = OMIT,
1046
+ truncate: typing.Optional[bool] = OMIT,
1047
+ from_time: typing.Optional[dt.datetime] = OMIT,
1048
+ to_time: typing.Optional[dt.datetime] = OMIT,
1049
+ request_options: typing.Optional[RequestOptions] = None,
1050
+ ) -> typing.Iterator[HttpResponse[typing.Iterator[bytes]]]:
1051
+ """
1052
+ Search spans
1053
+
1054
+ Parameters
1055
+ ----------
1056
+ trace_id : typing.Optional[str]
1057
+
1058
+ project_name : typing.Optional[str]
1059
+
1060
+ project_id : typing.Optional[str]
1061
+
1062
+ type : typing.Optional[SpanSearchStreamRequestPublicType]
1063
+
1064
+ filters : typing.Optional[typing.Sequence[SpanFilterPublic]]
1065
+
1066
+ limit : typing.Optional[int]
1067
+ Max number of spans to be streamed
1068
+
1069
+ last_retrieved_id : typing.Optional[str]
1070
+
1071
+ truncate : typing.Optional[bool]
1072
+ Truncate image included in either input, output or metadata
1073
+
1074
+ from_time : typing.Optional[dt.datetime]
1075
+ Filter spans created from this time (ISO-8601 format).
1076
+
1077
+ to_time : typing.Optional[dt.datetime]
1078
+ Filter spans created up to this time (ISO-8601 format). If not provided, defaults to current time. Must be after 'from_time'.
1079
+
1080
+ request_options : typing.Optional[RequestOptions]
1081
+ Request-specific configuration. You can pass in configuration such as `chunk_size`, and more to customize the request and response.
1082
+
1083
+ Returns
1084
+ -------
1085
+ typing.Iterator[HttpResponse[typing.Iterator[bytes]]]
1086
+ Spans stream or error during process
1087
+ """
1088
+ with self._client_wrapper.httpx_client.stream(
1089
+ "v1/private/spans/search",
1090
+ method="POST",
1091
+ json={
1092
+ "trace_id": trace_id,
1093
+ "project_name": project_name,
1094
+ "project_id": project_id,
1095
+ "type": type,
1096
+ "filters": convert_and_respect_annotation_metadata(
1097
+ object_=filters, annotation=typing.Sequence[SpanFilterPublic], direction="write"
1098
+ ),
1099
+ "limit": limit,
1100
+ "last_retrieved_id": last_retrieved_id,
1101
+ "truncate": truncate,
1102
+ "from_time": from_time,
1103
+ "to_time": to_time,
1104
+ },
1105
+ headers={
1106
+ "content-type": "application/json",
1107
+ },
1108
+ request_options=request_options,
1109
+ omit=OMIT,
1110
+ ) as _response:
1111
+
1112
+ def stream() -> HttpResponse[typing.Iterator[bytes]]:
1113
+ try:
1114
+ if 200 <= _response.status_code < 300:
1115
+ _chunk_size = request_options.get("chunk_size", None) if request_options is not None else None
1116
+ return HttpResponse(
1117
+ response=_response, data=(_chunk for _chunk in _response.iter_bytes(chunk_size=_chunk_size))
1118
+ )
1119
+ _response.read()
1120
+ if _response.status_code == 400:
1121
+ raise BadRequestError(
1122
+ headers=dict(_response.headers),
1123
+ body=typing.cast(
1124
+ typing.Optional[typing.Any],
1125
+ parse_obj_as(
1126
+ type_=typing.Optional[typing.Any], # type: ignore
1127
+ object_=_response.json(),
1128
+ ),
1129
+ ),
1130
+ )
1131
+ _response_json = _response.json()
1132
+ except JSONDecodeError:
1133
+ raise ApiError(
1134
+ status_code=_response.status_code, headers=dict(_response.headers), body=_response.text
1135
+ )
1136
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
1137
+
1138
+ yield stream()
1139
+
1140
+ def update_span_comment(
1141
+ self,
1142
+ comment_id: str,
1143
+ *,
1144
+ text: str,
1145
+ id: typing.Optional[str] = OMIT,
1146
+ created_at: typing.Optional[dt.datetime] = OMIT,
1147
+ last_updated_at: typing.Optional[dt.datetime] = OMIT,
1148
+ created_by: typing.Optional[str] = OMIT,
1149
+ last_updated_by: typing.Optional[str] = OMIT,
1150
+ request_options: typing.Optional[RequestOptions] = None,
1151
+ ) -> HttpResponse[None]:
1152
+ """
1153
+ Update span comment by id
1154
+
1155
+ Parameters
1156
+ ----------
1157
+ comment_id : str
1158
+
1159
+ text : str
1160
+
1161
+ id : typing.Optional[str]
1162
+
1163
+ created_at : typing.Optional[dt.datetime]
1164
+
1165
+ last_updated_at : typing.Optional[dt.datetime]
1166
+
1167
+ created_by : typing.Optional[str]
1168
+
1169
+ last_updated_by : typing.Optional[str]
1170
+
1171
+ request_options : typing.Optional[RequestOptions]
1172
+ Request-specific configuration.
1173
+
1174
+ Returns
1175
+ -------
1176
+ HttpResponse[None]
1177
+ """
1178
+ _response = self._client_wrapper.httpx_client.request(
1179
+ f"v1/private/spans/comments/{jsonable_encoder(comment_id)}",
1180
+ method="PATCH",
1181
+ json={
1182
+ "id": id,
1183
+ "text": text,
1184
+ "created_at": created_at,
1185
+ "last_updated_at": last_updated_at,
1186
+ "created_by": created_by,
1187
+ "last_updated_by": last_updated_by,
1188
+ },
1189
+ headers={
1190
+ "content-type": "application/json",
1191
+ },
1192
+ request_options=request_options,
1193
+ omit=OMIT,
1194
+ )
1195
+ try:
1196
+ if 200 <= _response.status_code < 300:
1197
+ return HttpResponse(response=_response, data=None)
1198
+ if _response.status_code == 404:
1199
+ raise NotFoundError(
1200
+ headers=dict(_response.headers),
1201
+ body=typing.cast(
1202
+ typing.Optional[typing.Any],
1203
+ parse_obj_as(
1204
+ type_=typing.Optional[typing.Any], # type: ignore
1205
+ object_=_response.json(),
1206
+ ),
1207
+ ),
1208
+ )
1209
+ _response_json = _response.json()
1210
+ except JSONDecodeError:
1211
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
1212
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
1213
+
1214
+
1215
+ class AsyncRawSpansClient:
1216
+ def __init__(self, *, client_wrapper: AsyncClientWrapper):
1217
+ self._client_wrapper = client_wrapper
1218
+
1219
+ async def add_span_comment(
1220
+ self,
1221
+ id_: str,
1222
+ *,
1223
+ text: str,
1224
+ id: typing.Optional[str] = OMIT,
1225
+ created_at: typing.Optional[dt.datetime] = OMIT,
1226
+ last_updated_at: typing.Optional[dt.datetime] = OMIT,
1227
+ created_by: typing.Optional[str] = OMIT,
1228
+ last_updated_by: typing.Optional[str] = OMIT,
1229
+ request_options: typing.Optional[RequestOptions] = None,
1230
+ ) -> AsyncHttpResponse[None]:
1231
+ """
1232
+ Add span comment
1233
+
1234
+ Parameters
1235
+ ----------
1236
+ id_ : str
1237
+
1238
+ text : str
1239
+
1240
+ id : typing.Optional[str]
1241
+
1242
+ created_at : typing.Optional[dt.datetime]
1243
+
1244
+ last_updated_at : typing.Optional[dt.datetime]
1245
+
1246
+ created_by : typing.Optional[str]
1247
+
1248
+ last_updated_by : typing.Optional[str]
1249
+
1250
+ request_options : typing.Optional[RequestOptions]
1251
+ Request-specific configuration.
1252
+
1253
+ Returns
1254
+ -------
1255
+ AsyncHttpResponse[None]
1256
+ """
1257
+ _response = await self._client_wrapper.httpx_client.request(
1258
+ f"v1/private/spans/{jsonable_encoder(id_)}/comments",
1259
+ method="POST",
1260
+ json={
1261
+ "id": id,
1262
+ "text": text,
1263
+ "created_at": created_at,
1264
+ "last_updated_at": last_updated_at,
1265
+ "created_by": created_by,
1266
+ "last_updated_by": last_updated_by,
1267
+ },
1268
+ headers={
1269
+ "content-type": "application/json",
1270
+ },
1271
+ request_options=request_options,
1272
+ omit=OMIT,
1273
+ )
1274
+ try:
1275
+ if 200 <= _response.status_code < 300:
1276
+ return AsyncHttpResponse(response=_response, data=None)
1277
+ _response_json = _response.json()
1278
+ except JSONDecodeError:
1279
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
1280
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
1281
+
1282
+ async def add_span_feedback_score(
1283
+ self,
1284
+ id: str,
1285
+ *,
1286
+ name: str,
1287
+ value: float,
1288
+ source: FeedbackScoreSource,
1289
+ category_name: typing.Optional[str] = OMIT,
1290
+ reason: typing.Optional[str] = OMIT,
1291
+ created_at: typing.Optional[dt.datetime] = OMIT,
1292
+ last_updated_at: typing.Optional[dt.datetime] = OMIT,
1293
+ created_by: typing.Optional[str] = OMIT,
1294
+ last_updated_by: typing.Optional[str] = OMIT,
1295
+ value_by_author: typing.Optional[typing.Dict[str, ValueEntry]] = OMIT,
1296
+ request_options: typing.Optional[RequestOptions] = None,
1297
+ ) -> AsyncHttpResponse[None]:
1298
+ """
1299
+ Add span feedback score
1300
+
1301
+ Parameters
1302
+ ----------
1303
+ id : str
1304
+
1305
+ name : str
1306
+
1307
+ value : float
1308
+
1309
+ source : FeedbackScoreSource
1310
+
1311
+ category_name : typing.Optional[str]
1312
+
1313
+ reason : typing.Optional[str]
1314
+
1315
+ created_at : typing.Optional[dt.datetime]
1316
+
1317
+ last_updated_at : typing.Optional[dt.datetime]
1318
+
1319
+ created_by : typing.Optional[str]
1320
+
1321
+ last_updated_by : typing.Optional[str]
1322
+
1323
+ value_by_author : typing.Optional[typing.Dict[str, ValueEntry]]
1324
+
1325
+ request_options : typing.Optional[RequestOptions]
1326
+ Request-specific configuration.
1327
+
1328
+ Returns
1329
+ -------
1330
+ AsyncHttpResponse[None]
1331
+ """
1332
+ _response = await self._client_wrapper.httpx_client.request(
1333
+ f"v1/private/spans/{jsonable_encoder(id)}/feedback-scores",
1334
+ method="PUT",
1335
+ json={
1336
+ "name": name,
1337
+ "category_name": category_name,
1338
+ "value": value,
1339
+ "reason": reason,
1340
+ "source": source,
1341
+ "created_at": created_at,
1342
+ "last_updated_at": last_updated_at,
1343
+ "created_by": created_by,
1344
+ "last_updated_by": last_updated_by,
1345
+ "value_by_author": convert_and_respect_annotation_metadata(
1346
+ object_=value_by_author, annotation=typing.Dict[str, ValueEntry], direction="write"
1347
+ ),
1348
+ },
1349
+ headers={
1350
+ "content-type": "application/json",
1351
+ },
1352
+ request_options=request_options,
1353
+ omit=OMIT,
1354
+ )
1355
+ try:
1356
+ if 200 <= _response.status_code < 300:
1357
+ return AsyncHttpResponse(response=_response, data=None)
1358
+ _response_json = _response.json()
1359
+ except JSONDecodeError:
1360
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
1361
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
1362
+
1363
+ async def create_spans(
1364
+ self, *, spans: typing.Sequence[SpanWrite], request_options: typing.Optional[RequestOptions] = None
1365
+ ) -> AsyncHttpResponse[None]:
1366
+ """
1367
+ Create spans
1368
+
1369
+ Parameters
1370
+ ----------
1371
+ spans : typing.Sequence[SpanWrite]
1372
+
1373
+ request_options : typing.Optional[RequestOptions]
1374
+ Request-specific configuration.
1375
+
1376
+ Returns
1377
+ -------
1378
+ AsyncHttpResponse[None]
1379
+ """
1380
+ _response = await self._client_wrapper.httpx_client.request(
1381
+ "v1/private/spans/batch",
1382
+ method="POST",
1383
+ json={
1384
+ "spans": convert_and_respect_annotation_metadata(
1385
+ object_=spans, annotation=typing.Sequence[SpanWrite], direction="write"
1386
+ ),
1387
+ },
1388
+ headers={
1389
+ "content-type": "application/json",
1390
+ },
1391
+ request_options=request_options,
1392
+ omit=OMIT,
1393
+ )
1394
+ try:
1395
+ if 200 <= _response.status_code < 300:
1396
+ return AsyncHttpResponse(response=_response, data=None)
1397
+ _response_json = _response.json()
1398
+ except JSONDecodeError:
1399
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
1400
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
1401
+
1402
+ async def batch_update_spans(
1403
+ self,
1404
+ *,
1405
+ ids: typing.Sequence[str],
1406
+ update: SpanUpdate,
1407
+ merge_tags: typing.Optional[bool] = OMIT,
1408
+ request_options: typing.Optional[RequestOptions] = None,
1409
+ ) -> AsyncHttpResponse[None]:
1410
+ """
1411
+ Update multiple spans
1412
+
1413
+ Parameters
1414
+ ----------
1415
+ ids : typing.Sequence[str]
1416
+ List of span IDs to update (max 1000)
1417
+
1418
+ update : SpanUpdate
1419
+
1420
+ merge_tags : typing.Optional[bool]
1421
+ If true, merge tags with existing tags instead of replacing them. Default: false
1422
+
1423
+ request_options : typing.Optional[RequestOptions]
1424
+ Request-specific configuration.
1425
+
1426
+ Returns
1427
+ -------
1428
+ AsyncHttpResponse[None]
1429
+ """
1430
+ _response = await self._client_wrapper.httpx_client.request(
1431
+ "v1/private/spans/batch",
1432
+ method="PATCH",
1433
+ json={
1434
+ "ids": ids,
1435
+ "update": convert_and_respect_annotation_metadata(
1436
+ object_=update, annotation=SpanUpdate, direction="write"
1437
+ ),
1438
+ "merge_tags": merge_tags,
1439
+ },
1440
+ headers={
1441
+ "content-type": "application/json",
1442
+ },
1443
+ request_options=request_options,
1444
+ omit=OMIT,
1445
+ )
1446
+ try:
1447
+ if 200 <= _response.status_code < 300:
1448
+ return AsyncHttpResponse(response=_response, data=None)
1449
+ if _response.status_code == 400:
1450
+ raise BadRequestError(
1451
+ headers=dict(_response.headers),
1452
+ body=typing.cast(
1453
+ typing.Optional[typing.Any],
1454
+ parse_obj_as(
1455
+ type_=typing.Optional[typing.Any], # type: ignore
1456
+ object_=_response.json(),
1457
+ ),
1458
+ ),
1459
+ )
1460
+ _response_json = _response.json()
1461
+ except JSONDecodeError:
1462
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
1463
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
1464
+
1465
+ async def get_spans_by_project(
1466
+ self,
1467
+ *,
1468
+ page: typing.Optional[int] = None,
1469
+ size: typing.Optional[int] = None,
1470
+ project_name: typing.Optional[str] = None,
1471
+ project_id: typing.Optional[str] = None,
1472
+ trace_id: typing.Optional[str] = None,
1473
+ type: typing.Optional[GetSpansByProjectRequestType] = None,
1474
+ filters: typing.Optional[str] = None,
1475
+ truncate: typing.Optional[bool] = None,
1476
+ strip_attachments: typing.Optional[bool] = None,
1477
+ sorting: typing.Optional[str] = None,
1478
+ exclude: typing.Optional[str] = None,
1479
+ from_time: typing.Optional[dt.datetime] = None,
1480
+ to_time: typing.Optional[dt.datetime] = None,
1481
+ request_options: typing.Optional[RequestOptions] = None,
1482
+ ) -> AsyncHttpResponse[SpanPagePublic]:
1483
+ """
1484
+ Get spans by project_name or project_id and optionally by trace_id and/or type
1485
+
1486
+ Parameters
1487
+ ----------
1488
+ page : typing.Optional[int]
1489
+
1490
+ size : typing.Optional[int]
1491
+
1492
+ project_name : typing.Optional[str]
1493
+
1494
+ project_id : typing.Optional[str]
1495
+
1496
+ trace_id : typing.Optional[str]
1497
+
1498
+ type : typing.Optional[GetSpansByProjectRequestType]
1499
+
1500
+ filters : typing.Optional[str]
1501
+
1502
+ truncate : typing.Optional[bool]
1503
+
1504
+ strip_attachments : typing.Optional[bool]
1505
+
1506
+ sorting : typing.Optional[str]
1507
+
1508
+ exclude : typing.Optional[str]
1509
+
1510
+ from_time : typing.Optional[dt.datetime]
1511
+
1512
+ to_time : typing.Optional[dt.datetime]
1513
+
1514
+ request_options : typing.Optional[RequestOptions]
1515
+ Request-specific configuration.
1516
+
1517
+ Returns
1518
+ -------
1519
+ AsyncHttpResponse[SpanPagePublic]
1520
+ Spans resource
1521
+ """
1522
+ _response = await self._client_wrapper.httpx_client.request(
1523
+ "v1/private/spans",
1524
+ method="GET",
1525
+ params={
1526
+ "page": page,
1527
+ "size": size,
1528
+ "project_name": project_name,
1529
+ "project_id": project_id,
1530
+ "trace_id": trace_id,
1531
+ "type": type,
1532
+ "filters": filters,
1533
+ "truncate": truncate,
1534
+ "strip_attachments": strip_attachments,
1535
+ "sorting": sorting,
1536
+ "exclude": exclude,
1537
+ "from_time": serialize_datetime(from_time) if from_time is not None else None,
1538
+ "to_time": serialize_datetime(to_time) if to_time is not None else None,
1539
+ },
1540
+ request_options=request_options,
1541
+ )
1542
+ try:
1543
+ if 200 <= _response.status_code < 300:
1544
+ _data = typing.cast(
1545
+ SpanPagePublic,
1546
+ parse_obj_as(
1547
+ type_=SpanPagePublic, # type: ignore
1548
+ object_=_response.json(),
1549
+ ),
1550
+ )
1551
+ return AsyncHttpResponse(response=_response, data=_data)
1552
+ _response_json = _response.json()
1553
+ except JSONDecodeError:
1554
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
1555
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
1556
+
1557
+ async def create_span(
1558
+ self,
1559
+ *,
1560
+ start_time: dt.datetime,
1561
+ id: typing.Optional[str] = OMIT,
1562
+ project_name: typing.Optional[str] = OMIT,
1563
+ trace_id: typing.Optional[str] = OMIT,
1564
+ parent_span_id: typing.Optional[str] = OMIT,
1565
+ name: typing.Optional[str] = OMIT,
1566
+ type: typing.Optional[SpanWriteType] = OMIT,
1567
+ end_time: typing.Optional[dt.datetime] = OMIT,
1568
+ input: typing.Optional[JsonListStringWrite] = OMIT,
1569
+ output: typing.Optional[JsonListStringWrite] = OMIT,
1570
+ metadata: typing.Optional[JsonListStringWrite] = OMIT,
1571
+ model: typing.Optional[str] = OMIT,
1572
+ provider: typing.Optional[str] = OMIT,
1573
+ tags: typing.Optional[typing.Sequence[str]] = OMIT,
1574
+ usage: typing.Optional[typing.Dict[str, int]] = OMIT,
1575
+ error_info: typing.Optional[ErrorInfoWrite] = OMIT,
1576
+ last_updated_at: typing.Optional[dt.datetime] = OMIT,
1577
+ total_estimated_cost: typing.Optional[float] = OMIT,
1578
+ total_estimated_cost_version: typing.Optional[str] = OMIT,
1579
+ request_options: typing.Optional[RequestOptions] = None,
1580
+ ) -> AsyncHttpResponse[None]:
1581
+ """
1582
+ Create span
1583
+
1584
+ Parameters
1585
+ ----------
1586
+ start_time : dt.datetime
1587
+
1588
+ id : typing.Optional[str]
1589
+
1590
+ project_name : typing.Optional[str]
1591
+ If null, the default project is used
1592
+
1593
+ trace_id : typing.Optional[str]
1594
+
1595
+ parent_span_id : typing.Optional[str]
1596
+
1597
+ name : typing.Optional[str]
1598
+
1599
+ type : typing.Optional[SpanWriteType]
1600
+
1601
+ end_time : typing.Optional[dt.datetime]
1602
+
1603
+ input : typing.Optional[JsonListStringWrite]
1604
+
1605
+ output : typing.Optional[JsonListStringWrite]
1606
+
1607
+ metadata : typing.Optional[JsonListStringWrite]
1608
+
1609
+ model : typing.Optional[str]
1610
+
1611
+ provider : typing.Optional[str]
1612
+
1613
+ tags : typing.Optional[typing.Sequence[str]]
1614
+
1615
+ usage : typing.Optional[typing.Dict[str, int]]
1616
+
1617
+ error_info : typing.Optional[ErrorInfoWrite]
1618
+
1619
+ last_updated_at : typing.Optional[dt.datetime]
1620
+
1621
+ total_estimated_cost : typing.Optional[float]
1622
+
1623
+ total_estimated_cost_version : typing.Optional[str]
1624
+
1625
+ request_options : typing.Optional[RequestOptions]
1626
+ Request-specific configuration.
1627
+
1628
+ Returns
1629
+ -------
1630
+ AsyncHttpResponse[None]
1631
+ """
1632
+ _response = await self._client_wrapper.httpx_client.request(
1633
+ "v1/private/spans",
1634
+ method="POST",
1635
+ json={
1636
+ "id": id,
1637
+ "project_name": project_name,
1638
+ "trace_id": trace_id,
1639
+ "parent_span_id": parent_span_id,
1640
+ "name": name,
1641
+ "type": type,
1642
+ "start_time": start_time,
1643
+ "end_time": end_time,
1644
+ "input": convert_and_respect_annotation_metadata(
1645
+ object_=input, annotation=JsonListStringWrite, direction="write"
1646
+ ),
1647
+ "output": convert_and_respect_annotation_metadata(
1648
+ object_=output, annotation=JsonListStringWrite, direction="write"
1649
+ ),
1650
+ "metadata": convert_and_respect_annotation_metadata(
1651
+ object_=metadata, annotation=JsonListStringWrite, direction="write"
1652
+ ),
1653
+ "model": model,
1654
+ "provider": provider,
1655
+ "tags": tags,
1656
+ "usage": usage,
1657
+ "error_info": convert_and_respect_annotation_metadata(
1658
+ object_=error_info, annotation=ErrorInfoWrite, direction="write"
1659
+ ),
1660
+ "last_updated_at": last_updated_at,
1661
+ "total_estimated_cost": total_estimated_cost,
1662
+ "total_estimated_cost_version": total_estimated_cost_version,
1663
+ },
1664
+ headers={
1665
+ "content-type": "application/json",
1666
+ },
1667
+ request_options=request_options,
1668
+ omit=OMIT,
1669
+ )
1670
+ try:
1671
+ if 200 <= _response.status_code < 300:
1672
+ return AsyncHttpResponse(response=_response, data=None)
1673
+ if _response.status_code == 409:
1674
+ raise ConflictError(
1675
+ headers=dict(_response.headers),
1676
+ body=typing.cast(
1677
+ typing.Optional[typing.Any],
1678
+ parse_obj_as(
1679
+ type_=typing.Optional[typing.Any], # type: ignore
1680
+ object_=_response.json(),
1681
+ ),
1682
+ ),
1683
+ )
1684
+ _response_json = _response.json()
1685
+ except JSONDecodeError:
1686
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
1687
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
1688
+
1689
+ async def get_span_by_id(
1690
+ self,
1691
+ id: str,
1692
+ *,
1693
+ strip_attachments: typing.Optional[bool] = None,
1694
+ request_options: typing.Optional[RequestOptions] = None,
1695
+ ) -> AsyncHttpResponse[SpanPublic]:
1696
+ """
1697
+ Get span by id
1698
+
1699
+ Parameters
1700
+ ----------
1701
+ id : str
1702
+
1703
+ strip_attachments : typing.Optional[bool]
1704
+
1705
+ request_options : typing.Optional[RequestOptions]
1706
+ Request-specific configuration.
1707
+
1708
+ Returns
1709
+ -------
1710
+ AsyncHttpResponse[SpanPublic]
1711
+ Span resource
1712
+ """
1713
+ _response = await self._client_wrapper.httpx_client.request(
1714
+ f"v1/private/spans/{jsonable_encoder(id)}",
1715
+ method="GET",
1716
+ params={
1717
+ "strip_attachments": strip_attachments,
1718
+ },
1719
+ request_options=request_options,
1720
+ )
1721
+ try:
1722
+ if 200 <= _response.status_code < 300:
1723
+ _data = typing.cast(
1724
+ SpanPublic,
1725
+ parse_obj_as(
1726
+ type_=SpanPublic, # type: ignore
1727
+ object_=_response.json(),
1728
+ ),
1729
+ )
1730
+ return AsyncHttpResponse(response=_response, data=_data)
1731
+ if _response.status_code == 404:
1732
+ raise NotFoundError(
1733
+ headers=dict(_response.headers),
1734
+ body=typing.cast(
1735
+ typing.Optional[typing.Any],
1736
+ parse_obj_as(
1737
+ type_=typing.Optional[typing.Any], # type: ignore
1738
+ object_=_response.json(),
1739
+ ),
1740
+ ),
1741
+ )
1742
+ _response_json = _response.json()
1743
+ except JSONDecodeError:
1744
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
1745
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
1746
+
1747
+ async def delete_span_by_id(
1748
+ self, id: str, *, request_options: typing.Optional[RequestOptions] = None
1749
+ ) -> AsyncHttpResponse[None]:
1750
+ """
1751
+ Delete span by id
1752
+
1753
+ Parameters
1754
+ ----------
1755
+ id : str
1756
+
1757
+ request_options : typing.Optional[RequestOptions]
1758
+ Request-specific configuration.
1759
+
1760
+ Returns
1761
+ -------
1762
+ AsyncHttpResponse[None]
1763
+ """
1764
+ _response = await self._client_wrapper.httpx_client.request(
1765
+ f"v1/private/spans/{jsonable_encoder(id)}",
1766
+ method="DELETE",
1767
+ request_options=request_options,
1768
+ )
1769
+ try:
1770
+ if 200 <= _response.status_code < 300:
1771
+ return AsyncHttpResponse(response=_response, data=None)
1772
+ if _response.status_code == 501:
1773
+ raise NotImplementedError(
1774
+ headers=dict(_response.headers),
1775
+ body=typing.cast(
1776
+ typing.Optional[typing.Any],
1777
+ parse_obj_as(
1778
+ type_=typing.Optional[typing.Any], # type: ignore
1779
+ object_=_response.json(),
1780
+ ),
1781
+ ),
1782
+ )
1783
+ _response_json = _response.json()
1784
+ except JSONDecodeError:
1785
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
1786
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
1787
+
1788
+ async def update_span(
1789
+ self,
1790
+ id: str,
1791
+ *,
1792
+ trace_id: str,
1793
+ project_name: typing.Optional[str] = OMIT,
1794
+ project_id: typing.Optional[str] = OMIT,
1795
+ parent_span_id: typing.Optional[str] = OMIT,
1796
+ name: typing.Optional[str] = OMIT,
1797
+ type: typing.Optional[SpanUpdateType] = OMIT,
1798
+ end_time: typing.Optional[dt.datetime] = OMIT,
1799
+ input: typing.Optional[JsonListString] = OMIT,
1800
+ output: typing.Optional[JsonListString] = OMIT,
1801
+ metadata: typing.Optional[JsonListString] = OMIT,
1802
+ model: typing.Optional[str] = OMIT,
1803
+ provider: typing.Optional[str] = OMIT,
1804
+ tags: typing.Optional[typing.Sequence[str]] = OMIT,
1805
+ usage: typing.Optional[typing.Dict[str, int]] = OMIT,
1806
+ total_estimated_cost: typing.Optional[float] = OMIT,
1807
+ error_info: typing.Optional[ErrorInfo] = OMIT,
1808
+ request_options: typing.Optional[RequestOptions] = None,
1809
+ ) -> AsyncHttpResponse[None]:
1810
+ """
1811
+ Update span by id
1812
+
1813
+ Parameters
1814
+ ----------
1815
+ id : str
1816
+
1817
+ trace_id : str
1818
+
1819
+ project_name : typing.Optional[str]
1820
+ If null and project_id not specified, Default Project is assumed
1821
+
1822
+ project_id : typing.Optional[str]
1823
+ If null and project_name not specified, Default Project is assumed
1824
+
1825
+ parent_span_id : typing.Optional[str]
1826
+
1827
+ name : typing.Optional[str]
1828
+
1829
+ type : typing.Optional[SpanUpdateType]
1830
+
1831
+ end_time : typing.Optional[dt.datetime]
1832
+
1833
+ input : typing.Optional[JsonListString]
1834
+
1835
+ output : typing.Optional[JsonListString]
1836
+
1837
+ metadata : typing.Optional[JsonListString]
1838
+
1839
+ model : typing.Optional[str]
1840
+
1841
+ provider : typing.Optional[str]
1842
+
1843
+ tags : typing.Optional[typing.Sequence[str]]
1844
+
1845
+ usage : typing.Optional[typing.Dict[str, int]]
1846
+
1847
+ total_estimated_cost : typing.Optional[float]
1848
+
1849
+ error_info : typing.Optional[ErrorInfo]
1850
+
1851
+ request_options : typing.Optional[RequestOptions]
1852
+ Request-specific configuration.
1853
+
1854
+ Returns
1855
+ -------
1856
+ AsyncHttpResponse[None]
1857
+ """
1858
+ _response = await self._client_wrapper.httpx_client.request(
1859
+ f"v1/private/spans/{jsonable_encoder(id)}",
1860
+ method="PATCH",
1861
+ json={
1862
+ "project_name": project_name,
1863
+ "project_id": project_id,
1864
+ "trace_id": trace_id,
1865
+ "parent_span_id": parent_span_id,
1866
+ "name": name,
1867
+ "type": type,
1868
+ "end_time": end_time,
1869
+ "input": convert_and_respect_annotation_metadata(
1870
+ object_=input, annotation=JsonListString, direction="write"
1871
+ ),
1872
+ "output": convert_and_respect_annotation_metadata(
1873
+ object_=output, annotation=JsonListString, direction="write"
1874
+ ),
1875
+ "metadata": convert_and_respect_annotation_metadata(
1876
+ object_=metadata, annotation=JsonListString, direction="write"
1877
+ ),
1878
+ "model": model,
1879
+ "provider": provider,
1880
+ "tags": tags,
1881
+ "usage": usage,
1882
+ "total_estimated_cost": total_estimated_cost,
1883
+ "error_info": convert_and_respect_annotation_metadata(
1884
+ object_=error_info, annotation=ErrorInfo, direction="write"
1885
+ ),
1886
+ },
1887
+ headers={
1888
+ "content-type": "application/json",
1889
+ },
1890
+ request_options=request_options,
1891
+ omit=OMIT,
1892
+ )
1893
+ try:
1894
+ if 200 <= _response.status_code < 300:
1895
+ return AsyncHttpResponse(response=_response, data=None)
1896
+ if _response.status_code == 404:
1897
+ raise NotFoundError(
1898
+ headers=dict(_response.headers),
1899
+ body=typing.cast(
1900
+ typing.Optional[typing.Any],
1901
+ parse_obj_as(
1902
+ type_=typing.Optional[typing.Any], # type: ignore
1903
+ object_=_response.json(),
1904
+ ),
1905
+ ),
1906
+ )
1907
+ _response_json = _response.json()
1908
+ except JSONDecodeError:
1909
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
1910
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
1911
+
1912
+ async def delete_span_comments(
1913
+ self, *, ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
1914
+ ) -> AsyncHttpResponse[None]:
1915
+ """
1916
+ Delete span comments
1917
+
1918
+ Parameters
1919
+ ----------
1920
+ ids : typing.Sequence[str]
1921
+
1922
+ request_options : typing.Optional[RequestOptions]
1923
+ Request-specific configuration.
1924
+
1925
+ Returns
1926
+ -------
1927
+ AsyncHttpResponse[None]
1928
+ """
1929
+ _response = await self._client_wrapper.httpx_client.request(
1930
+ "v1/private/spans/comments/delete",
1931
+ method="POST",
1932
+ json={
1933
+ "ids": ids,
1934
+ },
1935
+ headers={
1936
+ "content-type": "application/json",
1937
+ },
1938
+ request_options=request_options,
1939
+ omit=OMIT,
1940
+ )
1941
+ try:
1942
+ if 200 <= _response.status_code < 300:
1943
+ return AsyncHttpResponse(response=_response, data=None)
1944
+ _response_json = _response.json()
1945
+ except JSONDecodeError:
1946
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
1947
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
1948
+
1949
+ async def delete_span_feedback_score(
1950
+ self,
1951
+ id: str,
1952
+ *,
1953
+ name: str,
1954
+ author: typing.Optional[str] = OMIT,
1955
+ request_options: typing.Optional[RequestOptions] = None,
1956
+ ) -> AsyncHttpResponse[None]:
1957
+ """
1958
+ Delete span feedback score
1959
+
1960
+ Parameters
1961
+ ----------
1962
+ id : str
1963
+
1964
+ name : str
1965
+
1966
+ author : typing.Optional[str]
1967
+
1968
+ request_options : typing.Optional[RequestOptions]
1969
+ Request-specific configuration.
1970
+
1971
+ Returns
1972
+ -------
1973
+ AsyncHttpResponse[None]
1974
+ """
1975
+ _response = await self._client_wrapper.httpx_client.request(
1976
+ f"v1/private/spans/{jsonable_encoder(id)}/feedback-scores/delete",
1977
+ method="POST",
1978
+ json={
1979
+ "name": name,
1980
+ "author": author,
1981
+ },
1982
+ headers={
1983
+ "content-type": "application/json",
1984
+ },
1985
+ request_options=request_options,
1986
+ omit=OMIT,
1987
+ )
1988
+ try:
1989
+ if 200 <= _response.status_code < 300:
1990
+ return AsyncHttpResponse(response=_response, data=None)
1991
+ _response_json = _response.json()
1992
+ except JSONDecodeError:
1993
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
1994
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
1995
+
1996
+ async def find_feedback_score_names_1(
1997
+ self,
1998
+ *,
1999
+ project_id: typing.Optional[str] = None,
2000
+ type: typing.Optional[FindFeedbackScoreNames1RequestType] = None,
2001
+ request_options: typing.Optional[RequestOptions] = None,
2002
+ ) -> AsyncHttpResponse[typing.List[str]]:
2003
+ """
2004
+ Find Feedback Score names
2005
+
2006
+ Parameters
2007
+ ----------
2008
+ project_id : typing.Optional[str]
2009
+
2010
+ type : typing.Optional[FindFeedbackScoreNames1RequestType]
2011
+
2012
+ request_options : typing.Optional[RequestOptions]
2013
+ Request-specific configuration.
2014
+
2015
+ Returns
2016
+ -------
2017
+ AsyncHttpResponse[typing.List[str]]
2018
+ Feedback Scores resource
2019
+ """
2020
+ _response = await self._client_wrapper.httpx_client.request(
2021
+ "v1/private/spans/feedback-scores/names",
2022
+ method="GET",
2023
+ params={
2024
+ "project_id": project_id,
2025
+ "type": type,
2026
+ },
2027
+ request_options=request_options,
2028
+ )
2029
+ try:
2030
+ if 200 <= _response.status_code < 300:
2031
+ _data = typing.cast(
2032
+ typing.List[str],
2033
+ parse_obj_as(
2034
+ type_=typing.List[str], # type: ignore
2035
+ object_=_response.json(),
2036
+ ),
2037
+ )
2038
+ return AsyncHttpResponse(response=_response, data=_data)
2039
+ _response_json = _response.json()
2040
+ except JSONDecodeError:
2041
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
2042
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
2043
+
2044
+ async def get_span_comment(
2045
+ self, comment_id: str, span_id: str, *, request_options: typing.Optional[RequestOptions] = None
2046
+ ) -> AsyncHttpResponse[Comment]:
2047
+ """
2048
+ Get span comment
2049
+
2050
+ Parameters
2051
+ ----------
2052
+ comment_id : str
2053
+
2054
+ span_id : str
2055
+
2056
+ request_options : typing.Optional[RequestOptions]
2057
+ Request-specific configuration.
2058
+
2059
+ Returns
2060
+ -------
2061
+ AsyncHttpResponse[Comment]
2062
+ Comment resource
2063
+ """
2064
+ _response = await self._client_wrapper.httpx_client.request(
2065
+ f"v1/private/spans/{jsonable_encoder(span_id)}/comments/{jsonable_encoder(comment_id)}",
2066
+ method="GET",
2067
+ request_options=request_options,
2068
+ )
2069
+ try:
2070
+ if 200 <= _response.status_code < 300:
2071
+ _data = typing.cast(
2072
+ Comment,
2073
+ parse_obj_as(
2074
+ type_=Comment, # type: ignore
2075
+ object_=_response.json(),
2076
+ ),
2077
+ )
2078
+ return AsyncHttpResponse(response=_response, data=_data)
2079
+ if _response.status_code == 404:
2080
+ raise NotFoundError(
2081
+ headers=dict(_response.headers),
2082
+ body=typing.cast(
2083
+ typing.Optional[typing.Any],
2084
+ parse_obj_as(
2085
+ type_=typing.Optional[typing.Any], # type: ignore
2086
+ object_=_response.json(),
2087
+ ),
2088
+ ),
2089
+ )
2090
+ _response_json = _response.json()
2091
+ except JSONDecodeError:
2092
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
2093
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
2094
+
2095
+ async def get_span_stats(
2096
+ self,
2097
+ *,
2098
+ project_id: typing.Optional[str] = None,
2099
+ project_name: typing.Optional[str] = None,
2100
+ trace_id: typing.Optional[str] = None,
2101
+ type: typing.Optional[GetSpanStatsRequestType] = None,
2102
+ filters: typing.Optional[str] = None,
2103
+ from_time: typing.Optional[dt.datetime] = None,
2104
+ to_time: typing.Optional[dt.datetime] = None,
2105
+ request_options: typing.Optional[RequestOptions] = None,
2106
+ ) -> AsyncHttpResponse[ProjectStatsPublic]:
2107
+ """
2108
+ Get span stats
2109
+
2110
+ Parameters
2111
+ ----------
2112
+ project_id : typing.Optional[str]
2113
+
2114
+ project_name : typing.Optional[str]
2115
+
2116
+ trace_id : typing.Optional[str]
2117
+
2118
+ type : typing.Optional[GetSpanStatsRequestType]
2119
+
2120
+ filters : typing.Optional[str]
2121
+
2122
+ from_time : typing.Optional[dt.datetime]
2123
+
2124
+ to_time : typing.Optional[dt.datetime]
2125
+
2126
+ request_options : typing.Optional[RequestOptions]
2127
+ Request-specific configuration.
2128
+
2129
+ Returns
2130
+ -------
2131
+ AsyncHttpResponse[ProjectStatsPublic]
2132
+ Span stats resource
2133
+ """
2134
+ _response = await self._client_wrapper.httpx_client.request(
2135
+ "v1/private/spans/stats",
2136
+ method="GET",
2137
+ params={
2138
+ "project_id": project_id,
2139
+ "project_name": project_name,
2140
+ "trace_id": trace_id,
2141
+ "type": type,
2142
+ "filters": filters,
2143
+ "from_time": serialize_datetime(from_time) if from_time is not None else None,
2144
+ "to_time": serialize_datetime(to_time) if to_time is not None else None,
2145
+ },
2146
+ request_options=request_options,
2147
+ )
2148
+ try:
2149
+ if 200 <= _response.status_code < 300:
2150
+ _data = typing.cast(
2151
+ ProjectStatsPublic,
2152
+ parse_obj_as(
2153
+ type_=ProjectStatsPublic, # type: ignore
2154
+ object_=_response.json(),
2155
+ ),
2156
+ )
2157
+ return AsyncHttpResponse(response=_response, data=_data)
2158
+ _response_json = _response.json()
2159
+ except JSONDecodeError:
2160
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
2161
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
2162
+
2163
+ async def score_batch_of_spans(
2164
+ self,
2165
+ *,
2166
+ scores: typing.Sequence[FeedbackScoreBatchItem],
2167
+ request_options: typing.Optional[RequestOptions] = None,
2168
+ ) -> AsyncHttpResponse[None]:
2169
+ """
2170
+ Batch feedback scoring for spans
2171
+
2172
+ Parameters
2173
+ ----------
2174
+ scores : typing.Sequence[FeedbackScoreBatchItem]
2175
+
2176
+ request_options : typing.Optional[RequestOptions]
2177
+ Request-specific configuration.
2178
+
2179
+ Returns
2180
+ -------
2181
+ AsyncHttpResponse[None]
2182
+ """
2183
+ _response = await self._client_wrapper.httpx_client.request(
2184
+ "v1/private/spans/feedback-scores",
2185
+ method="PUT",
2186
+ json={
2187
+ "scores": convert_and_respect_annotation_metadata(
2188
+ object_=scores, annotation=typing.Sequence[FeedbackScoreBatchItem], direction="write"
2189
+ ),
2190
+ },
2191
+ headers={
2192
+ "content-type": "application/json",
2193
+ },
2194
+ request_options=request_options,
2195
+ omit=OMIT,
2196
+ )
2197
+ try:
2198
+ if 200 <= _response.status_code < 300:
2199
+ return AsyncHttpResponse(response=_response, data=None)
2200
+ _response_json = _response.json()
2201
+ except JSONDecodeError:
2202
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
2203
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
2204
+
2205
+ @contextlib.asynccontextmanager
2206
+ async def search_spans(
2207
+ self,
2208
+ *,
2209
+ trace_id: typing.Optional[str] = OMIT,
2210
+ project_name: typing.Optional[str] = OMIT,
2211
+ project_id: typing.Optional[str] = OMIT,
2212
+ type: typing.Optional[SpanSearchStreamRequestPublicType] = OMIT,
2213
+ filters: typing.Optional[typing.Sequence[SpanFilterPublic]] = OMIT,
2214
+ limit: typing.Optional[int] = OMIT,
2215
+ last_retrieved_id: typing.Optional[str] = OMIT,
2216
+ truncate: typing.Optional[bool] = OMIT,
2217
+ from_time: typing.Optional[dt.datetime] = OMIT,
2218
+ to_time: typing.Optional[dt.datetime] = OMIT,
2219
+ request_options: typing.Optional[RequestOptions] = None,
2220
+ ) -> typing.AsyncIterator[AsyncHttpResponse[typing.AsyncIterator[bytes]]]:
2221
+ """
2222
+ Search spans
2223
+
2224
+ Parameters
2225
+ ----------
2226
+ trace_id : typing.Optional[str]
2227
+
2228
+ project_name : typing.Optional[str]
2229
+
2230
+ project_id : typing.Optional[str]
2231
+
2232
+ type : typing.Optional[SpanSearchStreamRequestPublicType]
2233
+
2234
+ filters : typing.Optional[typing.Sequence[SpanFilterPublic]]
2235
+
2236
+ limit : typing.Optional[int]
2237
+ Max number of spans to be streamed
2238
+
2239
+ last_retrieved_id : typing.Optional[str]
2240
+
2241
+ truncate : typing.Optional[bool]
2242
+ Truncate image included in either input, output or metadata
2243
+
2244
+ from_time : typing.Optional[dt.datetime]
2245
+ Filter spans created from this time (ISO-8601 format).
2246
+
2247
+ to_time : typing.Optional[dt.datetime]
2248
+ Filter spans created up to this time (ISO-8601 format). If not provided, defaults to current time. Must be after 'from_time'.
2249
+
2250
+ request_options : typing.Optional[RequestOptions]
2251
+ Request-specific configuration. You can pass in configuration such as `chunk_size`, and more to customize the request and response.
2252
+
2253
+ Returns
2254
+ -------
2255
+ typing.AsyncIterator[AsyncHttpResponse[typing.AsyncIterator[bytes]]]
2256
+ Spans stream or error during process
2257
+ """
2258
+ async with self._client_wrapper.httpx_client.stream(
2259
+ "v1/private/spans/search",
2260
+ method="POST",
2261
+ json={
2262
+ "trace_id": trace_id,
2263
+ "project_name": project_name,
2264
+ "project_id": project_id,
2265
+ "type": type,
2266
+ "filters": convert_and_respect_annotation_metadata(
2267
+ object_=filters, annotation=typing.Sequence[SpanFilterPublic], direction="write"
2268
+ ),
2269
+ "limit": limit,
2270
+ "last_retrieved_id": last_retrieved_id,
2271
+ "truncate": truncate,
2272
+ "from_time": from_time,
2273
+ "to_time": to_time,
2274
+ },
2275
+ headers={
2276
+ "content-type": "application/json",
2277
+ },
2278
+ request_options=request_options,
2279
+ omit=OMIT,
2280
+ ) as _response:
2281
+
2282
+ async def stream() -> AsyncHttpResponse[typing.AsyncIterator[bytes]]:
2283
+ try:
2284
+ if 200 <= _response.status_code < 300:
2285
+ _chunk_size = request_options.get("chunk_size", None) if request_options is not None else None
2286
+ return AsyncHttpResponse(
2287
+ response=_response,
2288
+ data=(_chunk async for _chunk in _response.aiter_bytes(chunk_size=_chunk_size)),
2289
+ )
2290
+ await _response.aread()
2291
+ if _response.status_code == 400:
2292
+ raise BadRequestError(
2293
+ headers=dict(_response.headers),
2294
+ body=typing.cast(
2295
+ typing.Optional[typing.Any],
2296
+ parse_obj_as(
2297
+ type_=typing.Optional[typing.Any], # type: ignore
2298
+ object_=_response.json(),
2299
+ ),
2300
+ ),
2301
+ )
2302
+ _response_json = _response.json()
2303
+ except JSONDecodeError:
2304
+ raise ApiError(
2305
+ status_code=_response.status_code, headers=dict(_response.headers), body=_response.text
2306
+ )
2307
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
2308
+
2309
+ yield await stream()
2310
+
2311
+ async def update_span_comment(
2312
+ self,
2313
+ comment_id: str,
2314
+ *,
2315
+ text: str,
2316
+ id: typing.Optional[str] = OMIT,
2317
+ created_at: typing.Optional[dt.datetime] = OMIT,
2318
+ last_updated_at: typing.Optional[dt.datetime] = OMIT,
2319
+ created_by: typing.Optional[str] = OMIT,
2320
+ last_updated_by: typing.Optional[str] = OMIT,
2321
+ request_options: typing.Optional[RequestOptions] = None,
2322
+ ) -> AsyncHttpResponse[None]:
2323
+ """
2324
+ Update span comment by id
2325
+
2326
+ Parameters
2327
+ ----------
2328
+ comment_id : str
2329
+
2330
+ text : str
2331
+
2332
+ id : typing.Optional[str]
2333
+
2334
+ created_at : typing.Optional[dt.datetime]
2335
+
2336
+ last_updated_at : typing.Optional[dt.datetime]
2337
+
2338
+ created_by : typing.Optional[str]
2339
+
2340
+ last_updated_by : typing.Optional[str]
2341
+
2342
+ request_options : typing.Optional[RequestOptions]
2343
+ Request-specific configuration.
2344
+
2345
+ Returns
2346
+ -------
2347
+ AsyncHttpResponse[None]
2348
+ """
2349
+ _response = await self._client_wrapper.httpx_client.request(
2350
+ f"v1/private/spans/comments/{jsonable_encoder(comment_id)}",
2351
+ method="PATCH",
2352
+ json={
2353
+ "id": id,
2354
+ "text": text,
2355
+ "created_at": created_at,
2356
+ "last_updated_at": last_updated_at,
2357
+ "created_by": created_by,
2358
+ "last_updated_by": last_updated_by,
2359
+ },
2360
+ headers={
2361
+ "content-type": "application/json",
2362
+ },
2363
+ request_options=request_options,
2364
+ omit=OMIT,
2365
+ )
2366
+ try:
2367
+ if 200 <= _response.status_code < 300:
2368
+ return AsyncHttpResponse(response=_response, data=None)
2369
+ if _response.status_code == 404:
2370
+ raise NotFoundError(
2371
+ headers=dict(_response.headers),
2372
+ body=typing.cast(
2373
+ typing.Optional[typing.Any],
2374
+ parse_obj_as(
2375
+ type_=typing.Optional[typing.Any], # type: ignore
2376
+ object_=_response.json(),
2377
+ ),
2378
+ ),
2379
+ )
2380
+ _response_json = _response.json()
2381
+ except JSONDecodeError:
2382
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
2383
+ raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)