aljaz/Auto-GPT
1
0
Fork 0
mirror of https://github.com/aljazceru/Auto-GPT.git synced 2025-12-18 14:34:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

Rework local file address blocks

Browse Source 
add `def check_local_file_access`, which defines and checks for local file address prefixes; use it to restrict access
This commit is contained in:
onekum
2023-04-10 12:10:28 -04:00
committed by GitHub
parent fa8461be9d
commit b60c7518b0
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
scripts/browse.py
View File

@@ -5,6 +5,11 @@ from llm_utils import create_chat_completion
cfg = Config() cfg = Config()
# Define and check for local file address prefixes
def check_local_file_access(url):
local_prefixes = ['file:///', 'file://localhost', 'http://localhost', 'https://localhost']
return any(url.startswith(prefix) for prefix in local_prefixes)
def scrape_text(url): def scrape_text(url):
"""Scrape text from a webpage""" """Scrape text from a webpage"""
# Most basic check if the URL is valid: # Most basic check if the URL is valid:
@@ -12,7 +17,7 @@ def scrape_text(url):
return "Error: Invalid URL" return "Error: Invalid URL"
# Restrict access to local files # Restrict access to local files
if url.startswith('file://') or url.startswith('file://localhost'): if check_local_file_access(url):
return "Error: Access to local files is restricted" return "Error: Access to local files is restricted"
try: try: