latitude-sdk 0.1.0b4__tar.gz → 0.1.0b5__tar.gz

{latitude_sdk-0.1.0b4 → latitude_sdk-0.1.0b5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: latitude-sdk
-Version: 0.1.0b4
+Version: 0.1.0b5
 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-0.1.0b4 → latitude_sdk-0.1.0b5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "latitude-sdk"
-version = "0.1.0-beta.4"
+version = "0.1.0-beta.5"
 description = "Latitude SDK for Python"
 authors = [{ name = "Latitude Data SL", email = "hello@latitude.so" }]
 maintainers = [{ name = "Latitude Data SL", email = "hello@latitude.so" }]

{latitude_sdk-0.1.0b4 → latitude_sdk-0.1.0b5}/src/latitude_sdk/sdk/types.py

@@ -54,15 +54,15 @@ class FileContent(Model):
 
 class ToolCallContent(Model):
     type: Literal[ContentType.ToolCall] = ContentType.ToolCall
-    tool_call_id: str = Field(alias=str("toolCallId"))
-    tool_name: str = Field(alias=str("toolName"))
-    tool_arguments: Dict[str, Any] = Field(alias=str("toolArguments"))
+    id: str = Field(alias=str("toolCallId"))
+    name: str = Field(alias=str("toolName"))
+    arguments: Dict[str, Any] = Field(alias=str("args"))
 
 
 class ToolResultContent(Model):
     type: Literal[ContentType.ToolResult] = ContentType.ToolResult
-    tool_call_id: str = Field(alias=str("toolCallId"))
-    tool_name: str = Field(alias=str("toolName"))
+    id: str = Field(alias=str("toolCallId"))
+    name: str = Field(alias=str("toolName"))
     result: str
     is_error: Optional[bool] = Field(default=None, alias=str("isError"))
 
@@ -95,16 +95,9 @@ class UserMessage(Model):
     name: Optional[str] = None
 
 
-class ToolCall(Model):
-    id: str
-    name: str
-    arguments: Dict[str, Any]
-
-
 class AssistantMessage(Model):
     role: Literal[MessageRole.Assistant] = MessageRole.Assistant
     content: Union[str, List[Union[TextContent, ToolCallContent]]]
-    tool_calls: Optional[List[ToolCall]] = Field(default=None, alias=str("toolCalls"))
 
 
 class ToolMessage(Model):
@@ -121,6 +114,12 @@ class ModelUsage(Model):
     total_tokens: int = Field(alias=str("totalTokens"))
 
 
+class ToolCall(Model):
+    id: str
+    name: str
+    arguments: Dict[str, Any]
+
+
 class StreamTypes(StrEnum):
     Text = "text"
     Object = "object"

{latitude_sdk-0.1.0b4 → latitude_sdk-0.1.0b5}/tests/evaluations/create_result_test.py

@@ -26,7 +26,6 @@ class TestCreateEvaluationResult(TestCase):
             body={
                 "result": options.result,
                 "reason": options.reason,
-                "__internal": {"source": "api"},
             },
         )
         self.assertEqual(endpoint_mock.call_count, 1)
@@ -53,7 +52,6 @@ class TestCreateEvaluationResult(TestCase):
                 body={
                     "result": options.result,
                     "reason": options.reason,
-                    "__internal": {"source": "api"},
                 },
             )
             for request in requests

{latitude_sdk-0.1.0b4 → latitude_sdk-0.1.0b5}/tests/evaluations/trigger_test.py

@@ -24,7 +24,6 @@ class TestTriggerEvaluation(TestCase):
             endpoint=endpoint,
             body={
                 "evaluationUuids": options.evaluation_uuids,
-                "__internal": {"source": "api"},
             },
         )
         self.assertEqual(endpoint_mock.call_count, 1)
@@ -47,9 +46,7 @@ class TestTriggerEvaluation(TestCase):
                 request,
                 method="POST",
                 endpoint=endpoint,
-                body={
-                    "__internal": {"source": "api"},
-                },
+                body={},
             )
             for request in requests
         ]

{latitude_sdk-0.1.0b4 → latitude_sdk-0.1.0b5}/tests/logs/create_test.py

@@ -28,7 +28,6 @@ class TestCreateLog(TestCase):
                 "path": path,
                 "messages": [json.loads(message.model_dump_json()) for message in messages],
                 "response": options.response,
-                "__internal": {"source": "api"},
             },
         )
         self.assertEqual(endpoint_mock.call_count, 1)
@@ -54,7 +53,6 @@ class TestCreateLog(TestCase):
                 "path": path,
                 "messages": [json.loads(message.model_dump_json()) for message in messages],
                 "response": options.response,
-                "__internal": {"source": "api"},
             },
         )
         self.assertEqual(endpoint_mock.call_count, 1)
@@ -81,7 +79,6 @@ class TestCreateLog(TestCase):
                 "path": path,
                 "messages": [json.loads(message.model_dump_json()) for message in messages],
                 "response": options.response,
-                "__internal": {"source": "api"},
             },
         )
         self.assertEqual(endpoint_mock.call_count, 1)
@@ -109,7 +106,6 @@ class TestCreateLog(TestCase):
                     "path": path,
                     "messages": [json.loads(message.model_dump_json()) for message in messages],
                     "response": options.response,
-                    "__internal": {"source": "api"},
                 },
             )
             for request in requests

{latitude_sdk-0.1.0b4 → latitude_sdk-0.1.0b5}/tests/prompts/chat_test.py

@@ -36,7 +36,6 @@ class TestChatPromptSync(TestCase):
             body={
                 "messages": [json.loads(message.model_dump_json()) for message in messages],
                 "stream": options.stream,
-                "__internal": {"source": "api"},
             },
         )
         self.assertEqual(endpoint_mock.call_count, 1)
@@ -73,7 +72,6 @@ class TestChatPromptSync(TestCase):
                 body={
                     "messages": [json.loads(message.model_dump_json()) for message in messages],
                     "stream": options.stream,
-                    "__internal": {"source": "api"},
                 },
             )
             for request in requests
@@ -110,7 +108,6 @@ class TestChatPromptSync(TestCase):
             body={
                 "messages": [json.loads(message.model_dump_json()) for message in messages],
                 "stream": options.stream,
-                "__internal": {"source": "api"},
             },
         )
         self.assertEqual(endpoint_mock.call_count, 1)
@@ -144,7 +141,6 @@ class TestChatPromptSync(TestCase):
             body={
                 "messages": [json.loads(message.model_dump_json()) for message in messages],
                 "stream": options.stream,
-                "__internal": {"source": "api"},
             },
         )
         self.assertEqual(endpoint_mock.call_count, 1)
@@ -183,7 +179,6 @@ class TestChatPromptStream(TestCase):
             body={
                 "messages": [json.loads(message.model_dump_json()) for message in messages],
                 "stream": options.stream,
-                "__internal": {"source": "api"},
             },
         )
         self.assertEqual(endpoint_mock.call_count, 1)
@@ -221,7 +216,6 @@ class TestChatPromptStream(TestCase):
                 body={
                     "messages": [json.loads(message.model_dump_json()) for message in messages],
                     "stream": options.stream,
-                    "__internal": {"source": "api"},
                 },
             )
             for request in requests
@@ -258,7 +252,6 @@ class TestChatPromptStream(TestCase):
             body={
                 "messages": [json.loads(message.model_dump_json()) for message in messages],
                 "stream": options.stream,
-                "__internal": {"source": "api"},
             },
         )
         self.assertEqual(endpoint_mock.call_count, 1)
@@ -292,7 +285,6 @@ class TestChatPromptStream(TestCase):
             body={
                 "messages": [json.loads(message.model_dump_json()) for message in messages],
                 "stream": options.stream,
-                "__internal": {"source": "api"},
             },
         )
         self.assertEqual(endpoint_mock.call_count, 1)

{latitude_sdk-0.1.0b4 → latitude_sdk-0.1.0b5}/tests/prompts/get_or_create_test.py

@@ -25,7 +25,6 @@ class TestGetOrCreatePrompt(TestCase):
             body={
                 "path": path,
                 "prompt": options.prompt,
-                "__internal": {"source": "api"},
             },
         )
         self.assertEqual(endpoint_mock.call_count, 1)
@@ -49,7 +48,6 @@ class TestGetOrCreatePrompt(TestCase):
             body={
                 "path": path,
                 "prompt": options.prompt,
-                "__internal": {"source": "api"},
             },
         )
         self.assertEqual(endpoint_mock.call_count, 1)
@@ -74,7 +72,6 @@ class TestGetOrCreatePrompt(TestCase):
             body={
                 "path": path,
                 "prompt": options.prompt,
-                "__internal": {"source": "api"},
             },
         )
         self.assertEqual(endpoint_mock.call_count, 1)
@@ -100,7 +97,6 @@ class TestGetOrCreatePrompt(TestCase):
                 body={
                     "path": path,
                     "prompt": options.prompt,
-                    "__internal": {"source": "api"},
                 },
             )
             for request in requests

{latitude_sdk-0.1.0b4 → latitude_sdk-0.1.0b5}/tests/prompts/run_test.py

@@ -38,7 +38,6 @@ class TestRunPromptSync(TestCase):
                 "customIdentifier": options.custom_identifier,
                 "parameters": options.parameters,
                 "stream": options.stream,
-                "__internal": {"source": "api"},
             },
         )
         self.assertEqual(endpoint_mock.call_count, 1)
@@ -79,7 +78,6 @@ class TestRunPromptSync(TestCase):
                 "customIdentifier": options.custom_identifier,
                 "parameters": options.parameters,
                 "stream": options.stream,
-                "__internal": {"source": "api"},
             },
         )
         self.assertEqual(endpoint_mock.call_count, 1)
@@ -119,7 +117,6 @@ class TestRunPromptSync(TestCase):
                 "customIdentifier": options.custom_identifier,
                 "parameters": options.parameters,
                 "stream": options.stream,
-                "__internal": {"source": "api"},
             },
         )
         self.assertEqual(endpoint_mock.call_count, 1)
@@ -159,7 +156,6 @@ class TestRunPromptSync(TestCase):
                     "customIdentifier": options.custom_identifier,
                     "parameters": options.parameters,
                     "stream": options.stream,
-                    "__internal": {"source": "api"},
                 },
             )
             for request in requests
@@ -199,7 +195,6 @@ class TestRunPromptSync(TestCase):
                 "customIdentifier": options.custom_identifier,
                 "parameters": options.parameters,
                 "stream": options.stream,
-                "__internal": {"source": "api"},
             },
         )
         self.assertEqual(endpoint_mock.call_count, 1)
@@ -236,7 +231,6 @@ class TestRunPromptSync(TestCase):
                 "customIdentifier": options.custom_identifier,
                 "parameters": options.parameters,
                 "stream": options.stream,
-                "__internal": {"source": "api"},
             },
         )
         self.assertEqual(endpoint_mock.call_count, 1)
@@ -278,7 +272,6 @@ class TestRunPromptStream(TestCase):
                 "customIdentifier": options.custom_identifier,
                 "parameters": options.parameters,
                 "stream": options.stream,
-                "__internal": {"source": "api"},
             },
         )
         self.assertEqual(endpoint_mock.call_count, 1)
@@ -321,7 +314,6 @@ class TestRunPromptStream(TestCase):
                 "customIdentifier": options.custom_identifier,
                 "parameters": options.parameters,
                 "stream": options.stream,
-                "__internal": {"source": "api"},
             },
         )
         self.assertEqual(endpoint_mock.call_count, 1)
@@ -363,7 +355,6 @@ class TestRunPromptStream(TestCase):
                 "customIdentifier": options.custom_identifier,
                 "parameters": options.parameters,
                 "stream": options.stream,
-                "__internal": {"source": "api"},
             },
         )
         self.assertEqual(endpoint_mock.call_count, 1)
@@ -404,7 +395,6 @@ class TestRunPromptStream(TestCase):
                     "customIdentifier": options.custom_identifier,
                     "parameters": options.parameters,
                     "stream": options.stream,
-                    "__internal": {"source": "api"},
                 },
             )
             for request in requests
@@ -444,7 +434,6 @@ class TestRunPromptStream(TestCase):
                 "customIdentifier": options.custom_identifier,
                 "parameters": options.parameters,
                 "stream": options.stream,
-                "__internal": {"source": "api"},
             },
         )
         self.assertEqual(endpoint_mock.call_count, 1)
@@ -481,7 +470,6 @@ class TestRunPromptStream(TestCase):
                 "customIdentifier": options.custom_identifier,
                 "parameters": options.parameters,
                 "stream": options.stream,
-                "__internal": {"source": "api"},
             },
         )
         self.assertEqual(endpoint_mock.call_count, 1)

{latitude_sdk-0.1.0b4 → latitude_sdk-0.1.0b5}/tests/utils/fixtures.py

@@ -24,6 +24,8 @@ from latitude_sdk import (
     StreamEvents,
     SystemMessage,
     TextContent,
+    ToolCall,
+    ToolCallContent,
     UserMessage,
 )
 
@@ -177,7 +179,7 @@ data: {json.dumps({
     "finishReason": "stop",
     "isContinued": False,
     "experimental_providerMetadata": {"openai": {"reasoningTokens": 0, "cachedPromptTokens": 0}},
-    "response": {"timestamp": "2025-01-02T12:29:13.000Z", "modelId": "gpt-4o-mini-2024-07-18"},
+    "response": {"timestamp": "2025-01-02T12:29:13.000Z", "modelId": "gpt-4o-mini-latest"},
     "usage": {"promptTokens": 31, "completionTokens": 9, "totalTokens": 40},
 })}""",
     f"""
@@ -186,7 +188,7 @@ data: {json.dumps({
     "type": "finish",
     "finishReason": "stop",
     "experimental_providerMetadata": {"openai": {"reasoningTokens": 0, "cachedPromptTokens": 0}},
-    "response": {"timestamp": "2025-01-02T12:29:13.000Z", "modelId": "gpt-4o-mini-2024-07-18"},
+    "response": {"timestamp": "2025-01-02T12:29:13.000Z", "modelId": "gpt-4o-mini-latest"},
     "usage": {"promptTokens": 31, "completionTokens": 9, "totalTokens": 40},
 })}""",
     f"""
@@ -199,32 +201,6 @@ data: {json.dumps({
         "text": "I should look at their decimals.",
         "toolCalls": [],
         "usage": {"promptTokens": 31, "completionTokens": 9, "totalTokens": 40},
-        "documentLogUuid": "bf7b0b97-6a3a-4147-b058-2588517dd209",
-        "providerLog": {
-            "id": 248559,
-            "workspaceId": 1,
-            "uuid": "4ae2e019-52ef-4595-9610-b38423c175f5",
-            "documentLogUuid": "bf7b0b97-6a3a-4147-b058-2588517dd209",
-            "providerId": 1,
-            "model": "gpt-4o-mini",
-            "finishReason": "stop",
-            "config": {"provider": "OpenAI", "model": "gpt-4o-mini"},
-            "messages": [
-                {"role": "system", "content": [{"type": "text", "text": "Reason before answering."}]},
-                {"role": "user", "content": [{"type": "text", "text": "My question is: Is 9.9 greater than 9.11?"}]},
-            ],
-            "responseObject": None,
-            "responseText": "I should look at their decimals.",
-            "toolCalls": [],
-            "tokens": 40,
-            "costInMillicents": 9,
-            "duration": 2,
-            "source": "api",
-            "apiKeyId": None,
-            "generatedAt": "2025-01-02T12:29:16.085Z",
-            "createdAt": "2025-01-02T12:29:16.086Z",
-            "updatedAt": "2025-01-02T12:29:16.086Z",
-        },
     },
 })}""",
     f"""
@@ -233,7 +209,26 @@ data: {json.dumps({
     "type": "chain-step",
     "uuid": "bf7b0b97-6a3a-4147-b058-2588517dd209",
     "isLastStep": True,
-    "config": {"provider": "OpenAI", "model": "gpt-4o-mini"},
+    "config": {
+        "provider": "OpenAI",
+        "model": "gpt-4o-mini",
+        "tools": {
+            "calculator": {
+                "description": "Calculates an expression.",
+                "parameters": {
+                    "type": "object",
+                    "properties": {
+                        "expression": {
+                            "type": "string",
+                            "description": "The expression to calculate, e.g., '1 + 1'.",
+                        }
+                    },
+                    "required": ["expression"],
+                    "additionalProperties": False,
+                },
+            },
+        },
+    },
     "messages": [
         {"role": "assistant", "content": [{"type": "text", "text": "I should look at their decimals."}]},
         {"role": "system", "content": [{"type": "text", "text": "Now answer succinctly."}]},
@@ -244,25 +239,33 @@ data: {json.dumps({
 event: provider-event
 data: {json.dumps({
     "type": "text-delta",
-    "textDelta": "Yes, 9.9 is greater than 9.11.",
+    "textDelta": "Yes, 9.9 is greater than 9.11. Use the calculator if you don't believe me.",
+})}""",
+    f"""
+event: provider-event
+data: {json.dumps({
+    "type": "tool-call",
+    "toolCallId": "toolu_01ARatRfRidTDshkg1UuQhW2",
+    "toolName": "calculator",
+    "args": {"expression": "9.9 > 9.11 ?"},
 })}""",
     f"""
 event: provider-event
 data: {json.dumps({
     "type": "step-finish",
-    "finishReason": "stop",
+    "finishReason": "tool-calls",
     "isContinued": False,
     "experimental_providerMetadata": {"openai": {"reasoningTokens": 0, "cachedPromptTokens": 0}},
-    "response": {"timestamp": "2025-01-02T12:29:16.000Z", "modelId": "gpt-4o-mini-2024-07-18"},
+    "response": {"timestamp": "2025-01-02T12:29:16.000Z", "modelId": "gpt-4o-mini-latest"},
     "usage": {"promptTokens": 61, "completionTokens": 9, "totalTokens": 70},
 })}""",
     f"""
 event: provider-event
 data: {json.dumps({
     "type": "finish",
-    "finishReason": "stop",
+    "finishReason": "tool-calls",
     "experimental_providerMetadata": {"openai": {"reasoningTokens": 0, "cachedPromptTokens": 0}},
-    "response": {"timestamp": "2025-01-02T12:29:16.000Z", "modelId": "gpt-4o-mini-2024-07-18"},
+    "response": {"timestamp": "2025-01-02T12:29:16.000Z", "modelId": "gpt-4o-mini-latest"},
     "usage": {"promptTokens": 61, "completionTokens": 9, "totalTokens": 70},
 })}""",
     f"""
@@ -272,38 +275,15 @@ data: {json.dumps({
     "uuid": "bf7b0b97-6a3a-4147-b058-2588517dd209",
     "response": {
         "streamType": "text",
-        "text": "Yes, 9.9 is greater than 9.11.",
-        "toolCalls": [],
+        "text": "Yes, 9.9 is greater than 9.11. Use the calculator if you don't believe me.",
+        "toolCalls": [
+            {
+                "id": "toolu_01ARatRfRidTDshkg1UuQhW2",
+                "name": "calculator",
+                "arguments": {"expression": "9.9 > 9.11 ?"},
+            },
+        ],
         "usage": {"promptTokens": 61, "completionTokens": 9, "totalTokens": 70},
-        "documentLogUuid": "bf7b0b97-6a3a-4147-b058-2588517dd209",
-        "providerLog": {
-            "id": 248560,
-            "workspaceId": 1,
-            "uuid": "5b7c7d44-c99b-4764-af9b-f2c4adf16b42",
-            "documentLogUuid": "bf7b0b97-6a3a-4147-b058-2588517dd209",
-            "providerId": 1,
-            "model": "gpt-4o-mini",
-            "finishReason": "stop",
-            "config": {"provider": "OpenAI", "model": "gpt-4o-mini"},
-            "messages": [
-                {"role": "system", "content": [{"type": "text", "text": "Reason before answering."}]},
-                {"role": "user", "content": [{"type": "text", "text": "My question is: Is 9.9 greater than 9.11?"}]},
-                {"role": "assistant", "content": [{"type": "text", "text": "I should look at their decimals."}]},
-                {"role": "system", "content": [{"type": "text", "text": "Now answer succinctly."}]},
-                {"role": "user", "content": [{"type": "text", "text": "My question was: Is 9.9 greater than 9.11?"}]},
-            ],
-            "responseObject": None,
-            "responseText": "Yes, 9.9 is greater than 9.11.",
-            "toolCalls": [],
-            "tokens": 70,
-            "costInMillicents": 9,
-            "duration": 2,
-            "source": "api",
-            "apiKeyId": None,
-            "generatedAt": "2025-01-02T12:29:16.607Z",
-            "createdAt": "2025-01-02T12:29:16.608Z",
-            "updatedAt": "2025-01-02T12:29:16.608Z",
-        },
     },
 })}""",
     f"""
@@ -311,14 +291,60 @@ event: latitude-event
 data: {json.dumps({
     "type": "chain-complete",
     "uuid": "bf7b0b97-6a3a-4147-b058-2588517dd209",
-    "config": {"provider": "OpenAI", "model": "gpt-4o-mini"},
+    "config": {
+        "provider": "OpenAI",
+        "model": "gpt-4o-mini",
+        "tools": {
+            "calculator": {
+                "description": "Calculates an expression.",
+                "parameters": {
+                    "type": "object",
+                    "properties": {
+                        "expression": {
+                            "type": "string",
+                            "description": "The expression to calculate, e.g., '1 + 1'.",
+                        }
+                    },
+                    "required": ["expression"],
+                    "additionalProperties": False,
+                },
+            },
+        },
+    },
     "messages": [
-        {"role": "assistant", "content": "Yes, 9.9 is greater than 9.11.", "toolCalls": []},
+        {
+            "role": "assistant",
+            "content": [
+                {
+                    "type": "text",
+                    "text": "Yes, 9.9 is greater than 9.11. Use the calculator if you don't believe me.",
+                },
+                {
+                    "type": "tool-call",
+                    "toolCallId": "toolu_01ARatRfRidTDshkg1UuQhW2",
+                    "toolName": "calculator",
+                    "args": {"expression": "9.9 > 9.11 ?"},
+                },
+            ],
+           "toolCalls": [
+                {
+                    "id": "toolu_01ARatRfRidTDshkg1UuQhW2",
+                    "name": "calculator",
+                    "arguments": {"expression": "9.9 > 9.11 ?"},
+                },
+            ],
+        },
     ],
     "response": {
         "streamType": "text",
-        "text": "Yes, 9.9 is greater than 9.11.",
-        "toolCalls": [],
+        "text": "Yes, 9.9 is greater than 9.11. Use the calculator if you don't believe me.",
+        "toolCalls": [
+            {
+                "id": "toolu_01ARatRfRidTDshkg1UuQhW2",
+                "name": "calculator",
+                "arguments": {"expression": "9.9 > 9.11 ?"},
+            },
+        ],
         "usage": {"promptTokens": 61, "completionTokens": 9, "totalTokens": 70},
     },
 })}""",
@@ -351,7 +377,7 @@ CONVERSATION_EVENTS: List[StreamEvent] = [
         "finishReason": "stop",
         "isContinued": False,
         "experimental_providerMetadata": {"openai": {"reasoningTokens": 0, "cachedPromptTokens": 0}},
-        "response": {"timestamp": "2025-01-02T12:29:13.000Z", "modelId": "gpt-4o-mini-2024-07-18"},
+        "response": {"timestamp": "2025-01-02T12:29:13.000Z", "modelId": "gpt-4o-mini-latest"},
         "usage": {"promptTokens": 31, "completionTokens": 9, "totalTokens": 40},
     },
     {
@@ -359,7 +385,7 @@ CONVERSATION_EVENTS: List[StreamEvent] = [
         "type": "finish",
         "finishReason": "stop",
         "experimental_providerMetadata": {"openai": {"reasoningTokens": 0, "cachedPromptTokens": 0}},
-        "response": {"timestamp": "2025-01-02T12:29:13.000Z", "modelId": "gpt-4o-mini-2024-07-18"},
+        "response": {"timestamp": "2025-01-02T12:29:13.000Z", "modelId": "gpt-4o-mini-latest"},
         "usage": {"promptTokens": 31, "completionTokens": 9, "totalTokens": 40},
     },
     ChainEventStepCompleted(
@@ -373,7 +399,26 @@ CONVERSATION_EVENTS: List[StreamEvent] = [
     ChainEventStep(
         uuid="bf7b0b97-6a3a-4147-b058-2588517dd209",
         is_last_step=True,
-        config={"provider": "OpenAI", "model": "gpt-4o-mini"},
+        config={
+            "provider": "OpenAI",
+            "model": "gpt-4o-mini",
+            "tools": {
+                "calculator": {
+                    "description": "Calculates an expression.",
+                    "parameters": {
+                        "type": "object",
+                        "properties": {
+                            "expression": {
+                                "type": "string",
+                                "description": "The expression to calculate, e.g., '1 + 1'.",
+                            }
+                        },
+                        "required": ["expression"],
+                        "additionalProperties": False,
+                    },
+                },
+            },
+        },
         messages=[
             AssistantMessage(content=[TextContent(text="I should look at their decimals.")]),
             SystemMessage(content=[TextContent(text="Now answer succinctly.")]),
@@ -383,43 +428,92 @@ CONVERSATION_EVENTS: List[StreamEvent] = [
     {
         "event": StreamEvents.Provider,
         "type": "text-delta",
-        "textDelta": "Yes, 9.9 is greater than 9.11.",
+        "textDelta": "Yes, 9.9 is greater than 9.11. Use the calculator if you don't believe me.",
+    },
+    {
+        "event": StreamEvents.Provider,
+        "type": "tool-call",
+        "toolCallId": "toolu_01ARatRfRidTDshkg1UuQhW2",
+        "toolName": "calculator",
+        "args": {"expression": "9.9 > 9.11 ?"},
     },
     {
         "event": StreamEvents.Provider,
         "type": "step-finish",
-        "finishReason": "stop",
+        "finishReason": "tool-calls",
         "isContinued": False,
         "experimental_providerMetadata": {"openai": {"reasoningTokens": 0, "cachedPromptTokens": 0}},
-        "response": {"timestamp": "2025-01-02T12:29:16.000Z", "modelId": "gpt-4o-mini-2024-07-18"},
+        "response": {"timestamp": "2025-01-02T12:29:16.000Z", "modelId": "gpt-4o-mini-latest"},
         "usage": {"promptTokens": 61, "completionTokens": 9, "totalTokens": 70},
     },
     {
         "event": StreamEvents.Provider,
         "type": "finish",
-        "finishReason": "stop",
+        "finishReason": "tool-calls",
         "experimental_providerMetadata": {"openai": {"reasoningTokens": 0, "cachedPromptTokens": 0}},
-        "response": {"timestamp": "2025-01-02T12:29:16.000Z", "modelId": "gpt-4o-mini-2024-07-18"},
+        "response": {"timestamp": "2025-01-02T12:29:16.000Z", "modelId": "gpt-4o-mini-latest"},
         "usage": {"promptTokens": 61, "completionTokens": 9, "totalTokens": 70},
     },
     ChainEventStepCompleted(
         uuid="bf7b0b97-6a3a-4147-b058-2588517dd209",
         response=ChainTextResponse(
-            text="Yes, 9.9 is greater than 9.11.",
-            tool_calls=[],
+            text="Yes, 9.9 is greater than 9.11. Use the calculator if you don't believe me.",
+            tool_calls=[
+                ToolCall(
+                    id="toolu_01ARatRfRidTDshkg1UuQhW2",
+                    name="calculator",
+                    arguments={"expression": "9.9 > 9.11 ?"},
+                )
+            ],
             usage=ModelUsage(prompt_tokens=61, completion_tokens=9, total_tokens=70),
         ),
     ),
     ChainEventCompleted(
         uuid="bf7b0b97-6a3a-4147-b058-2588517dd209",
-        config={"provider": "OpenAI", "model": "gpt-4o-mini"},
+        config={
+            "provider": "OpenAI",
+            "model": "gpt-4o-mini",
+            "tools": {
+                "calculator": {
+                    "description": "Calculates an expression.",
+                    "parameters": {
+                        "type": "object",
+                        "properties": {
+                            "expression": {
+                                "type": "string",
+                                "description": "The expression to calculate, e.g., '1 + 1'.",
+                            }
+                        },
+                        "required": ["expression"],
+                        "additionalProperties": False,
+                    },
+                },
+            },
+        },
         messages=[
-            AssistantMessage(content="Yes, 9.9 is greater than 9.11.", tool_calls=[]),
+            AssistantMessage(
+                content=[
+                    TextContent(
+                        text="Yes, 9.9 is greater than 9.11. Use the calculator if you don't believe me.",
+                    ),
+                    ToolCallContent(
+                        id="toolu_01ARatRfRidTDshkg1UuQhW2",
+                        name="calculator",
+                        arguments={"expression": "9.9 > 9.11 ?"},
+                    ),
+                ]
+            ),
         ],
         object=None,
         response=ChainTextResponse(
-            text="Yes, 9.9 is greater than 9.11.",
-            tool_calls=[],
+            text="Yes, 9.9 is greater than 9.11. Use the calculator if you don't believe me.",
+            tool_calls=[
+                ToolCall(
+                    id="toolu_01ARatRfRidTDshkg1UuQhW2",
+                    name="calculator",
+                    arguments={"expression": "9.9 > 9.11 ?"},
+                )
+            ],
             usage=ModelUsage(prompt_tokens=61, completion_tokens=9, total_tokens=70),
         ),
     ),
@@ -469,12 +563,39 @@ CONVERSATION_FINISHED_EVENT_RESPONSE: Dict[str, Any] = {
         {"role": "assistant", "content": [{"type": "text", "text": "I should look at their decimals."}]},
         {"role": "system", "content": [{"type": "text", "text": "Now answer succinctly."}]},
         {"role": "user", "content": [{"type": "text", "text": "My question was: Is 9.9 greater than 9.11?"}]},
-        {"role": "assistant", "content": "Yes, 9.9 is greater than 9.11.", "toolCalls": []},
+        {
+            "role": "assistant",
+            "content": [
+                {
+                    "type": "text",
+                    "text": "Yes, 9.9 is greater than 9.11. Use the calculator if you don't believe me.",
+                },
+                {
+                    "type": "tool-call",
+                    "toolCallId": "toolu_01ARatRfRidTDshkg1UuQhW2",
+                    "toolName": "calculator",
+                    "args": {"expression": "9.9 > 9.11 ?"},
+                },
+            ],
+            "toolCalls": [
+                {
+                    "id": "toolu_01ARatRfRidTDshkg1UuQhW2",
+                    "name": "calculator",
+                    "arguments": {"expression": "9.9 > 9.11 ?"},
+                },
+            ],
+        },
     ],
     "response": {
         "streamType": "text",
-        "text": "Yes, 9.9 is greater than 9.11.",
-        "toolCalls": [],
+        "text": "Yes, 9.9 is greater than 9.11. Use the calculator if you don't believe me.",
+        "toolCalls": [
+            {
+                "id": "toolu_01ARatRfRidTDshkg1UuQhW2",
+                "name": "calculator",
+                "arguments": {"expression": "9.9 > 9.11 ?"},
+            },
+        ],
         "usage": {"promptTokens": 61, "completionTokens": 9, "totalTokens": 70},
     },
 }
@@ -487,11 +608,28 @@ CONVERSATION_FINISHED_EVENT = FinishedEvent(
         AssistantMessage(content=[TextContent(text="I should look at their decimals.")]),
         SystemMessage(content=[TextContent(text="Now answer succinctly.")]),
         UserMessage(content=[TextContent(text="My question was: Is 9.9 greater than 9.11?")]),
-        AssistantMessage(content="Yes, 9.9 is greater than 9.11.", tool_calls=[]),
+        AssistantMessage(
+            content=[
+                TextContent(
+                    text="Yes, 9.9 is greater than 9.11. Use the calculator if you don't believe me.",
+                ),
+                ToolCallContent(
+                    id="toolu_01ARatRfRidTDshkg1UuQhW2",
+                    name="calculator",
+                    arguments={"expression": "9.9 > 9.11 ?"},
+                ),
+            ]
+        ),
     ],
     response=ChainTextResponse(
-        text="Yes, 9.9 is greater than 9.11.",
-        tool_calls=[],
+        text="Yes, 9.9 is greater than 9.11. Use the calculator if you don't believe me.",
+        tool_calls=[
+            ToolCall(
+                id="toolu_01ARatRfRidTDshkg1UuQhW2",
+                name="calculator",
+                arguments={"expression": "9.9 > 9.11 ?"},
+            )
+        ],
         usage=ModelUsage(prompt_tokens=61, completion_tokens=9, total_tokens=70),
     ),
 )

{latitude_sdk-0.1.0b4 → latitude_sdk-0.1.0b5}/tests/utils/utils.py

@@ -13,6 +13,7 @@ from latitude_sdk import (
     InternalOptions,
     Latitude,
     LatitudeOptions,
+    LogSources,
     Message,
     SystemMessage,
     TextContent,
@@ -81,7 +82,10 @@ class TestCase(IsolatedAsyncioTestCase):
             dict(request.headers),
         )
         try:
-            self.assertEqual(json.loads(request.content), body or {})
+            self.assertEqual(
+                json.loads(request.content),
+                {**{"__internal": {"source": LogSources.Api}}, **(body or {})},
+            )
         except json.JSONDecodeError:
             self.assertEqual(None, body)
 
@@ -100,17 +104,17 @@ class TestCase(IsolatedAsyncioTestCase):
                 content=[
                     TextContent(text="assistant message"),
                     ToolCallContent(
-                        tool_call_id="tool call id",
-                        tool_name="tool name",
-                        tool_arguments={"argument_1": "value 1", "argument_2": "value 2"},
+                        id="tool id",
+                        name="tool name",
+                        arguments={"argument_1": "value 1", "argument_2": "value 2"},
                     ),
                 ],
             ),
             ToolMessage(
                 content=[
                     ToolResultContent(
-                        tool_call_id="tool call id",
-                        tool_name="tool name",
+                        id="tool id",
+                        name="tool name",
                         result="tool result",
                         is_error=False,
                     ),

{latitude_sdk-0.1.0b4 → latitude_sdk-0.1.0b5}/uv.lock

@@ -158,7 +158,7 @@ wheels = [
 
 [[package]]
 name = "latitude-sdk"
-version = "0.1.0b4"
+version = "0.1.0b5"
 source = { editable = "." }
 dependencies = [
     { name = "httpx" },