Skip to content

Commit

Permalink
Fix
Browse files Browse the repository at this point in the history
  • Loading branch information
Dicklesworthstone committed May 21, 2024
1 parent 8627064 commit 513fd77
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion swiss_army_llama.py
Original file line number Diff line number Diff line change
Expand Up @@ -847,7 +847,7 @@ async def get_all_embedding_vectors_for_document(
lock = await shared_resources.lock_manager.lock(unique_id)
if lock.valid:
try:
async with AsyncSession() as session:
async with AsyncSessionLocal() as session:
result = await session.execute(select(DocumentEmbedding).filter(DocumentEmbedding.file_hash == file_hash, DocumentEmbedding.llm_model_name == llm_model_name))
existing_document_embedding = result.scalar_one_or_none()
if existing_document_embedding:
Expand Down

0 comments on commit 513fd77

Please sign in to comment.