Skip to content

Commit

Permalink
Fix lint errors
Browse files Browse the repository at this point in the history
  • Loading branch information
humpydonkey committed Jul 3, 2024
1 parent 2459449 commit 695334f
Showing 1 changed file with 2 additions and 5 deletions.
7 changes: 2 additions & 5 deletions vision_agent/tools/tool_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,10 +23,7 @@ def send_inference_request(
if "TOOL_ENDPOINT_URL" in os.environ:
url = os.environ["TOOL_ENDPOINT_URL"]

headers = {
"Content-Type": "application/json",
"apikey": _LND_API_KEY
}
headers = {"Content-Type": "application/json", "apikey": _LND_API_KEY}
if "TOOL_ENDPOINT_AUTH" in os.environ:
headers["Authorization"] = os.environ["TOOL_ENDPOINT_AUTH"]
headers.pop("apikey")
Expand All @@ -43,7 +40,7 @@ def send_inference_request(

resp = res.json()
# TODO: consider making the response schema the same between below two sources
return resp if "TOOL_ENDPOINT_AUTH" in os.environ else resp["data"]
return resp if "TOOL_ENDPOINT_AUTH" in os.environ else resp["data"] # type: ignore


def _create_requests_session(
Expand Down

0 comments on commit 695334f

Please sign in to comment.