feat: initial release

.flake8

@@ -4,6 +4,7 @@
 # E203 whitespace before ':' (triggered on list slices like xs[i : i + 5])
 # E704 multiple statements on one line (def)
 ignore = E501, W503, E203, E704
+max-line-length = 80
 exclude =
     .git,
     .tmp,

.github/CODEOWNERS

@@ -1,2 +1,2 @@
-* @adubovik
+* @adubovik @roman-romanov-o
 /.github/ @nepalevov @alexey-ban

.github/workflows/pr-title-check.yml

@@ -9,6 +9,6 @@ on:
 
 jobs:
   pr-title-check:
-    uses: epam/ai-dial-ci/.github/workflows/pr-title-check.yml@1.9.0
+    uses: epam/ai-dial-ci/.github/workflows/pr-title-check.yml@1.9.1
     secrets:
       ACTIONS_BOT_TOKEN: ${{ secrets.ACTIONS_BOT_TOKEN }}

.github/workflows/pr.yml

@@ -6,7 +6,7 @@ on:
 
 jobs:
   run_tests:
-    uses: epam/ai-dial-ci/.github/workflows/python_package_pr.yml@1.9.0
+    uses: epam/ai-dial-ci/.github/workflows/python_package_pr.yml@1.9.1
     secrets: inherit
     with:
       python_version: 3.8

.github/workflows/release.yml

@@ -6,7 +6,7 @@ on:
 
 jobs:
   release:
-    uses: epam/ai-dial-ci/.github/workflows/python_package_release.yml@1.9.0
+    uses: epam/ai-dial-ci/.github/workflows/python_package_release.yml@1.9.1
     secrets: inherit
     with:
       python_version: 3.8

.ort.yml

@@ -1,8 +1,5 @@
 ---
 excludes:
-  paths:
-    - pattern: "examples/**"
-      reason: "EXAMPLE_OF"
   scopes:
     - pattern: "lint"
       reason: "DEV_DEPENDENCY_OF"
@@ -12,9 +9,9 @@ excludes:
       comment: "Packages for testing only."
 resolutions:
   rule_violations:
-    - message: ".*PyPI::httpcore:0\\.18\\.0.*"
+    - message: ".*PyPI::httpcore:1\\.0\\.5.*"
       reason: "CANT_FIX_EXCEPTION"
-      comment: "BSD 3-Clause New or Revised License: https://github.com/encode/httpcore/blob/0.18.0/LICENSE.md"
-    - message: ".*PyPI::httpx:0\\.25\\.0.*"
+      comment: "BSD 3-Clause New or Revised License: https://github.com/encode/httpcore/blob/1.0.5/LICENSE.md"
+    - message: ".*PyPI::httpx:0\\.25\\.2.*"
       reason: "CANT_FIX_EXCEPTION"
       comment: "BSD 3-Clause New or Revised License: https://github.com/encode/httpx/blob/0.25.0/LICENSE.md"

Makefile

@@ -1,20 +1,16 @@
-all: build
+.PHONY: all install clean lint format test spell_check
 
-install:
-	poetry install --all-extras
+all: build
 
 build: install
 	poetry build
 
-clean:
-	rm -rf $$(poetry env info --path)
-	rm -rf .nox
-	rm -rf .pytest_cache
-	rm -rf dist
-	find . -type d -name __pycache__ | xargs rm -r
+install:
+	poetry install
 
-publish: build
-	poetry publish -u __token__ -p ${PYPI_TOKEN} --skip-existing
+clean:
+	poetry run clean
+	poetry env remove --all
 
 lint: install
 	poetry run nox -s lint
@@ -25,14 +21,20 @@ format: install
 test: install
 	poetry run nox -s test $(if $(PYTHON),--python=$(PYTHON),)
 
+integration_test: install
+	poetry run nox -s integration_test $(if $(PYTHON),--python=$(PYTHON),)
+
+coverage: install
+	poetry run nox -s coverage
+
+publish: build
+	poetry publish -u __token__ -p ${PYPI_TOKEN} --skip-existing
+
 help:
 	@echo '===================='
-	@echo 'build                        - build the library'
+	@echo 'install                      - install virtual env and dependencies'
 	@echo 'clean                        - clean virtual env and build artifacts'
-	@echo 'publish                      - publish the library to Pypi'
 	@echo '-- LINTING --'
 	@echo 'format                       - run code formatters'
 	@echo 'lint                         - run linters'
-	@echo '-- TESTS --'
-	@echo 'test                         - run unit tests'
-	@echo 'test PYTHON=<python_version> - run unit tests with the specific python version'
+	@echo 'spell_check                  - run spell check'

aidial_client/__init__.py

@@ -0,0 +1,24 @@
+from aidial_client._auth import AsyncAuthValue, AuthType, SyncAuthValue
+from aidial_client._client import AsyncDial, Dial
+from aidial_client._client_pool import AsyncDialClientPool, DialClientPool
+from aidial_client._exception import (
+    DialException,
+    InvalidDialURLError,
+    InvalidRequestError,
+    ParsingDataError,
+)
+
+__all__ = [
+    "Dial",
+    "AsyncDial",
+    "DialClientPool",
+    "AsyncDialClientPool",
+    "AuthType",
+    "SyncAuthValue",
+    "AsyncAuthValue",
+    # Exceptions
+    "DialException",
+    "InvalidDialURLError",
+    "InvalidRequestError",
+    "ParsingDataError",
+]

aidial_client/_auth.py

@@ -0,0 +1,99 @@
+from enum import Enum
+from inspect import isawaitable
+from typing import (
+    Awaitable,
+    Callable,
+    Dict,
+    Optional,
+    Tuple,
+    TypeVar,
+    Union,
+    overload,
+)
+
+from typing_extensions import assert_never
+
+
+class AuthType(Enum):
+    API_KEY = "API_KEY"
+    BEARER = "BEARER"
+
+
+SyncAuthValue = Union[str, Callable[[], str]]
+AsyncAuthValue = Union[SyncAuthValue, Callable[[], Awaitable[str]]]
+
+AuthValueT = TypeVar(
+    "AuthValueT",
+    bound=Union[SyncAuthValue, AsyncAuthValue],
+)
+
+
+@overload
+def get_auth_value(auth_value: SyncAuthValue) -> str: ...
+
+
+@overload
+def get_auth_value(
+    auth_value: AsyncAuthValue,
+) -> Union[str, Awaitable[str]]: ...
+
+
+def get_auth_value(
+    auth_value: Union[SyncAuthValue, AsyncAuthValue]
+) -> Union[str, Awaitable[str]]:
+    if isinstance(auth_value, str):
+        return auth_value
+    elif callable(auth_value):
+        return auth_value()
+    else:
+        assert_never(auth_value)
+
+
+async def aget_auth_value(auth_value: AsyncAuthValue) -> str:
+    processed_auth_value = get_auth_value(auth_value)
+    if isawaitable(processed_auth_value):
+        return await processed_auth_value
+    return processed_auth_value
+
+
+def _get_auth_headers(auth_type: AuthType, auth_value: str) -> Dict[str, str]:
+    if auth_type == AuthType.API_KEY:
+        return {"api-key": auth_value}
+    elif auth_type == AuthType.BEARER:
+        return {"Authorization": f"Bearer {auth_value}"}
+    else:
+        assert_never(auth_type)
+
+
+def get_auth_headers(
+    *,
+    auth_value: SyncAuthValue,
+    auth_type: AuthType,
+) -> Dict[str, str]:
+    processed_auth_value = get_auth_value(auth_value)
+    return _get_auth_headers(auth_type, processed_auth_value)
+
+
+async def aget_auth_headers(
+    auth_value: AsyncAuthValue,
+    auth_type: AuthType,
+) -> Dict[str, str]:
+    processed_auth_value = await aget_auth_value(auth_value)
+    return _get_auth_headers(auth_type, processed_auth_value)
+
+
+def process_auth(
+    *,
+    api_key: Optional[AuthValueT] = None,
+    bearer_token: Optional[AuthValueT] = None,
+) -> Tuple[AuthType, AuthValueT]:
+    if api_key and bearer_token:
+        raise ValueError(
+            "Either api_key or bearer_token must be provided, but not both"
+        )
+    elif api_key:
+        return AuthType.API_KEY, api_key
+    elif bearer_token:
+        return AuthType.BEARER, bearer_token
+    else:
+        raise ValueError("Either api_key or bearer_token must be provided")