Skip to content

Commit cd3e910

Browse files
feat(boxsdkgen): add confidence scores for structured extract (box/box-openapi#582) (#1310)
1 parent 83f7014 commit cd3e910

File tree

5 files changed

+15
-1
lines changed

5 files changed

+15
-1
lines changed

.codegen.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
{ "engineHash": "f9e2519", "specHash": "f8fb08c", "version": "4.3.0" }
1+
{ "engineHash": "f9e2519", "specHash": "ccdb456", "version": "4.3.0" }

box_sdk_gen/managers/ai.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -425,6 +425,7 @@ def create_ai_extract_structured(
425425
*,
426426
metadata_template: Optional[CreateAiExtractStructuredMetadataTemplate] = None,
427427
fields: Optional[List[CreateAiExtractStructuredFields]] = None,
428+
include_confidence_score: Optional[bool] = None,
428429
ai_agent: Optional[AiExtractStructuredAgent] = None,
429430
extra_headers: Optional[Dict[str, Optional[str]]] = None
430431
) -> AiExtractStructuredResponse:
@@ -450,6 +451,8 @@ def create_ai_extract_structured(
450451
:param fields: The fields to be extracted from the provided items.
451452
For your request to work, you must provide either `metadata_template` or `fields`, but not both., defaults to None
452453
:type fields: Optional[List[CreateAiExtractStructuredFields]], optional
454+
:param include_confidence_score: A flag to indicate whether confidence scores for every extracted field should be returned., defaults to None
455+
:type include_confidence_score: Optional[bool], optional
453456
:param extra_headers: Extra headers that will be included in the HTTP request., defaults to None
454457
:type extra_headers: Optional[Dict[str, Optional[str]]], optional
455458
"""
@@ -459,6 +462,7 @@ def create_ai_extract_structured(
459462
'items': items,
460463
'metadata_template': metadata_template,
461464
'fields': fields,
465+
'include_confidence_score': include_confidence_score,
462466
'ai_agent': ai_agent,
463467
}
464468
headers_map: Dict[str, str] = prepare_params({**extra_headers})

box_sdk_gen/schemas/ai_extract_structured.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -114,6 +114,7 @@ def __init__(
114114
*,
115115
metadata_template: Optional[AiExtractStructuredMetadataTemplateField] = None,
116116
fields: Optional[List[AiExtractStructuredFieldsField]] = None,
117+
include_confidence_score: Optional[bool] = None,
117118
ai_agent: Optional[AiExtractStructuredAgent] = None,
118119
**kwargs
119120
):
@@ -126,9 +127,12 @@ def __init__(
126127
:param fields: The fields to be extracted from the provided items.
127128
For your request to work, you must provide either `metadata_template` or `fields`, but not both., defaults to None
128129
:type fields: Optional[List[AiExtractStructuredFieldsField]], optional
130+
:param include_confidence_score: A flag to indicate whether confidence scores for every extracted field should be returned., defaults to None
131+
:type include_confidence_score: Optional[bool], optional
129132
"""
130133
super().__init__(**kwargs)
131134
self.items = items
132135
self.metadata_template = metadata_template
133136
self.fields = fields
137+
self.include_confidence_score = include_confidence_score
134138
self.ai_agent = ai_agent

box_sdk_gen/schemas/ai_extract_structured_response.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ def __init__(
1818
created_at: DateTime,
1919
*,
2020
completion_reason: Optional[str] = None,
21+
confidence_score: Optional[Dict] = None,
2122
ai_agent_info: Optional[AiAgentInfo] = None,
2223
**kwargs
2324
):
@@ -26,9 +27,12 @@ def __init__(
2627
:type created_at: DateTime
2728
:param completion_reason: The reason the response finishes., defaults to None
2829
:type completion_reason: Optional[str], optional
30+
:param confidence_score: The confidence score numeric values for each extracted field as a JSON dictionary. This can be empty if no field could be extracted., defaults to None
31+
:type confidence_score: Optional[Dict], optional
2932
"""
3033
super().__init__(**kwargs)
3134
self.answer = answer
3235
self.created_at = created_at
3336
self.completion_reason = completion_reason
37+
self.confidence_score = confidence_score
3438
self.ai_agent_info = ai_agent_info

docs/box_sdk_gen/ai.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -178,6 +178,8 @@ client.ai.create_ai_extract_structured([AiItemBase(id=file.id)], fields=[CreateA
178178
- The metadata template containing the fields to extract. For your request to work, you must provide either `metadata_template` or `fields`, but not both.
179179
- fields `Optional[List[CreateAiExtractStructuredFields]]`
180180
- The fields to be extracted from the provided items. For your request to work, you must provide either `metadata_template` or `fields`, but not both.
181+
- include_confidence_score `Optional[bool]`
182+
- A flag to indicate whether confidence scores for every extracted field should be returned.
181183
- ai_agent `Optional[AiExtractStructuredAgent]`
182184
- extra_headers `Optional[Dict[str, Optional[str]]]`
183185
- Extra headers that will be included in the HTTP request.

0 commit comments

Comments
 (0)