Skip to content

Commit

Permalink
remove debugging comments
Browse files Browse the repository at this point in the history
  • Loading branch information
Ismael Cruz authored and Ismael Cruz committed Dec 5, 2024
1 parent f9e6c4e commit 5637278
Showing 1 changed file with 0 additions and 6 deletions.
6 changes: 0 additions & 6 deletions utils/recipes.py
Original file line number Diff line number Diff line change
Expand Up @@ -88,21 +88,16 @@ def initialize_vector_db():
db = {}

embedding_model, chat = get_models()
print(f'### embedding model: {embedding_model} ###')
print(f'### chat: {chat} ###')

# This will create store tables if they don't exist
print('### create store tables ###')
for mem_type in similarity_cutoff.keys():
print(f'### {mem_type} ###')
COLLECTION_NAME = f"{mem_type}_embedding"
db[mem_type] = PGVector(
collection_name=COLLECTION_NAME,
connection_string=CONNECTION_STRING,
embedding_function=embedding_model,
create_extension=False,
)
print(f'### {mem_type} SUCCESS ###')

return db

Expand Down Expand Up @@ -182,7 +177,6 @@ def check_recipe_memory(intent, debug=True):
if debug:
print(f"======= Checking {mem_type} for intent: {intent}")
docs = db[mem_type].similarity_search_with_score(intent, k=3)
print('### docs SUCCESS ###')
for d in docs:
score = d[1]
content = d[0].page_content
Expand Down

0 comments on commit 5637278

Please sign in to comment.