mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-11 00:58:30 +00:00
feat: Reduce level for Bandit and fix code to adapt (#1604)
This commit is contained in:
committed by
GitHub
parent
e1b1a9361c
commit
5ca23ce757
2
.github/workflows/security-checker.yml
vendored
2
.github/workflows/security-checker.yml
vendored
@@ -19,5 +19,5 @@ jobs:
|
|||||||
run: pip install bandit
|
run: pip install bandit
|
||||||
|
|
||||||
- name: Run Bandit
|
- name: Run Bandit
|
||||||
run: bandit -c pyproject.toml -r src/ -lll
|
run: bandit -c pyproject.toml -r src/ -ll
|
||||||
|
|
||||||
|
|||||||
@@ -34,7 +34,9 @@ class AuthenticationCommand:
|
|||||||
"scope": "openid",
|
"scope": "openid",
|
||||||
"audience": AUTH0_AUDIENCE,
|
"audience": AUTH0_AUDIENCE,
|
||||||
}
|
}
|
||||||
response = requests.post(url=self.DEVICE_CODE_URL, data=device_code_payload)
|
response = requests.post(
|
||||||
|
url=self.DEVICE_CODE_URL, data=device_code_payload, timeout=20
|
||||||
|
)
|
||||||
response.raise_for_status()
|
response.raise_for_status()
|
||||||
return response.json()
|
return response.json()
|
||||||
|
|
||||||
@@ -54,7 +56,7 @@ class AuthenticationCommand:
|
|||||||
|
|
||||||
attempts = 0
|
attempts = 0
|
||||||
while True and attempts < 5:
|
while True and attempts < 5:
|
||||||
response = requests.post(self.TOKEN_URL, data=token_payload)
|
response = requests.post(self.TOKEN_URL, data=token_payload, timeout=30)
|
||||||
token_data = response.json()
|
token_data = response.json()
|
||||||
|
|
||||||
if response.status_code == 200:
|
if response.status_code == 200:
|
||||||
|
|||||||
@@ -103,7 +103,7 @@ class KickoffTaskOutputsSQLiteStorage:
|
|||||||
else value
|
else value
|
||||||
)
|
)
|
||||||
|
|
||||||
query = f"UPDATE latest_kickoff_task_outputs SET {', '.join(fields)} WHERE task_index = ?"
|
query = f"UPDATE latest_kickoff_task_outputs SET {', '.join(fields)} WHERE task_index = ?" # nosec
|
||||||
values.append(task_index)
|
values.append(task_index)
|
||||||
|
|
||||||
cursor.execute(query, tuple(values))
|
cursor.execute(query, tuple(values))
|
||||||
|
|||||||
@@ -83,7 +83,7 @@ class LTMSQLiteStorage:
|
|||||||
WHERE task_description = ?
|
WHERE task_description = ?
|
||||||
ORDER BY datetime DESC, score ASC
|
ORDER BY datetime DESC, score ASC
|
||||||
LIMIT {latest_n}
|
LIMIT {latest_n}
|
||||||
""",
|
""", # nosec
|
||||||
(task_description,),
|
(task_description,),
|
||||||
)
|
)
|
||||||
rows = cursor.fetchall()
|
rows = cursor.fetchall()
|
||||||
|
|||||||
@@ -16,7 +16,11 @@ class FileHandler:
|
|||||||
|
|
||||||
def log(self, **kwargs):
|
def log(self, **kwargs):
|
||||||
now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
||||||
message = f"{now}: " + ", ".join([f"{key}=\"{value}\"" for key, value in kwargs.items()]) + "\n"
|
message = (
|
||||||
|
f"{now}: "
|
||||||
|
+ ", ".join([f'{key}="{value}"' for key, value in kwargs.items()])
|
||||||
|
+ "\n"
|
||||||
|
)
|
||||||
with open(self._path, "a", encoding="utf-8") as file:
|
with open(self._path, "a", encoding="utf-8") as file:
|
||||||
file.write(message + "\n")
|
file.write(message + "\n")
|
||||||
|
|
||||||
@@ -63,7 +67,7 @@ class PickleHandler:
|
|||||||
|
|
||||||
with open(self.file_path, "rb") as file:
|
with open(self.file_path, "rb") as file:
|
||||||
try:
|
try:
|
||||||
return pickle.load(file)
|
return pickle.load(file) # nosec
|
||||||
except EOFError:
|
except EOFError:
|
||||||
return {} # Return an empty dictionary if the file is empty or corrupted
|
return {} # Return an empty dictionary if the file is empty or corrupted
|
||||||
except Exception:
|
except Exception:
|
||||||
|
|||||||
Reference in New Issue
Block a user