latitude-sdk 1.0.3__tar.gz → 1.1.0__tar.gz

{latitude_sdk-1.0.3 → latitude_sdk-1.1.0}/PKG-INFO

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

@@ -1,6 +1,6 @@
 [project]
 name = "latitude-sdk"
-version = "1.0.3"
+version = "1.1.0"
 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-1.0.3 → latitude_sdk-1.1.0}/src/latitude_sdk/client/payloads.py

@@ -23,6 +23,10 @@ class GetPromptRequestParams(PromptRequestParams, Model):
     path: str
 
 
+class GetAllPromptRequestParams(PromptRequestParams, Model):
+    pass
+
+
 class GetOrCreatePromptRequestParams(PromptRequestParams, Model):
     pass
 
@@ -90,6 +94,7 @@ class CreateEvaluationResultRequestBody(Model):
 
 RequestParams = Union[
     GetPromptRequestParams,
+    GetAllPromptRequestParams,
     GetOrCreatePromptRequestParams,
     RunPromptRequestParams,
     ChatPromptRequestParams,
@@ -111,6 +116,7 @@ RequestBody = Union[
 
 class RequestHandler(StrEnum):
     GetPrompt = "GET_PROMPT"
+    GetAllPrompts = "GET_ALL_PROMPTS"
     GetOrCreatePrompt = "GET_OR_CREATE_PROMPT"
     RunPrompt = "RUN_PROMPT"
     ChatPrompt = "CHAT_PROMPT"

{latitude_sdk-1.0.3 → latitude_sdk-1.1.0}/src/latitude_sdk/client/router.py

@@ -4,6 +4,7 @@ from latitude_sdk.client.payloads import (
     ChatPromptRequestParams,
     CreateEvaluationResultRequestParams,
     CreateLogRequestParams,
+    GetAllPromptRequestParams,
     GetOrCreatePromptRequestParams,
     GetPromptRequestParams,
     RequestHandler,
@@ -36,6 +37,14 @@ class Router:
                 version_uuid=params.version_uuid,
             ).prompt(params.path)
 
+        if handler == RequestHandler.GetAllPrompts:
+            assert isinstance(params, GetAllPromptRequestParams)
+
+            return "GET", self.prompts(
+                project_id=params.project_id,
+                version_uuid=params.version_uuid,
+            ).all_prompts
+
         elif handler == RequestHandler.GetOrCreatePrompt:
             assert isinstance(params, GetOrCreatePromptRequestParams)
 
@@ -94,6 +103,7 @@ class Router:
 
     class Prompts(Model):
         prompt: Callable[[str], str]
+        all_prompts: str
         get_or_create: str
         run: str
         logs: str
@@ -102,6 +112,7 @@ class Router:
         base_url = f"{self.commits_url(project_id, version_uuid)}/documents"
 
         return self.Prompts(
+            all_prompts=f"{base_url}",
             prompt=lambda path: f"{base_url}/{path}",
             get_or_create=f"{base_url}/get-or-create",
             run=f"{base_url}/run",

{latitude_sdk-1.0.3 → latitude_sdk-1.1.0}/src/latitude_sdk/sdk/prompts.py

@@ -1,5 +1,5 @@
 import asyncio
-from typing import Any, AsyncGenerator, Optional, Sequence, Union
+from typing import Any, AsyncGenerator, List, Optional, Sequence, Union
 
 from promptl_ai import Adapter, Message, MessageLike, Promptl, ToolMessage, ToolResultContent
 from promptl_ai.bindings.types import _Message
@@ -9,6 +9,7 @@ from latitude_sdk.client import (
     ChatPromptRequestParams,
     Client,
     ClientEvent,
+    GetAllPromptRequestParams,
     GetOrCreatePromptRequestBody,
     GetOrCreatePromptRequestParams,
     GetPromptRequestParams,
@@ -33,6 +34,7 @@ from latitude_sdk.sdk.types import (
     ToolResult,
     _LatitudeEvent,
 )
+from latitude_sdk.util import Adapter as AdapterUtil
 from latitude_sdk.util import Model
 
 _PROVIDER_TO_ADAPTER = {
@@ -67,6 +69,13 @@ class GetPromptResult(Prompt, Model):
     pass
 
 
+_GetAllPromptResults = AdapterUtil[List[GetPromptResult]](List[GetPromptResult])
+
+
+class GetAllPromptOptions(PromptOptions, Model):
+    pass
+
+
 class GetOrCreatePromptOptions(PromptOptions, Model):
     prompt: Optional[str] = None
 
@@ -283,6 +292,20 @@ class Prompts:
         ) as response:
             return GetPromptResult.model_validate_json(response.content)
 
+    async def get_all(self, options: Optional[GetAllPromptOptions] = None) -> List[GetPromptResult]:
+        options = GetAllPromptOptions(**{**dict(self._options), **dict(options or {})})
+        self._ensure_prompt_options(options)
+        assert options.project_id is not None
+
+        async with self._client.request(
+            handler=RequestHandler.GetAllPrompts,
+            params=GetAllPromptRequestParams(
+                project_id=options.project_id,
+                version_uuid=options.version_uuid,
+            ),
+        ) as response:
+            return _GetAllPromptResults.validate_json(response.content)
+
     async def get_or_create(
         self, path: str, options: Optional[GetOrCreatePromptOptions] = None
     ) -> GetOrCreatePromptResult:

{latitude_sdk-1.0.3 → latitude_sdk-1.1.0}/src/latitude_sdk/sdk/types.py

@@ -24,11 +24,22 @@ class Providers(StrEnum):
     Custom = "custom"
 
 
+class ParameterType(StrEnum):
+    Text = "text"
+    File = "file"
+    Image = "image"
+
+
+class PromptParameter(Model):
+    type: ParameterType
+
+
 class Prompt(Model):
     uuid: str
     path: str
     content: str
     config: dict[str, Any]
+    parameters: dict[str, PromptParameter]
     provider: Optional[Providers] = None
 
 

latitude_sdk-1.1.0/tests/prompts/get_all_test.py

@@ -0,0 +1,62 @@
+from typing import List, cast
+
+import httpx
+
+from latitude_sdk.sdk.prompts import GetAllPromptOptions, GetPromptResult
+from tests.utils import TestCase, fixtures
+
+
+class TestGetAllPrompts(TestCase):
+    async def test_success_global_options(self):
+        endpoint = f"/projects/{self.project_id}/versions/{self.version_uuid}/documents"
+        endpoint_mock = self.gateway_mock.get(endpoint).mock(
+            return_value=httpx.Response(200, json=[fixtures.PROMPT_RESPONSE])
+        )
+
+        result = await self.sdk.prompts.get_all()
+        request, _ = endpoint_mock.calls.last
+
+        self.assert_requested(request, method="GET", endpoint=endpoint)
+        self.assertEqual(endpoint_mock.call_count, 1)
+        self.assertEqual(result, [GetPromptResult(**dict(fixtures.PROMPT))])
+
+    async def test_success_overrides_options(self):
+        options = GetAllPromptOptions(project_id=21, version_uuid="version-uuid")
+        endpoint = f"/projects/{options.project_id}/versions/{options.version_uuid}/documents"
+        endpoint_mock = self.gateway_mock.get(endpoint).mock(
+            return_value=httpx.Response(200, json=[fixtures.PROMPT_RESPONSE])
+        )
+
+        result = await self.sdk.prompts.get_all(options)
+        request, _ = endpoint_mock.calls.last
+
+        self.assert_requested(request, method="GET", endpoint=endpoint)
+        self.assertEqual(endpoint_mock.call_count, 1)
+        self.assertEqual(result, [GetPromptResult(**dict(fixtures.PROMPT))])
+
+    async def test_success_default_version_uuid(self):
+        self.sdk._options.version_uuid = None  # pyright: ignore [reportPrivateUsage]
+        endpoint = f"/projects/{self.project_id}/versions/live/documents"
+        endpoint_mock = self.gateway_mock.get(endpoint).mock(
+            return_value=httpx.Response(200, json=[fixtures.PROMPT_RESPONSE])
+        )
+
+        result = await self.sdk.prompts.get_all()
+        request, _ = endpoint_mock.calls.last
+
+        self.assert_requested(request, method="GET", endpoint=endpoint)
+        self.assertEqual(endpoint_mock.call_count, 1)
+        self.assertEqual(result, [GetPromptResult(**dict(fixtures.PROMPT))])
+
+    async def test_fails(self):
+        endpoint = f"/projects/{self.project_id}/versions/{self.version_uuid}/documents"
+        endpoint_mock = self.gateway_mock.get(endpoint).mock(
+            return_value=httpx.Response(500, json=fixtures.ERROR_RESPONSE)
+        )
+
+        with self.assertRaisesRegex(type(fixtures.ERROR), fixtures.ERROR.message):
+            await self.sdk.prompts.get_all()
+        requests = cast(List[httpx.Request], [request for request, _ in endpoint_mock.calls])  # type: ignore
+
+        [self.assert_requested(request, method="GET", endpoint=endpoint) for request in requests]
+        self.assertEqual(endpoint_mock.call_count, self.internal_options.retries)

{latitude_sdk-1.0.3 → latitude_sdk-1.1.0}/tests/utils/fixtures.py

@@ -35,7 +35,9 @@ from latitude_sdk import (
     Log,
     LogSources,
     ModelUsage,
+    ParameterType,
     Prompt,
+    PromptParameter,
     Providers,
     StreamEvent,
     StreamEvents,
@@ -86,6 +88,7 @@ topP: 0.9
         "topP": 0.9,
     },
     "provider": "openai",
+    "parameters": {"question": {"type": "text"}},
 }
 
 PROMPT = Prompt(
@@ -116,6 +119,7 @@ topP: 0.9
         "topP": 0.9,
     },
     provider=Providers.OpenAI,
+    parameters={"question": PromptParameter(type=ParameterType.Text)},
 )
 
 

{latitude_sdk-1.0.3 → latitude_sdk-1.1.0}/uv.lock

@@ -316,7 +316,7 @@ wheels = [
 
 [[package]]
 name = "latitude-sdk"
-version = "1.0.3"
+version = "1.1.0"
 source = { editable = "." }
 dependencies = [
     { name = "httpx" },