JPM34 commited on
Commit
4e188ba
·
1 Parent(s): 6c9d559

Add DEBUG

Browse files
Files changed (1) hide show
  1. agent_gemini.py +3 -0
agent_gemini.py CHANGED
@@ -123,6 +123,7 @@ class BasicAgent:
123
  attempt = 0
124
 
125
  # Create a temporary directory that will be automatically cleaned up
 
126
  with tempfile.TemporaryDirectory() as temp_dir:
127
  while attempt < max_retries:
128
  # default_api_url = os.getenv("DEFAULT_API_URL")
@@ -130,6 +131,7 @@ class BasicAgent:
130
  file_url = f"{default_api_url}/files/{task_id}"
131
 
132
  try:
 
133
  # Download file to temporary directory
134
  file = download_file_from_url.invoke(
135
  {
@@ -142,6 +144,7 @@ class BasicAgent:
142
  file = None
143
 
144
  try:
 
145
  attempt += 1
146
  logger.info(f"Attempt {attempt} of {max_retries}")
147
 
 
123
  attempt = 0
124
 
125
  # Create a temporary directory that will be automatically cleaned up
126
+ print("HELLO")
127
  with tempfile.TemporaryDirectory() as temp_dir:
128
  while attempt < max_retries:
129
  # default_api_url = os.getenv("DEFAULT_API_URL")
 
131
  file_url = f"{default_api_url}/files/{task_id}"
132
 
133
  try:
134
+ print("HELLO-A")
135
  # Download file to temporary directory
136
  file = download_file_from_url.invoke(
137
  {
 
144
  file = None
145
 
146
  try:
147
+ print("HELLO-B")
148
  attempt += 1
149
  logger.info(f"Attempt {attempt} of {max_retries}")
150