smuod 0.9.2__tar.gz

smuod-0.9.2/.gitignore

@@ -0,0 +1,24 @@
+# Project
+*.env
+output*/
+
+# Python
+.venv/
+__pycache__/
+build/
+dist/
+*.egg-info/
+.mypy_cache/
+.pytest_cache/
+.coverage
+MANIFEST
+
+# Jupyter Notebook
+.ipynb_checkpoints/
+*.ipynb
+
+# IDE
+.vscode/settings.json
+
+# Autogenerated docs
+docs/site

smuod-0.9.2/.pre-commit-config.yaml

@@ -0,0 +1,45 @@
+repos:
+  - repo: https://github.com/astral-sh/uv-pre-commit
+    rev: 0.13.5
+    hooks:
+      # Audit project dependencies for known vulnerabilities
+      - id: uv-audit
+        name: Dependency vulnerability audit
+        args: [--locked, --no-dev]
+        always_run: true
+
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.17.2
+    hooks:
+      # Run the linter, automatically fixing issues when possible.
+      - id: ruff
+        name: Ruff linting
+        types_or: [python, pyi, jupyter]
+        args: [--fix]
+      # Run the formatter.
+      - id: ruff-format
+        name: Ruff formatting
+        types_or: [python, pyi, jupyter]
+
+  - repo: https://github.com/pre-commit/mirrors-mypy
+    rev: v2.4.0
+    hooks:
+      - id: mypy
+        name: Mypy static type checker
+        additional_dependencies:
+          - pydantic
+
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v6.0.0
+    hooks:
+      - id: check-merge-conflict
+      - id: check-added-large-files
+      - id: end-of-file-fixer
+      - id: mixed-line-ending
+        args: ["--fix", "lf"]
+
+  - repo: https://github.com/kynan/nbstripout
+    rev: 0.9.1
+    hooks:
+      - id: nbstripout
+        name: Strip output from Jupyter notebooks

smuod-0.9.2/.python-version

@@ -0,0 +1 @@
+3.14

smuod-0.9.2/LICENSE

@@ -0,0 +1,20 @@
+MIT License
+
+Copyright (c) 2026 Martin Høy (martin.hoy@dnv.com)
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of this
+software and associated documentation files (the "Software"), to deal in the Software
+without restriction, including without limitation the rights to use, copy, modify,
+merge, publish, distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to the following
+conditions:
+
+The above copyright notice and this permission notice shall be included in all copies or
+substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
+INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR
+PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT
+OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
+OTHER DEALINGS IN THE SOFTWARE.

smuod-0.9.2/PKG-INFO

@@ -0,0 +1,222 @@
+Metadata-Version: 2.5
+Name: smuod
+Version: 0.9.2
+Summary: Common utilities for SMUOD
+Author-email: Martin Høy <martin.hoy@dnv.com>
+License-Expression: MIT
+License-File: LICENSE
+Requires-Python: >=3.12
+Requires-Dist: azure-identity>=1.26
+Requires-Dist: colorama>=0.4
+Requires-Dist: logfire[system-metrics]>=5.3
+Requires-Dist: opentelemetry-exporter-otlp-proto-grpc>=1.45
+Requires-Dist: opentelemetry-instrumentation-fastapi>=0.66b1
+Requires-Dist: opentelemetry-instrumentation-httpx>=0.66b1
+Requires-Dist: pydantic-settings>=2
+Provides-Extra: langchain
+Requires-Dist: langchain[openai]>=1.4; extra == 'langchain'
+Provides-Extra: mssql
+Requires-Dist: opentelemetry-instrumentation-sqlalchemy>=0.66b1; extra == 'mssql'
+Requires-Dist: pyodbc>=5.3; extra == 'mssql'
+Requires-Dist: sqlalchemy<2.1.0,>=2.0; extra == 'mssql'
+Provides-Extra: postgres
+Requires-Dist: opentelemetry-instrumentation-sqlalchemy>=0.66b1; extra == 'postgres'
+Requires-Dist: psycopg[binary]>=3.3; extra == 'postgres'
+Requires-Dist: sqlalchemy<2.1.0,>=2.0; extra == 'postgres'
+Provides-Extra: pydantic-ai
+Requires-Dist: httpx2>=2.16; extra == 'pydantic-ai'
+Requires-Dist: pydantic-ai-harness>=0.61; extra == 'pydantic-ai'
+Requires-Dist: pydantic-ai[typesafe]>=2.55; extra == 'pydantic-ai'
+Provides-Extra: snowflake
+Requires-Dist: snowflake-sqlalchemy>=2.1; extra == 'snowflake'
+Provides-Extra: sqlalchemy
+Requires-Dist: opentelemetry-instrumentation-sqlalchemy>=0.66b1; extra == 'sqlalchemy'
+Requires-Dist: sqlalchemy<2.1.0,>=2.0; extra == 'sqlalchemy'
+Description-Content-Type: text/markdown
+
+# smuod - Common utilities for the SMUOD team
+
+Join the team, work SMOUDly.
+
+# About this project
+
+This repository contains common utilities and helpers used across the SMUOD team's
+projects, including Azure Foundry access, database connection management, Logging and
+OpenTelemetry instrumentation.
+
+
+## Installing
+
+You can install this package using `uv`:
+
+```bash
+uv add smuod
+```
+
+## Configuration variables
+
+Each configuration variable is searched for in the following locations (the first value
+wins):
+
+- environment variables in the current shell
+- configuration (.env) file given by environment variable `CONFIG_FILE` or `config.env`
+- defaults set in the `smuod.settings` module.
+
+Relevant environment variables:
+| Environment Variable | Description |
+|--------------------|-------------|
+| `CONFIG_FILE` | Path to the configuration file to load environment variables from (default: `config.env`). |
+| `AZURE_FOUNDRY_NAME` | Name of the Azure Foundry environment to use (default: `smuod-foundry-sweden`) |
+| `LOGLEVEL_CONSOLE` | Log level for console output (default: `INFO`). |
+| `LOGLEVEL_OTEL` | Log level for OpenTelemetry output (default: `INFO`). |
+| `OTEL_EXPORTER_OTLP_ENDPOINT` | Endpoint for the OpenTelemetry OTLP exporter (default: `""`). Example: `http://localhost:4317`. The port number is assumed to be the gRPC port, and the http endpoint is assumed to +1 higher. |
+
+# How to use this package
+
+## Logging and OpenTelemetry instrumentation
+
+Put the following code e.g. in your top level `__init__.py` file of your package.
+
+```python
+from smuod import configure_logging_console, configure_otel
+
+configure_logging_console(__package__)
+configure_otel()
+```
+
+
+## Database engines
+
+`create_db_engine` supports SQL Server (`mssql+pyodbc`), PostgreSQL
+(`postgresql+psycopg`), and Snowflake (`snowflake`) SQLAlchemy URLs.
+Other drivers raise `ValueError`. Install the corresponding extra:
+`smuod[mssql]`, `smuod[postgres]`, or `smuod[snowflake]`.
+
+SQL Server and PostgreSQL use Microsoft Entra tokens. Snowflake uses its
+connector's authentication options directly; the `credential` argument is unused.
+For example, to use Azure workload identity with Snowflake:
+
+```python
+from smuod import create_db_engine
+
+engine = create_db_engine(
+    "snowflake://@account.privatelink/database/schema"
+    "?authenticator=WORKLOAD_IDENTITY&workload_identity_provider=AZURE"
+)
+# Reuse the engine and dispose of it at application shutdown.
+engine.dispose()
+```
+
+## LLM example code
+
+Both the Pydantic AI and Langchain integrations are implmented with a `get_model`
+function to obtain the appropriate model instance. And `get_embedder` for embedding
+models. 
+
+The argument is a string that follows the Pydantic AI naming convention (e.g.
+`provider:model-version`). If `provider` is omitted, the default provider is
+`azure-responses`.
+
+
+### Pydantic AI integration
+
+The Pydantic AI integration is an optional dependency. Install the package with the
+following command:
+
+```bash
+uv add "smuod[pydantic_ai]"
+```
+
+The Pydantic AI client is configured with automatic retries for return codes 429, 502,
+503, 504 and connection failures. It will respect Retry-After headers (capped at 300
+seconds per wait), otherwise it uses exponential backoff. Upon retries, a WARNING will
+be logged. After 5 attempts, the client will stop retrying and raise an exception.
+
+Examples:
+```python
+from smuod.pydantic_ai import get_embedder, get_model
+
+embedder = get_embedder("text-embedding-3-large")
+gpt6 = get_model("gpt-6-luna-2026-09-22")
+grok = get_model("grok-4.7")
+# Decision models
+decision1 = get_model("typesafe:Decision-1")
+openai_decisions = get_model("openai-decisions:gpt-6-luna-2026-09-22")
+```
+
+An example Agent:
+
+```python
+from pydantic_ai import Agent
+
+from smuod.pydantic_ai import get_model
+
+agent = Agent(get_model("gpt-5.6-luna-2026-07-09"))
+response = await agent.run("Hello, world!")
+```
+
+
+An Agent that uses the Microsoft-Decision-1 model:
+```python
+from pydantic_ai import Agent
+
+from smuod.pydantic_ai import get_model
+
+agent = Agent(
+    get_model("typesafe:Decision-1"),
+    output_type=bool,
+    instructions="Does this request contain sensitive information?",
+)
+result = await agent.run("My credit card number is 1234-5678-9012-3456.")
+print(result.output)
+print(result.response.provider_details)
+```
+
+An Agent that uses the OpenAI Decisions API:
+
+```python
+from pydantic_ai import Agent
+
+from smuod.pydantic_ai import get_model
+
+agent = Agent(
+    get_model("openai-decisions:gpt-6-luna-2026-09-22"),
+    output_type=bool,
+    instructions="Does this request contain sensitive information?",
+)
+result = await agent.run("My credit card number is 1234-5678-9012-3456.")
+print(result.output)
+print(result.response.provider_details)
+```
+
+### Langchain integration
+
+The Langchain integration is an optional dependency. Install the package with the
+following command:
+
+```bash
+uv add "smuod[langchain]"
+```
+
+```python
+from langchain_core.output_parsers import StrOutputParser
+
+from smuod.langchain import get_model
+
+llm = get_model("gpt-6-luna-2026-09-22")
+chain = llm | StrOutputParser()
+chain.invoke("Hello, world!")
+```
+
+# Contributing
+
+Contributions to this repository are welcome. Please follow these steps:
+
+1. Clone / fork the repository.
+2. Create a new branch for your feature or bugfix.
+3. Commit your changes with clear and descriptive messages.
+4. Add appropriate tests for your changes.
+5. Push your changes to the remote repository.
+6. Open a pull request against the main repository.
+
+Please ensure that your code follows the existing style and includes appropriate tests.

smuod-0.9.2/README.md

@@ -0,0 +1,186 @@
+# smuod - Common utilities for the SMUOD team
+
+Join the team, work SMOUDly.
+
+# About this project
+
+This repository contains common utilities and helpers used across the SMUOD team's
+projects, including Azure Foundry access, database connection management, Logging and
+OpenTelemetry instrumentation.
+
+
+## Installing
+
+You can install this package using `uv`:
+
+```bash
+uv add smuod
+```
+
+## Configuration variables
+
+Each configuration variable is searched for in the following locations (the first value
+wins):
+
+- environment variables in the current shell
+- configuration (.env) file given by environment variable `CONFIG_FILE` or `config.env`
+- defaults set in the `smuod.settings` module.
+
+Relevant environment variables:
+| Environment Variable | Description |
+|--------------------|-------------|
+| `CONFIG_FILE` | Path to the configuration file to load environment variables from (default: `config.env`). |
+| `AZURE_FOUNDRY_NAME` | Name of the Azure Foundry environment to use (default: `smuod-foundry-sweden`) |
+| `LOGLEVEL_CONSOLE` | Log level for console output (default: `INFO`). |
+| `LOGLEVEL_OTEL` | Log level for OpenTelemetry output (default: `INFO`). |
+| `OTEL_EXPORTER_OTLP_ENDPOINT` | Endpoint for the OpenTelemetry OTLP exporter (default: `""`). Example: `http://localhost:4317`. The port number is assumed to be the gRPC port, and the http endpoint is assumed to +1 higher. |
+
+# How to use this package
+
+## Logging and OpenTelemetry instrumentation
+
+Put the following code e.g. in your top level `__init__.py` file of your package.
+
+```python
+from smuod import configure_logging_console, configure_otel
+
+configure_logging_console(__package__)
+configure_otel()
+```
+
+
+## Database engines
+
+`create_db_engine` supports SQL Server (`mssql+pyodbc`), PostgreSQL
+(`postgresql+psycopg`), and Snowflake (`snowflake`) SQLAlchemy URLs.
+Other drivers raise `ValueError`. Install the corresponding extra:
+`smuod[mssql]`, `smuod[postgres]`, or `smuod[snowflake]`.
+
+SQL Server and PostgreSQL use Microsoft Entra tokens. Snowflake uses its
+connector's authentication options directly; the `credential` argument is unused.
+For example, to use Azure workload identity with Snowflake:
+
+```python
+from smuod import create_db_engine
+
+engine = create_db_engine(
+    "snowflake://@account.privatelink/database/schema"
+    "?authenticator=WORKLOAD_IDENTITY&workload_identity_provider=AZURE"
+)
+# Reuse the engine and dispose of it at application shutdown.
+engine.dispose()
+```
+
+## LLM example code
+
+Both the Pydantic AI and Langchain integrations are implmented with a `get_model`
+function to obtain the appropriate model instance. And `get_embedder` for embedding
+models. 
+
+The argument is a string that follows the Pydantic AI naming convention (e.g.
+`provider:model-version`). If `provider` is omitted, the default provider is
+`azure-responses`.
+
+
+### Pydantic AI integration
+
+The Pydantic AI integration is an optional dependency. Install the package with the
+following command:
+
+```bash
+uv add "smuod[pydantic_ai]"
+```
+
+The Pydantic AI client is configured with automatic retries for return codes 429, 502,
+503, 504 and connection failures. It will respect Retry-After headers (capped at 300
+seconds per wait), otherwise it uses exponential backoff. Upon retries, a WARNING will
+be logged. After 5 attempts, the client will stop retrying and raise an exception.
+
+Examples:
+```python
+from smuod.pydantic_ai import get_embedder, get_model
+
+embedder = get_embedder("text-embedding-3-large")
+gpt6 = get_model("gpt-6-luna-2026-09-22")
+grok = get_model("grok-4.7")
+# Decision models
+decision1 = get_model("typesafe:Decision-1")
+openai_decisions = get_model("openai-decisions:gpt-6-luna-2026-09-22")
+```
+
+An example Agent:
+
+```python
+from pydantic_ai import Agent
+
+from smuod.pydantic_ai import get_model
+
+agent = Agent(get_model("gpt-5.6-luna-2026-07-09"))
+response = await agent.run("Hello, world!")
+```
+
+
+An Agent that uses the Microsoft-Decision-1 model:
+```python
+from pydantic_ai import Agent
+
+from smuod.pydantic_ai import get_model
+
+agent = Agent(
+    get_model("typesafe:Decision-1"),
+    output_type=bool,
+    instructions="Does this request contain sensitive information?",
+)
+result = await agent.run("My credit card number is 1234-5678-9012-3456.")
+print(result.output)
+print(result.response.provider_details)
+```
+
+An Agent that uses the OpenAI Decisions API:
+
+```python
+from pydantic_ai import Agent
+
+from smuod.pydantic_ai import get_model
+
+agent = Agent(
+    get_model("openai-decisions:gpt-6-luna-2026-09-22"),
+    output_type=bool,
+    instructions="Does this request contain sensitive information?",
+)
+result = await agent.run("My credit card number is 1234-5678-9012-3456.")
+print(result.output)
+print(result.response.provider_details)
+```
+
+### Langchain integration
+
+The Langchain integration is an optional dependency. Install the package with the
+following command:
+
+```bash
+uv add "smuod[langchain]"
+```
+
+```python
+from langchain_core.output_parsers import StrOutputParser
+
+from smuod.langchain import get_model
+
+llm = get_model("gpt-6-luna-2026-09-22")
+chain = llm | StrOutputParser()
+chain.invoke("Hello, world!")
+```
+
+# Contributing
+
+Contributions to this repository are welcome. Please follow these steps:
+
+1. Clone / fork the repository.
+2. Create a new branch for your feature or bugfix.
+3. Commit your changes with clear and descriptive messages.
+4. Add appropriate tests for your changes.
+5. Push your changes to the remote repository.
+6. Open a pull request against the main repository.
+
+Please ensure that your code follows the existing style and includes appropriate tests.

smuod-0.9.2/pyproject.toml

@@ -0,0 +1,127 @@
+[project]
+name = "smuod"
+dynamic = ["version"]
+description = "Common utilities for SMUOD"
+readme = "README.md"
+authors = [{ name = "Martin Høy", email = "martin.hoy@dnv.com" }]
+license = "MIT"
+license-files = ["LICENSE"]
+requires-python = ">=3.12"
+dependencies = [
+    # Main dependencies
+    "colorama>=0.4",
+    "azure-identity>=1.26",
+    "pydantic-settings>=2",
+    # Instrumentation dependencies
+    "logfire[system-metrics]>=5.3",
+    "opentelemetry-exporter-otlp-proto-grpc>=1.45",
+    "opentelemetry-instrumentation-httpx>=0.66b1",
+    "opentelemetry-instrumentation-fastapi>=0.66b1",
+]
+
+[project.optional-dependencies]
+sqlalchemy = [
+    "sqlalchemy>=2.0,<2.1.0",
+    "opentelemetry-instrumentation-sqlalchemy>=0.66b1",
+]
+mssql = ["smuod[sqlalchemy]", "pyodbc>=5.3"]
+postgres = ["smuod[sqlalchemy]", "psycopg[binary]>=3.3"]
+snowflake = ["snowflake-sqlalchemy>=2.1"]
+pydantic_ai = [
+    "pydantic-ai[typesafe]>=2.55",
+    "pydantic-ai-harness>=0.61",
+    "httpx2>=2.16",
+]
+langchain = ["langchain[openai]>=1.4"]
+
+[dependency-groups]
+# Development dependencies are not installed in production
+dev = [
+    "pre-commit>=4.1.0",
+    "pytest-cov>=6.0.0",
+    "ipykernel>=6.29.5",
+    "starlette>=1.10.0",
+]
+
+[build-system]
+requires = ["hatchling", "hatch-vcs"]
+build-backend = "hatchling.build"
+
+[tool.hatch.version]
+source = "vcs"
+
+[tool.ruff]
+# Consider packages in the src directory as first-party code
+src = ["src"]
+# Run ruff on Notebooks as well
+extend-include = ["*.ipynb"]
+
+[tool.ruff.lint.per-file-ignores]
+# Allow the use of assert in tests
+"tests/*" = ["S101"]
+
+[tool.ruff.lint]
+select = [
+    "F",   # Pyflakes
+    "E",   # pycodestyle
+    "W",   # pycodestyle
+    "C90", # mccabe
+    "I",   # isort
+    "ICN", # flake8-import-conventions
+    "N",   # pep8-naming
+    "D",   # pydocstyle
+    "UP",  # pyupgrade
+    "ANN", # flake8-annotations
+    "S",   # flake8-bandit
+    "B",   # flake8-bugbear
+    "A",   # flake8-builtins
+    "C4",  # flake8-comprehensions
+    "DTZ", # flake8-datetimez
+    "FA",  # flake8-future-annotations
+    "ISC", # flake8-implicit-str-concat
+    "ICN", # flake8-import-conventions
+    "PT",  # flake8-pytest-style
+    "Q",   # flake8-quotes
+    "RET", # flake8-return
+    "SIM", # flake8-simplify
+    "TID", # flake8-tidy-imports
+    "ARG", # flake8-unused-arguments
+    "PTH", # flake8-use-pathlib
+    "PD",  # pandas-vet
+    "PLC", # Pylint convention
+    "PLE", # Pylint error
+    # "PLR",  # Pylint refactor
+    "NPY",  # NumPy
+    "FAST", # FastAPI
+    "RUF",  # Ruff-specific rules
+]
+ignore = []
+
+[tool.ruff.lint.pydocstyle]
+convention = "google"
+
+
+# mypy configuration, taken from pydantic documentation
+[tool.mypy]
+plugins = ["pydantic.mypy"]
+
+follow_imports = "silent"
+warn_redundant_casts = true
+warn_unused_ignores = true
+disallow_any_generics = true
+check_untyped_defs = true
+no_implicit_reexport = true
+
+disallow_untyped_defs = true
+
+[tool.pydantic-mypy]
+init_forbid_extra = true
+init_typed = true
+warn_required_dynamic_aliases = true
+
+
+# pytest configuration
+[tool.pytest.ini_options]
+addopts = "--cov-report term-missing:skip-covered --cov=smuod"
+# Ignore deprecation warnings from third party packages
+filterwarnings = []

smuod-0.9.2/src/smuod/__init__.py

@@ -0,0 +1,8 @@
+"""Initialization of smuod."""
+
+from smuod.database import create_db_engine
+from smuod.logsetup import configure_logging_console
+from smuod.otelsetup import configure_otel
+
+# List of things to expose from this package.
+__all__ = ["configure_logging_console", "configure_otel", "create_db_engine"]