Update Modules/Agent_HCP.py

Modules/Agent_HCP.py

@@ -29,7 +29,7 @@ from langchain_core.runnables import (
 from langchain_core.runnables.history import RunnableWithMessageHistory
 from typing import Optional
 from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
-
+import sqlalchemy
 
 
 # Load the API_Keys
@@ -43,11 +43,12 @@ model = ChatGroq(temperature=0,
                  model_name="llama3-70b-8192" 
                  )
 try:
-    db = SQLDatabase.from_uri("sqlite:///"+Modules.DATA_BASE, sample_rows_in_table_info=5)
-except sqlalchemy.exc.DatabaseError as e:
+    db = SQLDatabase.from_uri(f"sqlite:///{Modules.DATA_BASE}", sample_rows_in_table_info=5)
+except DatabaseError as e:
     print(f"Database error: {e}")
     print(f"Database path: {Modules.DATA_BASE}")
-
+except sqlalchemy.exc.SQLAlchemyError as e:
+    print(f"SQLAlchemy error: {e}")
 
 custom_prompt = PromptTemplate(
     input_variables=['agent_scratchpad', 'input', 'history'],  # Ajout de 'history' comme variable d'entrée