Update util.py

util.py

@@ -22,8 +22,7 @@ octoai_client = octoai.client.Client(token=os.getenv('OCTOML_KEY'))
 
 from pinecone import Pinecone, ServerlessSpec
 pc = Pinecone(api_key=os.getenv('PINECONE_API_KEY'))
-pc_512 = pc.Index('prorata-postman-ds')
-pc_256 = pc.Index('prorata-postman-ds-256')
+pc_256 = pc.Index('prorata-postman-ds-256-v2')
 pc_128 = pc.Index('prorata-postman-ds-128-v2')
 
 
@@ -85,7 +84,7 @@ prompt_tf_template = "Given the context below, anwer the question that follows.
 
 def get_atoms_list(answer, file=None):
     prompt_af = prompt_af_template_llama3.format(answer)
-    response = None
+    response, atoms_l = None, []
     for _ in range(5):
         try:
             # response = octoai_client.chat.completions.create(
@@ -126,7 +125,7 @@ def get_atoms_list(answer, file=None):
             time.sleep(2)
     return atoms_l
 
-def get_topk_matches(atom, k=5, pc_index=pc_128):
+def get_topk_matches(atom, k=5, pc_index=pc_256):
     embed_atom = get_embedding(atom)
     res = pc_index.query(vector=embed_atom, top_k=k, include_metadata=True)
     return res['matches']