Merge pull request #269 from SmartAPI/update-mkg-parser

MetaKG Parser Update
SmartAPI · Oct 10, 2024 · 47da3ad · 47da3ad
2 parents 754292c + 2dfc3dd
commit 47da3ad
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/utils/metakg/endpoint.py b/src/utils/metakg/endpoint.py
@@ -82,6 +82,17 @@ def parse_individual_operation(self, op, method, path_params):
                     "response_mapping": response_mapping,
                     "tags": query_operation.tags,
                 }
+
+                # Add additional fields to update_info if they exist in the operation
+                if "agent_type" in op:
+                    update_info["agent_type"] = op["agent_type"]
+                if "knowledge_level" in op:
+                    update_info["knowledge_level"] = op["knowledge_level"]
+                if "testExamples" in op:
+                    update_info["testExamples"] = op["testExamples"]
+                if "useTemplating" in op:
+                    update_info["useTemplating"] = op["useTemplating"]
+
                 res.append(update_info)
         return res