latitude-sdk 2.0.2__tar.gz → 2.1.1__tar.gz

{latitude_sdk-2.0.2 → latitude_sdk-2.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: latitude-sdk
-Version: 2.0.2
+Version: 2.1.1
 Summary: Latitude SDK for Python
 Project-URL: repository, https://github.com/latitude-dev/latitude-llm/tree/main/packages/sdks/python
 Project-URL: homepage, https://github.com/latitude-dev/latitude-llm/tree/main/packages/sdks/python#readme
@@ -11,7 +11,7 @@ License-Expression: LGPL-3.0
 Requires-Python: <3.13,>=3.9
 Requires-Dist: httpx-sse>=0.4.0
 Requires-Dist: httpx>=0.28.1
-Requires-Dist: promptl-ai>=1.0.5
+Requires-Dist: promptl-ai>=1.0.6
 Requires-Dist: pydantic>=2.10.3
 Requires-Dist: typing-extensions>=4.12.2
 Description-Content-Type: text/markdown

{latitude_sdk-2.0.2 → latitude_sdk-2.1.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "latitude-sdk"
-version = "2.0.2"
+version = "2.1.1"
 description = "Latitude SDK for Python"
 authors = [{ name = "Latitude Data SL", email = "hello@latitude.so" }]
 maintainers = [{ name = "Latitude Data SL", email = "hello@latitude.so" }]
@@ -15,7 +15,7 @@ dependencies = [
     "httpx-sse>=0.4.0",
     "pydantic>=2.10.3",
     "typing-extensions>=4.12.2",
-    "promptl-ai>=1.0.5",
+    "promptl-ai>=1.0.6",
 ]
 
 [dependency-groups]

{latitude_sdk-2.0.2 → latitude_sdk-2.1.1}/src/latitude_sdk/sdk/prompts.py

@@ -1,5 +1,5 @@
 import asyncio
-from typing import Any, AsyncGenerator, List, Optional, Sequence, Union
+from typing import Any, AsyncGenerator, List, Optional, Sequence, Tuple, Union
 
 from promptl_ai import Adapter, Message, MessageLike, Promptl, ToolMessage, ToolResultContent
 from promptl_ai.bindings.types import _Message
@@ -19,6 +19,7 @@ from latitude_sdk.client import (
 )
 from latitude_sdk.sdk.errors import ApiError, ApiErrorCodes
 from latitude_sdk.sdk.types import (
+    AGENT_END_TOOL_NAME,
     ChainEvents,
     FinishedResult,
     OnStep,
@@ -29,7 +30,6 @@ from latitude_sdk.sdk.types import (
     SdkOptions,
     StreamCallbacks,
     StreamEvents,
-    StreamTypes,
     ToolCall,
     ToolResult,
     _LatitudeEvent,
@@ -110,7 +110,7 @@ class RenderPromptOptions(Model):
 
 
 class RenderPromptResult(Model):
-    messages: list[MessageLike]
+    messages: List[MessageLike]
     config: dict[str, Any]
 
 
@@ -142,13 +142,28 @@ class Prompts:
                 response="Project ID is required",
             )
 
+    async def _extract_agent_tool_requests(
+        self, tool_requests: List[ToolCall]
+    ) -> Tuple[List[ToolCall], List[ToolCall]]:
+        agent: List[ToolCall] = []
+        other: List[ToolCall] = []
+
+        for tool in tool_requests:
+            if tool.name == AGENT_END_TOOL_NAME:
+                agent.append(tool)
+            else:
+                other.append(tool)
+
+        return agent, other
+
     async def _handle_stream(
         self, stream: AsyncGenerator[ClientEvent, Any], on_event: Optional[StreamCallbacks.OnEvent]
     ) -> FinishedResult:
         uuid = None
-        conversation: list[Message] = []
+        conversation: List[Message] = []
         response = None
-        tool_requests: list[ToolCall] = []
+        agent_response = None
+        tool_requests: List[ToolCall] = []
 
         async for stream_event in stream:
             event = None
@@ -195,7 +210,17 @@ class Prompts:
                 response="Stream ended without a chain-complete event. Missing uuid or response.",
             )
 
-        return FinishedResult(uuid=uuid, conversation=conversation, response=response, tool_requests=tool_requests)
+        agent_requests, tool_requests = await self._extract_agent_tool_requests(tool_requests)
+        if len(agent_requests) > 0:
+            agent_response = agent_requests[0].arguments
+
+        return FinishedResult(
+            uuid=uuid,
+            conversation=conversation,
+            response=response,
+            agent_response=agent_response,
+            tool_requests=tool_requests,
+        )
 
     @staticmethod
     def _pause_tool_execution() -> Any:
@@ -220,9 +245,6 @@ class Prompts:
     async def _handle_tool_calls(
         self, result: FinishedResult, options: Union[RunPromptOptions, ChatPromptOptions]
     ) -> Optional[FinishedResult]:
-        # Seems Python cannot infer the type
-        assert result.response.type == StreamTypes.Text and result.tool_requests is not None
-
         if not options.tools:
             raise ApiError(
                 status=400,
@@ -350,7 +372,7 @@ class Prompts:
                 else:
                     result = RunPromptResult.model_validate_json(response.content)
 
-            if options.tools and result.response.type == StreamTypes.Text and result.tool_requests:
+            if options.tools and result.tool_requests:
                 try:
                     # NOTE: The last sdk.chat called will already call on_finished
                     final_result = await self._handle_tool_calls(result, options)
@@ -402,7 +424,7 @@ class Prompts:
                 else:
                     result = ChatPromptResult.model_validate_json(response.content)
 
-            if options.tools and result.response.type == StreamTypes.Text and result.tool_requests:
+            if options.tools and result.tool_requests:
                 try:
                     # NOTE: The last sdk.chat called will already call on_finished
                     final_result = await self._handle_tool_calls(result, options)

{latitude_sdk-2.0.2 → latitude_sdk-2.1.1}/src/latitude_sdk/sdk/types.py

@@ -1,5 +1,5 @@
 from datetime import datetime
-from typing import Any, Callable, Literal, Optional, Protocol, Sequence, Union, runtime_checkable
+from typing import Any, Callable, List, Literal, Optional, Protocol, Sequence, Union, runtime_checkable
 
 from promptl_ai import Message, MessageLike
 
@@ -59,6 +59,10 @@ class FinishReason(StrEnum):
     Unknown = "unknown"
 
 
+AGENT_START_TOOL_NAME = "start_autonomous_chain"
+AGENT_END_TOOL_NAME = "end_autonomous_chain"
+
+
 class ToolCall(Model):
     id: str
     name: str
@@ -80,7 +84,7 @@ class StreamTypes(StrEnum):
 class ChainTextResponse(Model):
     type: Literal[StreamTypes.Text] = Field(default=StreamTypes.Text, alias=str("streamType"))
     text: str
-    tool_calls: list[ToolCall] = Field(alias=str("toolCalls"))
+    tool_calls: List[ToolCall] = Field(alias=str("toolCalls"))
     usage: ModelUsage
 
 
@@ -122,7 +126,7 @@ class ChainEvents(StrEnum):
 
 class GenericChainEvent(Model):
     event: Literal[StreamEvents.Latitude] = StreamEvents.Latitude
-    messages: list[Message]
+    messages: List[Message]
     uuid: str
 
 
@@ -149,7 +153,7 @@ class ChainEventProviderCompleted(GenericChainEvent):
 
 class ChainEventToolsStarted(GenericChainEvent):
     type: Literal[ChainEvents.ToolsStarted] = ChainEvents.ToolsStarted
-    tools: list[ToolCall]
+    tools: List[ToolCall]
 
 
 class ChainEventToolCompleted(GenericChainEvent):
@@ -173,7 +177,7 @@ class ChainEventChainError(GenericChainEvent):
 
 class ChainEventToolsRequested(GenericChainEvent):
     type: Literal[ChainEvents.ToolsRequested] = ChainEvents.ToolsRequested
-    tools: list[ToolCall]
+    tools: List[ToolCall]
 
 
 ChainEvent = Union[
@@ -195,9 +199,10 @@ _LatitudeEvent = Adapter[LatitudeEvent](LatitudeEvent)
 
 class FinishedResult(Model):
     uuid: str
-    conversation: list[Message]
+    conversation: List[Message]
     response: ChainResponse
-    tool_requests: list[ToolCall] = Field(alias=str("toolRequests"))
+    agent_response: Optional[dict[str, Any]] = Field(default=None, alias=str("agentResponse"))
+    tool_requests: List[ToolCall] = Field(alias=str("toolRequests"))
 
 
 StreamEvent = Union[ProviderEvent, LatitudeEvent]
@@ -271,9 +276,9 @@ class OnToolCallDetails(Model):
     id: str
     name: str
     conversation_uuid: str
-    messages: list[Message]
+    messages: List[Message]
     pause_execution: Callable[[], ToolResult]
-    requested_tool_calls: list[ToolCall]
+    requested_tool_calls: List[ToolCall]
 
 
 @runtime_checkable
@@ -284,7 +289,7 @@ class OnToolCall(Protocol):
 @runtime_checkable
 class OnStep(Protocol):
     async def __call__(
-        self, messages: list[MessageLike], config: dict[str, Any]
+        self, messages: List[MessageLike], config: dict[str, Any]
     ) -> Union[str, MessageLike, Sequence[MessageLike]]: ...
 
 

{latitude_sdk-2.0.2 → latitude_sdk-2.1.1}/tests/utils/fixtures.py

@@ -1,6 +1,6 @@
 import json
 from datetime import datetime
-from typing import Any, cast
+from typing import Any
 from unittest import mock
 
 from promptl_ai import (
@@ -15,6 +15,7 @@ from promptl_ai import (
 )
 
 from latitude_sdk import (
+    AGENT_END_TOOL_NAME,
     ApiError,
     ApiErrorCodes,
     ChainError,
@@ -510,6 +511,19 @@ data: {
 """.strip(),
     f"""
 event: provider-event
+data: {
+        json.dumps(
+            {
+                "type": "tool-call",
+                "toolCallId": "toolu_K12398312kjadbsadZ77JAS4",
+                "toolName": AGENT_END_TOOL_NAME,
+                "args": {"response": "I used the calculator!"},
+            },
+        )
+    }
+""".strip(),
+    f"""
+event: provider-event
 data: {
         json.dumps(
             {
@@ -567,6 +581,11 @@ data: {
                             "name": "calculator",
                             "arguments": {"expression": "9.9 less than 9.11"},
                         },
+                        {
+                            "id": "toolu_K12398312kjadbsadZ77JAS4",
+                            "name": AGENT_END_TOOL_NAME,
+                            "arguments": {"response": "I used the calculator!"},
+                        },
                     ],
                     "usage": {"promptTokens": 61, "completionTokens": 9, "totalTokens": 70},
                 },
@@ -608,6 +627,11 @@ data: {
                                 "toolName": "calculator",
                                 "args": {"expression": "9.9 less than 9.11"},
                             },
+                            {
+                                "toolCallId": "toolu_K12398312kjadbsadZ77JAS4",
+                                "toolName": AGENT_END_TOOL_NAME,
+                                "args": {"response": "I used the calculator!"},
+                            },
                         ],
                     },
                 ],
@@ -647,6 +671,11 @@ data: {
                         "name": "calculator",
                         "arguments": {"expression": "9.9 less than 9.11"},
                     },
+                    {
+                        "id": "toolu_K12398312kjadbsadZ77JAS4",
+                        "name": AGENT_END_TOOL_NAME,
+                        "arguments": {"response": "I used the calculator!"},
+                    },
                 ],
             },
             ensure_ascii=False,
@@ -698,6 +727,11 @@ data: {
                                 "toolName": "calculator",
                                 "args": {"expression": "9.9 less than 9.11"},
                             },
+                            {
+                                "toolCallId": "toolu_K12398312kjadbsadZ77JAS4",
+                                "toolName": AGENT_END_TOOL_NAME,
+                                "args": {"response": "I used the calculator!"},
+                            },
                         ],
                     },
                 ],
@@ -752,6 +786,11 @@ data: {
                                 "toolName": "calculator",
                                 "args": {"expression": "9.9 less than 9.11"},
                             },
+                            {
+                                "toolCallId": "toolu_K12398312kjadbsadZ77JAS4",
+                                "toolName": AGENT_END_TOOL_NAME,
+                                "args": {"response": "I used the calculator!"},
+                            },
                         ],
                     },
                 ],
@@ -904,6 +943,13 @@ CONVERSATION_EVENTS: list[StreamEvent] = [
         "toolName": "calculator",
         "args": {"expression": "9.9 less than 9.11"},
     },
+    {
+        "event": StreamEvents.Provider,
+        "type": "tool-call",
+        "toolCallId": "toolu_K12398312kjadbsadZ77JAS4",
+        "toolName": AGENT_END_TOOL_NAME,
+        "args": {"response": "I used the calculator!"},
+    },
     {
         "event": StreamEvents.Provider,
         "type": "step-finish",
@@ -941,6 +987,11 @@ CONVERSATION_EVENTS: list[StreamEvent] = [
                     name="calculator",
                     arguments={"expression": "9.9 less than 9.11"},
                 ),
+                ToolCall(
+                    id="toolu_K12398312kjadbsadZ77JAS4",
+                    name=AGENT_END_TOOL_NAME,
+                    arguments={"response": "I used the calculator!"},
+                ),
             ],
             usage=ModelUsage(prompt_tokens=61, completion_tokens=9, total_tokens=70),
         ),
@@ -963,6 +1014,11 @@ CONVERSATION_EVENTS: list[StreamEvent] = [
                         name="calculator",
                         arguments={"expression": "9.9 less than 9.11"},
                     ),
+                    ToolCallContent(
+                        id="toolu_K12398312kjadbsadZ77JAS4",
+                        name=AGENT_END_TOOL_NAME,
+                        arguments={"response": "I used the calculator!"},
+                    ),
                 ]
             ),
         ],
@@ -981,6 +1037,11 @@ CONVERSATION_EVENTS: list[StreamEvent] = [
             ToolCall(
                 id="toolu_B0398l23AOdTDshkg1UuQhZ3", name="calculator", arguments={"expression": "9.9 less than 9.11"}
             ),
+            ToolCall(
+                id="toolu_K12398312kjadbsadZ77JAS4",
+                name=AGENT_END_TOOL_NAME,
+                arguments={"response": "I used the calculator!"},
+            ),
         ],
     ),
     ChainEventStepCompleted(
@@ -1006,6 +1067,11 @@ CONVERSATION_EVENTS: list[StreamEvent] = [
                         name="calculator",
                         arguments={"expression": "9.9 less than 9.11"},
                     ),
+                    ToolCallContent(
+                        id="toolu_K12398312kjadbsadZ77JAS4",
+                        name=AGENT_END_TOOL_NAME,
+                        arguments={"response": "I used the calculator!"},
+                    ),
                 ]
             ),
         ],
@@ -1033,6 +1099,11 @@ CONVERSATION_EVENTS: list[StreamEvent] = [
                         name="calculator",
                         arguments={"expression": "9.9 less than 9.11"},
                     ),
+                    ToolCallContent(
+                        id="toolu_K12398312kjadbsadZ77JAS4",
+                        name=AGENT_END_TOOL_NAME,
+                        arguments={"response": "I used the calculator!"},
+                    ),
                 ]
             ),
         ],
@@ -1126,6 +1197,12 @@ CONVERSATION_FINISHED_RESULT_RESPONSE: dict[str, Any] = {
                     "toolName": "calculator",
                     "args": {"expression": "9.9 less than 9.11"},
                 },
+                {
+                    "type": "tool-call",
+                    "toolCallId": "toolu_K12398312kjadbsadZ77JAS4",
+                    "toolName": AGENT_END_TOOL_NAME,
+                    "args": {"response": "I used the calculator!"},
+                },
             ],
             "toolCalls": [
                 {
@@ -1138,6 +1215,11 @@ CONVERSATION_FINISHED_RESULT_RESPONSE: dict[str, Any] = {
                     "name": "calculator",
                     "arguments": {"expression": "9.9 less than 9.11"},
                 },
+                {
+                    "id": "toolu_K12398312kjadbsadZ77JAS4",
+                    "name": AGENT_END_TOOL_NAME,
+                    "arguments": {"response": "I used the calculator!"},
+                },
             ],
         },
     ],
@@ -1155,9 +1237,15 @@ CONVERSATION_FINISHED_RESULT_RESPONSE: dict[str, Any] = {
                 "name": "calculator",
                 "arguments": {"expression": "9.9 less than 9.11"},
             },
+            {
+                "id": "toolu_K12398312kjadbsadZ77JAS4",
+                "name": AGENT_END_TOOL_NAME,
+                "arguments": {"response": "I used the calculator!"},
+            },
         ],
         "usage": {"promptTokens": 61, "completionTokens": 9, "totalTokens": 70},
     },
+    "agent_response": {"response": "I used the calculator!"},
 }
 
 CONVERSATION_FINISHED_RESULT = FinishedResult(
@@ -1183,6 +1271,11 @@ CONVERSATION_FINISHED_RESULT = FinishedResult(
                     name="calculator",
                     arguments={"expression": "9.9 less than 9.11"},
                 ),
+                ToolCallContent(
+                    id="toolu_K12398312kjadbsadZ77JAS4",
+                    name=AGENT_END_TOOL_NAME,
+                    arguments={"response": "I used the calculator!"},
+                ),
             ]
         ),
     ],
@@ -1199,6 +1292,11 @@ CONVERSATION_FINISHED_RESULT = FinishedResult(
                 name="calculator",
                 arguments={"expression": "9.9 less than 9.11"},
             ),
+            ToolCall(
+                id="toolu_K12398312kjadbsadZ77JAS4",
+                name=AGENT_END_TOOL_NAME,
+                arguments={"response": "I used the calculator!"},
+            ),
         ],
         usage=ModelUsage(prompt_tokens=61, completion_tokens=9, total_tokens=70),
     ),
@@ -1214,11 +1312,10 @@ CONVERSATION_FINISHED_RESULT = FinishedResult(
             arguments={"expression": "9.9 less than 9.11"},
         ),
     ],
+    agent_response={"response": "I used the calculator!"},
 )
 
-CONVERSATION_TOOL_CALLS_MESSAGE = CONVERSATION_FINISHED_RESULT.conversation[-1]
-
-CONVERSATION_TOOL_CALLS = cast(ChainTextResponse, CONVERSATION_FINISHED_RESULT.response).tool_calls
+CONVERSATION_TOOL_CALLS = CONVERSATION_FINISHED_RESULT.tool_requests
 
 CONVERSATION_TOOL_RESULTS = [
     ToolResult(

{latitude_sdk-2.0.2 → latitude_sdk-2.1.1}/uv.lock

@@ -1,5 +1,4 @@
 version = 1
-revision = 1
 requires-python = ">=3.9, <3.13"
 
 [[package]]
@@ -140,7 +139,7 @@ wheels = [
 
 [[package]]
 name = "latitude-sdk"
-version = "2.0.2"
+version = "2.1.1"
 source = { editable = "." }
 dependencies = [
     { name = "httpx" },
@@ -165,7 +164,7 @@ dev = [
 requires-dist = [
     { name = "httpx", specifier = ">=0.28.1" },
     { name = "httpx-sse", specifier = ">=0.4.0" },
-    { name = "promptl-ai", specifier = ">=1.0.5" },
+    { name = "promptl-ai", specifier = ">=1.0.6" },
     { name = "pydantic", specifier = ">=2.10.3" },
     { name = "typing-extensions", specifier = ">=4.12.2" },
 ]
@@ -210,16 +209,16 @@ wheels = [
 
 [[package]]
 name = "promptl-ai"
-version = "1.0.5"
+version = "1.0.6"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "pydantic" },
     { name = "typing-extensions" },
     { name = "wasmtime" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/76/05/b54238992166d23402e44e556d3d09bc7a063942daa4c6eeab8abfff728c/promptl_ai-1.0.5.tar.gz", hash = "sha256:6536f023c119bd365ea85c2c16cb4124c832af691e1377ca8f88337836ec90a3", size = 1013545 }
+sdist = { url = "https://files.pythonhosted.org/packages/81/cf/25354b4f34a028840729b9a33421e6e2adca9ea5920450f63fc606ed4fbd/promptl_ai-1.0.6.tar.gz", hash = "sha256:caf8ab08b77341384cd6f68e68b305447a7ecb86b161acb9a56c6b1d68b6d1b8", size = 1013653 }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/2e/d7/aee27ededbca1d8f11172cefd5f46eac42c2fd8d5299be822853f18a637a/promptl_ai-1.0.5-py3-none-any.whl", hash = "sha256:94513e660d277e69a8f431cc827244951ee4042711e6873b3f442f52f34836a5", size = 1001969 },
+    { url = "https://files.pythonhosted.org/packages/12/3b/64dba92f43a697c04a7fc2a11988bde9d48ef7ced9d232b709d059b25bd1/promptl_ai-1.0.6-py3-none-any.whl", hash = "sha256:d991ef54fb979c789a927fadb6c6f7038ab08635875c355e002ba33025cd1843", size = 1002087 },
 ]
 
 [[package]]