agentic-blocks 0.1.16__tar.gz → 0.1.17__tar.gz

{agentic_blocks-0.1.16/src/agentic_blocks.egg-info → agentic_blocks-0.1.17}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentic-blocks
-Version: 0.1.16
+Version: 0.1.17
 Summary: Simple building blocks for agentic AI systems with MCP client and conversation management
 Author-email: Magnus Bjelkenhed <bjelkenhed@gmail.com>
 License: MIT

{agentic_blocks-0.1.16 → agentic_blocks-0.1.17}/pyproject.toml

@@ -14,7 +14,7 @@ agentic_blocks = []
 
 [project]
 name = "agentic-blocks"
-version = "0.1.16"
+version = "0.1.17"
 description = "Simple building blocks for agentic AI systems with MCP client and conversation management"
 readme = "README.md"
 requires-python = ">=3.11"

agentic_blocks-0.1.17/src/agentic_blocks/agent.py

@@ -0,0 +1,89 @@
+from pocketflow import Node, Flow
+from agentic_blocks.utils.tools_utils import (
+    create_tool_registry,
+    execute_pending_tool_calls,
+)
+from agentic_blocks import call_llm, Messages
+
+
+class Agent:
+    def __init__(self, system_prompt: str, tools: list):
+        self.system_prompt = system_prompt
+        self.tools = tools
+        self.tool_registry = create_tool_registry(tools)
+
+        # Create nodes
+        self.llm_node = self._create_llm_node()
+        self.tool_node = self._create_tool_node()
+        self.answer_node = self._create_answer_node()
+
+        # Set up flow
+        self.llm_node - "tool_node" >> self.tool_node
+        self.tool_node - "llm_node" >> self.llm_node
+        self.llm_node - "answer_node" >> self.answer_node
+
+        self.flow = Flow(self.llm_node)
+
+    def _create_llm_node(self):
+        class LLMNode(Node):
+            def __init__(self, system_prompt, tools):
+                super().__init__()
+                self.system_prompt = system_prompt
+                self.tools = tools
+
+            def prep(self, shared):
+                return shared["messages"]
+
+            def exec(self, messages) -> Messages:
+                model = "qwen/qwen3-235b-a22b-2507"
+                response = call_llm(model=model, messages=messages, tools=self.tools)
+                messages.add_response_message(response)
+                return messages
+
+            def post(self, shared, prep_res, messages):
+                if messages.has_pending_tool_calls():
+                    return "tool_node"
+                else:
+                    return "answer_node"
+
+        return LLMNode(self.system_prompt, self.tools)
+
+    def _create_tool_node(self):
+        class ToolNode(Node):
+            def __init__(self, tool_registry):
+                super().__init__()
+                self.tool_registry = tool_registry
+
+            def prep(self, shared):
+                return shared["messages"]
+
+            def exec(self, messages) -> Messages:
+                tool_responses = execute_pending_tool_calls(
+                    messages, self.tool_registry
+                )
+                messages.add_tool_responses(tool_responses)
+                return messages
+
+            def post(self, shared, prep_res, messages):
+                return "llm_node"
+
+        return ToolNode(self.tool_registry)
+
+    def _create_answer_node(self):
+        class AnswerNode(Node):
+            def prep(self, shared):
+                messages = shared["messages"]
+                shared["answer"] = messages.get_messages()[-1]["content"]
+                return messages
+
+        return AnswerNode()
+
+    def invoke(self, user_prompt: str) -> str:
+        messages = Messages(user_prompt=user_prompt)
+        if self.system_prompt:
+            messages.add_system_message(self.system_prompt)
+
+        shared = {"messages": messages}
+        self.flow.run(shared)
+
+        return shared["answer"]

{agentic_blocks-0.1.16 → agentic_blocks-0.1.17/src/agentic_blocks.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentic-blocks
-Version: 0.1.16
+Version: 0.1.17
 Summary: Simple building blocks for agentic AI systems with MCP client and conversation management
 Author-email: Magnus Bjelkenhed <bjelkenhed@gmail.com>
 License: MIT

{agentic_blocks-0.1.16 → agentic_blocks-0.1.17}/src/agentic_blocks.egg-info/SOURCES.txt

@@ -2,6 +2,7 @@ LICENSE
 README.md
 pyproject.toml
 src/agentic_blocks/__init__.py
+src/agentic_blocks/agent.py
 src/agentic_blocks/llm.py
 src/agentic_blocks/mcp_client.py
 src/agentic_blocks/messages.py