jaywadekar commited on
Commit
bd402e1
·
1 Parent(s): 31325d7

fixed reference extraction

Browse files
Files changed (1) hide show
  1. rag.py +6 -0
rag.py CHANGED
@@ -365,6 +365,12 @@ def extract_reference(url):
365
  return url.split("blob/main/")[-1]
366
  elif "tree/main" in url:
367
  return url.split("tree/main/")[-1] or "root"
 
 
 
 
 
 
368
  return url
369
 
370
  # Join content pages for processing
 
365
  return url.split("blob/main/")[-1]
366
  elif "tree/main" in url:
367
  return url.split("tree/main/")[-1] or "root"
368
+ elif "blob/master" in url:
369
+ return url.split("blob/master/")[-1]
370
+ elif "tree/master" in url:
371
+ return url.split("tree/master/")[-1] or "root"
372
+ elif "refs/heads/master" in url:
373
+ return url.split("refs/heads/master/")[-1]
374
  return url
375
 
376
  # Join content pages for processing