latitude-sdk 0.1.0b1__tar.gz

latitude_sdk-0.1.0b1/.gitignore

@@ -0,0 +1,141 @@
+# Custom
+.vscode/
+.superinvoke_cache/
+.idea/
+*.tmlanguage.cache
+*.tmPreferences.cache
+*.stTheme.cache
+*.sublime-workspace
+.DS_Store
+*.env
+
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+-
+
+# pdm
+.pdm.toml
+
+# PEP 582
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+venv/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# Ruff
+.ruff_cache/

latitude_sdk-0.1.0b1/.python-version

@@ -0,0 +1 @@
+3.8.20

latitude_sdk-0.1.0b1/PKG-INFO

@@ -0,0 +1,63 @@
+Metadata-Version: 2.4
+Name: latitude-sdk
+Version: 0.1.0b1
+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
+Project-URL: documentation, https://github.com/latitude-dev/latitude-llm/tree/main/packages/sdks/python#readme
+Author-email: Latitude Data SL <hello@latitude.so>
+Maintainer-email: Latitude Data SL <hello@latitude.so>
+License-Expression: LGPL-3.0
+Requires-Python: >=3.8
+Requires-Dist: httpx-sse>=0.4.0
+Requires-Dist: httpx>=0.28.1
+Requires-Dist: pydantic>=2.10.3
+Requires-Dist: typing-extensions>=4.12.2
+Description-Content-Type: text/markdown
+
+# Latitude SDK for Python
+
+```sh
+pip install latitude-sdk
+```
+
+Requires Python `3.8` or higher.
+
+Go to the [documentation](https://docs.latitude.so/guides/sdk/python) to learn more.
+
+## Usage
+
+```python
+from latitude_sdk import Latitude, LatitudeOptions, RunPromptOptions
+
+sdk = Latitude("my-api-key", LatitudeOptions(
+    project_id="my-project-id",
+    version_uuid="my-version-uuid",
+))
+
+await sdk.prompts.run("joke-teller", RunPromptOptions(
+    parameters={"topic": "Python"},
+    on_event=lambda event: print(event),
+    on_finished=lambda event: print(event),
+    on_error=lambda error: print(error),
+    stream=True,
+))
+```
+
+Find more examples [here](https://github.com/latitude-dev/latitude-llm/tree/main/examples/sdks/python).
+
+## Development
+
+Requires uv `0.5.10` or higher.
+
+- Install dependencies: `uv venv && uv sync --all-extras --all-groups`
+- Add [dev] dependencies: `uv add [--dev] <package>`
+- Run linter: `uv run scripts/lint.py`
+- Run formatter: `uv run scripts/format.py`
+- Run tests: `uv run scripts/test.py`
+- Build package: `uv build`
+- Publish package: `uv publish`
+
+## License
+
+The SDK is licensed under the [LGPL-3.0 License](https://opensource.org/licenses/LGPL-3.0) - read the [LICENSE](/LICENSE) file for details.

latitude_sdk-0.1.0b1/README.md

@@ -0,0 +1,46 @@
+# Latitude SDK for Python
+
+```sh
+pip install latitude-sdk
+```
+
+Requires Python `3.8` or higher.
+
+Go to the [documentation](https://docs.latitude.so/guides/sdk/python) to learn more.
+
+## Usage
+
+```python
+from latitude_sdk import Latitude, LatitudeOptions, RunPromptOptions
+
+sdk = Latitude("my-api-key", LatitudeOptions(
+    project_id="my-project-id",
+    version_uuid="my-version-uuid",
+))
+
+await sdk.prompts.run("joke-teller", RunPromptOptions(
+    parameters={"topic": "Python"},
+    on_event=lambda event: print(event),
+    on_finished=lambda event: print(event),
+    on_error=lambda error: print(error),
+    stream=True,
+))
+```
+
+Find more examples [here](https://github.com/latitude-dev/latitude-llm/tree/main/examples/sdks/python).
+
+## Development
+
+Requires uv `0.5.10` or higher.
+
+- Install dependencies: `uv venv && uv sync --all-extras --all-groups`
+- Add [dev] dependencies: `uv add [--dev] <package>`
+- Run linter: `uv run scripts/lint.py`
+- Run formatter: `uv run scripts/format.py`
+- Run tests: `uv run scripts/test.py`
+- Build package: `uv build`
+- Publish package: `uv publish`
+
+## License
+
+The SDK is licensed under the [LGPL-3.0 License](https://opensource.org/licenses/LGPL-3.0) - read the [LICENSE](/LICENSE) file for details.

latitude_sdk-0.1.0b1/examples/all.py

@@ -0,0 +1,120 @@
+import asyncio
+from pprint import pp  # type: ignore
+
+from latitude_sdk import (
+    ApiError,
+    ChatPromptOptions,
+    CreateEvaluationResultOptions,
+    CreateLogOptions,
+    GatewayOptions,
+    GetOrCreatePromptOptions,
+    GetPromptOptions,
+    InternalOptions,
+    Latitude,
+    LatitudeOptions,
+    RunPromptOptions,
+    TextContent,
+    TriggerEvaluationOptions,
+    UserMessage,
+)
+
+
+# TODO Move to root/examples when latitude-sdk is published
+async def main():
+    sdk = Latitude(
+        api_key="6f67407c-da6c-4a4d-9615-a3eb59e51d29",
+        options=LatitudeOptions(
+            project_id=3,
+            version_uuid="57502e00-20c2-4411-8b4b-44bc9008079e",
+            internal=InternalOptions(gateway=GatewayOptions(host="localhost", port=8787, ssl=False, api_version="v2")),
+        ),
+    )
+
+    try:
+        print("Getting prompt...")
+        get_prompt_result = await sdk.prompts.get("prompt", GetPromptOptions())
+        pp(get_prompt_result.model_dump())
+        print("\n" * 2)
+        print("-" * 100)
+
+        print("Getting or creating prompt...")
+        get_or_create_prompt_result = await sdk.prompts.get_or_create("prompt3", GetOrCreatePromptOptions())
+        pp(get_or_create_prompt_result.model_dump())
+        print("\n" * 2)
+        print("-" * 100)
+
+        print("Running prompt...")
+        run_prompt_result = await sdk.prompts.run(
+            "prompt",
+            RunPromptOptions(
+                on_event=lambda event: print(event, "\n" * 2),
+                on_finished=lambda event: print(event, "\n" * 2),
+                on_error=lambda error: print(error, "\n" * 2),
+                custom_identifier="custom!",
+                parameters={"topic": "Python"},
+                stream=True,
+            ),
+        )
+        assert run_prompt_result is not None
+        pp(run_prompt_result.model_dump())
+        print("\n" * 2)
+        print("-" * 100)
+
+        print("Chat prompt...")
+        chat_prompt_result = await sdk.prompts.chat(
+            run_prompt_result.uuid,
+            [
+                UserMessage(content=[TextContent(text="Hello, how are you?")]),
+                UserMessage(content="I'm fine btw"),
+            ],
+            ChatPromptOptions(
+                on_event=lambda event: print(event, "\n" * 2),
+                on_finished=lambda event: print(event, "\n" * 2),
+                on_error=lambda error: print(error, "\n" * 2),
+                stream=True,
+            ),
+        )
+        assert chat_prompt_result is not None
+        pp(chat_prompt_result.model_dump())
+        print("\n" * 2)
+        print("-" * 100)
+
+        print("Create log...")
+        create_log_result = await sdk.logs.create(
+            "prompt",
+            [
+                UserMessage(content=[TextContent(text="Hello, how are you?")]),
+                UserMessage(content=[TextContent(text="I'm fine btw")]),
+            ],
+            CreateLogOptions(),
+        )
+        pp(create_log_result.model_dump())
+        print("\n" * 2)
+        print("-" * 100)
+
+        print("Trigger evaluation...")
+        trigger_evaluation_result = await sdk.evaluations.trigger(
+            chat_prompt_result.uuid,
+            TriggerEvaluationOptions(evaluation_uuids=["46d29f2d-7086-44b8-9220-af1dea1e3692"]),
+        )
+        pp(trigger_evaluation_result.model_dump())
+        print("\n" * 2)
+        print("-" * 100)
+
+        print("Create evaluation result...")
+        create_evaluation_result_result = await sdk.evaluations.create_result(
+            chat_prompt_result.uuid,
+            "d7a04129-9df8-4047-ba93-6349029a1000",
+            CreateEvaluationResultOptions(result="I like it!", reason="Because I like it!"),
+        )
+        pp(create_evaluation_result_result.model_dump())
+        print("\n" * 2)
+        print("-" * 100)
+
+    except ApiError as error:
+        pp(error.__dict__)
+    except Exception as e:
+        raise e
+
+
+asyncio.run(main())

latitude_sdk-0.1.0b1/pyproject.toml

@@ -0,0 +1,64 @@
+[project]
+name = "latitude-sdk"
+version = "0.1.0-beta.1"
+description = "Latitude SDK for Python"
+authors = [{ name = "Latitude Data SL", email = "hello@latitude.so" }]
+maintainers = [{ name = "Latitude Data SL", email = "hello@latitude.so" }]
+readme = "README.md"
+license = "LGPL-3.0"
+urls.repository = "https://github.com/latitude-dev/latitude-llm/tree/main/packages/sdks/python"
+urls.homepage = "https://github.com/latitude-dev/latitude-llm/tree/main/packages/sdks/python#readme"
+urls.documentation = "https://github.com/latitude-dev/latitude-llm/tree/main/packages/sdks/python#readme"
+requires-python = ">=3.8"
+dependencies = [
+    "httpx>=0.28.1",
+    "httpx-sse>=0.4.0",
+    "pydantic>=2.10.3",
+    "typing-extensions>=4.12.2",
+]
+
+[dependency-groups]
+dev = [
+    "pytest-asyncio>=0.24.0",
+    "pytest-xdist>=3.6.1",
+    "pytest>=8.3.4",
+    "pyright>=1.1.391",
+    "ruff>=0.8.3",
+    "sh>=1.14.3",
+]
+
+[tool.pyright]
+pythonVersion = "3.8"
+typeCheckingMode = "strict"
+reportMissingTypeStubs = false
+reportUnnecessaryIsInstance = false
+reportPrivateUsage = false
+
+[tool.ruff]
+target-version = "py38"
+line-length = 120
+indent-width = 4
+
+[tool.ruff.lint]
+select = ["B", "C4", "E", "F", "I", "W", "UP"]
+ignore = [
+    "F401",
+    "F403",
+    # Needed because unnecessary str() on field aliases are needed
+    # https://docs.pydantic.dev/2.8/concepts/fields/#field-aliases
+    "UP018",
+]
+
+[tool.ruff.format]
+quote-style = "double"
+indent-style = "space"
+
+[tool.pytest.ini_options]
+addopts = "-p no:warnings -n auto"
+xfail_strict = true
+asyncio_mode = "auto"
+asyncio_default_fixture_loop_scope = "function"
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"

latitude_sdk-0.1.0b1/scripts/format.py

@@ -0,0 +1,8 @@
+import sys
+
+from sh import ruff  # type: ignore
+
+files = sys.argv[1:] or ["."]
+
+ruff("check", "--fix", *files, _out=sys.stdout)
+ruff("format", *files, _out=sys.stdout)

latitude_sdk-0.1.0b1/scripts/lint.py

@@ -0,0 +1,9 @@
+import sys
+
+from sh import pyright, ruff  # type: ignore
+
+files = sys.argv[1:] or ["."]
+
+pyright(*files, _out=sys.stdout)
+ruff("check", *files, _out=sys.stdout)
+ruff("format", "--check", *files, _out=sys.stdout)

latitude_sdk-0.1.0b1/scripts/test.py

@@ -0,0 +1,7 @@
+import sys
+
+from sh import pytest  # type: ignore
+
+files = sys.argv[1:] or ["."]
+
+pytest(*files, _out=sys.stdout)

latitude_sdk-0.1.0b1/src/latitude_sdk/__init__.py

@@ -0,0 +1 @@
+from .sdk import *

latitude_sdk-0.1.0b1/src/latitude_sdk/client/__init__.py

@@ -0,0 +1,3 @@
+from .client import *
+from .payloads import *
+from .router import *

latitude_sdk-0.1.0b1/src/latitude_sdk/client/client.py

@@ -0,0 +1,140 @@
+import asyncio
+import json
+from contextlib import asynccontextmanager
+from typing import Any, AsyncGenerator, Optional
+
+import httpx
+import httpx_sse
+
+from latitude_sdk.client.payloads import ErrorResponse, RequestBody, RequestHandler, RequestParams
+from latitude_sdk.client.router import Router, RouterOptions
+from latitude_sdk.sdk.errors import (
+    ApiError,
+    ApiErrorCodes,
+    ApiErrorDbRef,
+)
+from latitude_sdk.sdk.types import LogSources
+from latitude_sdk.util import Model
+
+RETRIABLE_STATUSES = [429, 500, 502, 503, 504]
+
+ClientEvent = httpx_sse.ServerSentEvent
+
+
+class ClientResponse(httpx.Response):
+    async def sse(self: httpx.Response) -> AsyncGenerator[ClientEvent, Any]:
+        source = httpx_sse.EventSource(self)
+
+        async for event in source.aiter_sse():
+            yield event
+
+
+httpx.Response.sse = ClientResponse.sse  # pyright: ignore [reportAttributeAccessIssue]
+
+
+class ClientOptions(Model):
+    api_key: str
+    retries: int
+    delay: float
+    timeout: float
+    source: LogSources
+    router: RouterOptions
+
+
+class Client:
+    options: ClientOptions
+    router: Router
+
+    def __init__(self, options: ClientOptions):
+        self.options = options
+        self.router = Router(options.router)
+
+    @asynccontextmanager
+    async def request(
+        self, handler: RequestHandler, params: RequestParams, body: Optional[RequestBody] = None
+    ) -> AsyncGenerator[ClientResponse, Any]:
+        client = httpx.AsyncClient(
+            headers={
+                "Authorization": f"Bearer {self.options.api_key}",
+                "Content-Type": "application/json",
+            },
+            timeout=self.options.timeout,
+            follow_redirects=False,
+            max_redirects=0,
+        )
+        response = None
+        attempt = 1
+
+        try:
+            method, url = self.router.resolve(handler, params)
+            content = None
+            if body:
+                content = json.dumps(
+                    {
+                        **json.loads(body.model_dump_json()),
+                        "__internal": {"source": self.options.source},
+                    }
+                )
+
+            while attempt <= self.options.retries:
+                try:
+                    response = await client.request(method=method, url=url, content=content)
+                    response.raise_for_status()
+
+                    yield response  # pyright: ignore [reportReturnType]
+                    break
+
+                except Exception as exception:
+                    if isinstance(exception, ApiError):
+                        raise exception
+
+                    if attempt >= self.options.retries:
+                        raise await self._exception(exception, response) from exception
+
+                    if response and response.status_code in RETRIABLE_STATUSES:
+                        await asyncio.sleep(self.options.delay * (2 ** (attempt - 1)))
+                    else:
+                        raise await self._exception(exception, response) from exception
+
+                finally:
+                    if response:
+                        await response.aclose()
+
+                    attempt += 1
+
+        except Exception as exception:
+            if isinstance(exception, ApiError):
+                raise exception
+
+            raise await self._exception(exception, response) from exception
+
+        finally:
+            await client.aclose()
+
+    async def _exception(self, exception: Exception, response: Optional[httpx.Response] = None) -> ApiError:
+        if not response:
+            return ApiError(
+                status=500,
+                code=ApiErrorCodes.InternalServerError,
+                message=str(exception),
+                response=str(exception),
+            )
+
+        try:
+            error = ErrorResponse.model_validate_json(response.content)
+
+            return ApiError(
+                status=response.status_code,
+                code=error.code,
+                message=error.message,
+                response=response.text,
+                db_ref=ApiErrorDbRef(**dict(error.db_ref)) if error.db_ref else None,
+            )
+
+        except Exception:
+            return ApiError(
+                status=response.status_code,
+                code=ApiErrorCodes.InternalServerError,
+                message=str(exception),
+                response=response.text,
+            )