Skip to content

Commit 9e6b66d

Browse files
authored
fix critical security bug (open-edge-platform#220)
1 parent e9bca53 commit 9e6b66d

File tree

1 file changed

+5
-3
lines changed
  • microservices/document-ingestion/pgvector

1 file changed

+5
-3
lines changed

microservices/document-ingestion/pgvector/main.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -172,9 +172,11 @@ def ingest_url_to_pgvector(url_list: List[str]) -> None:
172172
try:
173173
invalid_urls = 0
174174
for url in url_list:
175-
response = requests.get(url, timeout=5, allow_redirects=True)
176-
if response.status_code != 200:
177-
invalid_urls += 1
175+
if url.startswith('http://') or url.startswith('https://'):
176+
response = requests.get(url, timeout=5, allow_redirects=True)
177+
if response.status_code == 200:
178+
continue
179+
invalid_urls += 1
178180

179181
if invalid_urls > 0:
180182
raise Exception(

0 commit comments

Comments
 (0)