Parcourir la source

feat(garden): add inputSchema for remaining garden_* tools

Lukas Goldschmidt il y a 1 mois
Parent
commit
48ae510ce7
1 fichiers modifiés avec 74 ajouts et 0 suppressions
  1. 74 0
      src/garden_layer/__init__.py

+ 74 - 0
src/garden_layer/__init__.py

@@ -74,3 +74,77 @@ def register_layer(
         },
         ["mother_label", "new_label"],
     )
+
+    tool_schemas["garden_add_seedling"] = _obj_schema(
+        {
+            "plant_uri": {"type": "string"},
+            "seed_product_uri": {"type": "string"},
+            "cycle_uri": {"type": "string"},
+            "label": {"type": "string"},
+            "strain_uri": {"type": "string"},
+            "graph": {"type": "string"},
+        },
+        ["plant_uri", "seed_product_uri", "cycle_uri", "label"],
+    )
+
+    tool_schemas["garden_describe_subject"] = _obj_schema(
+        {
+            "subject_uri": {"type": "string"},
+            "limit": {"type": "integer", "minimum": 1, "maximum": 500},
+        },
+        ["subject_uri"],
+    )
+
+    tool_schemas["garden_path_traverse"] = _obj_schema(
+        {
+            "subject_uri": {"type": "string"},
+            "property_path": {
+                "type": "string",
+                "description": "Comma-separated predicate URIs",
+            },
+            "properties": {
+                "type": "array",
+                "items": {"type": "string"},
+            },
+            "direction": {"type": "string", "enum": ["outgoing", "incoming"]},
+            "limit": {"type": "integer", "minimum": 1, "maximum": 500},
+        },
+        ["subject_uri"],
+    )
+
+    tool_schemas["garden_property_usage_statistics"] = _obj_schema(
+        {
+            "property_uri": {"type": "string"},
+            "examples_limit": {"type": "integer", "minimum": 1, "maximum": 500},
+        },
+        ["property_uri"],
+    )
+
+    tool_schemas["garden_batch_insert"] = _obj_schema(
+        {
+            "ttl": {"type": "string"},
+            "graph": {"type": "string"},
+        },
+        ["ttl"],
+    )
+
+    tool_schemas["garden_load_examples"] = _obj_schema(
+        {
+            "files": {
+                "type": "array",
+                "items": {"type": "string"},
+            },
+            "graph": {"type": "string"},
+        },
+        [],
+    )
+
+    tool_schemas["garden_reassign_cycle"] = _obj_schema(
+        {
+            "subject": {"type": "string"},
+            "new_cycle": {"type": "string"},
+            "old_cycle": {"type": "string"},
+            "graph": {"type": "string"},
+        },
+        ["subject", "new_cycle"],
+    )