openai-sdk-helpers 0.4.2__py3-none-any.whl → 0.5.0__py3-none-any.whl

openai_sdk_helpers/__init__.py

@@ -2,15 +2,9 @@
 
 from __future__ import annotations
 
-from .environment import get_data_path, get_model
+from .environment import get_data_path
 from .utils.async_utils import run_coroutine_thread_safe, run_coroutine_with_fallback
-from .context_manager import (
-    AsyncManagedResource,
-    ManagedResource,
-    async_context,
-    ensure_closed,
-    ensure_closed_async,
-)
+
 from .errors import (
     OpenAISDKError,
     ConfigurationError,
@@ -22,8 +16,9 @@ from .errors import (
     InputValidationError,
     AsyncExecutionError,
     ResourceCleanupError,
+    ExtractionError,
 )
-from .retry import with_exponential_backoff
+
 from .utils.validation import (
     validate_choice,
     validate_dict_mapping,
@@ -46,12 +41,17 @@ from .structure import (
     ExtendedSummaryStructure,
     ValidationResultStructure,
     AgentBlueprint,
+    AnnotatedDocumentStructure,
+    AttributeStructure,
+    DocumentStructure,
+    ExampleDataStructure,
+    ExtractionStructure,
     create_plan,
     execute_task,
     execute_plan,
 )
 from .prompt import PromptRenderer
-from .config import OpenAISettings
+from .settings import OpenAISettings
 from .files_api import FilesAPIManager, FilePurpose
 from .vector_storage import VectorStorage, VectorStorageFileInfo, VectorStorageFileStats
 from .agent import (
@@ -73,22 +73,17 @@ from .response import (
     ResponseConfiguration,
     ResponseRegistry,
     get_default_registry,
-    parse_tool_arguments,
     attach_vector_store,
 )
 from .tools import (
-    serialize_tool_result,
     tool_handler_factory,
     StructureType,
+    ToolHandler,
+    ToolHandlerRegistration,
     ToolSpec,
-    build_tool_definitions,
-)
-from .config import build_openai_settings
-from .utils.deprecation import (
-    deprecated,
-    warn_deprecated,
-    DeprecationHelper,
+    build_tool_definition_list,
 )
+from .settings import build_openai_settings
 from .utils.output_validation import (
     ValidationResult,
     ValidationRule,
@@ -98,15 +93,21 @@ from .utils.output_validation import (
     OutputValidator,
     validate_output,
 )
-from .types import (
-    SupportsOpenAIClient,
-    OpenAIClient,
+from .utils.langextract import LangExtractAdapter, build_langextract_adapter
+from .extract import (
+    DocumentExtractor,
+    EXTRACTOR_CONFIG_AGENT_INSTRUCTIONS,
+    EXTRACTOR_CONFIG_GENERATOR,
+    PROMPT_OPTIMIZER_AGENT_INSTRUCTIONS,
+    generate_document_extractor_config,
+    generate_document_extractor_config_with_agent,
+    optimize_extractor_prompt,
+    optimize_extractor_prompt_with_agent,
 )
 
 __all__ = [
     # Environment utilities
     "get_data_path",
-    "get_model",
     # Async utilities
     "run_coroutine_thread_safe",
     "run_coroutine_with_fallback",
@@ -121,14 +122,7 @@ __all__ = [
     "InputValidationError",
     "AsyncExecutionError",
     "ResourceCleanupError",
-    # Retry utilities
-    "with_exponential_backoff",
-    # Context managers
-    "ManagedResource",
-    "AsyncManagedResource",
-    "ensure_closed",
-    "ensure_closed_async",
-    "async_context",
+    "ExtractionError",
     # Validation
     "validate_non_empty_string",
     "validate_max_length",
@@ -166,6 +160,11 @@ __all__ = [
     "WebSearchStructure",
     "VectorSearchStructure",
     "ValidationResultStructure",
+    "AnnotatedDocumentStructure",
+    "AttributeStructure",
+    "DocumentStructure",
+    "ExampleDataStructure",
+    "ExtractionStructure",
     "ResponseBase",
     "ResponseMessage",
     "ResponseMessages",
@@ -173,24 +172,17 @@ __all__ = [
     "ResponseConfiguration",
     "ResponseRegistry",
     "get_default_registry",
-    "parse_tool_arguments",
     "attach_vector_store",
-    "serialize_tool_result",
     "tool_handler_factory",
     "StructureType",
+    "ToolHandler",
+    "ToolHandlerRegistration",
     "ToolSpec",
-    "build_tool_definitions",
+    "build_tool_definition_list",
     "build_openai_settings",
     "create_plan",
     "execute_task",
     "execute_plan",
-    # Type definitions
-    "SupportsOpenAIClient",
-    "OpenAIClient",
-    # Deprecation utilities
-    "deprecated",
-    "warn_deprecated",
-    "DeprecationHelper",
     # Output validation
     "ValidationResult",
     "ValidationRule",
@@ -199,4 +191,16 @@ __all__ = [
     "LengthValidator",
     "OutputValidator",
     "validate_output",
+    # LangExtract
+    "LangExtractAdapter",
+    "build_langextract_adapter",
+    # Extraction helpers
+    "DocumentExtractor",
+    "EXTRACTOR_CONFIG_AGENT_INSTRUCTIONS",
+    "EXTRACTOR_CONFIG_GENERATOR",
+    "PROMPT_OPTIMIZER_AGENT_INSTRUCTIONS",
+    "generate_document_extractor_config",
+    "generate_document_extractor_config_with_agent",
+    "optimize_extractor_prompt",
+    "optimize_extractor_prompt_with_agent",
 ]

openai_sdk_helpers/agent/__init__.py

@@ -1,16 +1,15 @@
 """Shared agent helpers built on the OpenAI Agents SDK."""
 
 from __future__ import annotations
-
 from .base import AgentBase
-from .config import AgentConfiguration, AgentRegistry, get_default_registry
+from .configuration import AgentConfiguration, AgentRegistry, get_default_registry
 from ..structure.plan.enum import AgentEnum
-from .coordination import CoordinatorAgent
-from .runner import run_sync, run_async, run_streamed
+from .coordinator import CoordinatorAgent
+from .runner import run_sync, run_async
 from .search.base import SearchPlanner, SearchToolAgent, SearchWriter
 from .summarizer import SummarizerAgent
 from .translator import TranslatorAgent
-from .validation import ValidatorAgent
+from .validator import ValidatorAgent
 from .utils import run_coroutine_agent_sync
 from .search.vector import VectorAgentSearch
 from .search.web import WebAgentSearch
@@ -24,7 +23,6 @@ __all__ = [
     "CoordinatorAgent",
     "run_sync",
     "run_async",
-    "run_streamed",
     "run_coroutine_agent_sync",
     "SearchPlanner",
     "SearchToolAgent",