Samad2023 commited on
Commit
d5df257
·
verified ·
1 Parent(s): 9d98dd8

Update Modules/Agent_HCP.py

Browse files
Files changed (1) hide show
  1. Modules/Agent_HCP.py +6 -1
Modules/Agent_HCP.py CHANGED
@@ -42,7 +42,12 @@ model = ChatGroq(temperature=0,
42
  #model_name="llama3-8b-8192"
43
  model_name="llama3-70b-8192"
44
  )
45
- db = SQLDatabase.from_uri("sqlite:///"+Modules.DATA_BASE, sample_rows_in_table_info=5)
 
 
 
 
 
46
 
47
  custom_prompt = PromptTemplate(
48
  input_variables=['agent_scratchpad', 'input', 'history'], # Ajout de 'history' comme variable d'entrée
 
42
  #model_name="llama3-8b-8192"
43
  model_name="llama3-70b-8192"
44
  )
45
+ try:
46
+ db = SQLDatabase.from_uri("sqlite:///"+Modules.DATA_BASE, sample_rows_in_table_info=5)
47
+ except sqlalchemy.exc.DatabaseError as e:
48
+ print(f"Database error: {e}")
49
+ print(f"Database path: {Modules.DATA_BASE}")
50
+
51
 
52
  custom_prompt = PromptTemplate(
53
  input_variables=['agent_scratchpad', 'input', 'history'], # Ajout de 'history' comme variable d'entrée