Skip to content
Merged
Show file tree
Hide file tree
Changes from 3 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion mindsdb_sdk/__about__.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
__title__ = 'mindsdb_sdk'
__package_name__ = 'mindsdb_sdk'
__version__ = '3.4.5'
__version__ = '3.4.6'
__description__ = "MindsDB Python SDK, provides an SDK to use a remote mindsdb instance"
__email__ = "[email protected]"
__author__ = 'MindsDB Inc'
Expand Down
9 changes: 9 additions & 0 deletions mindsdb_sdk/agents.py
Original file line number Diff line number Diff line change
Expand Up @@ -87,6 +87,7 @@ def __init__(
data: dict = {},
prompt_template: str = None,
params: dict = {},
skills_extra_parameters: dict = {},
collection: CollectionBase = None
):
self.name = name
Expand All @@ -98,6 +99,7 @@ def __init__(
self.data = data
self.prompt_template = prompt_template
self.params = params
self.skills_extra_parameters = skills_extra_parameters
self.collection = collection

def completion(self, messages: List[dict]) -> AgentCompletion:
Expand Down Expand Up @@ -190,10 +192,16 @@ def __eq__(self, other):
return False
if self.provider != other.provider:
return False
if self.data != other.data:
return False
if self.prompt_template != other.prompt_template:
return False
if self.skills != other.skills:
return False
if self.params != other.params:
return False
if self.skills_extra_parameters != other.skills_extra_parameters:
return False
if self.created_at != other.created_at:
return False
return self.updated_at == other.updated_at
Expand All @@ -216,6 +224,7 @@ def from_json(cls, json: dict, collection: CollectionBase):
json.get('data', {}),
json.get('prompt_template'),
json.get('params', {}),
json.get('skills_extra_parameters', {}),
collection
)

Expand Down
Loading