aip-agents-binary 0.0.0b2__py3-none-any.whl → 0.5.11__py3-none-any.whl

aip_agents/types/a2a_events.pyi

@@ -1,74 +1,3 @@
-from enum import Enum
-from typing import Any
-from typing_extensions import TypedDict
+from aip_agents.schema.a2a import A2AEvent as A2AEvent, A2AStreamEventType as A2AStreamEventType, ToolCallInfo as ToolCallInfo, ToolResultInfo as ToolResultInfo
 
-class A2AStreamEventType(Enum):
-    """Semantic event types for A2A agent-executor communication.
-
-    These event types preserve the semantic meaning of LangChain message types
-    and eliminate the need for string-based content classification downstream.
-    """
-    STATUS_UPDATE = 'status_update'
-    CONTENT_CHUNK = 'content_chunk'
-    FINAL_RESPONSE = 'final_response'
-    TOOL_CALL = 'tool_call'
-    TOOL_RESULT = 'tool_result'
-    ERROR = 'error'
-
-class A2AEvent(TypedDict):
-    """Structured event data for A2A agent-executor communication.
-
-    This TypedDict replaces the generic dictionary format previously used,
-    providing type safety and semantic clarity for agent streaming events.
-
-    Attributes:
-        event_type: The semantic type of this event (tool call, content, etc.).
-        content: The main text content of the event.
-        metadata: Additional metadata associated with the event.
-        tool_info: Tool-specific information (only present for tool-related events).
-        is_final: Whether this event represents the final output.
-        artifacts: List of artifacts generated by tools or agents.
-        references: Reference data from tools that support reference collection.
-        step_usage: Usage metadata associated with the event.
-        total_usage: Accumulated usage metadata associated with the event.
-    """
-    event_type: A2AStreamEventType
-    content: str
-    metadata: dict[str, Any]
-    tool_info: dict[str, Any] | None
-    is_final: bool
-    artifacts: list[dict[str, Any]] | None
-    references: list[Any] | None
-    step_usage: dict[str, Any] | None
-    total_usage: dict[str, Any] | None
-    thinking_and_activity_info: dict[str, Any] | None
-
-class ToolCallInfo(TypedDict):
-    '''Structured information for tool invocation events.
-
-    This provides a standardized format for tool call information,
-    eliminating the need for JSON encoding/decoding.
-
-    Attributes:
-        tool_calls: List of tool calls with names and arguments.
-        status: Current status of the tool invocation ("running", "finished").
-    '''
-    tool_calls: list[dict[str, Any]]
-    status: str
-
-class ToolResultInfo(TypedDict):
-    """Structured information for tool completion events.
-
-    This provides a standardized format for tool execution results,
-    eliminating the need for JSON encoding/decoding.
-
-    Attributes:
-        name: Name of the executed tool.
-        args: Arguments passed to the tool.
-        output: Output returned by the tool.
-        execution_time: Time taken for tool execution in seconds.
-    """
-    name: str
-    args: dict[str, Any]
-    output: str
-    execution_time: float | None
+__all__ = ['A2AStreamEventType', 'A2AEvent', 'ToolCallInfo', 'ToolResultInfo']

aip_agents/utils/__init__.py

@@ -0,0 +1,79 @@
+# flake8: noqa: F401
+"""AIP Agents Utils.
+
+This module contains utility functions and classes for the AIP Agents package.
+
+Authors:
+    Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
+    Reinhart Linanda (reinhart.linanda@gdplabs.id)
+"""
+
+from aip_agents.utils.artifact_helpers import (
+    create_artifact_command,
+    create_artifact_response,
+    create_error_response,
+    create_multiple_artifacts_command,
+    create_text_artifact_response,
+)
+from aip_agents.utils.file_prompt_utils import augment_query_with_file_paths
+from aip_agents.utils.final_response_builder import assemble_final_response
+from aip_agents.utils.logger import LoggerManager, get_logger
+from aip_agents.utils.reference_helper import (
+    add_references_chunks,
+    serialize_references_for_metadata,
+    validate_references,
+)
+from aip_agents.utils.step_limit_manager import (
+    _DELEGATION_CHAIN_CVAR,
+    _DELEGATION_DEPTH_CVAR,
+    _REMAINING_STEP_BUDGET_CVAR,
+    _STEP_LIMIT_CONFIG_CVAR,
+    StepExecutionContext,
+    StepLimitManager,
+)
+
+# Optional PII imports - only import if dependencies are available
+try:
+    from aip_agents.utils.pii.pii_handler import ToolPIIHandler
+    from aip_agents.utils.pii.pii_helper import (
+        add_pii_mappings,
+        extract_pii_mapping_from_agent_response,
+    )
+
+    _PII_AVAILABLE = True
+except ImportError:
+    _PII_AVAILABLE = False
+
+# Build __all__ conditionally based on available dependencies
+from aip_agents.utils.sse_chunk_transformer import SSEChunkTransformer
+
+__all__ = [
+    "get_logger",
+    "LoggerManager",
+    "create_artifact_response",
+    "create_error_response",
+    "create_artifact_command",
+    "create_multiple_artifacts_command",
+    "create_text_artifact_response",
+    "validate_references",
+    "serialize_references_for_metadata",
+    "add_references_chunks",
+    "assemble_final_response",
+    "augment_query_with_file_paths",
+    "SSEChunkTransformer",
+    "StepExecutionContext",
+    "StepLimitManager",
+    "_DELEGATION_CHAIN_CVAR",
+    "_DELEGATION_DEPTH_CVAR",
+    "_REMAINING_STEP_BUDGET_CVAR",
+    "_STEP_LIMIT_CONFIG_CVAR",
+]
+
+if _PII_AVAILABLE:
+    __all__.extend(
+        [
+            "ToolPIIHandler",
+            "add_pii_mappings",
+            "extract_pii_mapping_from_agent_response",
+        ]
+    )

aip_agents/utils/__init__.pyi

@@ -1,5 +1,11 @@
 from aip_agents.utils.artifact_helpers import create_artifact_command as create_artifact_command, create_artifact_response as create_artifact_response, create_error_response as create_error_response, create_multiple_artifacts_command as create_multiple_artifacts_command, create_text_artifact_response as create_text_artifact_response
-from aip_agents.utils.logger_manager import LoggerManager as LoggerManager
+from aip_agents.utils.file_prompt_utils import augment_query_with_file_paths as augment_query_with_file_paths
+from aip_agents.utils.final_response_builder import assemble_final_response as assemble_final_response
+from aip_agents.utils.logger import LoggerManager as LoggerManager, get_logger as get_logger
+from aip_agents.utils.pii.pii_handler import ToolPIIHandler as ToolPIIHandler
+from aip_agents.utils.pii.pii_helper import add_pii_mappings as add_pii_mappings, extract_pii_mapping_from_agent_response as extract_pii_mapping_from_agent_response
 from aip_agents.utils.reference_helper import add_references_chunks as add_references_chunks, serialize_references_for_metadata as serialize_references_for_metadata, validate_references as validate_references
+from aip_agents.utils.sse_chunk_transformer import SSEChunkTransformer as SSEChunkTransformer
+from aip_agents.utils.step_limit_manager import StepExecutionContext as StepExecutionContext, StepLimitManager as StepLimitManager, _DELEGATION_CHAIN_CVAR as _DELEGATION_CHAIN_CVAR, _DELEGATION_DEPTH_CVAR as _DELEGATION_DEPTH_CVAR, _REMAINING_STEP_BUDGET_CVAR as _REMAINING_STEP_BUDGET_CVAR, _STEP_LIMIT_CONFIG_CVAR as _STEP_LIMIT_CONFIG_CVAR
 
-__all__ = ['LoggerManager', 'create_artifact_response', 'create_error_response', 'create_artifact_command', 'create_multiple_artifacts_command', 'create_text_artifact_response', 'validate_references', 'serialize_references_for_metadata', 'add_references_chunks']
+__all__ = ['get_logger', 'LoggerManager', 'create_artifact_response', 'create_error_response', 'create_artifact_command', 'create_multiple_artifacts_command', 'create_text_artifact_response', 'validate_references', 'serialize_references_for_metadata', 'add_references_chunks', 'assemble_final_response', 'augment_query_with_file_paths', 'SSEChunkTransformer', 'StepExecutionContext', 'StepLimitManager', '_DELEGATION_CHAIN_CVAR', '_DELEGATION_DEPTH_CVAR', '_REMAINING_STEP_BUDGET_CVAR', '_STEP_LIMIT_CONFIG_CVAR', 'ToolPIIHandler', 'add_pii_mappings', 'extract_pii_mapping_from_agent_response']