livellm 1.5.1__tar.gz → 1.5.3__tar.gz

{livellm-1.5.1 → livellm-1.5.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: livellm
-Version: 1.5.1
+Version: 1.5.3
 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.1 → livellm-1.5.3}/livellm/livellm.py

@@ -36,6 +36,7 @@ class BaseLivellmClient(ABC):
         model: str,
         messages: list,
         tools: Optional[list] = None,
+        include_history: bool = False,
         **kwargs
     ) -> AgentResponse:
         ...
@@ -53,6 +54,7 @@ class BaseLivellmClient(ABC):
         model: Optional[str] = None,
         messages: Optional[list] = None,
         tools: Optional[list] = None,
+        include_history: bool = False,
         **kwargs
     ) -> AgentResponse:
         """
@@ -69,7 +71,8 @@ class BaseLivellmClient(ABC):
                provider_uid="...",
                model="gpt-4",
                messages=[TextMessage(...)],
-               tools=[]
+               tools=[],
+               include_history=False
            )
         
         Args:
@@ -79,6 +82,7 @@ class BaseLivellmClient(ABC):
             messages: List of messages
             tools: Optional list of tools
             gen_config: Optional generation configuration
+            include_history: Whether to include full conversation history in the response
             
         Returns:
             AgentResponse with the agent's output
@@ -103,7 +107,8 @@ class BaseLivellmClient(ABC):
             model=model,
             messages=messages,
             tools=tools or [],
-            gen_config=kwargs or None
+            gen_config=kwargs or None,
+            include_history=include_history
         )
         return await self.handle_agent_run(agent_request)
     
@@ -122,6 +127,7 @@ class BaseLivellmClient(ABC):
         model: str,
         messages: list,
         tools: Optional[list] = None,
+        include_history: bool = False,
         **kwargs
     ) -> AsyncIterator[AgentResponse]:
         ...
@@ -139,6 +145,7 @@ class BaseLivellmClient(ABC):
         model: Optional[str] = None,
         messages: Optional[list] = None,
         tools: Optional[list] = None,
+        include_history: bool = False,
         **kwargs
     ) -> AsyncIterator[AgentResponse]:
         """
@@ -157,7 +164,8 @@ class BaseLivellmClient(ABC):
                provider_uid="...",
                model="gpt-4",
                messages=[TextMessage(...)],
-               tools=[]
+               tools=[],
+               include_history=False
            ):
                ...
         
@@ -168,6 +176,7 @@ class BaseLivellmClient(ABC):
             messages: List of messages
             tools: Optional list of tools
             gen_config: Optional generation configuration
+            include_history: Whether to include full conversation history in the response
             
         Returns:
             AsyncIterator of AgentResponse chunks
@@ -192,7 +201,8 @@ class BaseLivellmClient(ABC):
                 model=model,
                 messages=messages,
                 tools=tools or [],
-                gen_config=kwargs or None
+                gen_config=kwargs or None,
+                include_history=include_history
             )
             stream = self.handle_agent_run_stream(agent_request)
         

{livellm-1.5.1 → livellm-1.5.3}/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.1 → livellm-1.5.3}/pyproject.toml

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