fix the bug in example

Signed-off-by: Eric <eric@epsilla.com>

Author: eric-epsilla

examples/hello_epsilla_cloud.py

@@ -6,33 +6,40 @@
 # 2. create a table with schema in db
 # 3. get the epsilla cloud api key with project id, run this program
 
-import sys
 import os
+import sys
+
 from pyepsilla import cloud
 
-project_id = os.getenv("EPSILLA_PROJECT_ID", "Your-Project-ID")
-epsilla_api_key = os.getenv("EPSILLA_API_KEY", "Your-Epsilla-API-Key")
-db_id = os.getenv("EPSILLA_DB_ID", "Your-DB-ID")
+EPSILLA_PROJECT_ID = os.getenv("EPSILLA_PROJECT_ID", "Your-Project-ID")
+EPSILLA_API_KEY = os.getenv("EPSILLA_API_KEY", "Your-Epsilla-API-Key")
+
+DB_ID = os.getenv("DB_ID", "Your-DB-ID")
+DB_NAME = os.getenv("DB_NAME", "MyDB")
+DB_PATH = os.getenv("DB_PATH", "/tmp/epsilla_demo")
+TABLE_NAME = os.getenv("TABLE_NAME", "MyTable")
+
 
-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")
+if not EPSILLA_PROJECT_ID or not EPSILLA_API_KEY or not DB_ID:
+    print(
+        "Please set the environment variables: EPSILLA_PROJECT_ID, EPSILLA_API_KEY, DB_ID"
+    )
     sys.exit(1)
 
 # Connect to Epsilla Cloud
 cloud_client = cloud.Client(
-    project_id=project_id,
-    api_key=epsilla_api_key,
+    project_id=EPSILLA_PROJECT_ID,
+    api_key=EPSILLA_API_KEY,
 )
 
 # Connect to Vectordb
-db_cloud = cloud_client.vectordb(db_id=db_id)
-
+db_client = cloud_client.vectordb(db_id=DB_ID)
 
 
 try:
     # Create a table with schema
-    status_code, response = db.create_table(
-        table_name="MyTable",
+    status_code, response = db_client.create_table(
+        table_name=TABLE_NAME,
         table_fields=[
             {"name": "ID", "dataType": "INT", "primaryKey": True},
             {"name": "Doc", "dataType": "STRING"},
@@ -44,8 +51,8 @@
         raise Exception("Failed to create table")
 
     # Insert new vector records into table
-    status_code, response = db.insert(
-        table_name="MyTable",
+    status_code, response = db_client.insert(
+        table_name=TABLE_NAME,
         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]},
@@ -59,8 +66,8 @@
         raise Exception("Failed to insert records")
 
     # Query Vectors with specific response field, otherwise it will return all fields
-    status_code, response = db.query(
-        table_name="MyTable",
+    status_code, response = db_client.query(
+        table_name=TABLE_NAME,
         query_field="Embedding",
         query_vector=[0.35, 0.55, 0.47, 0.94],
         response_fields=["Doc"],
@@ -70,23 +77,24 @@
     if status_code != 200:
         raise Exception("Failed to query table")
 
-
     # Delete specific records from table
-    status_code, response = db.delete(table_name="MyTable", primary_keys=[4, 5])
+    status_code, response = db_client.delete(table_name=TABLE_NAME, primary_keys=[4, 5])
     print(status_code, response)
     if status_code != 200:
         raise Exception("Failed to delete records by primary keys")
 
-    status_code, response = db.delete(table_name="MyTable", filter="Doc <> 'San Francisco'")
+    status_code, response = db_client.delete(
+        table_name=TABLE_NAME, 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")
+    status_code, response = db_client.drop_table(table_name=TABLE_NAME)
     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)
+    sys.exit(1)