lite-agent 0.6.0__py3-none-any.whl → 0.8.0__py3-none-any.whl

lite_agent/types/__init__.py

@@ -30,6 +30,7 @@ from .messages import (
     AssistantToolCall,
     AssistantToolCallResult,
     BasicMessageMeta,
+    FlexibleInputMessage,
     FlexibleRunnerMessage,
     FunctionCallDict,
     FunctionCallOutputDict,
@@ -86,6 +87,7 @@ __all__ = [
     "CompletionRawEvent",
     "ContentDeltaEvent",
     "EventUsage",
+    "FlexibleInputMessage",
     "FlexibleRunnerMessage",
     "FunctionCallDeltaEvent",
     "FunctionCallDict",

lite_agent/types/messages.py

@@ -46,8 +46,6 @@ class AssistantMessageMeta(MessageMeta):
     total_time_ms: int | None = None
     latency_ms: int | None = None
     output_time_ms: int | None = None
-    input_tokens: int | None = None
-    output_tokens: int | None = None
 
 
 class ResponseInputImageDict(TypedDict):
@@ -273,13 +271,16 @@ class AssistantMessage(BaseModel):
 
 
 # Enhanced type definitions for better type hints
-# Supports new message format, legacy messages, and dict (for backward compatibility)
-FlexibleRunnerMessage = NewMessage | AgentUserMessage | AgentAssistantMessage | dict[str, Any]
+# FlexibleRunnerMessage for internal storage - only NewMessage types
+FlexibleRunnerMessage = NewMessage
 RunnerMessages = Sequence[FlexibleRunnerMessage]
 
+# Input types that can be converted - includes dict for backward compatibility
+FlexibleInputMessage = NewMessage | dict[str, Any]
+InputMessages = Sequence[FlexibleInputMessage]
 
 # Type alias for user input - supports string, single message, or sequence of messages
-UserInput = str | FlexibleRunnerMessage | RunnerMessages
+UserInput = str | FlexibleInputMessage | InputMessages
 
 
 def user_message_to_llm_dict(message: NewUserMessage) -> dict[str, Any]:

lite_agent/utils/message_builder.py

@@ -0,0 +1,211 @@
+import json
+from typing import Any
+
+from lite_agent.types import (
+    AssistantMessageContent,
+    AssistantMessageMeta,
+    AssistantTextContent,
+    AssistantToolCall,
+    AssistantToolCallResult,
+    MessageMeta,
+    NewAssistantMessage,
+    NewSystemMessage,
+    NewUserMessage,
+    UserImageContent,
+    UserMessageContent,
+    UserTextContent,
+)
+
+
+class MessageBuilder:
+    """Utility class for building and converting messages from various formats."""
+
+    @staticmethod
+    def build_user_message_from_dict(message: dict[str, Any]) -> NewUserMessage:
+        """Build a NewUserMessage from a dictionary.
+
+        Args:
+            message: Dictionary containing user message data
+
+        Returns:
+            NewUserMessage instance
+        """
+        content = message.get("content", "")
+
+        # Preserve meta information if present
+        meta_data = message.get("meta", {})
+        meta = MessageMeta(**meta_data) if meta_data else MessageMeta()
+
+        if isinstance(content, str):
+            return NewUserMessage(content=[UserTextContent(text=content)], meta=meta)
+
+        if isinstance(content, list):
+            return NewUserMessage(content=MessageBuilder._build_user_content_items(content), meta=meta)
+
+        # Handle non-string, non-list content
+        return NewUserMessage(content=[UserTextContent(text=str(content))], meta=meta)
+
+    @staticmethod
+    def _build_user_content_items(content_list: list[Any]) -> list[UserMessageContent]:
+        """Build user content items from a list of content data.
+
+        Args:
+            content_list: List of content items (dicts or objects)
+
+        Returns:
+            List of UserMessageContent items
+        """
+        user_content_items: list[UserMessageContent] = []
+
+        for item in content_list:
+            if isinstance(item, dict):
+                user_content_items.append(MessageBuilder._build_user_content_from_dict(item))
+            elif hasattr(item, "type"):
+                user_content_items.append(MessageBuilder._build_user_content_from_object(item))
+            else:
+                # Fallback: convert to text
+                user_content_items.append(UserTextContent(text=str(item)))
+
+        return user_content_items
+
+    @staticmethod
+    def _build_user_content_from_dict(item: dict[str, Any]) -> UserMessageContent:
+        """Build user content from a dictionary item.
+
+        Args:
+            item: Dictionary containing content item data
+
+        Returns:
+            UserMessageContent instance
+        """
+        item_type = item.get("type")
+
+        if item_type in {"input_text", "text"}:
+            return UserTextContent(text=item.get("text", ""))
+
+        if item_type in {"input_image", "image_url"}:
+            if item_type == "image_url":
+                # Handle completion API format
+                image_url_data = item.get("image_url", {})
+                url = image_url_data.get("url", "") if isinstance(image_url_data, dict) else str(image_url_data)
+                return UserImageContent(image_url=url)
+
+            # Handle response API format
+            return UserImageContent(
+                image_url=item.get("image_url"),
+                file_id=item.get("file_id"),
+                detail=item.get("detail", "auto"),
+            )
+
+        # Fallback: treat as text
+        return UserTextContent(text=str(item.get("text", item)))
+
+    @staticmethod
+    def _build_user_content_from_object(item: Any) -> UserMessageContent:  # noqa: ANN401
+        """Build user content from an object with attributes.
+
+        Args:
+            item: Object with type attribute and other properties
+
+        Returns:
+            UserMessageContent instance
+        """
+        if item.type == "input_text":
+            return UserTextContent(text=item.text)
+
+        if item.type == "input_image":
+            return UserImageContent(
+                image_url=getattr(item, "image_url", None),
+                file_id=getattr(item, "file_id", None),
+                detail=getattr(item, "detail", "auto"),
+            )
+
+        # Fallback: convert to text
+        return UserTextContent(text=str(item))
+
+    @staticmethod
+    def build_system_message_from_dict(message: dict[str, Any]) -> NewSystemMessage:
+        """Build a NewSystemMessage from a dictionary.
+
+        Args:
+            message: Dictionary containing system message data
+
+        Returns:
+            NewSystemMessage instance
+        """
+        content = message.get("content", "")
+
+        # Preserve meta information if present
+        meta_data = message.get("meta", {})
+        meta = MessageMeta(**meta_data) if meta_data else MessageMeta()
+
+        return NewSystemMessage(content=str(content), meta=meta)
+
+    @staticmethod
+    def build_assistant_message_from_dict(message: dict[str, Any]) -> NewAssistantMessage:
+        """Build a NewAssistantMessage from a dictionary.
+
+        Args:
+            message: Dictionary containing assistant message data
+
+        Returns:
+            NewAssistantMessage instance
+        """
+        content = message.get("content", "")
+        assistant_content_items: list[AssistantMessageContent] = []
+
+        if content:
+            if isinstance(content, str):
+                assistant_content_items = [AssistantTextContent(text=content)]
+            elif isinstance(content, list):
+                # Handle array content (from new format messages)
+                for item in content:
+                    if isinstance(item, dict):
+                        item_type = item.get("type")
+                        if item_type == "text":
+                            assistant_content_items.append(AssistantTextContent(text=item.get("text", "")))
+                        elif item_type == "tool_call":
+                            assistant_content_items.append(
+                                AssistantToolCall(
+                                    call_id=item.get("call_id", ""),
+                                    name=item.get("name", ""),
+                                    arguments=item.get("arguments", "{}"),
+                                ),
+                            )
+                        elif item_type == "tool_call_result":
+                            assistant_content_items.append(
+                                AssistantToolCallResult(
+                                    call_id=item.get("call_id", ""),
+                                    output=item.get("output", ""),
+                                    execution_time_ms=item.get("execution_time_ms"),
+                                ),
+                            )
+                        # Add more content types as needed
+                    else:
+                        # Fallback for unknown item format
+                        assistant_content_items.append(AssistantTextContent(text=str(item)))
+            else:
+                # Fallback for other content types
+                assistant_content_items = [AssistantTextContent(text=str(content))]
+
+        # Handle tool calls if present (legacy format)
+        if "tool_calls" in message:
+            for tool_call in message.get("tool_calls", []):
+                try:
+                    arguments = json.loads(tool_call["function"]["arguments"]) if isinstance(tool_call["function"]["arguments"], str) else tool_call["function"]["arguments"]
+                except (json.JSONDecodeError, TypeError):
+                    arguments = tool_call["function"]["arguments"]
+
+                assistant_content_items.append(
+                    AssistantToolCall(
+                        call_id=tool_call["id"],
+                        name=tool_call["function"]["name"],
+                        arguments=arguments,
+                    ),
+                )
+
+        # Preserve meta information if present
+        meta_data = message.get("meta", {})
+        meta = AssistantMessageMeta(**meta_data) if meta_data else AssistantMessageMeta()
+
+        return NewAssistantMessage(content=assistant_content_items, meta=meta)

lite_agent/utils/metrics.py

@@ -0,0 +1,50 @@
+from datetime import datetime
+
+
+class TimingMetrics:
+    """Utility class for calculating timing metrics in LLM responses."""
+
+    @staticmethod
+    def calculate_latency_ms(start_time: datetime | None, first_output_time: datetime | None) -> int | None:
+        """Calculate latency from start to first output.
+
+        Args:
+            start_time: When the request started
+            first_output_time: When the first output was received
+
+        Returns:
+            Latency in milliseconds, or None if either time is missing
+        """
+        if start_time is None or first_output_time is None:
+            return None
+        return int((first_output_time - start_time).total_seconds() * 1000)
+
+    @staticmethod
+    def calculate_output_time_ms(first_output_time: datetime | None, output_complete_time: datetime | None) -> int | None:
+        """Calculate time from first output to completion.
+
+        Args:
+            first_output_time: When the first output was received
+            output_complete_time: When output was completed
+
+        Returns:
+            Output time in milliseconds, or None if either time is missing
+        """
+        if first_output_time is None or output_complete_time is None:
+            return None
+        return int((output_complete_time - first_output_time).total_seconds() * 1000)
+
+    @staticmethod
+    def calculate_total_time_ms(start_time: datetime | None, output_complete_time: datetime | None) -> int | None:
+        """Calculate total time from start to completion.
+
+        Args:
+            start_time: When the request started
+            output_complete_time: When output was completed
+
+        Returns:
+            Total time in milliseconds, or None if either time is missing
+        """
+        if start_time is None or output_complete_time is None:
+            return None
+        return int((output_complete_time - start_time).total_seconds() * 1000)

{lite_agent-0.6.0.dist-info → lite_agent-0.8.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lite-agent
-Version: 0.6.0
+Version: 0.8.0
 Summary: A lightweight, extensible framework for building AI agent.
 Author-email: Jianqi Pan <jannchie@gmail.com>
 License: MIT
@@ -19,6 +19,7 @@ Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.10
 Requires-Dist: aiofiles>=24.1.0
 Requires-Dist: funcall>=0.10.0
+Requires-Dist: openai<=1.99.5
 Requires-Dist: prompt-toolkit>=3.0.51
 Requires-Dist: rich>=14.0.0
 Description-Content-Type: text/markdown

lite_agent-0.8.0.dist-info/RECORD

@@ -0,0 +1,31 @@
+lite_agent/__init__.py,sha256=Swuefee0etSiaDnn30K2hBNV9UI3hIValW3A-pRE7e0,338
+lite_agent/agent.py,sha256=XEBgUrR9yqSnqipA70hGVjkrEFy29pl-7lcABL_cDE4,29345
+lite_agent/chat_display.py,sha256=cb8tNCqSoTqyyDc8HXQhEUV9g53FelFMMZ9Pt5WvTn4,31131
+lite_agent/client.py,sha256=PTsic12TVYklUKJzb9gk8-FWl2yhNkSYRS_Cs5dosEU,8677
+lite_agent/constants.py,sha256=_xIDdQwaJrWk8N_62o-KYEo3jj1waPJ0ZOd3hHybKNo,718
+lite_agent/loggers.py,sha256=XkNkdqwD_nQGfhQJ-bBWT7koci_mMkNw3aBpyMhOICw,57
+lite_agent/message_transfers.py,sha256=wdr9ABr_V4TfXAFxO1RBe2uYIj0sFo6xc77SOcGbvVk,4504
+lite_agent/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lite_agent/runner.py,sha256=fHahn060SE5aiEPHFNCLvlyKANpeVO4FgeJ-BylfxHw,42872
+lite_agent/processors/__init__.py,sha256=ybpAzpMBIE9v5I24wIBZRXeaOaPNTmoKH13aofgNI6Q,234
+lite_agent/processors/completion_event_processor.py,sha256=zoWvs8dfrIkCSITGtS-4Hpve3WFCA0UUsMvYifL2fw0,13010
+lite_agent/processors/response_event_processor.py,sha256=-Q5USpv2ZNp4zoHfE9QpwHR2yX3SRkJFjaiXfvchVE8,8417
+lite_agent/response_handlers/__init__.py,sha256=za1pV3DwYhiX4lqo4TafpKlZSBGz81TViWsYebo6Qo4,386
+lite_agent/response_handlers/base.py,sha256=rSs3ImmInBG7aaGv47pYnfHblEs3DIKkT58y7nCUgOc,1682
+lite_agent/response_handlers/completion.py,sha256=e14etxhunRo-T68HgWOE-9JcmMXi5TPvPzyVwE8H7_s,3230
+lite_agent/response_handlers/responses.py,sha256=_Mb2s6RWLYynEncfPis_jRM5WDDEJik8npJvqG6qq9o,3170
+lite_agent/stream_handlers/__init__.py,sha256=a5s1GZr42uvndtcQqEhK2cnjGkK8ZFTAZCj3J61Bb5E,209
+lite_agent/stream_handlers/litellm.py,sha256=3D0u7R2ADA8kDwpFImZlw20o-CsmFXVLvq4nvwwD0Rk,2922
+lite_agent/templates/handoffs_source_instructions.xml.j2,sha256=2XsXQlBzk38qbxGrfyt8y2b0KlZmsV_1xavLufcdkHc,428
+lite_agent/templates/handoffs_target_instructions.xml.j2,sha256=gSbWVYYcovPKbGpFc0kqGSJ5Y5UC3fOHyUmZfcrDgSE,356
+lite_agent/templates/wait_for_user_instructions.xml.j2,sha256=wXbcYD5Q1FaCGVBm3Hz_Cp7nnoK7KzloP0ao-jYMwPk,231
+lite_agent/types/__init__.py,sha256=Zk7Uf1ScOJ5IGfeiJ3PM2FzPD4wmiq-9jzQzLpPpecI,3147
+lite_agent/types/events.py,sha256=mFMqV55WWJbPDyb_P61nd3qMLpEnwZgVY6NTKFkINkg,2389
+lite_agent/types/messages.py,sha256=QjWL2McEWckVy7AIogl2HQkUy-XXdNAAcB0oCNnojbg,9922
+lite_agent/types/tool_calls.py,sha256=Xnut8-2-Ld9vgA2GKJY6BbFlBaAv_n4W7vo7Jx21A-E,260
+lite_agent/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lite_agent/utils/message_builder.py,sha256=3O80MVCAxifDv1BVh4YNg-7XA4udR8m-z8fIcTxfmh8,8008
+lite_agent/utils/metrics.py,sha256=RzOEhCWxbLmmIEkzaxOJ6tAdthI8dv2Foc98Lq8afOQ,1915
+lite_agent-0.8.0.dist-info/METADATA,sha256=K_kN9VfzXcHRmHk4KEeYi3G9CZB4kRU7arZ-EcThmK0,3486
+lite_agent-0.8.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lite_agent-0.8.0.dist-info/RECORD,,

lite_agent-0.6.0.dist-info/RECORD

@@ -1,27 +0,0 @@
-lite_agent/__init__.py,sha256=Swuefee0etSiaDnn30K2hBNV9UI3hIValW3A-pRE7e0,338
-lite_agent/agent.py,sha256=9stxur0iqdG9NUDXdk1ElxenjYsRsurt36hGhZcz_-c,23323
-lite_agent/chat_display.py,sha256=Pfg6ZgTeIuzRZMVxOUzlwZU18rfOLD9-8I1lqUd_fXc,30516
-lite_agent/client.py,sha256=QhtZZGX54ha9-HKHcbx0qUsaAUi4-TAO-YckCH_itQY,8633
-lite_agent/loggers.py,sha256=XkNkdqwD_nQGfhQJ-bBWT7koci_mMkNw3aBpyMhOICw,57
-lite_agent/message_transfers.py,sha256=9qucjc-uSIXvVfhcmVRC_0lp0Q8sWp99dV4ReCh6ZlI,4428
-lite_agent/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lite_agent/runner.py,sha256=B0KYE0Wfta4X85kPm_hMMGrLz8o1-TmGUnOG1cUZeBM,40985
-lite_agent/processors/__init__.py,sha256=ybpAzpMBIE9v5I24wIBZRXeaOaPNTmoKH13aofgNI6Q,234
-lite_agent/processors/completion_event_processor.py,sha256=8fQYRofgBd8t0V3oUakTOmZdv5Q9tCuzADGCGvVgy0k,13442
-lite_agent/processors/response_event_processor.py,sha256=CElJMUzLs8mklVqJtoLiVu-NTq0Dz2NNd9YdAKpjgE0,8088
-lite_agent/response_handlers/__init__.py,sha256=2xe8YngMpjdp1B5tE8C3IiimYI30TnqQHj9KTtg6wCI,385
-lite_agent/response_handlers/base.py,sha256=amQSnhUdoMyaacL7BlGfIUJDYDgqH6seYlfOl6loy-w,1566
-lite_agent/response_handlers/completion.py,sha256=X-sBM-ZBxodppcCXAwoN8Lslda5QYSoK7DdKEdgaYnM,2026
-lite_agent/response_handlers/responses.py,sha256=KEKnnsel8HLiF2Ob8TzVSXuRjudCpvyZ_GMrg3ME2g0,1915
-lite_agent/stream_handlers/__init__.py,sha256=a5s1GZr42uvndtcQqEhK2cnjGkK8ZFTAZCj3J61Bb5E,209
-lite_agent/stream_handlers/litellm.py,sha256=3D0u7R2ADA8kDwpFImZlw20o-CsmFXVLvq4nvwwD0Rk,2922
-lite_agent/templates/handoffs_source_instructions.xml.j2,sha256=2XsXQlBzk38qbxGrfyt8y2b0KlZmsV_1xavLufcdkHc,428
-lite_agent/templates/handoffs_target_instructions.xml.j2,sha256=gSbWVYYcovPKbGpFc0kqGSJ5Y5UC3fOHyUmZfcrDgSE,356
-lite_agent/templates/wait_for_user_instructions.xml.j2,sha256=wXbcYD5Q1FaCGVBm3Hz_Cp7nnoK7KzloP0ao-jYMwPk,231
-lite_agent/types/__init__.py,sha256=QKuhjFWRcpAlsBK9JYgoCABpoQExwhuyGudJoiiqQfs,3093
-lite_agent/types/events.py,sha256=mFMqV55WWJbPDyb_P61nd3qMLpEnwZgVY6NTKFkINkg,2389
-lite_agent/types/messages.py,sha256=c7nTIWqXNo562het_vaWcZvsoy-adkARwAYn4JNqm0c,9897
-lite_agent/types/tool_calls.py,sha256=Xnut8-2-Ld9vgA2GKJY6BbFlBaAv_n4W7vo7Jx21A-E,260
-lite_agent-0.6.0.dist-info/METADATA,sha256=_gfjiwA85XKoQdB9TCJx3BI2D21gNkw-C5pL3CaiZz8,3456
-lite_agent-0.6.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lite_agent-0.6.0.dist-info/RECORD,,