Update agent.py

agent.py

@@ -344,12 +344,17 @@ class BERTEmbeddings(Embedding):
 
 # Example usage of BERTEmbedding with LangChain
 
-embedding_model = BERTEmbedding(model_name="bert-base-uncased")
+embedding_model = BERTEmbeddings(model_name="bert-base-uncased")
 
 # Sample text (replace with your own text)
 docs = [
     Document(page_content="Mercedes Sosa was an Argentine singer and musician."),
-    Document(page_content="The 2000s were a significant decade for music in Latin America.")
+    Document(page_content="Examine the video at https://www.youtube.com/watch?v=1htKBjuUWec.\n\nWhat does Teal'c say in response to the question \"Isn't that hot?\"),
+    Document(page_content="Who nominated the only Featured Article on English Wikipedia about a dinosaur that was promoted in November 2016?"),
+    Document(page_content="What is the surname of the equine veterinarian mentioned in 1.E Exercises from the chemistry materials licensed by Marisa Alviar-Agnew & Henry \
+                           Agnew under the CK-12 license in LibreText's Introductory Chemistry materials as compiled 08/21/2023?"),
+    Document(page_content="What is the final numeric output from the attached Python code?"),
+    Document(page_content="Who did the actor who played Ray in the Polish-language version of Everybody Loves Raymond play in Magda M.? Give only the first name.")
 ]
 
 # Get the embeddings for the documents