Fix ollama NameError

neo4j · Dec 17, 2024 · 17db6b1 · 17db6b1
1 parent 775447f
commit 17db6b1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/neo4j_graphrag/llm/ollama_llm.py b/src/neo4j_graphrag/llm/ollama_llm.py
@@ -12,6 +12,7 @@
 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
+from __future__ import annotations
 from typing import Any, Optional, Sequence, TYPE_CHECKING, cast
 
 from pydantic import ValidationError