Skip to content

Commit e56d6dd

Browse files
added skills_extra_parameters to agent
1 parent db19045 commit e56d6dd

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

mindsdb_sdk/agents.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -87,6 +87,7 @@ def __init__(
8787
data: dict = {},
8888
prompt_template: str = None,
8989
params: dict = {},
90+
skills_extra_parameters: dict = {},
9091
collection: CollectionBase = None
9192
):
9293
self.name = name
@@ -98,6 +99,7 @@ def __init__(
9899
self.data = data
99100
self.prompt_template = prompt_template
100101
self.params = params
102+
self.skills_extra_parameters = skills_extra_parameters
101103
self.collection = collection
102104

103105
def completion(self, messages: List[dict]) -> AgentCompletion:
@@ -216,6 +218,7 @@ def from_json(cls, json: dict, collection: CollectionBase):
216218
json.get('data', {}),
217219
json.get('prompt_template'),
218220
json.get('params', {}),
221+
json.get('skills_extra_parameters', {}),
219222
collection
220223
)
221224

0 commit comments

Comments
 (0)