update to kb

.gitignore

@@ -1,5 +1,6 @@
 bin/
 lib/
 __pycache__/
+include/
 .env
 pyvenv.cfg

kb.py

@@ -9,5 +9,8 @@ NAMESPACE = "knowedgebase"
 EMBEDDING_DIMENSION = 1536
 EMBEDDING_MODEL_NAME = "amazon.titan-embed-text-v1"
 
-embeddings = BedrockEmbeddings(model_id=EMBEDDING_MODEL_NAME)
-vectorstore = PineconeVectorStore(index_name=INDEX_NAME, embedding=embeddings)
+def getkb():
+    vectorstore = PineconeVectorStore(index_name=INDEX_NAME, 
+                            embedding=BedrockEmbeddings(model_id=EMBEDDING_MODEL_NAME), 
+                            namespace=NAMESPACE)
+    return vectorstore

main.py

@@ -2,12 +2,13 @@ from fastapi import FastAPI, Request, Query
 import logging
 from whatsapp_client import WhatsAppWrapper
 import json
-from kb import vectorstore
+from kb import getkb
 
 logger = logging.getLogger()
 
 app = FastAPI()
 
+vectorstore = getkb()
 
 @app.get('/')
 def ok():
@@ -36,13 +37,13 @@ async def callback(request: Request):
     data = json.loads(body)
     value = data['entry'][0]['changes'][0]['value']
 
-    if 'messages' in value and 'from ' in value['messages'][0] and 'text' in value['messages'][0] and 'text' in value['messages'][0]:
+    if 'messages' in value and 'from' in value['messages'][0] and 'text' in value['messages'][0]:
         message = value['messages'][0]
         phone_num = message['from']
-        mesg_body = value['messages'][0]['text']['body']
+        mesg_body = message['text']['body']
         print("SEARCH: " + str(mesg_body))
-        documents = vectorstore.similarity_search(mesg_body)
-        print(f"DOCUMENT: {len(documents)}")
+        documents = '\n'.join([doc[0].page_content for doc in vectorstore.similarity_search_with_score(mesg_body, k=6)])
+        print(f"CONTENT: {documents}")
         client = WhatsAppWrapper()
         client.send_text_message("hello, right back at ya ..." + mesg_body,  phone_num)
 

test.py

@@ -1,4 +1,8 @@
 import json
+from kb import getkb
+from dotenv import load_dotenv
+
+load_dotenv()
 
 data = """{
   "object": "whatsapp_business_account",
@@ -42,8 +46,21 @@ data = """{
 
 data = json.loads(data)
 value = data['entry'][0]['changes'][0]['value']
-phone_num = value['messages'][0]['from']
-mesg_body = value['messages'][0]['text']['body']
-phone_number_id = value['metadata']['phone_number_id']
 
-print(f"PHONE# {phone_num} \nMSG: {mesg_body}\nPONE_NUM_ID: {phone_number_id}")
+def main():
+    vectorstore = getkb()
+    print('here i am!')
+    if 'messages' in value and 'from' in value['messages'][0] and 'text' in value['messages'][0]:
+        print('i got here!')
+        message = value['messages'][0]
+        phone_num = message['from']
+        mesg_body = message['text']['body']
+        print(f"SEARCH: {mesg_body}, PHONE: {phone_num}")
+        
+        documents = '\n'.join([doc[0].page_content for doc in vectorstore.similarity_search_with_score(mesg_body, k=6)])
+        print(f"CONTENT: {documents}")
+        #client = WhatsAppWrapper()
+        #client.send_text_message("hello, right back at ya ..." + mesg_body,  phone_num)
+
+if __name__ == "__main__":
+    main()