diff --git a/litellm/llms/ollama.py b/litellm/llms/ollama.py
index a7ca01db02..03fb013454 100644
--- a/litellm/llms/ollama.py
+++ b/litellm/llms/ollama.py
@@ -1,3 +1,4 @@
+from itertools import chain
 import requests, types, time
 import json, uuid
 import traceback