Update app.py

app.py

@@ -43,9 +43,23 @@ def data_ingestion():
     splits = text_splitter.split_documents(documents)
 
     # create embeddings here
-    embeddings = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
-    vectordb = FAISS.from_documents(splits, embeddings)
-    vectordb.save_local("faiss_index")
+    #embeddings = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
+    openai.api_key = os.getenv("OPENAI_API_KEY")
+                model = "text-embedding-ada-002"
+
+                embeddings = []
+                for split in splits:
+                    completion = openai.Completion.create(
+                        model=model,
+                        prompt=split,
+                        temperature=0.7,
+                        max_tokens=100,
+                    )
+
+                    embedding = completion.choices[0].embedding
+                    embeddings.append(embedding)
+                    vectordb = FAISS.from_documents(splits, embeddings)
+                    vectordb.save_local("faiss_index")
 
 
 @st.cache_resource