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

fix: update library_client for workspace-based tenancy (#951)

Browse source 
Replace removed `user` parameter with `workspace` support following
the tenancy axis change in #840. Adds -w/--workspace flag and
$TRUSTGRAPH_WORKSPACE env var.
This commit is contained in:
cybermaggedon 2026-05-26 14:35:54 +01:00 • committed by GitHub
parent 6d07310d2b
commit 4200b5d683
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 10 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

21
dev-tools/library_client.py
View file

@ -25,7 +25,7 @@ BUCKET_URL = "https://storage.googleapis.com/trustgraph-library"
INDEX_URL = f"{BUCKET_URL}/index.json" INDEX_URL = f"{BUCKET_URL}/index.json"
default_url = os.getenv("TRUSTGRAPH_URL", "http://localhost:8088/") default_url = os.getenv("TRUSTGRAPH_URL", "http://localhost:8088/")
default_user = "trustgraph" default_workspace = os.getenv("TRUSTGRAPH_WORKSPACE", "default")
default_token = os.getenv("TRUSTGRAPH_TOKEN", None) default_token = os.getenv("TRUSTGRAPH_TOKEN", None)
@ -113,7 +113,7 @@ def convert_metadata(metadata_json):
return triples return triples
def load_document(api, user, doc_entry): def load_document(api, doc_entry):
"""Fetch metadata and content for a document, then load into TrustGraph.""" """Fetch metadata and content for a document, then load into TrustGraph."""
doc_id = doc_entry["id"] doc_id = doc_entry["id"]
title = doc_entry["title"] title = doc_entry["title"]
@ -133,7 +133,6 @@ def load_document(api, user, doc_entry):
api.add_document( api.add_document(
id=doc["id"], id=doc["id"],
metadata=metadata, metadata=metadata,
user=user,
kind=doc["kind"], kind=doc["kind"],
title=doc["title"], title=doc["title"],
comments=doc["comments"], comments=doc["comments"],
@ -144,12 +143,12 @@ def load_document(api, user, doc_entry):
print(f" done.") print(f" done.")
def load_documents(api, user, docs): def load_documents(api, docs):
"""Load a list of documents.""" """Load a list of documents."""
print(f"Loading {len(docs)} document(s)...\n") print(f"Loading {len(docs)} document(s)...\n")
for doc in docs: for doc in docs:
try: try:
load_document(api, user, doc) load_document(api, doc)
except Exception as e: except Exception as e:
print(f" FAILED: {e}", file=sys.stderr) print(f" FAILED: {e}", file=sys.stderr)
print() print()
@ -166,8 +165,8 @@ def main():
help=f"TrustGraph API URL (default: {default_url})", help=f"TrustGraph API URL (default: {default_url})",
) )
parser.add_argument( parser.add_argument(
"-U", "--user", default=default_user, "-w", "--workspace", default=default_workspace,
help=f"User ID (default: {default_user})", help=f"Workspace (default: {default_workspace})",
) )
parser.add_argument( parser.add_argument(
"-t", "--token", default=default_token, "-t", "--token", default=default_token,
@ -212,22 +211,22 @@ def main():
return return
# Load commands need the API # Load commands need the API
api = Api(args.url, token=args.token).library() api = Api(args.url, token=args.token, workspace=args.workspace).library()
if args.command == "load-all": if args.command == "load-all":
load_documents(api, args.user, index) load_documents(api, index)
elif args.command == "load-doc": elif args.command == "load-doc":
matches = [d for d in index if str(d.get("id")) == args.id] matches = [d for d in index if str(d.get("id")) == args.id]
if not matches: if not matches:
print(f"No document with ID '{args.id}' found.", file=sys.stderr) print(f"No document with ID '{args.id}' found.", file=sys.stderr)
sys.exit(1) sys.exit(1)
load_documents(api, args.user, matches) load_documents(api, matches)
elif args.command == "load-match": elif args.command == "load-match":
results = search_index(index, args.query) results = search_index(index, args.query)
if results: if results:
load_documents(api, args.user, results) load_documents(api, results)
else: else:
print("No matches found.", file=sys.stderr) print("No matches found.", file=sys.stderr)
sys.exit(1) sys.exit(1)