not-again-ai 0.13.0__tar.gz → 0.15.0__tar.gz

{not_again_ai-0.13.0 → not_again_ai-0.15.0}/PKG-INFO

@@ -1,12 +1,11 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: not-again-ai
-Version: 0.13.0
+Version: 0.15.0
 Summary: Designed to once and for all collect all the little things that come up over and over again in AI projects and put them in one place.
-Home-page: https://github.com/DaveCoDev/not-again-ai
 License: MIT
 Author: DaveCoDev
 Author-email: dave.co.dev@gmail.com
-Requires-Python: >=3.11,<4.0
+Requires-Python: >=3.11, <3.13
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Science/Research
@@ -17,26 +16,27 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Typing :: Typed
+Provides-Extra: data
 Provides-Extra: llm
-Provides-Extra: local-llm
 Provides-Extra: statistics
 Provides-Extra: viz
-Requires-Dist: azure-ai-inference (==1.0.0b4) ; extra == "llm"
-Requires-Dist: azure-identity (>=1.18,<2.0) ; extra == "llm"
-Requires-Dist: jinja2 (>=3.1,<4.0) ; extra == "local-llm"
-Requires-Dist: loguru (==0.7.2)
-Requires-Dist: numpy (>=2.1,<3.0) ; extra == "statistics" or extra == "viz"
-Requires-Dist: ollama (>=0.3,<0.4) ; extra == "local-llm"
-Requires-Dist: openai (>=1.51,<2.0) ; extra == "llm"
-Requires-Dist: pandas (>=2.2,<3.0) ; extra == "viz"
-Requires-Dist: pydantic (>=2.9,<3.0) ; extra == "llm"
-Requires-Dist: python-liquid (>=1.12,<2.0) ; extra == "llm"
-Requires-Dist: scikit-learn (>=1.5,<2.0) ; extra == "statistics"
-Requires-Dist: scipy (>=1.14,<2.0) ; extra == "statistics"
-Requires-Dist: seaborn (>=0.13,<0.14) ; extra == "viz"
-Requires-Dist: tiktoken (>=0.7,<0.8) ; extra == "llm"
-Requires-Dist: transformers (>=4.45,<5.0) ; extra == "local-llm"
-Project-URL: Documentation, https://github.com/DaveCoDev/not-again-ai
+Requires-Dist: azure-identity (>=1.19) ; extra == "llm"
+Requires-Dist: loguru (>=0.7)
+Requires-Dist: numpy (>=2.2) ; extra == "statistics"
+Requires-Dist: numpy (>=2.2) ; extra == "viz"
+Requires-Dist: ollama (>=0.4) ; extra == "llm"
+Requires-Dist: openai (>=1.60) ; extra == "llm"
+Requires-Dist: pandas (>=2.2) ; extra == "viz"
+Requires-Dist: playwright (>=1.49) ; extra == "data"
+Requires-Dist: pydantic (>=2.10)
+Requires-Dist: pytest-playwright (>=0.6) ; extra == "data"
+Requires-Dist: python-liquid (>=1.12) ; extra == "llm"
+Requires-Dist: scikit-learn (>=1.6) ; extra == "statistics"
+Requires-Dist: scipy (>=1.15) ; extra == "statistics"
+Requires-Dist: seaborn (>=0.13) ; extra == "viz"
+Requires-Dist: tiktoken (>=0.8) ; extra == "llm"
+Project-URL: Documentation, https://davecodev.github.io/not-again-ai/
+Project-URL: Homepage, https://github.com/DaveCoDev/not-again-ai
 Project-URL: Repository, https://github.com/DaveCoDev/not-again-ai
 Description-Content-Type: text/markdown
 
@@ -66,40 +66,47 @@ Requires: Python 3.11, or 3.12
 Install the entire package from [PyPI](https://pypi.org/project/not-again-ai/) with: 
 
 ```bash
-$ pip install not_again_ai[llm,local_llm,statistics,viz]
+$ pip install not_again_ai[data,llm,statistics,viz]
 ```
 
-Note that local LLM requires separate installations and will not work out of the box due to how hardware dependent it is. Be sure to check the [notebooks](notebooks/local_llm/) for more details.
-
 The package is split into subpackages, so you can install only the parts you need.
-* **Base only**: `pip install not_again_ai`
-* **LLM**: `pip install not_again_ai[llm]`
-    1. OpenAI API
-        1. Go to https://platform.openai.com/settings/profile?tab=api-keys to get your API key.
-        1. (Optional) Set the `OPENAI_API_KEY` and the `OPENAI_ORG_ID` environment variables.
-    1. Azure OpenAI (AOAI)
-       1. Using AOAI requires using Entra ID authentication. See https://learn.microsoft.com/en-us/azure/ai-services/openai/how-to/managed-identity for how to set this up for your AOAI deployment.
-          1. Requires the correct role assigned to your user account and being signed into the Azure CLI.
-       1. (Optional) Set the `AZURE_OPENAI_ENDPOINT` environment variable.
-    1. GitHub Models
-       1. Get a Personal Access Token from https://github.com/settings/tokens and set the `GITHUB_TOKEN` environment variable. The token does not need any permissions.
-       1. Check the [Github Marketplace](https://github.com/marketplace/models) to see which models are available.
-* **Local LLM**: `pip install not_again_ai[llm,local_llm]`
-    1. Some HuggingFace transformers tokenizers are gated behind access requests. If you wish to use these, you will need to request access from HuggingFace on the model card.
-       1. Then set the `HF_TOKEN` environment variable to your HuggingFace API token which can be found here: https://huggingface.co/settings/tokens
-    2. If you wish to use Ollama:
-        1. Follow the instructions at https://github.com/ollama/ollama to install Ollama for your system. 
-        2. (Optional) [Add Ollama as a startup service (recommended)](https://github.com/ollama/ollama/blob/main/docs/linux.md#adding-ollama-as-a-startup-service-recommended)
-        3. (Optional) To make the Ollama service accessible on your local network from a Linux server, add the following to the `/etc/systemd/system/ollama.service` file which will make Ollama available at `http://<local_address>:11434`:
-            ```bash
-            [Service]
-            ...
-            Environment="OLLAMA_HOST=0.0.0.0"
-            ```
-        4. It is recommended to always have the latest version of Ollama. To update Ollama check the [docs](https://github.com/ollama/ollama/blob/main/docs/). The command for Linux is: `curl -fsSL https://ollama.com/install.sh | sh`
-    3. HuggingFace transformers and other requirements are hardware dependent so for providers other than Ollama, this only installs some generic dependencies. Check the [notebooks](notebooks/local_llm/) for more details on what is available and how to install it.
-* **Statistics**: `pip install not_again_ai[statistics]`
-* **Visualization**: `pip install not_again_ai[viz]`
+
+### Base
+1. `pip install not_again_ai`
+
+
+### Data
+1. `pip install not_again_ai[data]`
+1. `playwright install` to download the browser binaries.
+
+
+### LLM
+1. `pip install not_again_ai[llm]`
+1. Setup OpenAI API
+   1. Go to https://platform.openai.com/settings/profile?tab=api-keys to get your API key.
+   1. (Optional) Set the `OPENAI_API_KEY` and the `OPENAI_ORG_ID` environment variables.
+1. Setup Azure OpenAI (AOAI)
+   1. Using AOAI requires using Entra ID authentication. See https://learn.microsoft.com/en-us/azure/ai-services/openai/how-to/managed-identity for how to set this up for your AOAI deployment.
+      * Requires the correct role assigned to your user account and being signed into the Azure CLI.
+   1. (Optional) Set the `AZURE_OPENAI_ENDPOINT` environment variable.
+1. If you wish to use Ollama:
+     1. Follow the instructions at https://github.com/ollama/ollama to install Ollama for your system. 
+     1. (Optional) [Add Ollama as a startup service (recommended)](https://github.com/ollama/ollama/blob/main/docs/linux.md#adding-ollama-as-a-startup-service-recommended)
+     1. (Optional) To make the Ollama service accessible on your local network from a Linux server, add the following to the `/etc/systemd/system/ollama.service` file which will make Ollama available at `http://<local_address>:11434`:
+         ```bash
+         [Service]
+         ...
+         Environment="OLLAMA_HOST=0.0.0.0"
+         ```
+     1. It is recommended to always have the latest version of Ollama. To update Ollama check the [docs](https://github.com/ollama/ollama/blob/main/docs/). The command for Linux is: `curl -fsSL https://ollama.com/install.sh | sh`
+
+
+### Statistics
+1. `pip install not_again_ai[statistics]`
+
+
+### Visualization
+1. `pip install not_again_ai[viz]`
 
 
 # Development Information
@@ -135,10 +142,8 @@ $ poetry update
 
 To install all dependencies (with all extra dependencies) into an isolated virtual environment:
 
-> Append `--sync` to uninstall dependencies that are no longer in use from the virtual environment.
-
 ```bash
-$ poetry install --all-extras
+$ poetry sync --all-extras
 ```
 
 To [activate](https://python-poetry.org/docs/basic-usage#activating-the-virtual-environment) the
@@ -194,7 +199,7 @@ Automated code quality checks are performed using
 environments and run commands based on [`noxfile.py`](./noxfile.py) for unit testing, PEP 8 style
 guide checking, type checking and documentation generation.
 
-> Note: `nox` is installed into the virtual environment automatically by the `poetry install`
+> Note: `nox` is installed into the virtual environment automatically by the `poetry sync`
 > command above. Run `poetry shell` to activate the virtual environment.
 
 To run all default sessions:
@@ -229,10 +234,10 @@ areas of the project that are currently not tested.
 
 pytest and code coverage are configured in [`pyproject.toml`](./pyproject.toml).
 
-To pass arguments to `pytest` through `nox`:
+To run selected tests:
 
 ```bash
-(.venv) $ nox -s test -- -k invalid_factorial
+(.venv) $ nox -s test -- -k "test_web"
 ```
 
 ## Code Style Checking

{not_again_ai-0.13.0 → not_again_ai-0.15.0}/README.md

@@ -24,40 +24,47 @@ Requires: Python 3.11, or 3.12
 Install the entire package from [PyPI](https://pypi.org/project/not-again-ai/) with: 
 
 ```bash
-$ pip install not_again_ai[llm,local_llm,statistics,viz]
+$ pip install not_again_ai[data,llm,statistics,viz]
 ```
 
-Note that local LLM requires separate installations and will not work out of the box due to how hardware dependent it is. Be sure to check the [notebooks](notebooks/local_llm/) for more details.
-
 The package is split into subpackages, so you can install only the parts you need.
-* **Base only**: `pip install not_again_ai`
-* **LLM**: `pip install not_again_ai[llm]`
-    1. OpenAI API
-        1. Go to https://platform.openai.com/settings/profile?tab=api-keys to get your API key.
-        1. (Optional) Set the `OPENAI_API_KEY` and the `OPENAI_ORG_ID` environment variables.
-    1. Azure OpenAI (AOAI)
-       1. Using AOAI requires using Entra ID authentication. See https://learn.microsoft.com/en-us/azure/ai-services/openai/how-to/managed-identity for how to set this up for your AOAI deployment.
-          1. Requires the correct role assigned to your user account and being signed into the Azure CLI.
-       1. (Optional) Set the `AZURE_OPENAI_ENDPOINT` environment variable.
-    1. GitHub Models
-       1. Get a Personal Access Token from https://github.com/settings/tokens and set the `GITHUB_TOKEN` environment variable. The token does not need any permissions.
-       1. Check the [Github Marketplace](https://github.com/marketplace/models) to see which models are available.
-* **Local LLM**: `pip install not_again_ai[llm,local_llm]`
-    1. Some HuggingFace transformers tokenizers are gated behind access requests. If you wish to use these, you will need to request access from HuggingFace on the model card.
-       1. Then set the `HF_TOKEN` environment variable to your HuggingFace API token which can be found here: https://huggingface.co/settings/tokens
-    2. If you wish to use Ollama:
-        1. Follow the instructions at https://github.com/ollama/ollama to install Ollama for your system. 
-        2. (Optional) [Add Ollama as a startup service (recommended)](https://github.com/ollama/ollama/blob/main/docs/linux.md#adding-ollama-as-a-startup-service-recommended)
-        3. (Optional) To make the Ollama service accessible on your local network from a Linux server, add the following to the `/etc/systemd/system/ollama.service` file which will make Ollama available at `http://<local_address>:11434`:
-            ```bash
-            [Service]
-            ...
-            Environment="OLLAMA_HOST=0.0.0.0"
-            ```
-        4. It is recommended to always have the latest version of Ollama. To update Ollama check the [docs](https://github.com/ollama/ollama/blob/main/docs/). The command for Linux is: `curl -fsSL https://ollama.com/install.sh | sh`
-    3. HuggingFace transformers and other requirements are hardware dependent so for providers other than Ollama, this only installs some generic dependencies. Check the [notebooks](notebooks/local_llm/) for more details on what is available and how to install it.
-* **Statistics**: `pip install not_again_ai[statistics]`
-* **Visualization**: `pip install not_again_ai[viz]`
+
+### Base
+1. `pip install not_again_ai`
+
+
+### Data
+1. `pip install not_again_ai[data]`
+1. `playwright install` to download the browser binaries.
+
+
+### LLM
+1. `pip install not_again_ai[llm]`
+1. Setup OpenAI API
+   1. Go to https://platform.openai.com/settings/profile?tab=api-keys to get your API key.
+   1. (Optional) Set the `OPENAI_API_KEY` and the `OPENAI_ORG_ID` environment variables.
+1. Setup Azure OpenAI (AOAI)
+   1. Using AOAI requires using Entra ID authentication. See https://learn.microsoft.com/en-us/azure/ai-services/openai/how-to/managed-identity for how to set this up for your AOAI deployment.
+      * Requires the correct role assigned to your user account and being signed into the Azure CLI.
+   1. (Optional) Set the `AZURE_OPENAI_ENDPOINT` environment variable.
+1. If you wish to use Ollama:
+     1. Follow the instructions at https://github.com/ollama/ollama to install Ollama for your system. 
+     1. (Optional) [Add Ollama as a startup service (recommended)](https://github.com/ollama/ollama/blob/main/docs/linux.md#adding-ollama-as-a-startup-service-recommended)
+     1. (Optional) To make the Ollama service accessible on your local network from a Linux server, add the following to the `/etc/systemd/system/ollama.service` file which will make Ollama available at `http://<local_address>:11434`:
+         ```bash
+         [Service]
+         ...
+         Environment="OLLAMA_HOST=0.0.0.0"
+         ```
+     1. It is recommended to always have the latest version of Ollama. To update Ollama check the [docs](https://github.com/ollama/ollama/blob/main/docs/). The command for Linux is: `curl -fsSL https://ollama.com/install.sh | sh`
+
+
+### Statistics
+1. `pip install not_again_ai[statistics]`
+
+
+### Visualization
+1. `pip install not_again_ai[viz]`
 
 
 # Development Information
@@ -93,10 +100,8 @@ $ poetry update
 
 To install all dependencies (with all extra dependencies) into an isolated virtual environment:
 
-> Append `--sync` to uninstall dependencies that are no longer in use from the virtual environment.
-
 ```bash
-$ poetry install --all-extras
+$ poetry sync --all-extras
 ```
 
 To [activate](https://python-poetry.org/docs/basic-usage#activating-the-virtual-environment) the
@@ -152,7 +157,7 @@ Automated code quality checks are performed using
 environments and run commands based on [`noxfile.py`](./noxfile.py) for unit testing, PEP 8 style
 guide checking, type checking and documentation generation.
 
-> Note: `nox` is installed into the virtual environment automatically by the `poetry install`
+> Note: `nox` is installed into the virtual environment automatically by the `poetry sync`
 > command above. Run `poetry shell` to activate the virtual environment.
 
 To run all default sessions:
@@ -187,10 +192,10 @@ areas of the project that are currently not tested.
 
 pytest and code coverage are configured in [`pyproject.toml`](./pyproject.toml).
 
-To pass arguments to `pytest` through `nox`:
+To run selected tests:
 
 ```bash
-(.venv) $ nox -s test -- -k invalid_factorial
+(.venv) $ nox -s test -- -k "test_web"
 ```
 
 ## Code Style Checking

{not_again_ai-0.13.0 → not_again_ai-0.15.0}/pyproject.toml

@@ -1,8 +1,10 @@
-[tool.poetry]
+[project]
 name = "not-again-ai"
-version = "0.13.0"
+version = "0.15.0"
 description = "Designed to once and for all collect all the little things that come up over and over again in AI projects and put them in one place."
-authors = ["DaveCoDev <dave.co.dev@gmail.com>"]
+authors = [
+    { name = "DaveCoDev", email = "dave.co.dev@gmail.com" }
+]
 license = "MIT"
 readme = "README.md"
 repository = "https://github.com/DaveCoDev/not-again-ai"
@@ -19,38 +21,49 @@ classifiers = [
     "Programming Language :: Python :: 3.12",
     "Typing :: Typed",
 ]
+requires-python = ">=3.11, <3.13"
+# Lets us define dependencies completely in tool.poetry.dependencies
+dynamic = [ "dependencies" ]
+dependencies = [
+    "loguru>=0.7",
+    "pydantic>=2.10"
+]
+
+[project.urls]
+Homepage = "https://github.com/DaveCoDev/not-again-ai"
+Documentation = "https://davecodev.github.io/not-again-ai/"
+Repository = "https://github.com/DaveCoDev/not-again-ai"
+
+[tool.poetry]
+requires-poetry = ">=2.0.1"
+
+[tool.poetry.requires-plugins]
+poetry-plugin-export = ">=1.8"
 
-[tool.poetry.dependencies]
-# Some packages, such as scipy, constrain their upper bound of Python versions they support.
-# Without also constraining the upper bound here, Poetry will not select those versions and will
-# result in an old version being resolved/locked.
-python = "^3.11 || ^3.12"
-
-loguru = { version = "==0.7.2" }
-
-# Optional dependencies are defined here, and groupings are defined below.
-azure-ai-inference = { version = "==1.0.0b4", optional = true }
-azure-identity = { version = "^1.18", optional = true }
-jinja2 = { version = "^3.1", optional = true }
-numpy = { version = "^2.1", optional = true }
-ollama = { version = "^0.3", optional = true }
-openai = { version = "^1.51", optional = true }
-pandas = { version = "^2.2", optional = true }
-pydantic = { version = "^2.9", optional = true }
-python-liquid = { version = "^1.12", optional = true }
-scipy = { version = "^1.14", optional = true }
-scikit-learn = { version = "^1.5", optional = true }
-seaborn = { version = "^0.13", optional = true }
-tiktoken = { version = "^0.7", optional = true }
-transformers = { version = "^4.45", optional = true }
-
-[tool.poetry.extras]
-llm = ["azure-ai-inference", "azure-identity", "openai", "pydantic", "python-liquid", "tiktoken"]
-local_llm = ["jinja2", "ollama", "transformers"]
-statistics = ["numpy", "scikit-learn", "scipy"]
-viz = ["numpy", "pandas", "seaborn"]
-
-[tool.poetry.dev-dependencies]
+[project.optional-dependencies]
+data = [
+    "playwright>=1.49",
+    "pytest-playwright>=0.6"
+]
+llm = [
+    "azure-identity>=1.19",
+    "ollama>=0.4",
+    "openai>=1.60",
+    "python-liquid>=1.12",
+    "tiktoken>=0.8"
+]
+statistics = [
+    "numpy>=2.2",
+    "scikit-learn>=1.6",
+    "scipy>=1.15"
+]
+viz = [
+    "numpy>=2.2",
+    "pandas>=2.2",
+    "seaborn>=0.13"
+]
+
+[tool.poetry.group.dev.dependencies]
 ipykernel = "*"
 ipywidgets = "*"
 
@@ -84,11 +97,8 @@ mkdocs-literate-nav = "*"
 [tool.poetry.group.typos.dependencies]
 typos = "*"
 
-[tool.poetry.scripts]
-not-again-ai = "not_again_ai.cli:entry_point"
-
 [build-system]
-requires = ["poetry-core"]
+requires = ["poetry-core>=2.0.0,<3.0.0"]
 build-backend = "poetry.core.masonry.api"
 
 [tool.mypy]
@@ -114,6 +124,8 @@ select = [
     "B", # flake8-bugbear
     "C4", # flake8-comprehensions
     "ISC", # flake8-implicit-str-concat
+    "PIE", # flake8-pie
+    "PT", # flake-pytest-style
     "PTH", # flake8-use-pathlib
     "SIM", # flake8-simplify
     "TID", # flake8-tidy-imports
@@ -138,7 +150,7 @@ filterwarnings = [
     # When running tests, treat warnings as errors (e.g. -Werror).
     # See: https://docs.pytest.org/en/latest/reference/reference.html#confval-filterwarnings
     "error",
-    # Add additional warning supressions as needed here. For example, if a third-party library
+    # Add additional warning suppressions as needed here. For example, if a third-party library
     # is throwing a deprecation warning that needs to be fixed upstream:
     # "ignore::DeprecationWarning:typer",
     "ignore::pytest.PytestUnraisableExceptionWarning"

not_again_ai-0.15.0/src/not_again_ai/data/__init__.py

@@ -0,0 +1,7 @@
+import importlib.util
+
+if importlib.util.find_spec("playwright") is None:
+    raise ImportError(
+        "not_again_ai.data requires the 'data' extra to be installed. "
+        "You can install it using 'pip install not_again_ai[data]'."
+    )

not_again_ai-0.15.0/src/not_again_ai/data/web.py

@@ -0,0 +1,56 @@
+from loguru import logger
+from playwright.sync_api import Browser, Playwright, sync_playwright
+
+
+def create_browser(headless: bool = True) -> tuple[Playwright, Browser]:
+    """Creates and returns a new Playwright instance and browser.
+
+    Args:
+        headless (bool, optional): Whether to run the browser in headless mode. Defaults to True.
+
+    Returns:
+        tuple[Playwright, Browser]: A tuple containing the Playwright instance and browser.
+    """
+    pwright = sync_playwright().start()
+    browser = pwright.chromium.launch(
+        headless=headless,
+        chromium_sandbox=False,
+        timeout=15000,
+    )
+    return pwright, browser
+
+
+def get_raw_web_content(url: str, browser: Browser | None = None, headless: bool = True) -> str:
+    """Fetches raw web content from a given URL using Playwright.
+
+    Args:
+        url (str): The URL to fetch content from.
+        browser (Browser | None, optional): An existing browser instance to use. Defaults to None.
+        headless (bool, optional): Whether to run the browser in headless mode. Defaults to True.
+
+    Returns:
+        str: The raw web content.
+    """
+    p = None
+    try:
+        if browser is None:
+            p, browser = create_browser(headless)
+
+        page = browser.new_page(
+            accept_downloads=False,
+            java_script_enabled=True,
+            viewport={"width": 1366, "height": 768},
+            user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/129.0.0.0 Safari/537.3",
+        )
+        page.goto(url)
+        content = page.content()
+        page.close()
+        return content
+    except Exception as e:
+        logger.error(f"Failed to get web content: {e}")
+        return ""
+    finally:
+        if browser:
+            browser.close()
+        if p:
+            p.stop()

not_again_ai-0.15.0/src/not_again_ai/llm/chat_completion/__init__.py

@@ -0,0 +1,4 @@
+from not_again_ai.llm.chat_completion.interface import chat_completion
+from not_again_ai.llm.chat_completion.types import ChatCompletionRequest
+
+__all__ = ["ChatCompletionRequest", "chat_completion"]

not_again_ai-0.15.0/src/not_again_ai/llm/chat_completion/interface.py

@@ -0,0 +1,32 @@
+from collections.abc import Callable
+from typing import Any
+
+from not_again_ai.llm.chat_completion.providers.ollama_api import ollama_chat_completion
+from not_again_ai.llm.chat_completion.providers.openai_api import openai_chat_completion
+from not_again_ai.llm.chat_completion.types import ChatCompletionRequest, ChatCompletionResponse
+
+
+def chat_completion(
+    request: ChatCompletionRequest,
+    provider: str,
+    client: Callable[..., Any],
+) -> ChatCompletionResponse:
+    """Get a chat completion response from the given provider. Currently supported providers:
+    - `openai` - OpenAI
+    - `azure_openai` - Azure OpenAI
+    - `ollama` - Ollama
+
+    Args:
+        request: Request parameter object
+        provider: The supported provider name
+        client: Client information, see the provider's implementation for what can be provided
+
+    Returns:
+        ChatCompletionResponse: The chat completion response.
+    """
+    if provider == "openai" or provider == "azure_openai":
+        return openai_chat_completion(request, client)
+    elif provider == "ollama":
+        return ollama_chat_completion(request, client)
+    else:
+        raise ValueError(f"Provider {provider} not supported")