File tree Expand file tree Collapse file tree 1 file changed +0
-6
lines changed Expand file tree Collapse file tree 1 file changed +0
-6
lines changed Original file line number Diff line number Diff line change @@ -88,21 +88,16 @@ def initialize_vector_db():
8888 db = {}
8989
9090 embedding_model , chat = get_models ()
91- print (f'### embedding model: { embedding_model } ###' )
92- print (f'### chat: { chat } ###' )
9391
9492 # This will create store tables if they don't exist
95- print ('### create store tables ###' )
9693 for mem_type in similarity_cutoff .keys ():
97- print (f'### { mem_type } ###' )
9894 COLLECTION_NAME = f"{ mem_type } _embedding"
9995 db [mem_type ] = PGVector (
10096 collection_name = COLLECTION_NAME ,
10197 connection_string = CONNECTION_STRING ,
10298 embedding_function = embedding_model ,
10399 create_extension = False ,
104100 )
105- print (f'### { mem_type } SUCCESS ###' )
106101
107102 return db
108103
@@ -182,7 +177,6 @@ def check_recipe_memory(intent, debug=True):
182177 if debug :
183178 print (f"======= Checking { mem_type } for intent: { intent } " )
184179 docs = db [mem_type ].similarity_search_with_score (intent , k = 3 )
185- print ('### docs SUCCESS ###' )
186180 for d in docs :
187181 score = d [1 ]
188182 content = d [0 ].page_content
You can’t perform that action at this time.
0 commit comments