Fix context retireval
This commit is contained in:
parent
f41f09171c
commit
fb542a3267
1 changed files with 8 additions and 4 deletions
12
main.py
12
main.py
|
|
@ -39,7 +39,7 @@ from sqlalchemy.ext.asyncio import AsyncSession
|
|||
from sqlalchemy.future import select
|
||||
from cognitive_architecture.utils import get_document_names, generate_letter_uuid, get_memory_name_by_doc_id, get_unsumarized_vector_db_namespace, get_vectordb_namespace, get_vectordb_document_name
|
||||
from cognitive_architecture.shared.language_processing import translate_text, detect_language
|
||||
|
||||
from cognitive_architecture.classifiers.classifier import classify_user_input
|
||||
|
||||
async def fetch_document_vectordb_namespace(session: AsyncSession, user_id: str, namespace_id:str, doc_id:str=None):
|
||||
logging.info("user id is", user_id)
|
||||
|
|
@ -609,10 +609,14 @@ async def unlink_user_from_memory(user_id: str=None, labels:list=None, topic:str
|
|||
logging.error(f"Error creating public memory node: {e}")
|
||||
return None
|
||||
|
||||
async def relevance_feedback( query: str, input_type: str):
|
||||
from cognitive_architecture.classifiers.classifier import classify_user_input
|
||||
async def relevance_feedback(query: str, input_type: str):
|
||||
|
||||
result = await classify_user_input( query, input_type=input_type)
|
||||
max_attempts = 6
|
||||
result = None
|
||||
for attempt in range(1, max_attempts + 1):
|
||||
result = await classify_user_input(query, input_type=input_type)
|
||||
if isinstance(result, bool):
|
||||
break # Exit the loop if a result of type bool is obtained
|
||||
return result
|
||||
|
||||
async def main():
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue