NeoAnd/evo-ai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(custom_tools): URL encode path parameters and improve response handling

Browse Source
This commit is contained in:
Guilherme Gomes 2025-05-19 01:21:30 -03:00
parent ef4e4ee1c7
commit 9135aa59d6
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/services/custom_tools.py
View File

@ -31,6 +31,7 @@ from typing import Any, Dict, List
from google.adk.tools import FunctionTool from google.adk.tools import FunctionTool
import requests import requests
import json import json
import urllib.parse
from src.utils.logger import setup_logger from src.utils.logger import setup_logger
logger = setup_logger(__name__) logger = setup_logger(__name__)
@ -70,7 +71,9 @@ class CustomToolBuilder:
url = endpoint url = endpoint
for param, value in path_params.items(): for param, value in path_params.items():
if param in all_values: if param in all_values:
url = url.replace(f"{{{param}}}", str(all_values[param])) # URL encode the value for URL safe characters
replacement_value = urllib.parse.quote(str(all_values[param]), safe='')
url = url.replace(f"{{{param}}}", replacement_value)
# Process query parameters # Process query parameters
query_params_dict = {} query_params_dict = {}
@ -119,8 +122,12 @@ class CustomToolBuilder:
f"Error in the request: {response.status_code} - {response.text}" f"Error in the request: {response.status_code} - {response.text}"
) )
# Always returns the response as a string # Try to parse the response as JSON, if it fails, return the text content
return json.dumps(response.json()) try:
return json.dumps(response.json())
except ValueError:
# Response is not JSON, return the text content
return json.dumps({"content": response.text})
except Exception as e: except Exception as e:
logger.error(f"Error executing tool {name}: {str(e)}") logger.error(f"Error executing tool {name}: {str(e)}")