diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml
index f60c4ca..290dccf 100644
--- a/.github/workflows/test.yaml
+++ b/.github/workflows/test.yaml
@@ -9,11 +9,15 @@ on:
 jobs:
   test:
     runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ['3.8', '3.9', '3.10', '3.11', '3.12', '3.13']
     steps:
       - uses: actions/checkout@v5
       - uses: astral-sh/setup-uv@v5
         with:
           enable-cache: true
+          python-version: ${{ matrix.python-version }}
       - run: uvx hatch test -acp
         if: ${{ always() }}
   lint:
diff --git a/ollama/_utils.py b/ollama/_utils.py
index 15f1cc0..b32525b 100644
--- a/ollama/_utils.py
+++ b/ollama/_utils.py
@@ -3,14 +3,14 @@ from __future__ import annotations
 import inspect
 import re
 from collections import defaultdict
-from typing import Callable, Union
+from typing import Callable, Dict, Union
 
 import pydantic
 
 from ollama._types import Tool
 
 
-def _parse_docstring(doc_string: Union[str, None]) -> dict[str, str]:
+def _parse_docstring(doc_string: Union[str, None]) -> Dict[str, str]:
   parsed_docstring = defaultdict(str)
   if not doc_string:
     return parsed_docstring