To v0.6.0; use embed_openai store

app.py

@@ -14,7 +14,7 @@ from openworm_ai.utils.llms import OPENAI_LLMS
 
 import socket
 
-__version__ = "0.5.5"
+__version__ = "0.6.0"
 
 
 def requires_openai_key(llm_ver):

corpus_query.py

@@ -17,7 +17,8 @@ LLM_GPT4o = "GPT4o"
 def print_(text):
     print(text)
 
-
+from openworm_ai.graphrag.GraphRAG_test import load_index
+'''
 def load_index(model):
     OLLAMA_MODEL = model.replace("Ollama:", "") if model is not LLM_GPT4o else None
 
@@ -43,7 +44,7 @@ def load_index(model):
 
     index_reloaded = load_index_from_storage(storage_context)
 
-    return index_reloaded
+    return index_reloaded'''
 
 
 def get_query_engine(index_reloaded, model, similarity_top_k=4):
@@ -131,6 +132,7 @@ def process_query(query, model=llm_ver):
     files_used = []
     for sn in response.source_nodes:
         # print(sn)
+        print(sn.metadata)
         sd = sn.metadata["source document"]
         if "et_al_" in sd:
             sd = sd.replace("WormAtlas Handbook:", "Paper: ")
@@ -182,6 +184,7 @@ if __name__ == "__main__":
     queries = [
         "What are the main types of neurons and muscles in the C. elegans pharynx?",
         "Tell me about the egg laying apparatus",
+        "Tell me about locomotion in C. elegans",
     ]
 
     for query in queries:

openworm.ai_store/default__vector_store.json

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:3b0017f9d77d51651eb4d2c45f08128c9702d15deb6a8a1bd619c80c66b0cebf
-size 13779731

openworm.ai_store/docstore.json

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:ad8615ba2fbc0fb330fec021f32d258af73e7cf353f5c43f2df2b3a2d0afb93b
-size 2187676

openworm.ai_store/graph_store.json

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:8e0a77744010862225c69da83c585f4f8a42fd551b044ce530dbb1eb6e16742c
-size 18

openworm.ai_store/image__vector_store.json

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:d17ed74c1649a438e518a8dc56a7772913dfe1ea7a7605bce069c63872431455
-size 72

openworm.ai_store/index_store.json

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:584307257581210ab2b3f426cb81983b71f8471071b2495e1f804dfd206a3f12
-size 33427

store/embed_openai/graph_store.json

@@ -0,0 +1 @@
+{"graph_dict": {}}

store/embed_openai/image__vector_store.json

@@ -0,0 +1 @@
+{"embedding_dict": {}, "text_id_to_ref_doc_id": {}, "metadata_dict": {}}