flock-core 0.1.1__py3-none-any.whl → 0.2.1__py3-none-any.whl

flock/core/util/input_resolver.py

@@ -0,0 +1,156 @@
+"""Utility functions for resolving input keys to their corresponding values."""
+
+from flock.core.context.context import FlockContext
+
+
+def split_top_level(s: str) -> list[str]:
+    """Split a string on commas that are not enclosed within brackets, parentheses, or quotes.
+
+    This function iterates over the string while keeping track of the nesting level. It
+    only splits on commas when the nesting level is zero. It also properly handles quoted
+    substrings.
+
+    Args:
+        s (str): The input string.
+
+    Returns:
+        List[str]: A list of substrings split at top-level commas.
+    """
+    parts = []
+    current = []
+    level = 0
+    in_quote = False
+    quote_char = ""
+
+    for char in s:
+        # If inside a quote, only exit when the matching quote is found.
+        if in_quote:
+            current.append(char)
+            if char == quote_char:
+                in_quote = False
+            elif char == "\\":
+                # Include escape sequences
+                continue
+            continue
+
+        # Check for the start of a quote.
+        if char in ('"', "'"):
+            in_quote = True
+            quote_char = char
+            current.append(char)
+            continue
+
+        # Track nesting.
+        if char in "([{":
+            level += 1
+        elif char in ")]}":
+            level -= 1
+
+        # Split on commas if not nested.
+        if char == "," and level == 0:
+            parts.append("".join(current).strip())
+            current = []
+        else:
+            current.append(char)
+    if current:
+        parts.append("".join(current).strip())
+    return parts
+
+
+def _parse_keys(keys: list[str]) -> list[str]:
+    """Split a comma‐separated string and strip any type annotations.
+
+    For example, "a, b: list[str]" becomes ["a", "b"].
+    """
+    res_keys = []
+    for key in keys:
+        if "|" in key:
+            key = key.split("|")[0].strip()
+        if ":" in key:
+            key = key.split(":")[0].strip()
+        res_keys.append(key)
+    return res_keys
+
+
+def top_level_to_keys(s: str) -> list[str]:
+    """Convert a top-level comma-separated string to a list of keys."""
+    top_level_split = split_top_level(s)
+    return _parse_keys(top_level_split)
+
+
+def resolve_inputs(
+    input_spec: str, context: FlockContext, previous_agent_name: str
+) -> dict:
+    """Build a dictionary of inputs based on the input specification string and the provided context.
+
+    The lookup rules are:
+      - "context" (case-insensitive): returns the entire context.
+      - "context.property": returns an attribute from the context.
+      - "def.agent_name": returns the agent definition for the given agent.
+      - "agent_name": returns the most up2date record from the given agent's history.
+      - "agent_name.property": returns the value of a property from the state variable keyed by "agent_name.property".
+      - "property": searches the history for the most recent value of a property.
+      - Otherwise, if no matching value is found, fallback to the FLOCK_INITIAL_INPUT.
+
+    -> Recommendations:
+        - prefix your agent variables with the agent name or a short handle to avoid conflicts.
+        eg. agent name: "idea_agent", variable: "ia_idea" (ia = idea agent)
+        - or set hand off mode to strict to avoid conflicts.
+        with strict mode, the agent will only accept inputs from the previous agent.
+
+    Args:
+        input_spec: Comma-separated input keys (e.g., "query" or "agent_name.property").
+        context: A FlockContext instance.
+
+    Returns:
+        A dictionary mapping each input key to its resolved value.
+    """
+    split_input = split_top_level(input_spec)
+    keys = _parse_keys(split_input)
+    inputs = {}
+
+    for key in keys:
+        split_key = key.split(".")
+
+        # Case 1: A single key
+        if len(split_key) == 1:
+            # Special keyword: "context"
+            if key.lower() == "context":
+                inputs[key] = context
+                continue
+
+            # Try to get a historic record for an agent (if any)
+            historic_records = context.get_agent_history(key)
+            if historic_records:
+                # You may choose to pass the entire record or just its data.
+                inputs[key] = historic_records[0].data
+                continue
+
+            # Fallback to the most recent value in the state
+            historic_value = context.get_most_recent_value(key)
+            if historic_value is not None:
+                inputs[key] = historic_value
+                continue
+
+            # Fallback to the initial input
+            inputs[key] = context.get_variable("flock." + key)
+
+        # Case 2: A compound key (e.g., "agent_name.property" or "context.property")
+        elif len(split_key) == 2:
+            entity_name, property_name = split_key
+
+            if entity_name.lower() == "context":
+                # Try to fetch the attribute from the context
+                inputs[key] = getattr(context, property_name, None)
+                continue
+
+            if entity_name.lower() == "def":
+                # Return the agent definition for the given property name
+                inputs[key] = context.get_agent_definition(property_name)
+                continue
+
+            # Otherwise, attempt to look up a state variable with the key "agent_name.property"
+            inputs[key] = context.get_variable(f"{entity_name}.{property_name}")
+            continue
+
+    return inputs

flock/core/util/serializable.py

@@ -0,0 +1,93 @@
+"""Module for serializable objects in the system."""
+
+import json
+from abc import ABC, abstractmethod
+from pathlib import Path
+from typing import Any, TypeVar
+
+import cloudpickle
+import msgpack
+
+T = TypeVar("T", bound="Serializable")
+
+
+class Serializable(ABC):
+    """Base class for all serializable objects in the system.
+
+    Provides methods for serializing/deserializing objects to various formats.
+    """
+
+    @abstractmethod
+    def to_dict(self) -> dict[str, Any]:
+        """Convert instance to dictionary representation."""
+        pass
+
+    @classmethod
+    @abstractmethod
+    def from_dict(cls: type[T], data: dict[str, Any]) -> T:
+        """Create instance from dictionary representation."""
+        pass
+
+    def to_json(self) -> str:
+        """Serialize to JSON string."""
+        try:
+            return json.dumps(self.to_dict())
+        except Exception:
+            raise
+
+    @classmethod
+    def from_json(cls: type[T], json_str: str) -> T:
+        """Create instance from JSON string."""
+        try:
+            return cls.from_dict(json.loads(json_str))
+        except Exception:
+            raise
+
+    def to_msgpack(self, path: Path | None = None) -> bytes:
+        """Serialize to msgpack bytes."""
+        try:
+            msgpack_bytes = msgpack.packb(self.to_dict())
+            if path:
+                path.write_bytes(msgpack_bytes)
+            return msgpack_bytes
+        except Exception:
+            raise
+
+    @classmethod
+    def from_msgpack(cls: type[T], msgpack_bytes: bytes) -> T:
+        """Create instance from msgpack bytes."""
+        try:
+            return cls.from_dict(msgpack.unpackb(msgpack_bytes))
+        except Exception:
+            raise
+
+    @classmethod
+    def from_msgpack_file(cls: type[T], path: Path) -> T:
+        """Create instance from msgpack file."""
+        try:
+            return cls.from_msgpack(path.read_bytes())
+        except Exception:
+            raise
+
+    def to_pickle(self) -> bytes:
+        """Serialize to pickle bytes."""
+        try:
+            return cloudpickle.dumps(self)
+        except Exception:
+            raise
+
+    @classmethod
+    def from_pickle(cls, pickle_bytes: bytes) -> T:
+        """Create instance from pickle bytes."""
+        try:
+            return cloudpickle.loads(pickle_bytes)
+        except Exception:
+            raise
+
+    @classmethod
+    def from_pickle_file(cls: type[T], path: Path) -> T:
+        """Create instance from pickle file."""
+        try:
+            return cls.from_pickle(path.read_bytes())
+        except Exception:
+            raise

flock/themes/3024-day.toml

@@ -0,0 +1,39 @@
+[rich]
+table_show_lines = false
+table_box = "SIMPLE"
+table_header_align = "left"
+table_row_styles = [ "", "italic",]
+panel_padding = [ 1, 1,]
+panel_title_align = "right"
+
+[colors.bright]
+black = "#5c5855"
+blue = "#807d7c"
+cyan = "#cdab53"
+green = "#3a3432"
+magenta = "#d6d5d4"
+red = "#e8bbd0"
+white = "#f7f7f7"
+yellow = "#4a4543"
+
+[colors.cursor]
+cursor = "#4a4543"
+text = "#f7f7f7"
+
+[colors.normal]
+black = "#090300"
+blue = "#01a0e4"
+cyan = "#b5e4f4"
+green = "#01a252"
+magenta = "#a16a94"
+red = "#db2d20"
+white = "#a5a2a2"
+yellow = "#fded02"
+
+[colors.primary]
+background = "#f7f7f7"
+foreground = "#4a4543"
+
+[colors.selection]
+background = "#a5a2a2"
+text = "#4a4543"

flock/themes/3024-night.toml

@@ -0,0 +1,77 @@
+[rich]
+panel_style = "on #090300"
+table_header_style = "bold #a5a2a2 on #4a4543"
+table_title_style = "bold #a5a2a2"
+table_border_style = "#807d7c"
+panel_border_style = "#807d7c"
+column_output = "bold #a5a2a2"
+column_value = "#a5a2a2"
+bright_black = "#5c5855"
+bright_blue = "#807d7c"
+bright_cyan = "#cdab53"
+bright_green = "#3a3432"
+bright_magenta = "#d6d5d4"
+bright_red = "#e8bbd0"
+bright_white = "#f7f7f7"
+bright_yellow = "#4a4543"
+normal_black = "#090300"
+normal_blue = "#01a0e4"
+normal_cyan = "#b5e4f4"
+normal_green = "#01a252"
+normal_magenta = "#a16a94"
+normal_red = "#db2d20"
+normal_white = "#a5a2a2"
+normal_yellow = "#fded02"
+cursor_cursor = "#a5a2a2"
+cursor_text = "#090300"
+table_show_lines = false
+table_box = "MINIMAL"
+panel_padding = [ 1, 1,]
+panel_title_align = "center"
+table_row_styles = [ "", "dim",]
+table_safe_box = true
+table_padding = [ 0, 1,]
+table_collapse_padding = false
+table_pad_edge = true
+table_expand = false
+table_show_footer = false
+table_show_edge = true
+table_leading = 0
+table_style = "none"
+table_footer_style = "none"
+table_caption_style = "none"
+table_title_justify = "center"
+table_caption_justify = "center"
+table_highlight = false
+
+[colors.bright]
+black = "#5c5855"
+blue = "#807d7c"
+cyan = "#cdab53"
+green = "#3a3432"
+magenta = "#d6d5d4"
+red = "#e8bbd0"
+white = "#f7f7f7"
+yellow = "#4a4543"
+
+[colors.cursor]
+cursor = "#a5a2a2"
+text = "#090300"
+
+[colors.normal]
+black = "#090300"
+blue = "#01a0e4"
+cyan = "#b5e4f4"
+green = "#01a252"
+magenta = "#a16a94"
+red = "#db2d20"
+white = "#a5a2a2"
+yellow = "#fded02"
+
+[colors.primary]
+background = "#090300"
+foreground = "#a5a2a2"
+
+[colors.selection]
+background = "#4a4543"
+text = "#a5a2a2"

flock/themes/aardvark-blue.toml

@@ -0,0 +1,77 @@
+[rich]
+panel_style = "on #102040"
+table_header_style = "bold #000000 on #bfdbfe"
+table_title_style = "bold #dddddd"
+table_border_style = "#60a4ec"
+panel_border_style = "#60a4ec"
+column_output = "bold #dddddd"
+column_value = "#dddddd"
+bright_black = "#454545"
+bright_blue = "#60a4ec"
+bright_cyan = "#60b6cb"
+bright_green = "#95dc55"
+bright_magenta = "#e26be2"
+bright_red = "#f05b50"
+bright_white = "#f7f7f7"
+bright_yellow = "#ffe763"
+normal_black = "#191919"
+normal_blue = "#1370d3"
+normal_cyan = "#008eb0"
+normal_green = "#4b8c0f"
+normal_magenta = "#c43ac3"
+normal_red = "#aa342e"
+normal_white = "#bebebe"
+normal_yellow = "#dbba00"
+cursor_cursor = "#007acc"
+cursor_text = "#bfdbfe"
+table_show_lines = true
+table_box = "ROUNDED"
+panel_padding = [ 1, 2,]
+panel_title_align = "center"
+table_row_styles = [ "", "underline",]
+table_safe_box = true
+table_padding = [ 0, 1,]
+table_collapse_padding = false
+table_pad_edge = true
+table_expand = false
+table_show_footer = false
+table_show_edge = true
+table_leading = 0
+table_style = "none"
+table_footer_style = "none"
+table_caption_style = "none"
+table_title_justify = "center"
+table_caption_justify = "center"
+table_highlight = false
+
+[colors.bright]
+black = "#454545"
+blue = "#60a4ec"
+cyan = "#60b6cb"
+green = "#95dc55"
+magenta = "#e26be2"
+red = "#f05b50"
+white = "#f7f7f7"
+yellow = "#ffe763"
+
+[colors.cursor]
+cursor = "#007acc"
+text = "#bfdbfe"
+
+[colors.normal]
+black = "#191919"
+blue = "#1370d3"
+cyan = "#008eb0"
+green = "#4b8c0f"
+magenta = "#c43ac3"
+red = "#aa342e"
+white = "#bebebe"
+yellow = "#dbba00"
+
+[colors.primary]
+background = "#102040"
+foreground = "#dddddd"
+
+[colors.selection]
+background = "#bfdbfe"
+text = "#000000"

flock/themes/abernathy.toml

@@ -0,0 +1,77 @@
+[rich]
+panel_style = "on #111416"
+table_header_style = "bold #333333 on #eeeeec"
+table_title_style = "bold #eeeeec"
+table_border_style = "#11b5f6"
+panel_border_style = "#11b5f6"
+column_output = "bold #eeeeec"
+column_value = "#eeeeec"
+bright_black = "#404040"
+bright_blue = "#11b5f6"
+bright_cyan = "#00ffff"
+bright_green = "#00ff00"
+bright_magenta = "#ff00ff"
+bright_red = "#ff0000"
+bright_white = "#ffffff"
+bright_yellow = "#ffff00"
+normal_black = "#000000"
+normal_blue = "#1093f5"
+normal_cyan = "#00cdcd"
+normal_green = "#00cd00"
+normal_magenta = "#cd00cd"
+normal_red = "#cd0000"
+normal_white = "#faebd7"
+normal_yellow = "#cdcd00"
+cursor_cursor = "#bbbbbb"
+cursor_text = "#ffffff"
+table_show_lines = true
+table_box = "HEAVY"
+panel_padding = [ 2, 2,]
+panel_title_align = "center"
+table_row_styles = [ "", "italic",]
+table_safe_box = true
+table_padding = [ 0, 1,]
+table_collapse_padding = false
+table_pad_edge = true
+table_expand = false
+table_show_footer = false
+table_show_edge = true
+table_leading = 0
+table_style = "none"
+table_footer_style = "none"
+table_caption_style = "none"
+table_title_justify = "center"
+table_caption_justify = "center"
+table_highlight = false
+
+[colors.bright]
+black = "#404040"
+blue = "#11b5f6"
+cyan = "#00ffff"
+green = "#00ff00"
+magenta = "#ff00ff"
+red = "#ff0000"
+white = "#ffffff"
+yellow = "#ffff00"
+
+[colors.cursor]
+cursor = "#bbbbbb"
+text = "#ffffff"
+
+[colors.normal]
+black = "#000000"
+blue = "#1093f5"
+cyan = "#00cdcd"
+green = "#00cd00"
+magenta = "#cd00cd"
+red = "#cd0000"
+white = "#faebd7"
+yellow = "#cdcd00"
+
+[colors.primary]
+background = "#111416"
+foreground = "#eeeeec"
+
+[colors.selection]
+background = "#eeeeec"
+text = "#333333"

flock/themes/adventure.toml

@@ -0,0 +1,77 @@
+[rich]
+panel_style = "on #040404"
+table_header_style = "bold #ffffff on #606060"
+table_title_style = "bold #feffff"
+table_border_style = "#97d7ef"
+panel_border_style = "#97d7ef"
+column_output = "bold #feffff"
+column_value = "#feffff"
+bright_black = "#685656"
+bright_blue = "#97d7ef"
+bright_cyan = "#bdcfe5"
+bright_green = "#99b52c"
+bright_magenta = "#aa7900"
+bright_red = "#d76b42"
+bright_white = "#e4d5c7"
+bright_yellow = "#ffb670"
+normal_black = "#040404"
+normal_blue = "#417ab3"
+normal_cyan = "#bdcfe5"
+normal_green = "#5da602"
+normal_magenta = "#e5c499"
+normal_red = "#d84a33"
+normal_white = "#dbded8"
+normal_yellow = "#eebb6e"
+cursor_cursor = "#feffff"
+cursor_text = "#000000"
+table_show_lines = false
+table_box = "SIMPLE"
+panel_padding = [ 0, 2,]
+panel_title_align = "center"
+table_row_styles = [ ""]
+table_safe_box = true
+table_padding = [ 0, 1,]
+table_collapse_padding = false
+table_pad_edge = true
+table_expand = false
+table_show_footer = false
+table_show_edge = true
+table_leading = 0
+table_style = "none"
+table_footer_style = "none"
+table_caption_style = "none"
+table_title_justify = "center"
+table_caption_justify = "center"
+table_highlight = false
+
+[colors.bright]
+black = "#685656"
+blue = "#97d7ef"
+cyan = "#bdcfe5"
+green = "#99b52c"
+magenta = "#aa7900"
+red = "#d76b42"
+white = "#e4d5c7"
+yellow = "#ffb670"
+
+[colors.cursor]
+cursor = "#feffff"
+text = "#000000"
+
+[colors.normal]
+black = "#040404"
+blue = "#417ab3"
+cyan = "#bdcfe5"
+green = "#5da602"
+magenta = "#e5c499"
+red = "#d84a33"
+white = "#dbded8"
+yellow = "#eebb6e"
+
+[colors.primary]
+background = "#040404"
+foreground = "#feffff"
+
+[colors.selection]
+background = "#606060"
+text = "#ffffff"

flock/themes/adventuretime.toml

@@ -0,0 +1,77 @@
+[rich]
+panel_style = "on #1f1d45"
+table_header_style = "bold #f3d9c4 on #706b4e"
+table_title_style = "bold #f8dcc0"
+table_border_style = "#1997c6"
+panel_border_style = "#1997c6"
+column_output = "bold #f8dcc0"
+column_value = "#f8dcc0"
+bright_black = "#4e7cbf"
+bright_blue = "#1997c6"
+bright_cyan = "#c8faf4"
+bright_green = "#9eff6e"
+bright_magenta = "#9b5953"
+bright_red = "#fc5f5a"
+bright_white = "#f6f5fb"
+bright_yellow = "#efc11a"
+normal_black = "#050404"
+normal_blue = "#0f4ac6"
+normal_cyan = "#70a598"
+normal_green = "#4ab118"
+normal_magenta = "#665993"
+normal_red = "#bd0013"
+normal_white = "#f8dcc0"
+normal_yellow = "#e7741e"
+cursor_cursor = "#efbf38"
+cursor_text = "#08080a"
+table_show_lines = true
+table_box = "MINIMAL"
+panel_padding = [ 1, 2,]
+panel_title_align = "right"
+table_row_styles = [ "", "dim",]
+table_safe_box = true
+table_padding = [ 0, 1,]
+table_collapse_padding = false
+table_pad_edge = true
+table_expand = false
+table_show_footer = false
+table_show_edge = true
+table_leading = 0
+table_style = "none"
+table_footer_style = "none"
+table_caption_style = "none"
+table_title_justify = "center"
+table_caption_justify = "center"
+table_highlight = true
+
+[colors.bright]
+black = "#4e7cbf"
+blue = "#1997c6"
+cyan = "#c8faf4"
+green = "#9eff6e"
+magenta = "#9b5953"
+red = "#fc5f5a"
+white = "#f6f5fb"
+yellow = "#efc11a"
+
+[colors.cursor]
+cursor = "#efbf38"
+text = "#08080a"
+
+[colors.normal]
+black = "#050404"
+blue = "#0f4ac6"
+cyan = "#70a598"
+green = "#4ab118"
+magenta = "#665993"
+red = "#bd0013"
+white = "#f8dcc0"
+yellow = "#e7741e"
+
+[colors.primary]
+background = "#1f1d45"
+foreground = "#f8dcc0"
+
+[colors.selection]
+background = "#706b4e"
+text = "#f3d9c4"