latitude-sdk 2.0.2__py3-none-any.whl → 2.1.0__py3-none-any.whl

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/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.dist-info → latitude_sdk-2.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: latitude-sdk
-Version: 2.0.2
+Version: 2.1.0
 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

{latitude_sdk-2.0.2.dist-info → latitude_sdk-2.1.0.dist-info}/RECORD

@@ -11,10 +11,10 @@ latitude_sdk/sdk/errors.py,sha256=9GlGdDE8LGy3dE2Ry_BipBg-tDbQx7LWXJfSnTJSSBE,17
 latitude_sdk/sdk/evaluations.py,sha256=fDGtAWjdPG9OuKLit6u-jufVleC1EnshRplK6RN8iyg,2277
 latitude_sdk/sdk/latitude.py,sha256=UpWHxZlaa7vYIiy35Lv12FpjeYqj5EtO8LWO_upxF9c,2626
 latitude_sdk/sdk/logs.py,sha256=CyHkRJvPl_p7wTSvR9bgxEI5akS0Tjc9FeQRb2C2vMg,1997
-latitude_sdk/sdk/prompts.py,sha256=VaBZr58aQtREWFlpZnKD8tWRFgGwy8Q7TP0fLUElC48,16853
-latitude_sdk/sdk/types.py,sha256=_VMTl2BEpdIfKLZexKGALzd6ml4ayYQf6_yly088BXo,8371
+latitude_sdk/sdk/prompts.py,sha256=WztCpDSt0mxng0N2hyIrrzEw1TanqGRIl7Cpc_5ei4M,17369
+latitude_sdk/sdk/types.py,sha256=wGxwFEzHWShy6a2eGUt4NwRkKP_snvSrlH0rYwNYdgY,8568
 latitude_sdk/util/__init__.py,sha256=alIDGBnxWH4JvP-UW-7N99seBBi0r1GV1h8f1ERFBec,21
 latitude_sdk/util/utils.py,sha256=hMOmF-u1QaDgOwXN6ME6n4TaQ70yZKLvijDUqNCMwXI,2844
-latitude_sdk-2.0.2.dist-info/METADATA,sha256=lZrHjAbB6o4BXa2g3H4whmTgahly2Fwa93kWukaQ3J0,2327
-latitude_sdk-2.0.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-latitude_sdk-2.0.2.dist-info/RECORD,,
+latitude_sdk-2.1.0.dist-info/METADATA,sha256=RILx765JUFhrChObo-yVfoLBmYswqHufQ9h3ldqS480,2327
+latitude_sdk-2.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+latitude_sdk-2.1.0.dist-info/RECORD,,