hello_epsilla_cloud add environment variables, add try expect

Author: destinyrobot

examples/hello_epsilla_cloud.py

@@ -7,60 +7,85 @@
 # 3. get the api key with project id, run this program
 
 import sys
-
+import os
 from pyepsilla import cloud
 
+project_id = os.getenv("epsilla_project_id")
+api_key = os.getenv("epsilla_api_key")
+db_id = os.getenv("epsilla_db_id")
+
+if not project_id or not api_key or not db_id:
+    print("Please set the environment variables: epsilla_project_id, epsilla_api_key, epsilla_db_id")
+    sys.exit(1)
+
 # Connect to Epsilla Cloud
 client = cloud.Client(
-    project_id="7a68814c-f839-4a67-9ec6-93c027c865e6",
-    api_key="epsilla-cloud-api-key",
+    project_id=project_id,
+    api_key=api_key,
 )
 
 # Connect to Vectordb
-db = client.vectordb(db_id="6accafb1-476d-43b0-aa64-12ebfbf7442d")
+db = client.vectordb(db_id=db_id)
 
 
-# Create a table with schema
-status_code, response = db.create_table(
-    table_name="MyTable",
-    table_fields=[
-        {"name": "ID", "dataType": "INT", "primaryKey": True},
-        {"name": "Doc", "dataType": "STRING"},
-        {"name": "Embedding", "dataType": "VECTOR_FLOAT", "dimensions": 4},
-    ],
-)
-print(status_code, response)
+try:
+    # Create a table with schema
+    status_code, response = db.create_table(
+        table_name="MyTable",
+        table_fields=[
+            {"name": "ID", "dataType": "INT", "primaryKey": True},
+            {"name": "Doc", "dataType": "STRING"},
+            {"name": "Embedding", "dataType": "VECTOR_FLOAT", "dimensions": 4},
+        ],
+    )
+    print(status_code, response)
+    if status_code != 200:
+        raise Exception("Failed to create table")
 
+    # Insert new vector records into table
+    status_code, response = db.insert(
+        table_name="MyTable",
+        records=[
+            {"ID": 1, "Doc": "Berlin", "Embedding": [0.05, 0.61, 0.76, 0.74]},
+            {"ID": 2, "Doc": "London", "Embedding": [0.19, 0.81, 0.75, 0.11]},
+            {"ID": 3, "Doc": "Moscow", "Embedding": [0.36, 0.55, 0.47, 0.94]},
+            {"ID": 4, "Doc": "San Francisco", "Embedding": [0.18, 0.01, 0.85, 0.80]},
+            {"ID": 5, "Doc": "Shanghai", "Embedding": [0.24, 0.18, 0.22, 0.44]},
+        ],
+    )
+    print(status_code, response)
+    if status_code != 200:
+        raise Exception("Failed to insert records")
 
-# Insert new vector records into table
-status_code, response = db.insert(
-    table_name="MyTable",
-    records=[
-        {"ID": 1, "Doc": "Berlin", "Embedding": [0.05, 0.61, 0.76, 0.74]},
-        {"ID": 2, "Doc": "London", "Embedding": [0.19, 0.81, 0.75, 0.11]},
-        {"ID": 3, "Doc": "Moscow", "Embedding": [0.36, 0.55, 0.47, 0.94]},
-        {"ID": 4, "Doc": "San Francisco", "Embedding": [0.18, 0.01, 0.85, 0.80]},
-        {"ID": 5, "Doc": "Shanghai", "Embedding": [0.24, 0.18, 0.22, 0.44]},
-    ],
-)
-print(status_code, response)
+    # Query Vectors with specific response field, otherwise it will return all fields
+    status_code, response = db.query(
+        table_name="MyTable",
+        query_field="Embedding",
+        query_vector=[0.35, 0.55, 0.47, 0.94],
+        response_fields=["Doc"],
+        limit=2,
+    )
+    print(status_code, response)
+    if status_code != 200:
+        raise Exception("Failed to query table")
 
-# Query Vectors with specific response field, otherwise it will return all fields
-status_code, response = db.query(
-    table_name="MyTable",
-    query_field="Embedding",
-    query_vector=[0.35, 0.55, 0.47, 0.94],
-    response_fields=["Doc"],
-    limit=2,
-)
-print(status_code, response)
 
+    # Delete specific records from table
+    status_code, response = db.delete(table_name="MyTable", primary_keys=[4, 5])
+    print(status_code, response)
+    if status_code != 200:
+        raise Exception("Failed to delete records by primary keys")
 
-# Delete specific records from table
-status_code, response = db.delete(table_name="MyTable", primary_keys=[4, 5])
-status_code, response = db.delete(table_name="MyTable", filter="Doc <> 'San Francisco'")
-print(status_code, response)
+    status_code, response = db.delete(table_name="MyTable", filter="Doc <> 'San Francisco'")
+    print(status_code, response)
+    if status_code != 200:
+        raise Exception("Failed to delete records by filter")
 
-# Drop table
-status_code, response = db.drop_table(table_name="MyTable")
-print(status_code, response)
+    # Drop table
+    status_code, response = db.drop_table(table_name="MyTable")
+    print(status_code, response)
+    if status_code != 200:
+        raise Exception("Failed to drop table")
+except Exception as e:
+    print(f"An error occurred: {e}")
+    sys.exit(1)