* Fix LMStudio, cache documents with tg-load-sample-documents

* Fix Mistral
This commit is contained in:
cybermaggedon 2025-05-06 16:17:16 +01:00 committed by GitHub
parent 54e475fa3a
commit d0da122bed
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 25 additions and 7 deletions

View file

@ -11,7 +11,7 @@ import datetime
import requests
from trustgraph.api import Api
from trustgraph.api.types import Uri, Literal, Triple
from trustgraph.api.types import hash, Uri, Literal, Triple
default_url = os.getenv("TRUSTGRAPH_URL", 'http://localhost:8088/')
default_user = 'trustgraph'
@ -29,6 +29,11 @@ session = requests.session()
session.mount('file://', FileAdapter())
try:
os.mkdir("doc-cache")
except:
pass
documents = [
{
@ -669,11 +674,19 @@ class Loader:
try:
print(doc["title"], ":")
print(" downloading...")
print(" done.")
resp = session.get(doc["url"])
content = resp.content
hid = hash(doc["url"])
cache_file = f"doc-cache/{hid}"
if os.path.isfile(cache_file):
print(" (use cache file)")
content = open(cache_file, "rb").read()
else:
print(" downloading...")
resp = session.get(doc["url"])
content = resp.content
open(cache_file, "wb").write(content)
print(" done.")
print(" adding...")