Add AIMLAPI workflow and update examples

Introduces a new GitHub Actions workflow for AIMLAPI integration testing. Updates example scripts to add noqa comments for print statements and refines message list construction. Modifies AIMLAPIChatGenerator to include an explicit 'openai_endpoint' parameter and clarifies a type ignore comment.

Author: D1m7asis

.github/workflows/aimlapi.yml

@@ -0,0 +1,84 @@
+# This workflow comes from https://github.com/ofek/hatch-mypyc
+# https://github.com/ofek/hatch-mypyc/blob/5a198c0ba8660494d02716cfc9d79ce4adfb1442/.github/workflows/test.yml
+name: Test / aimlapi
+
+on:
+  schedule:
+    - cron: "0 0 * * *"
+  pull_request:
+    paths:
+      - "integrations/aimlapi/**"
+      - "!integrations/aimlapi/*.md"
+      - ".github/workflows/aimlapi.yml"
+
+defaults:
+  run:
+    working-directory: integrations/aimlapi
+
+concurrency:
+  group: aimlapi-${{ github.head_ref }}
+  cancel-in-progress: true
+
+env:
+  PYTHONUNBUFFERED: "1"
+  FORCE_COLOR: "1"
+  AIMLAPI_API_KEY: ${{ secrets.AIMLAPI_API_KEY }}
+
+jobs:
+  run:
+    name: Python ${{ matrix.python-version }} on ${{ startsWith(matrix.os, 'macos-') && 'macOS' || startsWith(matrix.os, 'windows-') && 'Windows' || 'Linux' }}
+    runs-on: ${{ matrix.os }}
+    strategy:
+      fail-fast: false
+      matrix:
+        os: [ubuntu-latest, windows-latest, macos-latest]
+        python-version: ["3.9", "3.13"]
+
+    steps:
+      - name: Support longpaths
+        if: matrix.os == 'windows-latest'
+        working-directory: .
+        run: git config --system core.longpaths true
+
+      - uses: actions/checkout@v5
+
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v6
+        with:
+          python-version: ${{ matrix.python-version }}
+
+      - name: Install Hatch
+        run: pip install --upgrade hatch
+
+      - name: Lint
+        if: matrix.python-version == '3.9' && runner.os == 'Linux'
+        run: hatch run fmt-check && hatch run test:types
+
+      - name: Generate docs
+        if: matrix.python-version == '3.9' && runner.os == 'Linux'
+        run: hatch run docs
+
+      - name: Run tests
+        run: hatch run test:cov-retry
+
+      - name: Run unit tests with lowest direct dependencies
+        run: |
+          hatch run uv pip compile pyproject.toml --resolution lowest-direct --output-file requirements_lowest_direct.txt
+          hatch run uv pip install -r requirements_lowest_direct.txt
+          hatch run test:unit
+
+      # Since this integration inherits from OpenAIChatGenerator, we run ALL tests with Haystack main branch to catch regressions
+      - name: Nightly - run tests with Haystack main branch
+        if: github.event_name == 'schedule'
+        run: |
+          hatch env prune
+          hatch run uv pip install git+https://github.com/deepset-ai/haystack.git@main
+          hatch run test:all
+
+      - name: Send event to Datadog for nightly failures
+        if: failure() && github.event_name == 'schedule'
+        uses: ./.github/actions/send_failure
+        with:
+          title: |
+            Core integrations nightly tests failure: ${{ github.workflow }}
+          api-key: ${{ secrets.CORE_DATADOG_API_KEY }}

integrations/aimlapi/examples/aimlapi_basic_example.py

@@ -23,7 +23,7 @@ def main() -> None:
 
     reply = generator.run(messages=messages)["replies"][0]
 
-    print(f"assistant response: {reply.text}")
+    print(f"assistant response: {reply.text}")  # noqa: T201
 
 
 if __name__ == "__main__":

integrations/aimlapi/examples/aimlapi_with_tools_example.py

@@ -46,7 +46,7 @@ def main() -> None:
         ChatMessage.from_user("What's the weather in Tokyo today?"),
     ]
 
-    print("Requesting a tool call from the model...")
+    print("Requesting a tool call from the model...")  # noqa: T201
     tool_request = client.run(
         messages=messages,
         tools=[weather_tool],
@@ -55,26 +55,26 @@ def main() -> None:
         },
     )["replies"][0]
 
-    print(f"assistant tool request: {tool_request}")
+    print(f"assistant tool request: {tool_request}")  # noqa: T201
 
     if not tool_request.tool_calls:
-        print("No tool call was produced by the model.")
+        print("No tool call was produced by the model.")  # noqa: T201
         return
 
     tool_messages = tool_invoker.run(messages=[tool_request])["tool_messages"]
     for tool_message in tool_messages:
         for tool_result in tool_message.tool_call_results:
-            print(f"tool output: {tool_result.result}")
+            print(f"tool output: {tool_result.result}")  # noqa: T201
 
-    follow_up_messages = messages + [tool_request, *tool_messages]
+    follow_up_messages = [*messages, tool_request, *tool_messages]
 
     final_reply = client.run(
         messages=follow_up_messages,
         tools=[weather_tool],
         generation_kwargs={"tool_choice": "none"},
     )["replies"][0]
 
-    print(f"assistant final answer: {final_reply.text}")
+    print(f"assistant final answer: {final_reply.text}")  # noqa: T201
 
 
 if __name__ == "__main__":

integrations/aimlapi/src/haystack_integrations/components/generators/aimlapi/chat/chat_generator.py

@@ -189,10 +189,11 @@ def _prepare_api_call(
 
         return {
             "model": self.model,
-            "messages": aimlapi_formatted_messages,
+            "messages": aimlapi_formatted_messages,  # type: ignore[arg-type] # openai expects list of specific message types
             "stream": streaming_callback is not None,
             "n": num_responses,
             **aimlapi_tools,
             "extra_body": {**generation_kwargs},
             "extra_headers": {**extra_headers},
+            "openai_endpoint": "create",
         }