feat: add export step to knowledge

lavague-ai · Oct 14, 2024 · 2c43980 · 2c43980
1 parent ddaabc7
commit 2c43980
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/lavague-sdk/lavague/sdk/trajectory/model.py b/lavague-sdk/lavague/sdk/trajectory/model.py
@@ -101,10 +101,13 @@ def deserialize_action(cls, values: Dict[str, Any]) -> Dict[str, Any]:
         return values
 
 
+class StepKnowledge(ActionWrapper):
+    instruction: Instruction
+
+
 class StepCompletion(ActionWrapper):
     run_status: RunStatus
     run_mode: RunMode
 
-
-class StepKnowledge(ActionWrapper):
-    instruction: Instruction
+    def to_knowledge(self, instruction: Instruction) -> StepKnowledge:
+        return StepKnowledge(instruction=instruction, action=self.action)