orq-ai-sdk 4.2.0rc28__py3-none-any.whl → 4.2.6__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 (167) hide show
  1. orq_ai_sdk/_hooks/globalhook.py +0 -1
  2. orq_ai_sdk/_version.py +3 -3
  3. orq_ai_sdk/audio.py +30 -0
  4. orq_ai_sdk/basesdk.py +20 -6
  5. orq_ai_sdk/chat.py +22 -0
  6. orq_ai_sdk/completions.py +332 -0
  7. orq_ai_sdk/contacts.py +43 -855
  8. orq_ai_sdk/deployments.py +61 -0
  9. orq_ai_sdk/edits.py +258 -0
  10. orq_ai_sdk/embeddings.py +238 -0
  11. orq_ai_sdk/generations.py +272 -0
  12. orq_ai_sdk/identities.py +1037 -0
  13. orq_ai_sdk/images.py +28 -0
  14. orq_ai_sdk/models/__init__.py +5341 -737
  15. orq_ai_sdk/models/actionreviewedstreamingevent.py +18 -1
  16. orq_ai_sdk/models/actionreviewrequestedstreamingevent.py +44 -1
  17. orq_ai_sdk/models/agenterroredstreamingevent.py +18 -1
  18. orq_ai_sdk/models/agentinactivestreamingevent.py +168 -70
  19. orq_ai_sdk/models/agentmessagecreatedstreamingevent.py +18 -2
  20. orq_ai_sdk/models/agentresponsemessage.py +18 -2
  21. orq_ai_sdk/models/agentstartedstreamingevent.py +127 -2
  22. orq_ai_sdk/models/agentthoughtstreamingevent.py +178 -211
  23. orq_ai_sdk/models/conversationresponse.py +31 -20
  24. orq_ai_sdk/models/conversationwithmessagesresponse.py +31 -20
  25. orq_ai_sdk/models/createagentrequestop.py +1922 -384
  26. orq_ai_sdk/models/createagentresponse.py +147 -91
  27. orq_ai_sdk/models/createagentresponserequestop.py +111 -2
  28. orq_ai_sdk/models/createchatcompletionop.py +1375 -861
  29. orq_ai_sdk/models/createchunkop.py +46 -19
  30. orq_ai_sdk/models/createcompletionop.py +1890 -0
  31. orq_ai_sdk/models/createcontactop.py +45 -56
  32. orq_ai_sdk/models/createconversationop.py +61 -39
  33. orq_ai_sdk/models/createconversationresponseop.py +68 -4
  34. orq_ai_sdk/models/createdatasetitemop.py +424 -80
  35. orq_ai_sdk/models/createdatasetop.py +19 -2
  36. orq_ai_sdk/models/createdatasourceop.py +92 -26
  37. orq_ai_sdk/models/createembeddingop.py +384 -0
  38. orq_ai_sdk/models/createevalop.py +552 -24
  39. orq_ai_sdk/models/createidentityop.py +176 -0
  40. orq_ai_sdk/models/createimageeditop.py +504 -0
  41. orq_ai_sdk/models/createimageop.py +208 -117
  42. orq_ai_sdk/models/createimagevariationop.py +486 -0
  43. orq_ai_sdk/models/createknowledgeop.py +186 -121
  44. orq_ai_sdk/models/creatememorydocumentop.py +50 -1
  45. orq_ai_sdk/models/creatememoryop.py +34 -21
  46. orq_ai_sdk/models/creatememorystoreop.py +34 -1
  47. orq_ai_sdk/models/createmoderationop.py +521 -0
  48. orq_ai_sdk/models/createpromptop.py +2748 -1252
  49. orq_ai_sdk/models/creatererankop.py +416 -0
  50. orq_ai_sdk/models/createresponseop.py +2567 -0
  51. orq_ai_sdk/models/createspeechop.py +316 -0
  52. orq_ai_sdk/models/createtoolop.py +537 -12
  53. orq_ai_sdk/models/createtranscriptionop.py +562 -0
  54. orq_ai_sdk/models/createtranslationop.py +540 -0
  55. orq_ai_sdk/models/datapart.py +18 -1
  56. orq_ai_sdk/models/deletechunksop.py +34 -1
  57. orq_ai_sdk/models/{deletecontactop.py → deleteidentityop.py} +9 -9
  58. orq_ai_sdk/models/deletepromptop.py +26 -0
  59. orq_ai_sdk/models/deploymentcreatemetricop.py +362 -76
  60. orq_ai_sdk/models/deploymentgetconfigop.py +635 -194
  61. orq_ai_sdk/models/deploymentinvokeop.py +168 -173
  62. orq_ai_sdk/models/deploymentsop.py +195 -58
  63. orq_ai_sdk/models/deploymentstreamop.py +652 -304
  64. orq_ai_sdk/models/errorpart.py +18 -1
  65. orq_ai_sdk/models/filecontentpartschema.py +18 -1
  66. orq_ai_sdk/models/filegetop.py +19 -2
  67. orq_ai_sdk/models/filelistop.py +35 -2
  68. orq_ai_sdk/models/filepart.py +50 -1
  69. orq_ai_sdk/models/fileuploadop.py +51 -2
  70. orq_ai_sdk/models/generateconversationnameop.py +31 -20
  71. orq_ai_sdk/models/get_v2_evaluators_id_versionsop.py +34 -1
  72. orq_ai_sdk/models/get_v2_tools_tool_id_versions_version_id_op.py +18 -1
  73. orq_ai_sdk/models/get_v2_tools_tool_id_versionsop.py +34 -1
  74. orq_ai_sdk/models/getallmemoriesop.py +34 -21
  75. orq_ai_sdk/models/getallmemorydocumentsop.py +42 -1
  76. orq_ai_sdk/models/getallmemorystoresop.py +34 -1
  77. orq_ai_sdk/models/getallpromptsop.py +1690 -230
  78. orq_ai_sdk/models/getalltoolsop.py +325 -8
  79. orq_ai_sdk/models/getchunkscountop.py +34 -1
  80. orq_ai_sdk/models/getevalsop.py +395 -43
  81. orq_ai_sdk/models/getonechunkop.py +14 -19
  82. orq_ai_sdk/models/getoneknowledgeop.py +116 -96
  83. orq_ai_sdk/models/getonepromptop.py +1673 -230
  84. orq_ai_sdk/models/getpromptversionop.py +1670 -216
  85. orq_ai_sdk/models/imagecontentpartschema.py +50 -1
  86. orq_ai_sdk/models/internal/globals.py +18 -1
  87. orq_ai_sdk/models/invokeagentop.py +140 -2
  88. orq_ai_sdk/models/invokedeploymentrequest.py +418 -80
  89. orq_ai_sdk/models/invokeevalop.py +160 -131
  90. orq_ai_sdk/models/listagentsop.py +793 -166
  91. orq_ai_sdk/models/listchunksop.py +32 -19
  92. orq_ai_sdk/models/listchunkspaginatedop.py +46 -19
  93. orq_ai_sdk/models/listconversationsop.py +18 -1
  94. orq_ai_sdk/models/listdatasetdatapointsop.py +252 -42
  95. orq_ai_sdk/models/listdatasetsop.py +35 -2
  96. orq_ai_sdk/models/listdatasourcesop.py +35 -26
  97. orq_ai_sdk/models/{listcontactsop.py → listidentitiesop.py} +89 -79
  98. orq_ai_sdk/models/listknowledgebasesop.py +132 -96
  99. orq_ai_sdk/models/listmodelsop.py +1 -0
  100. orq_ai_sdk/models/listpromptversionsop.py +1684 -216
  101. orq_ai_sdk/models/parseop.py +161 -17
  102. orq_ai_sdk/models/partdoneevent.py +19 -2
  103. orq_ai_sdk/models/post_v2_router_ocrop.py +408 -0
  104. orq_ai_sdk/models/publiccontact.py +27 -4
  105. orq_ai_sdk/models/publicidentity.py +62 -0
  106. orq_ai_sdk/models/reasoningpart.py +19 -2
  107. orq_ai_sdk/models/refusalpartschema.py +18 -1
  108. orq_ai_sdk/models/remoteconfigsgetconfigop.py +34 -1
  109. orq_ai_sdk/models/responsedoneevent.py +114 -84
  110. orq_ai_sdk/models/responsestartedevent.py +18 -1
  111. orq_ai_sdk/models/retrieveagentrequestop.py +787 -166
  112. orq_ai_sdk/models/retrievedatapointop.py +236 -42
  113. orq_ai_sdk/models/retrievedatasetop.py +19 -2
  114. orq_ai_sdk/models/retrievedatasourceop.py +17 -26
  115. orq_ai_sdk/models/{retrievecontactop.py → retrieveidentityop.py} +38 -41
  116. orq_ai_sdk/models/retrievememorydocumentop.py +18 -1
  117. orq_ai_sdk/models/retrievememoryop.py +18 -21
  118. orq_ai_sdk/models/retrievememorystoreop.py +18 -1
  119. orq_ai_sdk/models/retrievetoolop.py +309 -8
  120. orq_ai_sdk/models/runagentop.py +1451 -197
  121. orq_ai_sdk/models/searchknowledgeop.py +108 -1
  122. orq_ai_sdk/models/security.py +18 -1
  123. orq_ai_sdk/models/streamagentop.py +93 -2
  124. orq_ai_sdk/models/streamrunagentop.py +1428 -195
  125. orq_ai_sdk/models/textcontentpartschema.py +34 -1
  126. orq_ai_sdk/models/thinkingconfigenabledschema.py +18 -1
  127. orq_ai_sdk/models/toolcallpart.py +18 -1
  128. orq_ai_sdk/models/tooldoneevent.py +18 -1
  129. orq_ai_sdk/models/toolexecutionfailedstreamingevent.py +50 -1
  130. orq_ai_sdk/models/toolexecutionfinishedstreamingevent.py +34 -1
  131. orq_ai_sdk/models/toolexecutionstartedstreamingevent.py +34 -1
  132. orq_ai_sdk/models/toolresultpart.py +18 -1
  133. orq_ai_sdk/models/toolreviewrequestedevent.py +18 -1
  134. orq_ai_sdk/models/toolstartedevent.py +18 -1
  135. orq_ai_sdk/models/updateagentop.py +1951 -404
  136. orq_ai_sdk/models/updatechunkop.py +46 -19
  137. orq_ai_sdk/models/updateconversationop.py +61 -39
  138. orq_ai_sdk/models/updatedatapointop.py +424 -80
  139. orq_ai_sdk/models/updatedatasetop.py +51 -2
  140. orq_ai_sdk/models/updatedatasourceop.py +17 -26
  141. orq_ai_sdk/models/updateevalop.py +577 -16
  142. orq_ai_sdk/models/{updatecontactop.py → updateidentityop.py} +78 -68
  143. orq_ai_sdk/models/updateknowledgeop.py +234 -190
  144. orq_ai_sdk/models/updatememorydocumentop.py +50 -1
  145. orq_ai_sdk/models/updatememoryop.py +50 -21
  146. orq_ai_sdk/models/updatememorystoreop.py +66 -1
  147. orq_ai_sdk/models/updatepromptop.py +2844 -1450
  148. orq_ai_sdk/models/updatetoolop.py +592 -9
  149. orq_ai_sdk/models/usermessagerequest.py +18 -2
  150. orq_ai_sdk/moderations.py +218 -0
  151. orq_ai_sdk/orq_completions.py +660 -0
  152. orq_ai_sdk/orq_responses.py +398 -0
  153. orq_ai_sdk/prompts.py +28 -36
  154. orq_ai_sdk/rerank.py +232 -0
  155. orq_ai_sdk/router.py +89 -641
  156. orq_ai_sdk/sdk.py +3 -0
  157. orq_ai_sdk/speech.py +251 -0
  158. orq_ai_sdk/transcriptions.py +326 -0
  159. orq_ai_sdk/translations.py +298 -0
  160. orq_ai_sdk/utils/__init__.py +13 -1
  161. orq_ai_sdk/variations.py +254 -0
  162. orq_ai_sdk-4.2.6.dist-info/METADATA +888 -0
  163. orq_ai_sdk-4.2.6.dist-info/RECORD +263 -0
  164. {orq_ai_sdk-4.2.0rc28.dist-info → orq_ai_sdk-4.2.6.dist-info}/WHEEL +2 -1
  165. orq_ai_sdk-4.2.6.dist-info/top_level.txt +1 -0
  166. orq_ai_sdk-4.2.0rc28.dist-info/METADATA +0 -867
  167. orq_ai_sdk-4.2.0rc28.dist-info/RECORD +0 -233
@@ -4,7 +4,7 @@ from __future__ import annotations
4
4
  from dataclasses import dataclass, field
5
5
  import httpx
6
6
  from orq_ai_sdk.models import OrqError
7
- from orq_ai_sdk.types import BaseModel
7
+ from orq_ai_sdk.types import BaseModel, UNSET_SENTINEL
8
8
  from orq_ai_sdk.utils import (
9
9
  FieldMetadata,
10
10
  PathParamMetadata,
@@ -12,7 +12,7 @@ from orq_ai_sdk.utils import (
12
12
  get_discriminator,
13
13
  )
14
14
  import pydantic
15
- from pydantic import Discriminator, Tag
15
+ from pydantic import Discriminator, Tag, model_serializer
16
16
  from typing import Any, Dict, List, Literal, Optional, Union
17
17
  from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
18
18
 
@@ -49,6 +49,22 @@ class UpdateEvalGuardrailConfigEvalsRequestRequestBodyNumber(BaseModel):
49
49
 
50
50
  alert_on_failure: Optional[bool] = False
51
51
 
52
+ @model_serializer(mode="wrap")
53
+ def serialize_model(self, handler):
54
+ optional_fields = set(["alert_on_failure"])
55
+ serialized = handler(self)
56
+ m = {}
57
+
58
+ for n, f in type(self).model_fields.items():
59
+ k = f.alias or n
60
+ val = serialized.get(k)
61
+
62
+ if val != UNSET_SENTINEL:
63
+ if val is not None or k not in optional_fields:
64
+ m[k] = val
65
+
66
+ return m
67
+
52
68
 
53
69
  UpdateEvalGuardrailConfigEvalsRequestRequestBody4Type = Literal["boolean",]
54
70
 
@@ -69,6 +85,22 @@ class UpdateEvalGuardrailConfigEvalsRequestRequestBodyBoolean(BaseModel):
69
85
 
70
86
  alert_on_failure: Optional[bool] = False
71
87
 
88
+ @model_serializer(mode="wrap")
89
+ def serialize_model(self, handler):
90
+ optional_fields = set(["alert_on_failure"])
91
+ serialized = handler(self)
92
+ m = {}
93
+
94
+ for n, f in type(self).model_fields.items():
95
+ k = f.alias or n
96
+ val = serialized.get(k)
97
+
98
+ if val != UNSET_SENTINEL:
99
+ if val is not None or k not in optional_fields:
100
+ m[k] = val
101
+
102
+ return m
103
+
72
104
 
73
105
  UpdateEvalRequestBodyEvalsRequest4GuardrailConfigTypedDict = TypeAliasType(
74
106
  "UpdateEvalRequestBodyEvalsRequest4GuardrailConfigTypedDict",
@@ -140,6 +172,24 @@ class RequestBodyPython(BaseModel):
140
172
 
141
173
  key: Optional[str] = None
142
174
 
175
+ @model_serializer(mode="wrap")
176
+ def serialize_model(self, handler):
177
+ optional_fields = set(
178
+ ["guardrail_config", "output_type", "code", "description", "key"]
179
+ )
180
+ serialized = handler(self)
181
+ m = {}
182
+
183
+ for n, f in type(self).model_fields.items():
184
+ k = f.alias or n
185
+ val = serialized.get(k)
186
+
187
+ if val != UNSET_SENTINEL:
188
+ if val is not None or k not in optional_fields:
189
+ m[k] = val
190
+
191
+ return m
192
+
143
193
 
144
194
  UpdateEvalGuardrailConfigEvalsRequestRequestBody32Type = Literal["number",]
145
195
 
@@ -173,6 +223,22 @@ class UpdateEvalGuardrailConfigEvalsRequestNumber(BaseModel):
173
223
 
174
224
  alert_on_failure: Optional[bool] = False
175
225
 
226
+ @model_serializer(mode="wrap")
227
+ def serialize_model(self, handler):
228
+ optional_fields = set(["alert_on_failure"])
229
+ serialized = handler(self)
230
+ m = {}
231
+
232
+ for n, f in type(self).model_fields.items():
233
+ k = f.alias or n
234
+ val = serialized.get(k)
235
+
236
+ if val != UNSET_SENTINEL:
237
+ if val is not None or k not in optional_fields:
238
+ m[k] = val
239
+
240
+ return m
241
+
176
242
 
177
243
  UpdateEvalGuardrailConfigEvalsRequestRequestBody3Type = Literal["boolean",]
178
244
 
@@ -193,6 +259,22 @@ class UpdateEvalGuardrailConfigEvalsRequestBoolean(BaseModel):
193
259
 
194
260
  alert_on_failure: Optional[bool] = False
195
261
 
262
+ @model_serializer(mode="wrap")
263
+ def serialize_model(self, handler):
264
+ optional_fields = set(["alert_on_failure"])
265
+ serialized = handler(self)
266
+ m = {}
267
+
268
+ for n, f in type(self).model_fields.items():
269
+ k = f.alias or n
270
+ val = serialized.get(k)
271
+
272
+ if val != UNSET_SENTINEL:
273
+ if val is not None or k not in optional_fields:
274
+ m[k] = val
275
+
276
+ return m
277
+
196
278
 
197
279
  UpdateEvalRequestBodyEvalsRequestGuardrailConfigTypedDict = TypeAliasType(
198
280
  "UpdateEvalRequestBodyEvalsRequestGuardrailConfigTypedDict",
@@ -280,6 +362,33 @@ class RequestBodyHTTP(BaseModel):
280
362
 
281
363
  key: Optional[str] = None
282
364
 
365
+ @model_serializer(mode="wrap")
366
+ def serialize_model(self, handler):
367
+ optional_fields = set(
368
+ [
369
+ "guardrail_config",
370
+ "output_type",
371
+ "url",
372
+ "method",
373
+ "headers",
374
+ "payload",
375
+ "description",
376
+ "key",
377
+ ]
378
+ )
379
+ serialized = handler(self)
380
+ m = {}
381
+
382
+ for n, f in type(self).model_fields.items():
383
+ k = f.alias or n
384
+ val = serialized.get(k)
385
+
386
+ if val != UNSET_SENTINEL:
387
+ if val is not None or k not in optional_fields:
388
+ m[k] = val
389
+
390
+ return m
391
+
283
392
 
284
393
  UpdateEvalGuardrailConfigEvalsRequestRequestBodyType = Literal["number",]
285
394
 
@@ -313,6 +422,22 @@ class UpdateEvalGuardrailConfigEvalsNumber(BaseModel):
313
422
 
314
423
  alert_on_failure: Optional[bool] = False
315
424
 
425
+ @model_serializer(mode="wrap")
426
+ def serialize_model(self, handler):
427
+ optional_fields = set(["alert_on_failure"])
428
+ serialized = handler(self)
429
+ m = {}
430
+
431
+ for n, f in type(self).model_fields.items():
432
+ k = f.alias or n
433
+ val = serialized.get(k)
434
+
435
+ if val != UNSET_SENTINEL:
436
+ if val is not None or k not in optional_fields:
437
+ m[k] = val
438
+
439
+ return m
440
+
316
441
 
317
442
  UpdateEvalGuardrailConfigEvalsRequestType = Literal["boolean",]
318
443
 
@@ -333,6 +458,22 @@ class UpdateEvalGuardrailConfigEvalsBoolean(BaseModel):
333
458
 
334
459
  alert_on_failure: Optional[bool] = False
335
460
 
461
+ @model_serializer(mode="wrap")
462
+ def serialize_model(self, handler):
463
+ optional_fields = set(["alert_on_failure"])
464
+ serialized = handler(self)
465
+ m = {}
466
+
467
+ for n, f in type(self).model_fields.items():
468
+ k = f.alias or n
469
+ val = serialized.get(k)
470
+
471
+ if val != UNSET_SENTINEL:
472
+ if val is not None or k not in optional_fields:
473
+ m[k] = val
474
+
475
+ return m
476
+
336
477
 
337
478
  UpdateEvalRequestBodyEvalsGuardrailConfigTypedDict = TypeAliasType(
338
479
  "UpdateEvalRequestBodyEvalsGuardrailConfigTypedDict",
@@ -395,6 +536,24 @@ class RequestBodyJSON(BaseModel):
395
536
 
396
537
  key: Optional[str] = None
397
538
 
539
+ @model_serializer(mode="wrap")
540
+ def serialize_model(self, handler):
541
+ optional_fields = set(
542
+ ["guardrail_config", "output_type", "schema", "description", "key"]
543
+ )
544
+ serialized = handler(self)
545
+ m = {}
546
+
547
+ for n, f in type(self).model_fields.items():
548
+ k = f.alias or n
549
+ val = serialized.get(k)
550
+
551
+ if val != UNSET_SENTINEL:
552
+ if val is not None or k not in optional_fields:
553
+ m[k] = val
554
+
555
+ return m
556
+
398
557
 
399
558
  UpdateEvalGuardrailConfigEvalsType = Literal["number",]
400
559
 
@@ -428,6 +587,22 @@ class UpdateEvalGuardrailConfigNumber(BaseModel):
428
587
 
429
588
  alert_on_failure: Optional[bool] = False
430
589
 
590
+ @model_serializer(mode="wrap")
591
+ def serialize_model(self, handler):
592
+ optional_fields = set(["alert_on_failure"])
593
+ serialized = handler(self)
594
+ m = {}
595
+
596
+ for n, f in type(self).model_fields.items():
597
+ k = f.alias or n
598
+ val = serialized.get(k)
599
+
600
+ if val != UNSET_SENTINEL:
601
+ if val is not None or k not in optional_fields:
602
+ m[k] = val
603
+
604
+ return m
605
+
431
606
 
432
607
  UpdateEvalGuardrailConfigType = Literal["boolean",]
433
608
 
@@ -448,6 +623,22 @@ class UpdateEvalGuardrailConfigBoolean(BaseModel):
448
623
 
449
624
  alert_on_failure: Optional[bool] = False
450
625
 
626
+ @model_serializer(mode="wrap")
627
+ def serialize_model(self, handler):
628
+ optional_fields = set(["alert_on_failure"])
629
+ serialized = handler(self)
630
+ m = {}
631
+
632
+ for n, f in type(self).model_fields.items():
633
+ k = f.alias or n
634
+ val = serialized.get(k)
635
+
636
+ if val != UNSET_SENTINEL:
637
+ if val is not None or k not in optional_fields:
638
+ m[k] = val
639
+
640
+ return m
641
+
451
642
 
452
643
  UpdateEvalRequestBodyGuardrailConfigTypedDict = TypeAliasType(
453
644
  "UpdateEvalRequestBodyGuardrailConfigTypedDict",
@@ -521,6 +712,24 @@ class RequestBodyLLM(BaseModel):
521
712
 
522
713
  key: Optional[str] = None
523
714
 
715
+ @model_serializer(mode="wrap")
716
+ def serialize_model(self, handler):
717
+ optional_fields = set(
718
+ ["guardrail_config", "output_type", "prompt", "description", "model", "key"]
719
+ )
720
+ serialized = handler(self)
721
+ m = {}
722
+
723
+ for n, f in type(self).model_fields.items():
724
+ k = f.alias or n
725
+ val = serialized.get(k)
726
+
727
+ if val != UNSET_SENTINEL:
728
+ if val is not None or k not in optional_fields:
729
+ m[k] = val
730
+
731
+ return m
732
+
524
733
 
525
734
  UpdateEvalRequestBodyTypedDict = TypeAliasType(
526
735
  "UpdateEvalRequestBodyTypedDict",
@@ -559,6 +768,22 @@ class UpdateEvalRequest(BaseModel):
559
768
  FieldMetadata(request=RequestMetadata(media_type="application/json")),
560
769
  ] = None
561
770
 
771
+ @model_serializer(mode="wrap")
772
+ def serialize_model(self, handler):
773
+ optional_fields = set(["RequestBody"])
774
+ serialized = handler(self)
775
+ m = {}
776
+
777
+ for n, f in type(self).model_fields.items():
778
+ k = f.alias or n
779
+ val = serialized.get(k)
780
+
781
+ if val != UNSET_SENTINEL:
782
+ if val is not None or k not in optional_fields:
783
+ m[k] = val
784
+
785
+ return m
786
+
562
787
 
563
788
  class UpdateEvalEvalsResponseBodyData(BaseModel):
564
789
  message: str
@@ -624,6 +849,22 @@ class UpdateEvalGuardrailConfigEvalsResponse200ApplicationJSONResponseBody7Numbe
624
849
 
625
850
  alert_on_failure: Optional[bool] = False
626
851
 
852
+ @model_serializer(mode="wrap")
853
+ def serialize_model(self, handler):
854
+ optional_fields = set(["alert_on_failure"])
855
+ serialized = handler(self)
856
+ m = {}
857
+
858
+ for n, f in type(self).model_fields.items():
859
+ k = f.alias or n
860
+ val = serialized.get(k)
861
+
862
+ if val != UNSET_SENTINEL:
863
+ if val is not None or k not in optional_fields:
864
+ m[k] = val
865
+
866
+ return m
867
+
627
868
 
628
869
  UpdateEvalGuardrailConfigEvalsResponse200ApplicationJSONResponseBody7Type = Literal[
629
870
  "boolean",
@@ -650,6 +891,22 @@ class UpdateEvalGuardrailConfigEvalsResponse200ApplicationJSONResponseBody7Boole
650
891
 
651
892
  alert_on_failure: Optional[bool] = False
652
893
 
894
+ @model_serializer(mode="wrap")
895
+ def serialize_model(self, handler):
896
+ optional_fields = set(["alert_on_failure"])
897
+ serialized = handler(self)
898
+ m = {}
899
+
900
+ for n, f in type(self).model_fields.items():
901
+ k = f.alias or n
902
+ val = serialized.get(k)
903
+
904
+ if val != UNSET_SENTINEL:
905
+ if val is not None or k not in optional_fields:
906
+ m[k] = val
907
+
908
+ return m
909
+
653
910
 
654
911
  UpdateEvalResponseBodyEvalsResponse200ApplicationJSON7GuardrailConfigTypedDict = TypeAliasType(
655
912
  "UpdateEvalResponseBodyEvalsResponse200ApplicationJSON7GuardrailConfigTypedDict",
@@ -702,14 +959,30 @@ class ResponseBodyTypescript(BaseModel):
702
959
 
703
960
  key: str
704
961
 
705
- created: Optional[str] = "2026-01-14T08:04:34.218Z"
962
+ created: Optional[str] = "2026-01-25T15:02:08.775Z"
706
963
 
707
- updated: Optional[str] = "2026-01-14T08:04:34.218Z"
964
+ updated: Optional[str] = "2026-01-25T15:02:08.775Z"
708
965
 
709
966
  guardrail_config: Optional[
710
967
  UpdateEvalResponseBodyEvalsResponse200ApplicationJSON7GuardrailConfig
711
968
  ] = None
712
969
 
970
+ @model_serializer(mode="wrap")
971
+ def serialize_model(self, handler):
972
+ optional_fields = set(["created", "updated", "guardrail_config"])
973
+ serialized = handler(self)
974
+ m = {}
975
+
976
+ for n, f in type(self).model_fields.items():
977
+ k = f.alias or n
978
+ val = serialized.get(k)
979
+
980
+ if val != UNSET_SENTINEL:
981
+ if val is not None or k not in optional_fields:
982
+ m[k] = val
983
+
984
+ return m
985
+
713
986
 
714
987
  UpdateEvalGuardrailConfigEvalsResponse200ApplicationJSONResponseBody62Type = Literal[
715
988
  "number",
@@ -753,6 +1026,22 @@ class UpdateEvalGuardrailConfigEvalsResponse200ApplicationJSONResponseBody6Numbe
753
1026
 
754
1027
  alert_on_failure: Optional[bool] = False
755
1028
 
1029
+ @model_serializer(mode="wrap")
1030
+ def serialize_model(self, handler):
1031
+ optional_fields = set(["alert_on_failure"])
1032
+ serialized = handler(self)
1033
+ m = {}
1034
+
1035
+ for n, f in type(self).model_fields.items():
1036
+ k = f.alias or n
1037
+ val = serialized.get(k)
1038
+
1039
+ if val != UNSET_SENTINEL:
1040
+ if val is not None or k not in optional_fields:
1041
+ m[k] = val
1042
+
1043
+ return m
1044
+
756
1045
 
757
1046
  UpdateEvalGuardrailConfigEvalsResponse200ApplicationJSONResponseBody6Type = Literal[
758
1047
  "boolean",
@@ -779,6 +1068,22 @@ class UpdateEvalGuardrailConfigEvalsResponse200ApplicationJSONResponseBody6Boole
779
1068
 
780
1069
  alert_on_failure: Optional[bool] = False
781
1070
 
1071
+ @model_serializer(mode="wrap")
1072
+ def serialize_model(self, handler):
1073
+ optional_fields = set(["alert_on_failure"])
1074
+ serialized = handler(self)
1075
+ m = {}
1076
+
1077
+ for n, f in type(self).model_fields.items():
1078
+ k = f.alias or n
1079
+ val = serialized.get(k)
1080
+
1081
+ if val != UNSET_SENTINEL:
1082
+ if val is not None or k not in optional_fields:
1083
+ m[k] = val
1084
+
1085
+ return m
1086
+
782
1087
 
783
1088
  UpdateEvalResponseBodyEvalsResponse200ApplicationJSON6GuardrailConfigTypedDict = TypeAliasType(
784
1089
  "UpdateEvalResponseBodyEvalsResponse200ApplicationJSON6GuardrailConfigTypedDict",
@@ -850,14 +1155,30 @@ class ResponseBodyRagas(BaseModel):
850
1155
 
851
1156
  model: str
852
1157
 
853
- created: Optional[str] = "2026-01-14T08:04:34.218Z"
1158
+ created: Optional[str] = "2026-01-25T15:02:08.775Z"
854
1159
 
855
- updated: Optional[str] = "2026-01-14T08:04:34.218Z"
1160
+ updated: Optional[str] = "2026-01-25T15:02:08.775Z"
856
1161
 
857
1162
  guardrail_config: Optional[
858
1163
  UpdateEvalResponseBodyEvalsResponse200ApplicationJSON6GuardrailConfig
859
1164
  ] = None
860
1165
 
1166
+ @model_serializer(mode="wrap")
1167
+ def serialize_model(self, handler):
1168
+ optional_fields = set(["created", "updated", "guardrail_config"])
1169
+ serialized = handler(self)
1170
+ m = {}
1171
+
1172
+ for n, f in type(self).model_fields.items():
1173
+ k = f.alias or n
1174
+ val = serialized.get(k)
1175
+
1176
+ if val != UNSET_SENTINEL:
1177
+ if val is not None or k not in optional_fields:
1178
+ m[k] = val
1179
+
1180
+ return m
1181
+
861
1182
 
862
1183
  UpdateEvalGuardrailConfigEvalsResponse200ApplicationJSONResponseBody52Type = Literal[
863
1184
  "number",
@@ -901,6 +1222,22 @@ class UpdateEvalGuardrailConfigEvalsResponse200ApplicationJSONResponseBody5Numbe
901
1222
 
902
1223
  alert_on_failure: Optional[bool] = False
903
1224
 
1225
+ @model_serializer(mode="wrap")
1226
+ def serialize_model(self, handler):
1227
+ optional_fields = set(["alert_on_failure"])
1228
+ serialized = handler(self)
1229
+ m = {}
1230
+
1231
+ for n, f in type(self).model_fields.items():
1232
+ k = f.alias or n
1233
+ val = serialized.get(k)
1234
+
1235
+ if val != UNSET_SENTINEL:
1236
+ if val is not None or k not in optional_fields:
1237
+ m[k] = val
1238
+
1239
+ return m
1240
+
904
1241
 
905
1242
  UpdateEvalGuardrailConfigEvalsResponse200ApplicationJSONResponseBody5Type = Literal[
906
1243
  "boolean",
@@ -927,6 +1264,22 @@ class UpdateEvalGuardrailConfigEvalsResponse200ApplicationJSONResponseBody5Boole
927
1264
 
928
1265
  alert_on_failure: Optional[bool] = False
929
1266
 
1267
+ @model_serializer(mode="wrap")
1268
+ def serialize_model(self, handler):
1269
+ optional_fields = set(["alert_on_failure"])
1270
+ serialized = handler(self)
1271
+ m = {}
1272
+
1273
+ for n, f in type(self).model_fields.items():
1274
+ k = f.alias or n
1275
+ val = serialized.get(k)
1276
+
1277
+ if val != UNSET_SENTINEL:
1278
+ if val is not None or k not in optional_fields:
1279
+ m[k] = val
1280
+
1281
+ return m
1282
+
930
1283
 
931
1284
  UpdateEvalResponseBodyEvalsResponse200ApplicationJSONGuardrailConfigTypedDict = TypeAliasType(
932
1285
  "UpdateEvalResponseBodyEvalsResponse200ApplicationJSONGuardrailConfigTypedDict",
@@ -1499,14 +1852,30 @@ class ResponseBodyFunction(BaseModel):
1499
1852
 
1500
1853
  key: str
1501
1854
 
1502
- created: Optional[str] = "2026-01-14T08:04:34.218Z"
1855
+ created: Optional[str] = "2026-01-25T15:02:08.775Z"
1503
1856
 
1504
- updated: Optional[str] = "2026-01-14T08:04:34.218Z"
1857
+ updated: Optional[str] = "2026-01-25T15:02:08.775Z"
1505
1858
 
1506
1859
  guardrail_config: Optional[
1507
1860
  UpdateEvalResponseBodyEvalsResponse200ApplicationJSONGuardrailConfig
1508
1861
  ] = None
1509
1862
 
1863
+ @model_serializer(mode="wrap")
1864
+ def serialize_model(self, handler):
1865
+ optional_fields = set(["created", "updated", "guardrail_config"])
1866
+ serialized = handler(self)
1867
+ m = {}
1868
+
1869
+ for n, f in type(self).model_fields.items():
1870
+ k = f.alias or n
1871
+ val = serialized.get(k)
1872
+
1873
+ if val != UNSET_SENTINEL:
1874
+ if val is not None or k not in optional_fields:
1875
+ m[k] = val
1876
+
1877
+ return m
1878
+
1510
1879
 
1511
1880
  UpdateEvalGuardrailConfigEvalsResponse200ApplicationJSONResponseBody42Type = Literal[
1512
1881
  "number",
@@ -1550,6 +1919,22 @@ class UpdateEvalGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyNumber
1550
1919
 
1551
1920
  alert_on_failure: Optional[bool] = False
1552
1921
 
1922
+ @model_serializer(mode="wrap")
1923
+ def serialize_model(self, handler):
1924
+ optional_fields = set(["alert_on_failure"])
1925
+ serialized = handler(self)
1926
+ m = {}
1927
+
1928
+ for n, f in type(self).model_fields.items():
1929
+ k = f.alias or n
1930
+ val = serialized.get(k)
1931
+
1932
+ if val != UNSET_SENTINEL:
1933
+ if val is not None or k not in optional_fields:
1934
+ m[k] = val
1935
+
1936
+ return m
1937
+
1553
1938
 
1554
1939
  UpdateEvalGuardrailConfigEvalsResponse200ApplicationJSONResponseBody4Type = Literal[
1555
1940
  "boolean",
@@ -1576,6 +1961,22 @@ class UpdateEvalGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyBoolea
1576
1961
 
1577
1962
  alert_on_failure: Optional[bool] = False
1578
1963
 
1964
+ @model_serializer(mode="wrap")
1965
+ def serialize_model(self, handler):
1966
+ optional_fields = set(["alert_on_failure"])
1967
+ serialized = handler(self)
1968
+ m = {}
1969
+
1970
+ for n, f in type(self).model_fields.items():
1971
+ k = f.alias or n
1972
+ val = serialized.get(k)
1973
+
1974
+ if val != UNSET_SENTINEL:
1975
+ if val is not None or k not in optional_fields:
1976
+ m[k] = val
1977
+
1978
+ return m
1979
+
1579
1980
 
1580
1981
  UpdateEvalResponseBodyEvalsResponse200GuardrailConfigTypedDict = TypeAliasType(
1581
1982
  "UpdateEvalResponseBodyEvalsResponse200GuardrailConfigTypedDict",
@@ -1628,14 +2029,30 @@ class UpdateEvalResponseBodyPython(BaseModel):
1628
2029
 
1629
2030
  key: str
1630
2031
 
1631
- created: Optional[str] = "2026-01-14T08:04:34.218Z"
2032
+ created: Optional[str] = "2026-01-25T15:02:08.775Z"
1632
2033
 
1633
- updated: Optional[str] = "2026-01-14T08:04:34.218Z"
2034
+ updated: Optional[str] = "2026-01-25T15:02:08.775Z"
1634
2035
 
1635
2036
  guardrail_config: Optional[
1636
2037
  UpdateEvalResponseBodyEvalsResponse200GuardrailConfig
1637
2038
  ] = None
1638
2039
 
2040
+ @model_serializer(mode="wrap")
2041
+ def serialize_model(self, handler):
2042
+ optional_fields = set(["created", "updated", "guardrail_config"])
2043
+ serialized = handler(self)
2044
+ m = {}
2045
+
2046
+ for n, f in type(self).model_fields.items():
2047
+ k = f.alias or n
2048
+ val = serialized.get(k)
2049
+
2050
+ if val != UNSET_SENTINEL:
2051
+ if val is not None or k not in optional_fields:
2052
+ m[k] = val
2053
+
2054
+ return m
2055
+
1639
2056
 
1640
2057
  UpdateEvalGuardrailConfigEvalsResponse200ApplicationJSONResponseBody32Type = Literal[
1641
2058
  "number",
@@ -1673,6 +2090,22 @@ class UpdateEvalGuardrailConfigEvalsResponse200ApplicationJSONNumber(BaseModel):
1673
2090
 
1674
2091
  alert_on_failure: Optional[bool] = False
1675
2092
 
2093
+ @model_serializer(mode="wrap")
2094
+ def serialize_model(self, handler):
2095
+ optional_fields = set(["alert_on_failure"])
2096
+ serialized = handler(self)
2097
+ m = {}
2098
+
2099
+ for n, f in type(self).model_fields.items():
2100
+ k = f.alias or n
2101
+ val = serialized.get(k)
2102
+
2103
+ if val != UNSET_SENTINEL:
2104
+ if val is not None or k not in optional_fields:
2105
+ m[k] = val
2106
+
2107
+ return m
2108
+
1676
2109
 
1677
2110
  UpdateEvalGuardrailConfigEvalsResponse200ApplicationJSONResponseBody3Type = Literal[
1678
2111
  "boolean",
@@ -1697,6 +2130,22 @@ class UpdateEvalGuardrailConfigEvalsResponse200ApplicationJSONBoolean(BaseModel)
1697
2130
 
1698
2131
  alert_on_failure: Optional[bool] = False
1699
2132
 
2133
+ @model_serializer(mode="wrap")
2134
+ def serialize_model(self, handler):
2135
+ optional_fields = set(["alert_on_failure"])
2136
+ serialized = handler(self)
2137
+ m = {}
2138
+
2139
+ for n, f in type(self).model_fields.items():
2140
+ k = f.alias or n
2141
+ val = serialized.get(k)
2142
+
2143
+ if val != UNSET_SENTINEL:
2144
+ if val is not None or k not in optional_fields:
2145
+ m[k] = val
2146
+
2147
+ return m
2148
+
1700
2149
 
1701
2150
  UpdateEvalResponseBodyEvalsResponseGuardrailConfigTypedDict = TypeAliasType(
1702
2151
  "UpdateEvalResponseBodyEvalsResponseGuardrailConfigTypedDict",
@@ -1764,14 +2213,30 @@ class UpdateEvalResponseBodyHTTP(BaseModel):
1764
2213
 
1765
2214
  key: str
1766
2215
 
1767
- created: Optional[str] = "2026-01-14T08:04:34.218Z"
2216
+ created: Optional[str] = "2026-01-25T15:02:08.775Z"
1768
2217
 
1769
- updated: Optional[str] = "2026-01-14T08:04:34.218Z"
2218
+ updated: Optional[str] = "2026-01-25T15:02:08.775Z"
1770
2219
 
1771
2220
  guardrail_config: Optional[UpdateEvalResponseBodyEvalsResponseGuardrailConfig] = (
1772
2221
  None
1773
2222
  )
1774
2223
 
2224
+ @model_serializer(mode="wrap")
2225
+ def serialize_model(self, handler):
2226
+ optional_fields = set(["created", "updated", "guardrail_config"])
2227
+ serialized = handler(self)
2228
+ m = {}
2229
+
2230
+ for n, f in type(self).model_fields.items():
2231
+ k = f.alias or n
2232
+ val = serialized.get(k)
2233
+
2234
+ if val != UNSET_SENTINEL:
2235
+ if val is not None or k not in optional_fields:
2236
+ m[k] = val
2237
+
2238
+ return m
2239
+
1775
2240
 
1776
2241
  UpdateEvalGuardrailConfigEvalsResponse200ApplicationJSONResponseBodyType = Literal[
1777
2242
  "number",
@@ -1807,6 +2272,22 @@ class UpdateEvalGuardrailConfigEvalsResponse200Number(BaseModel):
1807
2272
 
1808
2273
  alert_on_failure: Optional[bool] = False
1809
2274
 
2275
+ @model_serializer(mode="wrap")
2276
+ def serialize_model(self, handler):
2277
+ optional_fields = set(["alert_on_failure"])
2278
+ serialized = handler(self)
2279
+ m = {}
2280
+
2281
+ for n, f in type(self).model_fields.items():
2282
+ k = f.alias or n
2283
+ val = serialized.get(k)
2284
+
2285
+ if val != UNSET_SENTINEL:
2286
+ if val is not None or k not in optional_fields:
2287
+ m[k] = val
2288
+
2289
+ return m
2290
+
1810
2291
 
1811
2292
  UpdateEvalGuardrailConfigEvalsResponse200ApplicationJSONType = Literal["boolean",]
1812
2293
 
@@ -1827,6 +2308,22 @@ class UpdateEvalGuardrailConfigEvalsResponse200Boolean(BaseModel):
1827
2308
 
1828
2309
  alert_on_failure: Optional[bool] = False
1829
2310
 
2311
+ @model_serializer(mode="wrap")
2312
+ def serialize_model(self, handler):
2313
+ optional_fields = set(["alert_on_failure"])
2314
+ serialized = handler(self)
2315
+ m = {}
2316
+
2317
+ for n, f in type(self).model_fields.items():
2318
+ k = f.alias or n
2319
+ val = serialized.get(k)
2320
+
2321
+ if val != UNSET_SENTINEL:
2322
+ if val is not None or k not in optional_fields:
2323
+ m[k] = val
2324
+
2325
+ return m
2326
+
1830
2327
 
1831
2328
  UpdateEvalResponseBodyEvalsGuardrailConfigTypedDict = TypeAliasType(
1832
2329
  "UpdateEvalResponseBodyEvalsGuardrailConfigTypedDict",
@@ -1871,12 +2368,28 @@ class UpdateEvalResponseBodyJSON(BaseModel):
1871
2368
 
1872
2369
  key: str
1873
2370
 
1874
- created: Optional[str] = "2026-01-14T08:04:34.218Z"
2371
+ created: Optional[str] = "2026-01-25T15:02:08.775Z"
1875
2372
 
1876
- updated: Optional[str] = "2026-01-14T08:04:34.218Z"
2373
+ updated: Optional[str] = "2026-01-25T15:02:08.775Z"
1877
2374
 
1878
2375
  guardrail_config: Optional[UpdateEvalResponseBodyEvalsGuardrailConfig] = None
1879
2376
 
2377
+ @model_serializer(mode="wrap")
2378
+ def serialize_model(self, handler):
2379
+ optional_fields = set(["created", "updated", "guardrail_config"])
2380
+ serialized = handler(self)
2381
+ m = {}
2382
+
2383
+ for n, f in type(self).model_fields.items():
2384
+ k = f.alias or n
2385
+ val = serialized.get(k)
2386
+
2387
+ if val != UNSET_SENTINEL:
2388
+ if val is not None or k not in optional_fields:
2389
+ m[k] = val
2390
+
2391
+ return m
2392
+
1880
2393
 
1881
2394
  UpdateEvalGuardrailConfigEvalsResponse200Type = Literal["number",]
1882
2395
 
@@ -1910,6 +2423,22 @@ class UpdateEvalGuardrailConfigEvalsResponseNumber(BaseModel):
1910
2423
 
1911
2424
  alert_on_failure: Optional[bool] = False
1912
2425
 
2426
+ @model_serializer(mode="wrap")
2427
+ def serialize_model(self, handler):
2428
+ optional_fields = set(["alert_on_failure"])
2429
+ serialized = handler(self)
2430
+ m = {}
2431
+
2432
+ for n, f in type(self).model_fields.items():
2433
+ k = f.alias or n
2434
+ val = serialized.get(k)
2435
+
2436
+ if val != UNSET_SENTINEL:
2437
+ if val is not None or k not in optional_fields:
2438
+ m[k] = val
2439
+
2440
+ return m
2441
+
1913
2442
 
1914
2443
  UpdateEvalGuardrailConfigEvalsResponseType = Literal["boolean",]
1915
2444
 
@@ -1930,6 +2459,22 @@ class UpdateEvalGuardrailConfigEvalsResponseBoolean(BaseModel):
1930
2459
 
1931
2460
  alert_on_failure: Optional[bool] = False
1932
2461
 
2462
+ @model_serializer(mode="wrap")
2463
+ def serialize_model(self, handler):
2464
+ optional_fields = set(["alert_on_failure"])
2465
+ serialized = handler(self)
2466
+ m = {}
2467
+
2468
+ for n, f in type(self).model_fields.items():
2469
+ k = f.alias or n
2470
+ val = serialized.get(k)
2471
+
2472
+ if val != UNSET_SENTINEL:
2473
+ if val is not None or k not in optional_fields:
2474
+ m[k] = val
2475
+
2476
+ return m
2477
+
1933
2478
 
1934
2479
  UpdateEvalResponseBodyGuardrailConfigTypedDict = TypeAliasType(
1935
2480
  "UpdateEvalResponseBodyGuardrailConfigTypedDict",
@@ -1977,12 +2522,28 @@ class UpdateEvalResponseBodyLLM(BaseModel):
1977
2522
 
1978
2523
  model: str
1979
2524
 
1980
- created: Optional[str] = "2026-01-14T08:04:34.218Z"
2525
+ created: Optional[str] = "2026-01-25T15:02:08.775Z"
1981
2526
 
1982
- updated: Optional[str] = "2026-01-14T08:04:34.218Z"
2527
+ updated: Optional[str] = "2026-01-25T15:02:08.775Z"
1983
2528
 
1984
2529
  guardrail_config: Optional[UpdateEvalResponseBodyGuardrailConfig] = None
1985
2530
 
2531
+ @model_serializer(mode="wrap")
2532
+ def serialize_model(self, handler):
2533
+ optional_fields = set(["created", "updated", "guardrail_config"])
2534
+ serialized = handler(self)
2535
+ m = {}
2536
+
2537
+ for n, f in type(self).model_fields.items():
2538
+ k = f.alias or n
2539
+ val = serialized.get(k)
2540
+
2541
+ if val != UNSET_SENTINEL:
2542
+ if val is not None or k not in optional_fields:
2543
+ m[k] = val
2544
+
2545
+ return m
2546
+
1986
2547
 
1987
2548
  UpdateEvalResponseBodyTypedDict = TypeAliasType(
1988
2549
  "UpdateEvalResponseBodyTypedDict",