openai-sdk-helpers 0.6.5__py3-none-any.whl → 0.6.6__py3-none-any.whl

openai_sdk_helpers/agent/classifier.py

@@ -26,6 +26,8 @@ from ..utils import ensure_list
 from .base import AgentBase
 from .configuration import AgentConfiguration
 
+_CONTINUE_CONFIDENCE_THRESHOLD = 0.7
+
 
 class TaxonomyClassifierAgent(AgentBase):
     """Classify text by recursively traversing a taxonomy.
@@ -394,6 +396,13 @@ class TaxonomyClassifierAgent(AgentBase):
 
         resolved_nodes = _resolve_nodes(node_paths, step)
 
+        should_continue = _should_continue_from_stop(step, resolved_nodes)
+        if should_continue:
+            step = step.model_copy(
+                update={"stop_reason": ClassificationStopReason.CONTINUE}
+            )
+            state.steps[-1] = step
+
         if step.stop_reason.is_terminal:
             if resolved_nodes:
                 state.final_nodes.extend(resolved_nodes)
@@ -407,7 +416,7 @@ class TaxonomyClassifierAgent(AgentBase):
             return
 
         base_steps_len = len(state.steps)
-        child_tasks: list[tuple[Awaitable["_TraversalState"], int]] = []
+        child_tasks: list[tuple[Awaitable["_TraversalState"], int, TaxonomyNode]] = []
         for node in resolved_nodes:
             if node.children:
                 sub_agent = self._build_sub_agent(list(node.children))
@@ -429,6 +438,7 @@ class TaxonomyClassifierAgent(AgentBase):
                             state=sub_state,
                         ),
                         base_final_nodes_len,
+                        node,
                     )
                 )
             else:
@@ -439,13 +449,19 @@ class TaxonomyClassifierAgent(AgentBase):
                 )
         if child_tasks:
             child_states = await asyncio.gather(
-                *(child_task for child_task, _ in child_tasks)
+                *(child_task for child_task, _, _ in child_tasks)
             )
-            for child_state, (_, base_final_nodes_len) in zip(
+            for child_state, (_, base_final_nodes_len, parent_node) in zip(
                 child_states, child_tasks, strict=True
             ):
                 state.steps.extend(child_state.steps[base_steps_len:])
-                state.final_nodes.extend(child_state.final_nodes[base_final_nodes_len:])
+                child_final_nodes = child_state.final_nodes[base_final_nodes_len:]
+                state.final_nodes.extend(child_final_nodes)
+                if should_continue and not child_final_nodes:
+                    state.final_nodes.append(parent_node)
+                    state.best_confidence = _max_confidence(
+                        state.best_confidence, step.confidence
+                    )
                 state.best_confidence = _max_confidence(
                     state.best_confidence, child_state.best_confidence
                 )
@@ -627,6 +643,34 @@ def _resolve_stop_reason(state: _TraversalState) -> ClassificationStopReason:
     return ClassificationStopReason.NO_MATCH
 
 
+def _should_continue_from_stop(
+    step: ClassificationStep,
+    resolved_nodes: Sequence[TaxonomyNode],
+    threshold: float = _CONTINUE_CONFIDENCE_THRESHOLD,
+) -> bool:
+    """Return True when a stop reason should continue traversal.
+
+    Parameters
+    ----------
+    step : ClassificationStep
+        Classification step to evaluate.
+    resolved_nodes : Sequence[TaxonomyNode]
+        Resolved taxonomy nodes from the classification step.
+    threshold : float, default=0.7
+        Confidence threshold for overriding a stop reason.
+
+    Returns
+    -------
+    bool
+        True when traversal should proceed despite a stop reason.
+    """
+    if step.stop_reason is not ClassificationStopReason.STOP:
+        return False
+    if step.confidence is None or step.confidence < threshold:
+        return False
+    return any(node.children for node in resolved_nodes)
+
+
 def _normalize_roots(
     taxonomy: TaxonomyNode | Sequence[TaxonomyNode],
 ) -> list[TaxonomyNode]:

openai_sdk_helpers/structure/classification.py

@@ -480,6 +480,8 @@ class ClassificationResult(StructureBase):
         Yield selected identifiers across all steps.
     selected_nodes
         Return the selected identifiers across all steps.
+    to_lightweight_summary
+        Return a lightweight summary of selected node paths.
 
     Examples
     --------
@@ -565,6 +567,52 @@ class ClassificationResult(StructureBase):
                 if normalized:
                     yield normalized
 
+    def to_lightweight_summary(self) -> "ClassificationSummary | None":
+        """Return a lightweight summary of selected node paths.
+
+        Returns
+        -------
+        ClassificationSummary or None
+            Summary containing the ``full_paths`` of selected nodes, or None
+            when no selections exist.
+
+        Examples
+        --------
+        >>> result = ClassificationResult(steps=[])
+        >>> result.to_lightweight_summary() is None
+        True
+        """
+        full_paths = [
+            format_path_identifier(taxonomy_enum_path(node))
+            for node in self.selected_nodes
+        ]
+        if not full_paths:
+            return None
+        return ClassificationSummary(full_paths=full_paths)
+
+
+class ClassificationSummary(StructureBase):
+    """Represent a lightweight summary of selected taxonomy paths.
+
+    Attributes
+    ----------
+    full_paths : list[str] or None
+        Selected taxonomy paths including parent segments.
+
+    Methods
+    -------
+    to_json()
+        Serialize the summary to a JSON-compatible dictionary.
+    from_json(data)
+        Construct a summary instance from JSON-compatible data.
+    """
+
+    full_paths: list[str] | None = spec_field(
+        "full_paths",
+        description="Selected taxonomy paths including parent segments.",
+        default=None,
+    )
+
 
 def taxonomy_enum_path(value: Enum | str | None) -> list[str]:
     """Return the taxonomy path segments for an enum value.
@@ -598,6 +646,7 @@ def taxonomy_enum_path(value: Enum | str | None) -> list[str]:
 
 __all__ = [
     "ClassificationResult",
+    "ClassificationSummary",
     "ClassificationStep",
     "ClassificationStopReason",
     "Taxonomy",

{openai_sdk_helpers-0.6.5.dist-info → openai_sdk_helpers-0.6.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openai-sdk-helpers
-Version: 0.6.5
+Version: 0.6.6
 Summary: Composable helpers for OpenAI SDK agents, prompts, and storage
 Author: openai-sdk-helpers maintainers
 License: MIT

{openai_sdk_helpers-0.6.5.dist-info → openai_sdk_helpers-0.6.6.dist-info}/RECORD

@@ -10,7 +10,7 @@ openai_sdk_helpers/tools.py,sha256=8hhcytpmDfoXV16UQbDmDVV0rhLOn8c_VjXO8XaTFLQ,1
 openai_sdk_helpers/types.py,sha256=ejCG0rYqJhjOQvKLoNnzq-TzcKCFt69GVfi7y805NkU,1451
 openai_sdk_helpers/agent/__init__.py,sha256=qyzKzPhD8KsEl6d79XERK32AK5It_BZNOqChOpBdmhg,1199
 openai_sdk_helpers/agent/base.py,sha256=vLs0oALhxsd_Xy5dGjSZTUFTug-YwZkF1LabQ2ruLxk,29508
-openai_sdk_helpers/agent/classifier.py,sha256=PHUnA5dSDWBQeRxwo0Qe8xIe7Ren3xSRAOmsQrRK_oA,33241
+openai_sdk_helpers/agent/classifier.py,sha256=PyeLNnu2cHx_zjY8ReLV2AN5olc0LZuvh6urNZgK6jc,34795
 openai_sdk_helpers/agent/configuration.py,sha256=ZeH4ErgVe-BZamjUeNONbQi60ViolgYAWh-c8hNAQTw,15810
 openai_sdk_helpers/agent/coordinator.py,sha256=lVjA0yI-GhGKlqbNR_k9GOCrUjFoZ0QoqRaafHckyME,18052
 openai_sdk_helpers/agent/files.py,sha256=H7UfSZSjFUbv1cjRvNld9kZwIjc5wPq4vynqU8HgGJE,4478
@@ -58,7 +58,7 @@ openai_sdk_helpers/streamlit_app/configuration.py,sha256=0KeJ4HqCNFthBHsedV6ptqH
 openai_sdk_helpers/structure/__init__.py,sha256=w27ezTYVLzZdDMFfA8mawE82h8zO53idFBCiCfYfh7s,4321
 openai_sdk_helpers/structure/agent_blueprint.py,sha256=VyJWkgPNzAYKRDMeR1M4kE6qqQURnwqtrrEn0TRJf0g,9698
 openai_sdk_helpers/structure/base.py,sha256=OVI305F2suG6c2Qh_ZD_wZ1mS1GpqPBC-4RqInXqiAU,28512
-openai_sdk_helpers/structure/classification.py,sha256=SYrrsv0Y2A2kXhL3jbn7lWnTb5jB_UE-cx-sJSRCxEA,17312
+openai_sdk_helpers/structure/classification.py,sha256=aIelEz3Ffj4CKd0P_EB4uOMr0yX4uqqdKcRP9hjY0nw,18769
 openai_sdk_helpers/structure/extraction.py,sha256=wODP0iLAhhsdQkMWRYPYTiLUMU8bFMKiBjPl3PKUleg,37335
 openai_sdk_helpers/structure/prompt.py,sha256=ZfsaHdA0hj5zmZDrOdpXjCsC8U-jjzwFG4JBsWYiaH4,1535
 openai_sdk_helpers/structure/responses.py,sha256=WUwh0DhXj24pkvgqH1FMkdx5V2ArdvdtrDN_fuMBtDU,4882
@@ -92,8 +92,8 @@ openai_sdk_helpers/vector_storage/__init__.py,sha256=L5LxO09puh9_yBB9IDTvc1CvVkA
 openai_sdk_helpers/vector_storage/cleanup.py,sha256=sZ4ZSTlnjF52o9Cc8A9dTX37ZYXXDxS_fdIpoOBWvrg,3666
 openai_sdk_helpers/vector_storage/storage.py,sha256=t_ukacaXRa9EXE4-3BxsrB4Rjhu6nTu7NA9IjCJBIpQ,24259
 openai_sdk_helpers/vector_storage/types.py,sha256=jTCcOYMeOpZWvcse0z4T3MVs-RBOPC-fqWTBeQrgafU,1639
-openai_sdk_helpers-0.6.5.dist-info/METADATA,sha256=xInZz82S0zSN3YkKs_T949-3h0E_8mJaslN1nBgvES8,24622
-openai_sdk_helpers-0.6.5.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-openai_sdk_helpers-0.6.5.dist-info/entry_points.txt,sha256=gEOD1ZeXe8d2OP-KzUlG-b_9D9yUZTCt-GFW3EDbIIY,63
-openai_sdk_helpers-0.6.5.dist-info/licenses/LICENSE,sha256=CUhc1NrE50bs45tcXF7OcTQBKEvkUuLqeOHgrWQ5jaA,1067
-openai_sdk_helpers-0.6.5.dist-info/RECORD,,
+openai_sdk_helpers-0.6.6.dist-info/METADATA,sha256=cyg4hIimWHNrkSxsEwWZGOOmaBFAWtSHfL9JkGQXHxw,24622
+openai_sdk_helpers-0.6.6.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+openai_sdk_helpers-0.6.6.dist-info/entry_points.txt,sha256=gEOD1ZeXe8d2OP-KzUlG-b_9D9yUZTCt-GFW3EDbIIY,63
+openai_sdk_helpers-0.6.6.dist-info/licenses/LICENSE,sha256=CUhc1NrE50bs45tcXF7OcTQBKEvkUuLqeOHgrWQ5jaA,1067
+openai_sdk_helpers-0.6.6.dist-info/RECORD,,