diff --git a/module.yaml b/module.yaml
index 4448cfd..3bd2d50 100644
--- a/module.yaml
+++ b/module.yaml
@@ -1,4 +1,5 @@
-schema_version: 1
+schema_version: 2
+provider: fai
 name: llm-chat
 version: 0.1.0