lite-agent 0.1.0__py3-none-any.whl → 0.2.0__py3-none-any.whl

lite_agent/types.py

@@ -1,152 +0,0 @@
-from typing import Literal, TypedDict
-
-import litellm
-from pydantic import BaseModel
-
-
-class ToolCallFunction(BaseModel):
-    name: str
-    arguments: str | None = None
-
-
-class ToolCall(BaseModel):
-    type: Literal["function"]
-    function: ToolCallFunction
-    id: str
-
-
-class AssistantMessage(BaseModel):
-    id: str
-    role: Literal["assistant"] = "assistant"
-    content: str = ""
-    tool_calls: list[ToolCall] | None = None
-
-
-class Message(TypedDict):
-    role: str
-    content: str
-
-
-class UserMessageContentItemText(TypedDict):
-    type: Literal["text"]
-    text: str
-
-
-class UserMessageContentItemImageURLImageURL(TypedDict):
-    url: str
-
-
-class UserMessageContentItemImageURL(TypedDict):
-    type: Literal["image_url"]
-    image_url: UserMessageContentItemImageURLImageURL
-
-
-class AgentUserMessage(TypedDict):
-    role: Literal["user"] = "user"
-    content: str | list[UserMessageContentItemText | UserMessageContentItemImageURL]
-
-
-class AssistantMessageToolCallFunction(TypedDict):
-    name: str
-    arguments: str
-
-
-class AssistantMessageToolCall(TypedDict):
-    id: str
-    type: Literal["function"]
-    function: AssistantMessageToolCallFunction
-    tool_call_id: str
-
-
-class AgentAssistantMessage(TypedDict):
-    role: Literal["assistant"] = "assistant"
-    content: str
-    tool_calls: list[AssistantMessageToolCall] | None
-
-
-class AgentSystemMessage(TypedDict):
-    role: Literal["system"] = "system"
-    content: str
-
-
-class AgentToolCallMessage(TypedDict):
-    role: Literal["tool"] = "tool"
-    tool_call_id: str
-    content: str
-
-
-RunnerMessage = AgentUserMessage | AgentAssistantMessage | AgentToolCallMessage
-AgentMessage = RunnerMessage | AgentSystemMessage
-RunnerMessages = list[RunnerMessage]
-
-
-class LiteLLMRawChunk(TypedDict):
-    """
-    Define the type of chunk
-    """
-
-    type: Literal["litellm_raw"]
-    raw: litellm.ModelResponseStream
-
-
-class UsageChunk(TypedDict):
-    """
-    Define the type of usage info chunk
-    """
-
-    type: Literal["usage"]
-    usage: litellm.Usage
-
-
-class FinalMessageChunk(TypedDict):
-    """
-    Define the type of final message chunk
-    """
-
-    type: Literal["final_message"]
-    message: AssistantMessage
-    finish_reason: Literal["stop", "tool_calls"]
-
-
-class ToolCallChunk(TypedDict):
-    """
-    Define the type of tool call chunk
-    """
-
-    type: Literal["tool_call"]
-    name: str
-    arguments: str
-
-
-class ToolCallResultChunk(TypedDict):
-    """
-    Define the type of tool call result chunk
-    """
-
-    type: Literal["tool_call_result"]
-    tool_call_id: str
-    name: str
-    content: str
-
-
-class ContentDeltaChunk(TypedDict):
-    """
-    Define the type of message chunk
-    """
-
-    type: Literal["content_delta"]
-    delta: str
-
-
-class ToolCallDeltaChunk(TypedDict):
-    """
-    Define the type of tool call delta chunk
-    """
-
-    type: Literal["tool_call_delta"]
-    tool_call_id: str
-    name: str
-    arguments_delta: str
-
-
-AgentChunk = LiteLLMRawChunk | UsageChunk | FinalMessageChunk | ToolCallChunk | ToolCallResultChunk | ContentDeltaChunk | ToolCallDeltaChunk

lite_agent-0.1.0.dist-info/METADATA

@@ -1,22 +0,0 @@
-Metadata-Version: 2.4
-Name: lite-agent
-Version: 0.1.0
-Summary: A lightweight, extensible framework for building AI agent.
-Author-email: Jianqi Pan <jannchie@gmail.com>
-License: MIT
-Keywords: AI,agent framework,assistant,chatbot,function call,openai,pydantic,rich,tooling
-Classifier: Intended Audience :: Developers
-Classifier: Intended Audience :: Science/Research
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Operating System :: OS Independent
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Classifier: Topic :: Communications :: Chat
-Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
-Classifier: Topic :: Software Development :: Libraries :: Python Modules
-Requires-Python: >=3.10
-Requires-Dist: funcall>=0.6.0
-Requires-Dist: prompt-toolkit>=3.0.51
-Requires-Dist: rich>=14.0.0

lite_agent-0.1.0.dist-info/RECORD

@@ -1,13 +0,0 @@
-lite_agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lite_agent/__main__.py,sha256=D4Yg5a6hDYMq6aCEGg3MjILF4pVWD6mBKjri-qs7Rzk,3766
-lite_agent/agent.py,sha256=LmC5aqMVWsq5AkE18Od0FEVUdiZN1GTs5OlO6mrvxUM,1187
-lite_agent/chunk_handler.py,sha256=MDk9rW_axZ412VOPnITbQGVr-mCGAe2jy3khUXkYbvM,5135
-lite_agent/loggers.py,sha256=0-yb_1Ec8TodfTI_nZWeCJSpSZZM9R53UUOE5--L2iw,57
-lite_agent/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lite_agent/runner.py,sha256=TaBm6pCFwdJtygNnmVZDlFbkK3-Z1s52YD9XviwrA6o,2257
-lite_agent/types.py,sha256=NUpoSDaaiWWYezWluDdWBb-DsYFvZFoNd4tLWqBzCZ4,3058
-lite_agent/processors/__init__.py,sha256=AIxxS9uLcNws1Erf-b8z8zZKaKA5NQ7ooZ7HaCIJTq4,115
-lite_agent/processors/stream_chunk_processor.py,sha256=WlVcfc4kdiCz0Fc1Ybe5GapN2stK0C_0DIymbiGpZfI,3420
-lite_agent-0.1.0.dist-info/METADATA,sha256=oFRlVtYwsr1P9RPOP2JNu309-Jpqc95f-TYrExmrFBc,972
-lite_agent-0.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lite_agent-0.1.0.dist-info/RECORD,,