1 files changed, 17 insertions, 0 deletions
diff --git a/extension/scripts/replace.py b/extension/scripts/replace.py
new file mode 100644
index 00000000..08810243
--- /dev/null
+++ b/extension/scripts/replace.py
@@ -0,0 +1,17 @@
+import sys
+from gpt_index import GPTSimpleVectorIndex, Document
+
+def replace_additional_index(info: str):
+    """Replace the additional index."""
+    with open('data/additional_context.txt', 'w') as f:
+        f.write(info)
+    documents = [Document(info)]
+    index = GPTSimpleVectorIndex(documents)
+    index.save_to_disk('data/additional_index.json')
+    print("Additional index replaced")
+
+if __name__ == "__main__":
+    """python3 replace.py <info>"""
+    info = sys.argv[1] if len(sys.argv) > 1 else None
+    if info:
+        replace_additional_index(info)
+\ No newline at end of file