Archives/DocsGPT
2
0
Fork 0
mirror of https://github.com/arc53/DocsGPT synced 2024-11-03 23:15:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: restore index back when continue process

Browse Source
This commit is contained in:
冯不游 2023-02-14 22:05:16 +08:00
parent c3037b0f53
commit 458f2a3ff3
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
scripts/parser/open_ai_func.py
View File

@ -32,11 +32,14 @@ def call_openai_api(docs):
print("Saving progress")
print(f"stopped at {c1} out of {len(docs)}")
faiss.write_index(store.index, "docs.index")
store_index_bak = store.index
store.index = None
with open("faiss_store.pkl", "wb") as f:
pickle.dump(store, f)
print("Sleeping for 10 seconds and trying again")
time.sleep(10)
faiss.write_index(store_index_bak, "docs.index")
store.index = store_index_bak
store.add_texts([i.page_content], metadatas=[i.metadata])
c1 += 1
@ -66,4 +69,4 @@ def get_user_permission(docs):
elif user_input == "":
call_openai_api(docs)
else:
print("The API was not called. No money was spent.")
print("The API was not called. No money was spent.")