agentic-blocks 0.1.16__py3-none-any.whl → 0.1.18__py3-none-any.whl

agentic_blocks/agent.py

@@ -0,0 +1,88 @@
+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:
+                response = call_llm(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/llm.py

@@ -22,7 +22,7 @@ def call_llm(
     messages: Union[Messages, List[Dict[str, Any]]],
     tools: Optional[Union[List[Dict[str, Any]], List]] = None,
     api_key: Optional[str] = None,
-    model: str = "gpt-4o-mini",
+    model: Optional[str] = None,
     base_url: Optional[str] = None,
     **kwargs,
 ) -> Any:
@@ -52,6 +52,11 @@ def call_llm(
     if not api_key:
         api_key = os.getenv("OPENROUTER_API_KEY")
 
+    if not base_url:
+        base_url = os.getenv("BASE_URL")
+    if not model:
+        model = os.getenv("MODEL_ID")
+
     if not api_key and not base_url:
         raise LLMError(
             "API key not found. Set OPENROUTER_API_KEY or OPENAI_API_KEY environment variable or pass api_key parameter."
@@ -64,12 +69,7 @@ def call_llm(
         api_key = "EMPTY"
 
     # Initialize OpenAI client
-    client_kwargs = {}
-    if api_key:
-        client_kwargs["api_key"] = api_key
-    if base_url:
-        client_kwargs["base_url"] = base_url
-    client = OpenAI(**client_kwargs)
+    client = OpenAI(api_key=api_key, base_url=base_url)
 
     # Handle different message input types
     if isinstance(messages, Messages):
@@ -84,7 +84,7 @@ def call_llm(
     openai_tools = None
     if tools:
         # Check if it's a list of LangChain StructuredTools
-        if tools and hasattr(tools[0], 'args_schema'):
+        if tools and hasattr(tools[0], "args_schema"):
             openai_tools = langchain_tools_to_openai_format(tools)
         else:
             openai_tools = tools

{agentic_blocks-0.1.16.dist-info → agentic_blocks-0.1.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentic-blocks
-Version: 0.1.16
+Version: 0.1.18
 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.18.dist-info/RECORD

@@ -0,0 +1,11 @@
+agentic_blocks/__init__.py,sha256=LJy2tzTwX9ZjPw8dqkXOWiude7ZDDIaBIvaLC8U4d_Y,435
+agentic_blocks/agent.py,sha256=nMmXkTlF3Ok8YNjOUYhf6UrU2t4WN_c_87MlwgV99Ck,2852
+agentic_blocks/llm.py,sha256=9OnpOFqZ7CoOoy8DECW7gxSGtpw-76hX-NvV003TKBU,5330
+agentic_blocks/mcp_client.py,sha256=15mIN_Qw0OVNJAvfgO3jVZS4-AU4TtvEQSFDlL9ruqA,9773
+agentic_blocks/messages.py,sha256=dxaR_-IiH8n7pZygFDiUt8n7mMOm--FRuN-xCQEuewo,11758
+agentic_blocks/utils/tools_utils.py,sha256=ZX6cqvtaV_evwMUnR_uukdvEtQah6xkDMBy7jwR7tJg,7145
+agentic_blocks-0.1.18.dist-info/licenses/LICENSE,sha256=r4IcBaAjTv3-yfjXgDPuRD953Qci0Y0nQn5JfHwLyBY,1073
+agentic_blocks-0.1.18.dist-info/METADATA,sha256=sXbL5cPil2DVH6bxnKFSC0BmR-Aly_t_--9WJAnQicU,11944
+agentic_blocks-0.1.18.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agentic_blocks-0.1.18.dist-info/top_level.txt,sha256=-1a4RAemqicXLU1rRzw4QHV3KlNeQDNxVs3m2gAT238,15
+agentic_blocks-0.1.18.dist-info/RECORD,,

agentic_blocks-0.1.16.dist-info/RECORD

@@ -1,10 +0,0 @@
-agentic_blocks/__init__.py,sha256=LJy2tzTwX9ZjPw8dqkXOWiude7ZDDIaBIvaLC8U4d_Y,435
-agentic_blocks/llm.py,sha256=q2Utx2NiffLrp7VHmQIRZtBtrnQSMFtqCeYs9ynj0Ww,5335
-agentic_blocks/mcp_client.py,sha256=15mIN_Qw0OVNJAvfgO3jVZS4-AU4TtvEQSFDlL9ruqA,9773
-agentic_blocks/messages.py,sha256=dxaR_-IiH8n7pZygFDiUt8n7mMOm--FRuN-xCQEuewo,11758
-agentic_blocks/utils/tools_utils.py,sha256=ZX6cqvtaV_evwMUnR_uukdvEtQah6xkDMBy7jwR7tJg,7145
-agentic_blocks-0.1.16.dist-info/licenses/LICENSE,sha256=r4IcBaAjTv3-yfjXgDPuRD953Qci0Y0nQn5JfHwLyBY,1073
-agentic_blocks-0.1.16.dist-info/METADATA,sha256=UN3LO_W7dBCgd8UwgE1_W5_ne61busgWB5zk04ioK9w,11944
-agentic_blocks-0.1.16.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agentic_blocks-0.1.16.dist-info/top_level.txt,sha256=-1a4RAemqicXLU1rRzw4QHV3KlNeQDNxVs3m2gAT238,15
-agentic_blocks-0.1.16.dist-info/RECORD,,