livellm 1.5.2__tar.gz → 1.5.4__tar.gz

{livellm-1.5.2 → livellm-1.5.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: livellm
-Version: 1.5.2
+Version: 1.5.4
 Summary: Python client for the LiveLLM Server
 Project-URL: Homepage, https://github.com/qalby-tech/livellm-client-py
 Project-URL: Repository, https://github.com/qalby-tech/livellm-client-py

{livellm-1.5.2 → livellm-1.5.4}/livellm/models/agent/agent.py

@@ -9,7 +9,7 @@ from ..common import BaseRequest
 
 class AgentRequest(BaseRequest):
     model: str = Field(..., description="The model to use")
-    messages: List[Union[TextMessage, BinaryMessage]] = Field(..., description="The messages to use")
+    messages: List[Union[TextMessage, BinaryMessage, ToolCallMessage, ToolReturnMessage]] = Field(..., description="The messages to use")
     tools: List[Union[WebSearchInput, MCPStreamableServerInput]] = Field(default_factory=list, description="The tools to use")
     gen_config: Optional[dict] = Field(default=None, description="The configuration for the generation")
     include_history: bool = Field(default=False, description="Whether to include full conversation history in the response")

{livellm-1.5.2 → livellm-1.5.4}/livellm/models/agent/chat.py

@@ -1,7 +1,7 @@
 # models for chat messages
 from pydantic import BaseModel, Field, model_validator, field_serializer
 from enum import Enum
-from typing import Optional, Union
+from typing import Optional, Union, Any, Dict
 
 class MessageRole(str, Enum):
     USER = "user"
@@ -32,10 +32,10 @@ class BinaryMessage(Message):
 class ToolCallMessage(Message):
     """Message representing a tool call made by the agent"""
     tool_name: str = Field(..., description="The name of the tool being called")
-    args: dict = Field(..., description="The arguments passed to the tool")
+    args: Union[Dict[str, Any], str] = Field(..., description="The arguments passed to the tool")
 
 class ToolReturnMessage(Message):
     """Message representing the return value from a tool call"""
     tool_name: str = Field(..., description="The name of the tool that was called")
-    content: str = Field(..., description="The return value from the tool")
+    content: Any = Field(..., description="The return value from the tool")
 

{livellm-1.5.2 → livellm-1.5.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "livellm"
-version = "1.5.2"
+version = "1.5.4"
 description = "Python client for the LiveLLM Server"
 readme = "README.md"
 requires-python = ">=3.10"