Merge pull request #206 from tcdent/langgraph

LangGraph support, baby!

Author: tcdent

agentstack/__init__.py

@@ -9,17 +9,42 @@
 from pathlib import Path
 from agentstack import conf
 from agentstack.utils import get_framework
+from agentstack.agents import get_agent
+from agentstack.tasks import get_task
 from agentstack.inputs import get_inputs
 from agentstack import frameworks
 
 ___all___ = [
-    "conf",
-    "tools",
-    "get_tags",
-    "get_framework",
-    "get_inputs",
+    "conf", 
+    "agent", 
+    "task", 
+    "tools", 
+    "get_tags", 
+    "get_framework", 
+    "get_agent", 
+    "get_task", 
+    "get_inputs", 
 ]
 
+def agent(func):
+    """
+    The `agent` decorator is used to mark a method that implements an Agent. 
+    """
+    def wrap(*args, **kwargs):
+        """Does not alter the function's behavior; this is just a marker."""
+        return func(*args, **kwargs)
+    return wrap
+
+
+def task(func):
+    """
+    The `task` decorator is used to mark a method that implements a Task.
+    """
+    def wrap(*args, **kwargs):
+        """Does not alter the function's behavior; this is just a marker."""
+        return func(*args, **kwargs)
+    return wrap
+
 
 def get_tags() -> list[str]:
     """

agentstack/_tools/__init__.py

@@ -33,9 +33,8 @@ class ToolConfig(pydantic.BaseModel):
     @classmethod
     def from_tool_name(cls, name: str) -> 'ToolConfig':
         path = TOOLS_DIR / name / TOOLS_CONFIG_FILENAME
-        if not os.path.exists(path):  # TODO raise exceptions and handle message/exit in cli
-            print(term_color(f'No known agentstack tool: {name}', 'red'))
-            sys.exit(1)
+        if not os.path.exists(path):
+            raise ValidationError(f'No known agentstack tool: {name}')
         return cls.from_json(path)
 
     @classmethod
@@ -44,11 +43,10 @@ def from_json(cls, path: Path) -> 'ToolConfig':
         try:
             return cls(**data)
         except pydantic.ValidationError as e:
-            # TODO raise exceptions and handle message/exit in cli
-            print(term_color(f"Error validating tool config JSON: \n{path}", 'red'))
+            error_str = "Error validating tool config:\n"
             for error in e.errors():
-                print(f"{' '.join([str(loc) for loc in error['loc']])}: {error['msg']}")
-            sys.exit(1)
+                error_str += f"{' '.join([str(loc) for loc in error['loc']])}: {error['msg']}\n"
+            raise ValidationError(f"Error loading tool from {path}.\n{error_str}")
 
     @property
     def type(self) -> type:

agentstack/agents.py

@@ -5,10 +5,12 @@
 from ruamel.yaml import YAML, YAMLError
 from ruamel.yaml.scalarstring import FoldedScalarString
 from agentstack import conf, log
+from agentstack import frameworks
 from agentstack.exceptions import ValidationError
 
 
 AGENTS_FILENAME: Path = Path("src/config/agents.yaml")
+AGENTS_PROMPT_TPL: str = "You are {role}. {backstory}\nYour personal goal is: {goal}"
 
 yaml = YAML()
 yaml.preserve_quotes = True  # Preserve quotes in existing data
@@ -67,6 +69,23 @@ def __init__(self, name: str):
                 error_str += f"{' '.join([str(loc) for loc in error['loc']])}: {error['msg']}\n"
             raise ValidationError(f"Error loading agent {name} from {filename}.\n{error_str}")
 
+    @property
+    def provider(self) -> str:
+        return frameworks.parse_llm(self.llm)[0]
+
+    @property
+    def model(self) -> str:
+        return frameworks.parse_llm(self.llm)[1]
+
+    @property
+    def prompt(self) -> str:
+        """Concatenate the prompts for role, goal, and backstory."""
+        return AGENTS_PROMPT_TPL.format(**{
+            'role': self.role,
+            'goal': self.goal,
+            'backstory': self.backstory,
+        })
+
     def model_dump(self, *args, **kwargs) -> dict:
         dump = super().model_dump(*args, **kwargs)
         dump.pop('name')  # name is the key, so keep it out of the data
@@ -106,3 +125,9 @@ def get_all_agent_names() -> list[str]:
 
 def get_all_agents() -> list[AgentConfig]:
     return [AgentConfig(name) for name in get_all_agent_names()]
+
+
+def get_agent(name: str) -> Optional[AgentConfig]:
+    """Get an agent configuration by name."""
+    return AgentConfig(name)
+

agentstack/cli/__init__.py

@@ -1,4 +1,7 @@
-from .cli import init_project_builder, configure_default_model, export_template, welcome_message
+from .cli import configure_default_model, welcome_message, get_validated_input
 from .init import init_project
-from .tools import list_tools, add_tool
+from .wizard import run_wizard
 from .run import run_project
+from .tools import list_tools, add_tool
+from .templates import insert_template, export_template
+

agentstack/cli/agentstack_data.py

@@ -58,6 +58,7 @@ def __init__(
     ):
         self.agents = []
         self.tasks = []
+        self.graph = []
         self.inputs = {}
         self.method = method
         self.manager_agent = manager_agent
@@ -68,6 +69,9 @@ def add_agent(self, agent):
     def add_task(self, task):
         self.tasks.append(task)
 
+    def add_edge(self, edge):
+        self.graph.append(edge)
+
     def set_inputs(self, inputs):
         self.inputs = inputs
 
@@ -77,6 +81,7 @@ def to_dict(self):
             'manager_agent': self.manager_agent,
             'agents': self.agents,
             'tasks': self.tasks,
+            'graph': self.graph,
             'inputs': self.inputs,
         }