Skip to content

Commit 5a6326f

Browse files
committed
fixed update method
1 parent 1644d4e commit 5a6326f

File tree

1 file changed

+6
-9
lines changed

1 file changed

+6
-9
lines changed

mindsdb_sdk/chatbots.py

Lines changed: 6 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,7 @@ def update(self, name: str = None, agent_name: str = None, model_name: str = Non
3030
:param agent_name: (Optional) New agent name to associate with the chatbot.
3131
:param model_name: (Optional) New model to use for the chatbot.
3232
:param database_name: (Optional) New database connection name.
33-
:param inplace: If True, updates the current object in-place; otherwise, returns a new Chatbot object.
34-
:return: Updated Chatbot object, or None if inplace is True.
33+
:return: Updated Chatbot object.
3534
"""
3635
payload = {}
3736

@@ -53,14 +52,12 @@ def update(self, name: str = None, agent_name: str = None, model_name: str = Non
5352
data=payload
5453
)
5554

56-
if inplace:
57-
self.name = updated_chatbot.get('name', self.name)
58-
self.database_name = updated_chatbot.get('database', self.database_name)
59-
self.agent_name = updated_chatbot.get('agent', self.agent_name)
60-
self.model_name = updated_chatbot.get('model_name', self.model_name)
61-
return None
55+
self.name = updated_chatbot.get('name', self.name)
56+
self.database_name = updated_chatbot.get('database', self.database_name)
57+
self.agent_name = updated_chatbot.get('agent', self.agent_name)
58+
self.model_name = updated_chatbot.get('model_name', self.model_name)
6259

63-
return Chatbot(self.api, self.project, updated_chatbot)
60+
return self
6461

6562
def delete(self):
6663
"""

0 commit comments

Comments
 (0)