updated the base usage examples

Author: MinuraPunchihewa

examples/base_usage.py

@@ -1,19 +1,22 @@
 from minds.client import Client
+from openai import OpenAI
 
-# --- connect ---
-client = Client("YOUR_API_KEY")
 
-# or use not default server
-base_url = 'https://custom_cloud.mdb.ai/'
-client = Client("YOUR_API_KEY", base_url)
+# Basic Setup and Workflow
 
+# connect
+API_KEY = "YOUR_API_KEY"
+BASE_URL = 'https://custom_cloud.mdb.ai/api/v1'  # optional, if you use custom server
 
-# create datasource
-from minds.datasources import DatabaseConfig
+client = Client(API_KEY)
 
-postgres_config = DatabaseConfig(
+# or with custom base URL
+client = Client(API_KEY, base_url=BASE_URL)
+
+# create Datasource
+datasource = client.datasources.create(
     name='my_datasource',
-	description='<DESCRIPTION-OF-YOUR-DATA>',
+	description='House sales data',
 	engine='postgres',
 	connection_data={
     	'user': 'demo_user',
@@ -22,80 +25,128 @@
     	'port': 5432,
     	'database': 'demo',
     	'schema': 'demo_data'
-	},
-	tables=['<TABLE-1>', '<TABLE-2>']
+	}
 )
 
-# using sample config
-from minds.datasources.examples import example_ds
-
-# create mind
-# with datasource at the same time
-mind = client.minds.create(name='mind_name', datasources=[postgres_config] )
-
-# or separately
-datasource = client.datasources.create(postgres_config)
-mind = client.minds.create(name='mind_name', datasources=[datasource] )
+# create Mind
+mind = client.minds.create(
+    name='mind_name',
+    datasources=[
+		{
+			'name': datasource.name,
+			'tables': ['house_sales']
+		}
+	]
+)
 
-# with prompt template
-mind = client.minds.create(name='mind_name', prompt_template='You are codding assistant')
+# or add to existing Mind
+mind = client.minds.create(name='mind_name')
+mind.add_datasource(datasource.name, tables=['house_sales'])
+
+# chat with the Mind using the OpenAI-compatible Completions API (without streaming)
+openai_client = OpenAI(api_key=API_KEY, base_url=BASE_URL)
+completion = openai_client.chat.completions.create(
+    model=mind.name,
+    messages=[
+        {'role': 'user', 'content': 'How many three-bedroom houses were sold in 2008?'}
+    ],
+    stream=False
+)
+print(completion.choices[0].message.content)
+
+# with streaming
+completion_stream = openai_client.chat.completions.create(
+	model=mind.name,
+	messages=[
+		{'role': 'user', 'content': 'How many three-bedroom houses were sold in 2008?'}
+	],
+	stream=True
+)
+for chunk in completion_stream:
+	print(chunk.choices[0].delta)
 
-# restrict tables for datasource in context of the mind:
-from minds.datasources.datasources import DatabaseTables
-datasource = DatabaseTables(name='my_datasource', tables=['table1', 'table1'])
-mind = client.minds.create(name='mind_name', datasources=[datasource])
+# or chat with the Mind directly (without streaming)
+response = mind.completion('How many three-bedroom houses were sold in 2008?')
+print(response)
 
-# or add to existed mind
-mind = client.minds.create(name='mind_name')
-# by config
-mind.add_datasource(postgres_config)
-# or by datasource
-mind.add_datasource(datasource)
+# with streaming
+for chunk in mind.completion('How many three-bedroom houses were sold in 2008?', stream=True):
+	print(chunk)
 
 
-# --- managing minds ---
+# Mind Management
 
 # create or replace
-mind = client.minds.create(name='mind_name', replace=True, datasources=[postgres_config] )
+mind = client.minds.create(
+    name='mind_name',
+    datasources=[
+		{
+			'name': datasource.name,
+			'tables': ['house_sales']
+		}
+	],
+    replace=True
+)
 
 # update
-mind.update(
-	name='mind_name', # is required
-	datasources=[postgres_config]  # it will replace current datasource list
+mind = client.minds.update(
+	name='mind_name', # required
+	new_name='new_mind_name', # optional
+	datasources=[  # optional
+		{
+			'name': datasource.name,
+			'tables': ['home_rentals']
+		}
+	],
 )
 
 # list
-print(client.minds.list())
+minds = client.minds.list()
 
-# get by name
+# get
 mind = client.minds.get('mind_name')
 
-# removing datasource
-mind.del_datasource(datasource)
-
-# remove mind
+# remove
 client.minds.drop('mind_name')
 
-# call completion
-print(mind.completion('2+3'))
-
-# stream completion
-for chunk in mind.completion('2+3', stream=True):
-	print(chunk.content)
-
-# --- managing datasources ---
+# Datasource Management
 
 # create or replace
-datasource = client.datasources.create(postgres_config, replace=True)
+datasource = client.datasources.create(
+    name='my_datasource',
+	description='House sales data',
+	engine='postgres',
+	connection_data={
+    	'user': 'demo_user',
+    	'password': 'demo_password',
+    	'host': 'samples.mindsdb.com',
+    	'port': 5432,
+    	'database': 'demo',
+    	'schema': 'demo_data'
+	},
+    replace=True
+)
 
+# update
+datasource = client.datasources.update(
+    name='my_datasource',
+    new_name='updated_datasource',
+    description='Updated House sales data',
+    connection_data={
+        'user': 'demo_user',
+        'password': 'demo_password',
+        'host': 'samples.mindsdb.com',
+        'port': 5432,
+        'database': 'demo',
+        'schema': 'demo_data'
+    }
+)
 
 # list
-print(client.datasources.list())
+datasources = client.datasources.list()
 
 # get
 datasource = client.datasources.get('my_datasource')
 
 # remove
 client.datasources.drop('my_datasource')
-
-