Fixing comments

Author: ankush-cohere

examples/fern_documentation/python/core/add_metadata.py

@@ -18,4 +18,4 @@
 )
 
 if error:
-    raise Exception(f"Failed to create indexes: {error}")
+    raise Exception(f"Failed to add attributes: {error}")

examples/fern_documentation/python/core/create_index.py

@@ -8,4 +8,4 @@
 )
 r = compass_client.create_index(index_name=os.getenv("INDEX_NAME"))
 if r.error:
-    raise Exception(f"Failed to create indexes: {r.error}")
+    raise Exception(f"Failed to create index: {r.error}")

examples/fern_documentation/python/core/delete_doc.py

@@ -11,4 +11,4 @@
 )
 
 if r.error:
-    raise Exception(f"Failed to create indexes: {r.error}")
+    raise Exception(f"Failed to delete doc: {r.error}")

examples/fern_documentation/python/core/delete_index.py

@@ -8,4 +8,4 @@
 )
 r = compass_client.delete_index(index_name=os.getenv("INDEX_NAME"))
 if r.error:
-    raise Exception(f"Failed to create indexes: {r.error}")
+    raise Exception(f"Failed to delete index: {r.error}")

examples/fern_documentation/python/core/get_doc_asset.py

@@ -1,3 +1,4 @@
+import json
 import os
 
 from cohere.compass.clients import CompassClient
@@ -7,10 +8,36 @@
     bearer_token=os.getenv("COMPASS_API_BEARER_TOKEN"),
 )
 
-r = compass_client.get_document_asset(
-    index_name=os.getenv("INDEX_NAME"),
+results = compass_client.search_chunks(
+    index_name=os.getenv("INDEX_NAME"), query=os.getenv("SEARCH_QUERY")
+)
+if results.error:
+    raise Exception(f"Failed to search chunks: {results.error}")
+
+hits = results.hits
+
+if not hits:
+    raise Exception("No hits found")
+
+hit = hits[0]
+
+asset_id: str = hit.assets_info[0].asset_id  # type: ignore
+document_id = hit.document_id
+# URL to fetch asset
+presigned_url = hit.assets_info[0].presigned_url
+
+
+### Get document asset again, after pre-signed url has expired
+asset, content_type = compass_client.get_document_asset(
+    index_name=document_id,
     document_id=os.getenv("DOCUMENT_ID"),
-    asset_id=os.getenv("ASSET_ID"),
+    asset_id=asset_id,
 )
 
-print(r)
+# Save the asset to a file.
+if content_type in ["image/jpeg", "image/png"]:
+    with open(f"{asset_id}", "wb") as f:
+        f.write(asset)
+# Print the asset as JSON
+elif "text/json" in content_type:
+    print(json.dumps(asset, indent=2))

examples/fern_documentation/python/core/get_document.py

@@ -11,6 +11,6 @@
 )
 
 if r.error:
-    raise Exception(f"Failed to create indexes: {r.error}")
+    raise Exception(f"Failed to get doc: {r.error}")
 
-print(r.hits)
+print(r.result.hits)

examples/fern_documentation/python/core/list_index.py

@@ -13,5 +13,5 @@
 indexes = r.result["indexes"]
 for index in indexes:
     print(f"Index name: {index['name']}")
-    print(f"Chunk counts: {index['count']}")
-    print(f"Document counts: {index['parent_doc_count']}")
+    print(f"Chunk count: {index['count']}")
+    print(f"Document count: {index['parent_doc_count']}")

examples/fern_documentation/python/core/refresh_index.py

@@ -8,4 +8,4 @@
 )
 r = compass_client.refresh_index(index_name=os.getenv("INDEX_NAME"))
 if r.error:
-    raise Exception(f"Failed to create indexes: {r.error}")
+    raise Exception(f"Failed to refresh indexes: {r.error}")

examples/fern_documentation/python/core/search_chunks.py

@@ -7,10 +7,10 @@
     bearer_token=os.getenv("COMPASS_API_BEARER_TOKEN"),
 )
 r = compass_client.search_chunks(
-    index_name=os.getenv("INDEX_NAME"), query="hello world"
+    index_name=os.getenv("INDEX_NAME"), query=os.getenv("SEARCH_QUERY")
 )
 
 if r.error:
-    raise Exception(f"Failed to create indexes: {r.error}")
+    raise Exception(f"Failed to search: {r.error}")
 
 print(r.hits)

examples/fern_documentation/python/core/search_doc.py examples/fern_documentation/python/core/search_docs.py

@@ -7,10 +7,10 @@
     bearer_token=os.getenv("COMPASS_API_BEARER_TOKEN"),
 )
 r = compass_client.search_documents(
-    index_name=os.getenv("INDEX_NAME"), query="hello world"
+    index_name=os.getenv("INDEX_NAME"), query=os.getenv("SEARCH_QUERY")
 )
 
 if r.error:
-    raise Exception(f"Failed to create indexes: {r.error}")
+    raise Exception(f"Failed to search doc: {r.error}")
 
 print(r.hits)