From dd1b6302316e1fed7fede51f55aa5d762e65ae62 Mon Sep 17 00:00:00 2001
From: Santiago Zanella-Beguelin <santiago@microsoft.com>
Date: Wed, 3 Jul 2024 14:20:39 +0100
Subject: [PATCH] Fix promptlib by using deprecated AST module

---
 py/promptlib/promptlib/gen.py    | 2 +-
 py/promptlib/promptlib/prompt.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/py/promptlib/promptlib/gen.py b/py/promptlib/promptlib/gen.py
index 5290aa0b..846ed232 100644
--- a/py/promptlib/promptlib/gen.py
+++ b/py/promptlib/promptlib/gen.py
@@ -1,7 +1,7 @@
 from .prompt import PromptNode, TextNode
 
 from typing import List
-from pyaici import ast
+from pyaici import ast_ as ast
 
 class GenNode(PromptNode):
 
diff --git a/py/promptlib/promptlib/prompt.py b/py/promptlib/promptlib/prompt.py
index 95ee827c..d47734e9 100644
--- a/py/promptlib/promptlib/prompt.py
+++ b/py/promptlib/promptlib/prompt.py
@@ -2,7 +2,7 @@
 from typing import List
 from .aici import AICI
 
-from pyaici import ast
+from pyaici import ast_ as ast
 
 class PromptNode:
     """