From a18e7ad51f0f1abaecf0e81405e846a3045c43c3 Mon Sep 17 00:00:00 2001
From: lpm0073 <lpm0073@gmail.com>
Date: Tue, 28 Nov 2023 19:51:54 -0600
Subject: [PATCH] refactor: restore LCChatMemory.messages validation

---
 grader/langchain.py | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/grader/langchain.py b/grader/langchain.py
index ef6b2f8..71acdc9 100644
--- a/grader/langchain.py
+++ b/grader/langchain.py
@@ -3,7 +3,7 @@
 
 from typing import List, Optional
 
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, Field, model_validator
 
 
 class LCRequestMetaData(BaseModel):
@@ -30,12 +30,12 @@ class LCChatMemory(BaseModel):
 
     messages: List[LCMessage]
 
-    # @model_validator(mode="after")
-    # def validate_messages(self) -> "LCChatMemory":
-    #     """Validate that chat memory contains at least 2 dicts"""
-    #     if len(self.messages) < 2:
-    #         raise ValueError("messages must contain at least 2 objects")
-    #     return self
+    @model_validator(mode="after")
+    def validate_messages(self) -> "LCChatMemory":
+        """Validate that chat memory contains at least 2 dicts"""
+        if len(self.messages) < 2:
+            raise ValueError("messages must contain at least 2 objects")
+        return self
 
 
 class LCBody(BaseModel):