From 27ecedb02ef79ce53bf533e016b00462c44541be Mon Sep 17 00:00:00 2001 From: Nate Sesti Date: Tue, 23 May 2023 23:45:12 -0400 Subject: copying from old repo --- extension/scripts/replace.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) create mode 100644 extension/scripts/replace.py (limited to 'extension/scripts/replace.py') 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 = sys.argv[1] if len(sys.argv) > 1 else None + if info: + replace_additional_index(info) \ No newline at end of file -- cgit v1.2.3-70-g09d2