fixes for #45, #46

ai_adapter.py

@@ -8,6 +8,7 @@
 from langchain_core.runnables import RunnableLambda, RunnablePassthrough
 from langchain.schema import format_document
 from langchain_core.messages import get_buffer_string
+from langchain_core.messages.ai import AIMessage
 from langchain_core.runnables import RunnableBranch
 
 from operator import itemgetter
@@ -254,6 +255,12 @@ async def query_chain(question, language, chat_history):
         loaded_memory | standalone_question | retrieved_documents_sa | answer,
     )
 
-    logger.debug(f"final chain {final_chain}\n")
-    result = await final_chain.ainvoke(question)
-    return {'answer': result['answer'], 'source_documents': result['docs']}
+    try:
+        logger.debug(f"final chain {final_chain}\n")
+        result = await final_chain.ainvoke(question)
+    except Exception as e:
+        logger.error(f"An error occurred while generating a response: {str(e)}")
+        # Handle the error appropriately here
+        return {'answer': AIMessage(content='An error occurred while generating a response.'), 'source_documents': []}
+    else:
+        return {'answer': result['answer'], 'source_documents': result['docs'] if result['docs'] else []}

guidance_engine.py

@@ -80,11 +80,7 @@ async def query(user_id, query, language_code):
         logger.info(f"\nQuery from user {user_id}: {query}\n")
 
         if user_id not in user_data:
-            user_data[user_id] = {}
-            user_data[user_id]['chat_history'] = ConversationBufferWindowMemory(k=3, return_messages=True, output_key="answer", input_key="question")
-            # user_chain[user_id]=ai_utils.setup_chain()
             reset(user_id)
-            # chat_history=[]
 
         user_data[user_id]['language'] = ai_adapter.get_language_by_code(language_code)
 
@@ -116,8 +112,10 @@ async def query(user_id, query, language_code):
         return response
 
 def reset(user_id):
+    if user_id not in user_data:
+        user_data[user_id] = {}
+        user_data[user_id]['chat_history'] = ConversationBufferWindowMemory(k=3, return_messages=True, output_key="answer", input_key="question")
     user_data[user_id]['chat_history'].clear()
-
     return "Reset function executed"
 
 async def ingest(source_url, website_repo, destination_path, source_path, source_url2, website_repo2, destination_path2, source_path2):
@@ -182,8 +180,10 @@ async def process_message(message: aio_pika.IncomingMessage):
                 else:
                     response = "Query parameter(s) not provided"
             elif operation == 'reset':
+                logger.info(f"reset user id: {user_id}\n\n")
                 response = reset(user_id)
             else:
+                logger.error(f"unknown function for user: {user_id}\n\n")
                 response = "Unknown function"
 
     try:

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "guidance-engine"
-version = "0.6.0"
+version = "0.6.1"
 description = "Alkemio Generative AI Guidance Engine"
 authors = ["Alkemio BV <[email protected]>"]
 license = "EUPL-1.2"