Rename messages aggregator into context

Signed-off-by: Louis Mandel <[email protected]>

Author: mandel

pdl-live-react/src/pdl_ast.d.ts

@@ -512,7 +512,7 @@ export type PdlId2 = string | null
 export type PdlIsLeaf2 = true
 export type Kind2 = "aggregator"
 export type Aggregator =
-  | ("messages" | "stdout" | "stderr")
+  | ("context" | "stdout" | "stderr")
   | FileAggregatorConfig
 /**
  * Documentation associated to the aggregator config.

src/pdl/pdl-schema.json

@@ -384,7 +384,7 @@
           "anyOf": [
             {
               "enum": [
-                "messages",
+                "context",
                 "stdout",
                 "stderr"
               ],

src/pdl/pdl_ast.py

@@ -978,7 +978,7 @@ class FileAggregatorConfig(AggregatorConfig):
 
 
 AggregatorType: TypeAlias = (
-    Literal["messages", "stdout", "stderr"] | FileAggregatorConfig
+    Literal["context", "stdout", "stderr"] | FileAggregatorConfig
 )
 
 

src/pdl/pdl_dumper.py

@@ -471,7 +471,7 @@ def contribute_to_list(
 
 def aggregator_to_dict(aggregator: AggregatorType):
     match aggregator:
-        case "messages" | "stdout" | "stderr":
+        case "context" | "stdout" | "stderr":
             result = aggregator
         case FileAggregatorConfig():
             result = aggregator.model_dump()

src/pdl/pdl_interpreter.py

@@ -2146,7 +2146,7 @@ def contribute(
         """
 
 
-class MessagesAggregator(Aggregator):
+class ContextAggregator(Aggregator):
     def __init__(self, messages: Optional[LazyMessages] = None):
         if messages is None:
             self.messages: LazyMessages = DependentContext([])
@@ -2159,7 +2159,7 @@ def contribute(
         role: Optional[RoleType] = None,
         loc: Optional[PdlLocationType] = None,
         block: Optional[BlockType] = None,
-    ) -> "MessagesAggregator":
+    ) -> "ContextAggregator":
         match block:
             case None | StructuredBlock():
                 return self
@@ -2170,7 +2170,7 @@ def contribute(
                 msg = {"role": role, "content": result}
         new_messages: LazyMessages = SingletonContext(PdlDict(msg))
         messages = DependentContext([self.messages, new_messages])
-        return MessagesAggregator(messages)
+        return ContextAggregator(messages)
 
 
 class FileAggregator(Aggregator):
@@ -2201,8 +2201,8 @@ def process_aggregator(
 ) -> tuple[PdlLazy[Aggregator], LazyMessages, ScopeType, AggregatorBlock]:
     aggregator: Aggregator
     match block.aggregator:
-        case "messages":
-            aggregator = MessagesAggregator()
+        case "context":
+            aggregator = ContextAggregator()
         case "stdout":
             aggregator = FileAggregator(sys.stdout)
         case "stderr":

tests/test_examples_run.yaml

@@ -1,5 +1,5 @@
 update_results: false
-check: []
+check: [examples/aggregators/debug.pdl]
 skip:
   - examples/cldk/cldk-assistant.pdl
   - examples/gsm8k/gsm8.pdl