jaf-py 2.4.1__py3-none-any.whl → 2.4.2__py3-none-any.whl

jaf/__init__.py

@@ -16,6 +16,21 @@ from .core.tools import (
 )
 from .core.tracing import ConsoleTraceCollector, TraceCollector
 from .core.types import *
+# Import attachment utilities for convenience
+from .utils.attachments import (
+    make_image_attachment,
+    make_file_attachment,
+    make_document_attachment,
+    validate_attachment,
+    AttachmentValidationError,
+)
+from .utils.document_processor import (
+    extract_document_content,
+    is_document_supported,
+    get_document_description,
+    ProcessedDocument,
+    DocumentProcessingError,
+)
 from .exceptions import (
     A2AException,
     A2AProtocolError,

jaf/core/agent_tool.py

@@ -27,6 +27,7 @@ from .types import (
     RunState,
     RunResult,
     Message,
+    get_text_content,
     ContentRole,
     generate_run_id,
     generate_trace_id,
@@ -201,12 +202,13 @@ def create_agent_tool(
                     return str(result.outcome.output)
                 else:
                     # Fall back to the last assistant message
+                    from .types import get_text_content
                     assistant_messages = [
                         msg for msg in result.final_state.messages 
-                        if msg.role == ContentRole.ASSISTANT and msg.content
+                        if msg.role == ContentRole.ASSISTANT and get_text_content(msg.content)
                     ]
                     if assistant_messages:
-                        return assistant_messages[-1].content
+                        return get_text_content(assistant_messages[-1].content)
                     return "Agent completed successfully but produced no output"
             else:
                 # Error case
@@ -278,8 +280,8 @@ def create_json_output_extractor() -> Callable[[RunResult], str]:
     def json_extractor(run_result: RunResult) -> str:
         # Scan the agent's outputs in reverse order until we find a JSON-like message
         for message in reversed(run_result.final_state.messages):
-            if message.role == ContentRole.ASSISTANT and message.content:
-                content = message.content.strip()
+            if message.role == ContentRole.ASSISTANT and get_text_content(message.content):
+                content = get_text_content(message.content).strip()
                 if content.startswith('{') or content.startswith('['):
                     try:
                         # Validate it's proper JSON

jaf/core/analytics.py

@@ -12,7 +12,7 @@ from typing import Dict, List, Optional, Any, Tuple, Set
 from collections import defaultdict, Counter
 from datetime import datetime, timedelta
 
-from .types import Message, ContentRole, RunState, TraceEvent
+from .types import Message, ContentRole, RunState, TraceEvent, get_text_content
 from .performance import PerformanceMetrics
 
 
@@ -129,10 +129,11 @@ class ConversationAnalyzer:
         total_words = 0
         
         for message in messages:
-            if not message.content:
+            content_text = get_text_content(message.content)
+            if not content_text:
                 continue
             
-            words = message.content.lower().split()
+            words = content_text.lower().split()
             total_words += len(words)
             
             for word in words: