diff --git a/nodes.py b/nodes.py
index 2376099..ce84b5e 100644
--- a/nodes.py
+++ b/nodes.py
@@ -5,8 +5,8 @@
 from openai import OpenAI
 from openai.types.chat import ChatCompletionMessageParam
 from .tools import VariantSupport
-from comfy.graph_utils import GraphBuilder, is_link
-from comfy.graph import DynamicPrompt
+from comfy_execution.graph_utils import GraphBuilder, is_link
+from comfy_execution.graph import DynamicPrompt
 
 indentation_regex = re.compile(r"^[ \t]+")
 
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..2351145
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1 @@
+openai