|
|
@@ -34,9 +34,7 @@ class R2R:
|
|
|
self.client_sync.users.login(
|
|
|
tool_settings.R2R_USERNAME, tool_settings.R2R_PASSWORD
|
|
|
)
|
|
|
- #print(
|
|
|
- "1111111111111111111111111111111122222vvdgdfdf" + tool_settings.R2R_USERNAME
|
|
|
- )
|
|
|
+ #print( "1111111111111111111111111111111122222vvdgdfdf" + tool_settings.R2R_USERNAME)
|
|
|
# #print(tool_settings.R2R_USERNAME)
|
|
|
# #print(tool_settings.R2R_PASSWORD)
|
|
|
#print(self.client_sync)
|
|
|
@@ -46,9 +44,7 @@ class R2R:
|
|
|
if not self.auth_enabled:
|
|
|
return
|
|
|
# if not self.client:
|
|
|
- #print(
|
|
|
- "1111111111111111111111111111111122222vvdgdfdf" + tool_settings.R2R_USERNAME
|
|
|
- )
|
|
|
+ #print( "1111111111111111111111111111111122222vvdgdfdf" + tool_settings.R2R_USERNAME)
|
|
|
#print(tool_settings.R2R_USERNAME)
|
|
|
#print(tool_settings.R2R_PASSWORD)
|
|
|
# client = R2RAsyncClient(tool_settings.R2R_BASE_URL)
|
|
|
@@ -78,9 +74,7 @@ class R2R:
|
|
|
|
|
|
def search(self, query: str, filters: dict[str, Any]):
|
|
|
client = self._check_login_sync()
|
|
|
- #print(
|
|
|
- "aaaaaaaaaaaaaaaaaaaaaaaaaaaasssssssssssssssssssssssssssssssssssssssssgggggggggggggggggggg"
|
|
|
- )
|
|
|
+ #print( "aaaaaaaaaaaaaaaaaaaaaaaaaaaasssssssssssssssssssssssssssssssssssssssssgggggggggggggggggggg")
|
|
|
search_response = client.retrieval.search(
|
|
|
query=query,
|
|
|
# search_mode="basic",
|
|
|
@@ -106,9 +100,7 @@ class R2R:
|
|
|
|
|
|
def list_chunks(self, ids: list[str] = []):
|
|
|
client = self._check_login_sync()
|
|
|
- #print(
|
|
|
- "retrieve_documentsretrieve_documentsretrieve_documentsretrieve_documentsretrieve_documents"
|
|
|
- )
|
|
|
+ #print( "retrieve_documentsretrieve_documentsretrieve_documentsretrieve_documentsretrieve_documents")
|
|
|
#print(ids)
|
|
|
allfile = []
|
|
|
for id in ids:
|
|
|
@@ -128,9 +120,7 @@ class R2R:
|
|
|
docs = client.collections.list_documents(empty_coll_id).results
|
|
|
assert len(docs) == 0, "Expected no documents in a new empty collection"
|
|
|
"""
|
|
|
- #print(
|
|
|
- "collectionscollectionscollectionscollectionscollectionscollectionscollectionscollectionscollectionscollectionscollectionscollections"
|
|
|
- )
|
|
|
+ #print( "collectionscollectionscollectionscollectionscollectionscollectionscollectionscollectionscollectionscollectionscollectionscollections")
|
|
|
if id != "":
|
|
|
try:
|
|
|
listed = client.collections.list_documents(
|