kweaver-dolphin 0.1.0__py3-none-any.whl

dolphin/core/code_block/assign_block.py

@@ -0,0 +1,98 @@
+import json
+
+from dolphin.core.code_block.basic_code_block import BasicCodeBlock
+from dolphin.core.logging.logger import console_block_start
+from dolphin.core.common.enums import CategoryBlock, count_occurrences, TypeStage
+from dolphin.core.context.context import Context
+from dolphin.core.context.var_output import SourceType
+
+
+class AssignBlock(BasicCodeBlock):
+    def __init__(self, context: Context, debug_infos=None):
+        super().__init__(context)
+        self.debug_info = debug_infos
+
+    async def execute(
+        self,
+        content,
+        category: CategoryBlock = CategoryBlock.ASSIGN,
+        replace_variables=False,
+    ):
+        # Call the parent class's execute method
+        async for _ in super().execute(content, category, replace_variables):
+            pass
+        num0 = count_occurrences(["->", ">>"], self.content)
+        num1 = count_occurrences(
+            ["/if/", "/prompt/", "/judge/", "/explore/"], self.content
+        )
+        if num0 > 1 or num1 > 0:
+            raise Exception(f"Syntax Error({self.content})，check the '->' or '>>'")
+
+        if self.assign_type == "->":
+            var_before_str = self.content.split("->")[0].strip()
+            if len(var_before_str) == 0 or len(self.output_var) == 0:
+                raise Exception(f"Syntax Error({self.content})，check the variable")
+
+            var_before = self.context.recognize_variable(var_before_str)
+            local_variables = {"json": json}
+            for i in range(len(var_before)):
+                var_before_str = var_before_str.replace(
+                    var_before[i][0], "temp" + str(i)
+                )
+                local_variables["temp" + str(i)] = self.context.get_variable_type(
+                    var_before[i][0]
+                )
+
+            try:
+                result = eval(var_before_str, local_variables)
+            except Exception as e:
+                filtered_local_vars = {
+                    k: v for k, v in local_variables.items() if k.startswith("temp")
+                }
+                raise Exception(
+                    f"Failed to parse variable: var_before_str = {var_before_str}; "
+                    f"local_variables = {filtered_local_vars}; error = {e}"
+                )
+            console_block_start(
+                "assign", self.output_var, str(result), verbose=self.context.verbose
+            )
+            self.recorder.update(
+                item=result,
+                stage=TypeStage.ASSIGN,
+                is_completed=True,
+                source_type=SourceType.ASSIGN,
+            )
+            yield result
+        elif self.assign_type == ">>":
+            var_before_str = self.content.split(">>")[0].strip()
+            if len(var_before_str) == 0 or len(self.output_var) == 0:
+                raise Exception(f"Syntax Error({self.content})，check the variable")
+            var_before = self.context.recognize_variable(var_before_str)
+            local_variables = {"json": json}
+            for i in range(len(var_before)):
+                var_before_str = var_before_str.replace(
+                    var_before[i][0], "temp" + str(i)
+                )
+                local_variables["temp" + str(i)] = self.context.get_variable_type(
+                    var_before[i][0]
+                )
+
+            try:
+                result = eval(var_before_str, local_variables)
+            except Exception as e:
+                # Filter local_variables, keeping only fields that start with 'temp'
+                filtered_local_vars = {
+                    k: v for k, v in local_variables.items() if k.startswith("temp")
+                }
+                raise Exception(
+                    f"Failed to parse variable: var_before_str = {var_before_str}; "
+                    f"local_variables = {filtered_local_vars}; error = {e}"
+                )
+            self.block_start_log("assign", result)
+            self.recorder.update(
+                item=result,
+                stage=TypeStage.ASSIGN,
+                is_completed=True,
+                source_type=SourceType.ASSIGN,
+            )
+            yield result