diff --git a/fam-edge/src/fam_edge/model_adapters/ollama_adapter.py b/fam-edge/src/fam_edge/model_adapters/ollama_adapter.py
index 44984ba..5336907 100644
--- a/fam-edge/src/fam_edge/model_adapters/ollama_adapter.py
+++ b/fam-edge/src/fam_edge/model_adapters/ollama_adapter.py
@@ -9,7 +9,7 @@ provider_name = "ollama"
 """
 import base64
 import requests
-from typing import List, Optional
+from typing import Dict, List, Optional
 
 from .base_adapter import BaseModelAdapter
 from .circuit_breaker import CircuitBreaker