Skip to content

Commit 47da3ad

Browse files
authored
Merge pull request #269 from SmartAPI/update-mkg-parser
MetaKG Parser Update
2 parents 754292c + 2dfc3dd commit 47da3ad

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

src/utils/metakg/endpoint.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,17 @@ def parse_individual_operation(self, op, method, path_params):
8282
"response_mapping": response_mapping,
8383
"tags": query_operation.tags,
8484
}
85+
86+
# Add additional fields to update_info if they exist in the operation
87+
if "agent_type" in op:
88+
update_info["agent_type"] = op["agent_type"]
89+
if "knowledge_level" in op:
90+
update_info["knowledge_level"] = op["knowledge_level"]
91+
if "testExamples" in op:
92+
update_info["testExamples"] = op["testExamples"]
93+
if "useTemplating" in op:
94+
update_info["useTemplating"] = op["useTemplating"]
95+
8596
res.append(update_info)
8697
return res
8798

0 commit comments

Comments
 (0)