apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-05-30 17:55:13 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Merge pull request #952 from jmolz/fix-load-sample-doc-cache

Browse source 
fix: avoid swallowing sample document cache errors
This commit is contained in:
cybermaggedon 2026-05-26 15:03:36 +01:00 • committed by GitHub
commit f8e8c0e9e6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

11
trustgraph-cli/trustgraph/cli/load_sample_documents.py
View file

@ -4,9 +4,10 @@ Loads a PDF document into the library
import argparse import argparse
import os import os
import uuid
import datetime import datetime
import requests import requests
from requests.adapters import HTTPAdapter
from urllib3.response import HTTPResponse
from trustgraph.api import Api from trustgraph.api import Api
from trustgraph.api.types import hash, Uri, Literal, Triple from trustgraph.api.types import hash, Uri, Literal, Triple
@ -16,9 +17,6 @@ default_token = os.getenv("TRUSTGRAPH_TOKEN", None)
default_workspace = os.getenv("TRUSTGRAPH_WORKSPACE", "default") default_workspace = os.getenv("TRUSTGRAPH_WORKSPACE", "default")
from requests.adapters import HTTPAdapter
from urllib3.response import HTTPResponse
class FileAdapter(HTTPAdapter): class FileAdapter(HTTPAdapter):
def send(self, request, *args, **kwargs): def send(self, request, *args, **kwargs):
resp = HTTPResponse(body=open(request.url[7:], 'rb'), status=200, preload_content=False) resp = HTTPResponse(body=open(request.url[7:], 'rb'), status=200, preload_content=False)
@ -28,10 +26,7 @@ session = requests.session()
session.mount('file://', FileAdapter()) session.mount('file://', FileAdapter())
try: os.makedirs("doc-cache", exist_ok=True)
os.mkdir("doc-cache")
except:
pass
documents = [ documents = [