wt002 commited on
Commit
5fb0b4f
·
verified ·
1 Parent(s): 427bb40

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -135,7 +135,7 @@ sys_msg = SystemMessage(content=system_prompt)
135
 
136
  # Initialize SentenceTransformer and set max_seq_length
137
  sentence_transformer = SentenceTransformer("sentence-transformers/all-mpnet-base-v2")
138
- sentence_transformer.max_seq_length = 2048 # Set max sequence length
139
 
140
  # Initialize embeddings with the model name (dim=768)
141
  embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
@@ -155,7 +155,7 @@ vector_store = SupabaseVectorStore(
155
  )
156
 
157
  # Create retriever tool
158
- create_retriever_tool = create_retriever_tool(
159
  retriever=vector_store.as_retriever(),
160
  name="Question Search",
161
  description="A tool to retrieve similar questions from a vector store."
 
135
 
136
  # Initialize SentenceTransformer and set max_seq_length
137
  sentence_transformer = SentenceTransformer("sentence-transformers/all-mpnet-base-v2")
138
+ sentence_transformer.max_seq_length = 512 # Set max sequence length
139
 
140
  # Initialize embeddings with the model name (dim=768)
141
  embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
 
155
  )
156
 
157
  # Create retriever tool
158
+ retriever_tool = create_retriever_tool(
159
  retriever=vector_store.as_retriever(),
160
  name="Question Search",
161
  description="A tool to retrieve similar questions from a vector store."