Skip to content

Commit a696cc1

Browse files
authored
Merge pull request #3 from mindsdb/default-prompt
Add default prompt template
2 parents a66c833 + 81fac36 commit a696cc1

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

minds/minds.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,8 @@
22

33
from minds.datasources import Datasource, DatabaseConfig
44

5+
DEFAULT_PROMPT_TEMPLATE = 'Use your database tools to answer the user\'s question: {{question}}'
6+
57

68
class Mind:
79
def __init__(
@@ -132,6 +134,11 @@ def create(
132134
raise ValueError(f'Unknown type of datasource: {ds}')
133135
ds_names.append(ds)
134136

137+
if parameters is None:
138+
parameters = {}
139+
if 'prompt_template' not in parameters:
140+
parameters['prompt_template'] = DEFAULT_PROMPT_TEMPLATE
141+
135142
self.api.post(
136143
f'/projects/{self.project}/minds',
137144
data={

0 commit comments

Comments
 (0)