Move policies to resources/policy

Author: l46kok

tools/ai/BUILD.bazel

@@ -13,5 +13,5 @@ java_library(
 alias(
     name = "test_policies",
     testonly = True,
-    actual = "//tools/src/test/resources:test_policies",
+    actual = "//tools/src/test/resources/policy:test_policies",
 )

tools/src/main/java/dev/cel/tools/ai/BUILD.bazel

@@ -47,3 +47,16 @@ java_proto_library(
     name = "agent_context_java_proto",
     deps = [":agent_context_proto"],
 )
+
+proto_library(
+    name = "agent_context_extensions_proto",
+    srcs = ["agent_context_extensions.proto"],
+    deps = [
+        ":agent_context_proto",
+    ],
+)
+
+java_proto_library(
+    name = "agent_context_extensions_java_proto",
+    deps = [":agent_context_extensions_proto"],
+)

tools/src/main/java/dev/cel/tools/ai/agent_context_extensions.proto

@@ -0,0 +1,11 @@
+edition = "2024";
+
+package cel.expr.ai;
+
+import "tools/src/main/java/dev/cel/tools/ai/agent_context.proto";
+
+// Extensions for the Agent-related policy protos.
+extend AgentContext {
+  repeated ClassificationLabel agent_context_classification_labels = 1000;
+  repeated AgentMessage agent_context_message_history = 1001;
+}

tools/src/test/java/dev/cel/tools/ai/AgenticPolicyCompilerTest.java

@@ -223,8 +223,8 @@ public class AgenticPolicyCompilerTest {
 
   @Test
   public void runAgenticPolicyTestCases(@TestParameter AgenticPolicyTestCase testCase) throws Exception {
-    CelAbstractSyntaxTree compiledPolicy = compilePolicy(testCase.policyFilePath);
-    PolicyTestSuite testSuite = PolicyTestSuiteHelper.readTestSuite(testCase.policyTestCaseFilePath);
+    CelAbstractSyntaxTree compiledPolicy = compilePolicy("policy/" + testCase.policyFilePath);
+    PolicyTestSuite testSuite = PolicyTestSuiteHelper.readTestSuite("policy/" + testCase.policyTestCaseFilePath);
     runTests(CEL, compiledPolicy, testSuite);
   }