openai-sdk-helpers 0.5.2__tar.gz → 0.6.1__tar.gz

{openai_sdk_helpers-0.5.2 → openai_sdk_helpers-0.6.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openai-sdk-helpers
-Version: 0.5.2
+Version: 0.6.1
 Summary: Composable helpers for OpenAI SDK agents, prompts, and storage
 Author: openai-sdk-helpers maintainers
 License: MIT
@@ -97,6 +97,7 @@ The `agent` module provides a higher-level abstraction for building agents, whil
   - **SummarizerAgent**: Generate concise summaries from provided text
   - **TranslatorAgent**: Translate text into target languages
   - **ValidatorAgent**: Check inputs and outputs against safety guardrails
+  - **TaxonomyClassifierAgent**: Classify text into taxonomy-driven labels
 
 #### Response Module (Built on `openai` SDK)
 - **Response handling utilities** for direct API control with fine-grained message management
@@ -210,12 +211,18 @@ These use the `agent` module built on `openai-agents` SDK:
 ```python
 from openai_sdk_helpers.agent import (
     SummarizerAgent,
+    TaxonomyClassifierAgent,
     TranslatorAgent,
     ValidatorAgent,
 )
+from openai_sdk_helpers.structure import TaxonomyNode
 
 # Initialize agents with a default model
 summarizer = SummarizerAgent(default_model="gpt-4o-mini")
+classifier = TaxonomyClassifierAgent(
+    model="gpt-4o-mini",
+    taxonomy=[TaxonomyNode(label="Billing"), TaxonomyNode(label="Support")],
+)
 translator = TranslatorAgent(default_model="gpt-4o-mini")
 validator = ValidatorAgent(default_model="gpt-4o-mini")
 
@@ -227,6 +234,10 @@ print(summary.text)
 translation = translator.run_sync("Bonjour", target_language="English")
 print(translation)
 
+# Classify text against a taxonomy
+classification = classifier.run_sync("I need help with my invoice")
+print(classification.final_node)
+
 # Validate against guardrails
 validation = validator.run_sync(
     "Share meeting notes with names removed",

{openai_sdk_helpers-0.5.2 → openai_sdk_helpers-0.6.1}/README.md

@@ -59,6 +59,7 @@ The `agent` module provides a higher-level abstraction for building agents, whil
   - **SummarizerAgent**: Generate concise summaries from provided text
   - **TranslatorAgent**: Translate text into target languages
   - **ValidatorAgent**: Check inputs and outputs against safety guardrails
+  - **TaxonomyClassifierAgent**: Classify text into taxonomy-driven labels
 
 #### Response Module (Built on `openai` SDK)
 - **Response handling utilities** for direct API control with fine-grained message management
@@ -172,12 +173,18 @@ These use the `agent` module built on `openai-agents` SDK:
 ```python
 from openai_sdk_helpers.agent import (
     SummarizerAgent,
+    TaxonomyClassifierAgent,
     TranslatorAgent,
     ValidatorAgent,
 )
+from openai_sdk_helpers.structure import TaxonomyNode
 
 # Initialize agents with a default model
 summarizer = SummarizerAgent(default_model="gpt-4o-mini")
+classifier = TaxonomyClassifierAgent(
+    model="gpt-4o-mini",
+    taxonomy=[TaxonomyNode(label="Billing"), TaxonomyNode(label="Support")],
+)
 translator = TranslatorAgent(default_model="gpt-4o-mini")
 validator = ValidatorAgent(default_model="gpt-4o-mini")
 
@@ -189,6 +196,10 @@ print(summary.text)
 translation = translator.run_sync("Bonjour", target_language="English")
 print(translation)
 
+# Classify text against a taxonomy
+classification = classifier.run_sync("I need help with my invoice")
+print(classification.final_node)
+
 # Validate against guardrails
 validation = validator.run_sync(
     "Share meeting notes with names removed",

{openai_sdk_helpers-0.5.2 → openai_sdk_helpers-0.6.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "openai-sdk-helpers"
-version = "0.5.2"
+version = "0.6.1"
 requires-python = ">=3.10"
 readme = "README.md"
 description = "Composable helpers for OpenAI SDK agents, prompts, and storage"

{openai_sdk_helpers-0.5.2 → openai_sdk_helpers-0.6.1}/src/openai_sdk_helpers/agent/__init__.py

@@ -7,6 +7,7 @@ from ..structure.plan.enum import AgentEnum
 from .coordinator import CoordinatorAgent
 from .runner import run_sync, run_async
 from .search.base import SearchPlanner, SearchToolAgent, SearchWriter
+from .classifier import TaxonomyClassifierAgent
 from .summarizer import SummarizerAgent
 from .translator import TranslatorAgent
 from .validator import ValidatorAgent
@@ -27,6 +28,7 @@ __all__ = [
     "SearchPlanner",
     "SearchToolAgent",
     "SearchWriter",
+    "TaxonomyClassifierAgent",
     "SummarizerAgent",
     "TranslatorAgent",
     "ValidatorAgent",

{openai_sdk_helpers-0.5.2 → openai_sdk_helpers-0.6.1}/src/openai_sdk_helpers/agent/base.py

@@ -429,9 +429,16 @@ class AgentBase(DataclassJSONSerializable):
         """
         return self._session
 
-    def get_agent(self) -> Agent:
+    def get_agent(
+        self, output_structure: Optional[type[StructureBase]] = None
+    ) -> Agent:
         """Construct and return the configured :class:`agents.Agent` instance.
 
+        Parameters
+        ----------
+        output_structure : type[StructureBase] or None, default=None
+            Optional override for the agent output schema.
+
         Returns
         -------
         Agent
@@ -442,8 +449,9 @@ class AgentBase(DataclassJSONSerializable):
             "instructions": self._configuration.instructions_text or ".",
             "model": self._model,
         }
-        if self._configuration.output_structure:
-            agent_config["output_type"] = self._configuration.output_structure
+        output_type = output_structure or self._configuration.output_structure
+        if output_type is not None:
+            agent_config["output_type"] = output_type
         if self._configuration.tools:
             agent_config["tools"] = self._configuration.tools
         if self._model_settings:
@@ -490,7 +498,7 @@ class AgentBase(DataclassJSONSerializable):
         session_to_use = session if session is not None else self._session
         try:
             return await run_async(
-                agent=self.get_agent(),
+                agent=self.get_agent(output_structure=output_structure),
                 input=input,
                 context=context,
                 output_structure=output_structure,
@@ -545,7 +553,7 @@ class AgentBase(DataclassJSONSerializable):
         session_to_use = session if session is not None else self._session
         try:
             return run_sync(
-                agent=self.get_agent(),
+                agent=self.get_agent(output_structure=output_structure),
                 input=input,
                 context=context,
                 output_structure=output_structure,