Spaces:
Sleeping
Sleeping
Commit
·
bd402e1
1
Parent(s):
31325d7
fixed reference extraction
Browse files
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
|