@@ -187,32 +187,48 @@ def init_server():
187187
188188 # Create component instances
189189 graph_db = GraphStoreFactory .from_config (graph_db_config )
190- vector_db = VecDBFactory .from_config (vector_db_config )
190+ vector_db = (
191+ VecDBFactory .from_config (vector_db_config )
192+ if os .getenv ("ENABLE_PREFERENCE_MEMORY" , "false" ).lower () == "true"
193+ else None
194+ )
191195 llm = LLMFactory .from_config (llm_config )
192196 embedder = EmbedderFactory .from_config (embedder_config )
193197 mem_reader = MemReaderFactory .from_config (mem_reader_config )
194198 reranker = RerankerFactory .from_config (reranker_config )
195199 internet_retriever = InternetRetrieverFactory .from_config (
196200 internet_retriever_config , embedder = embedder
197201 )
198- pref_extractor = ExtractorFactory .from_config (
199- config_factory = pref_extractor_config ,
200- llm_provider = llm ,
201- embedder = embedder ,
202- vector_db = vector_db ,
202+ pref_extractor = (
203+ ExtractorFactory .from_config (
204+ config_factory = pref_extractor_config ,
205+ llm_provider = llm ,
206+ embedder = embedder ,
207+ vector_db = vector_db ,
208+ )
209+ if os .getenv ("ENABLE_PREFERENCE_MEMORY" , "false" ).lower () == "true"
210+ else None
203211 )
204- pref_adder = AdderFactory .from_config (
205- config_factory = pref_adder_config ,
206- llm_provider = llm ,
207- embedder = embedder ,
208- vector_db = vector_db ,
212+ pref_adder = (
213+ AdderFactory .from_config (
214+ config_factory = pref_adder_config ,
215+ llm_provider = llm ,
216+ embedder = embedder ,
217+ vector_db = vector_db ,
218+ )
219+ if os .getenv ("ENABLE_PREFERENCE_MEMORY" , "false" ).lower () == "true"
220+ else None
209221 )
210- pref_retriever = RetrieverFactory .from_config (
211- config_factory = pref_retriever_config ,
212- llm_provider = llm ,
213- embedder = embedder ,
214- reranker = reranker ,
215- vector_db = vector_db ,
222+ pref_retriever = (
223+ RetrieverFactory .from_config (
224+ config_factory = pref_retriever_config ,
225+ llm_provider = llm ,
226+ embedder = embedder ,
227+ reranker = reranker ,
228+ vector_db = vector_db ,
229+ )
230+ if os .getenv ("ENABLE_PREFERENCE_MEMORY" , "false" ).lower () == "true"
231+ else None
216232 )
217233
218234 # Initialize memory manager
0 commit comments