Update agent.py

agent.py

@@ -122,16 +122,30 @@ with open("system_prompt.txt", "r", encoding="utf-8") as f:
 # System message
 sys_msg = SystemMessage(content=system_prompt)
 
-embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2", model_kwargs={"max_length": 2048}) #  dim=768
+# Initialize SentenceTransformer with max_seq_length
+sentence_transformer = SentenceTransformer(
+    "sentence-transformers/all-mpnet-base-v2",
+    max_seq_length=2048  # Set max sequence length here
+)
+
+# Initialize embeddings with the custom SentenceTransformer model
+embeddings = HuggingFaceEmbeddings(model=sentence_transformer)
+
+# Initialize Supabase client
 supabase: Client = create_client(
-    os.environ.get("SUPABASE_URL"), 
-    os.environ.get("SUPABASE_SERVICE_KEY"))
+    os.environ.get("SUPABASE_URL"),
+    os.environ.get("SUPABASE_SERVICE_KEY")
+)
+
+# Create vector store
 vector_store = SupabaseVectorStore(
     client=supabase,
-    embedding= embeddings,
+    embedding=embeddings,
     table_name="documents",
     query_name="match_documents_langchain",
 )
+
+# Create retriever tool
 create_retriever_tool = create_retriever_tool(
     retriever=vector_store.as_retriever(),
     name="Question Search",