Add DEBUG
Browse files- 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 |
|