Ensure the repair prompt is actually appended to our messages and upd…

…ate snapshots.

python/examples/healthData/translator.py

@@ -27,8 +27,8 @@ def __init__(
         self._additional_agent_instructions = additional_agent_instructions
 
     @override
-    async def translate(self, request: str, *, prompt_preamble: str | list[PromptSection] | None = None) -> Result[T]:
-        result = await super().translate(request=request, prompt_preamble=prompt_preamble)
+    async def translate(self, input: str, *, prompt_preamble: str | list[PromptSection] | None = None) -> Result[T]:
+        result = await super().translate(input=input, prompt_preamble=prompt_preamble)
         if not isinstance(result, Failure):
             self._chat_history.append(ChatMessage(source="assistant", body=result.value))
         return result

python/tests/__snapshots__/test_translator.ambr

@@ -3,24 +3,33 @@
   list([
     dict({
       'kind': 'CLIENT REQUEST',
-      'payload': '''
-
-        You are a service that translates user requests into JSON objects of type "ExampleABC" according to the following TypeScript definitions:
-        ```
-        interface ExampleABC {
-            a: string;
-            b: boolean;
-            c: number;
-        }
-
-        ```
-        The following is a user request:
-        '''
-        Get me stuff.
-        '''
-        The following is the user request translated into a JSON object with 2 spaces of indentation and no properties with the value undefined:
+      'payload': list([
+        dict({
+          'content': 'Get me stuff.',
+          'role': 'user',
+        }),
+        dict({
+          'content': '''
+
+            You are a service that translates user requests into JSON objects of type "ExampleABC" according to the following TypeScript definitions:
+            ```
+            interface ExampleABC {
+                a: string;
+                b: boolean;
+                c: number;
+            }
+
+            ```
+            The following is a user request:
+            '''
+            Get me stuff.
+            '''
+            The following is the user request translated into a JSON object with 2 spaces of indentation and no properties with the value undefined:
 
-      ''',
+          ''',
+          'role': 'user',
+        }),
+      ]),
     }),
     dict({
       'kind': 'MODEL RESPONSE',
@@ -32,49 +41,93 @@
   list([
     dict({
       'kind': 'CLIENT REQUEST',
-      'payload': '''
-
-        You are a service that translates user requests into JSON objects of type "ExampleABC" according to the following TypeScript definitions:
-        ```
-        interface ExampleABC {
-            a: string;
-            b: boolean;
-            c: number;
-        }
-
-        ```
-        The following is a user request:
-        '''
-        Get me stuff.
-        '''
-        The following is the user request translated into a JSON object with 2 spaces of indentation and no properties with the value undefined:
+      'payload': list([
+        dict({
+          'content': 'Get me stuff.',
+          'role': 'user',
+        }),
+        dict({
+          'content': '''
+
+            You are a service that translates user requests into JSON objects of type "ExampleABC" according to the following TypeScript definitions:
+            ```
+            interface ExampleABC {
+                a: string;
+                b: boolean;
+                c: number;
+            }
+
+            ```
+            The following is a user request:
+            '''
+            Get me stuff.
+            '''
+            The following is the user request translated into a JSON object with 2 spaces of indentation and no properties with the value undefined:
 
-      ''',
+          ''',
+          'role': 'user',
+        }),
+        dict({
+          'content': '''
+
+            The above JSON object is invalid for the following reason:
+            '''
+            Validation path `c` failed for value `{"a": "hello", "b": true}` because:
+              Field required
+            '''
+            The following is a revised JSON object:
+
+          ''',
+          'role': 'user',
+        }),
+      ]),
     }),
     dict({
       'kind': 'MODEL RESPONSE',
       'payload': '{ "a": "hello", "b": true }',
     }),
     dict({
       'kind': 'CLIENT REQUEST',
-      'payload': '''
-
-        You are a service that translates user requests into JSON objects of type "ExampleABC" according to the following TypeScript definitions:
-        ```
-        interface ExampleABC {
-            a: string;
-            b: boolean;
-            c: number;
-        }
-
-        ```
-        The following is a user request:
-        '''
-        Get me stuff.
-        '''
-        The following is the user request translated into a JSON object with 2 spaces of indentation and no properties with the value undefined:
+      'payload': list([
+        dict({
+          'content': 'Get me stuff.',
+          'role': 'user',
+        }),
+        dict({
+          'content': '''
+
+            You are a service that translates user requests into JSON objects of type "ExampleABC" according to the following TypeScript definitions:
+            ```
+            interface ExampleABC {
+                a: string;
+                b: boolean;
+                c: number;
+            }
+
+            ```
+            The following is a user request:
+            '''
+            Get me stuff.
+            '''
+            The following is the user request translated into a JSON object with 2 spaces of indentation and no properties with the value undefined:
+
+          ''',
+          'role': 'user',
+        }),
+        dict({
+          'content': '''
+
+            The above JSON object is invalid for the following reason:
+            '''
+            Validation path `c` failed for value `{"a": "hello", "b": true}` because:
+              Field required
+            '''
+            The following is a revised JSON object:
 
-      ''',
+          ''',
+          'role': 'user',
+        }),
+      ]),
     }),
     dict({
       'kind': 'MODEL RESPONSE',