agent-executor 0.1.0__py3-none-any.whl

agent_executer/__init__.py

@@ -0,0 +1,15 @@
+from .llm import ChatOllama
+from .prompt import ChatPromptTemplate
+from .parser import PydanticOutputParser
+from .agent import create_tool_calling_agent
+from .executor import AgentExecutor
+from .tools import Tool
+
+__all__ = [
+    "ChatOllama",
+    "ChatPromptTemplate",
+    "PydanticOutputParser",
+    "create_tool_calling_agent",
+    "AgentExecutor",
+    "Tool",
+]

agent_executer/agent.py

@@ -0,0 +1,19 @@
+def create_tool_calling_agent(llm, prompt, tools):
+    tool_names = [t.name for t in tools]
+
+    def agent(messages):
+        system = {
+            "role": "system",
+            "content": (
+                "You are an agent.\n"
+                f"Available tools: {', '.join(tool_names)}\n\n"
+                "When using tools, respond ONLY in JSON:\n"
+                "{type: tool, tool: name, args: {}}\n"
+                "or\n"
+                "{type: final, output: string}"
+            )
+        }
+
+        return llm([system] + messages)
+
+    return agent

agent_executer/config.py

@@ -0,0 +1,9 @@
+from dotenv import load_dotenv
+import os
+
+load_dotenv()  # MUST happen here
+
+class Config:
+    OLLAMA_HOST = os.getenv("OLLAMA_HOST", "http://localhost:11434")
+    OLLAMA_MODEL = os.getenv("OLLAMA_MODEL", "qwen2.5-coder")
+    DEBUG = os.getenv("DEBUG", "false").lower() == "true"

agent_executer/executor.py

@@ -0,0 +1,59 @@
+import json
+
+
+class AgentExecutor:
+    def __init__(self, agent, tools, verbose=False):
+        self.agent = agent
+        self.verbose = verbose
+
+        # map tools by name
+        self.tools = {t.name: t.func for t in tools}
+
+    def invoke(self, inputs: dict):
+        query = inputs["query"]
+
+        messages = [
+            {"role": "user", "content": query}
+        ]
+
+        while True:
+            result = self.agent(messages)
+            content = result["message"]["content"]
+
+            if self.verbose:
+                print("\n[RAW MODEL OUTPUT]\n", content)
+
+            try:
+                data = json.loads(content)
+            except Exception:
+                return {"output": [{"text": content}]}
+
+            # FINAL ANSWER
+            if data.get("type") == "final":
+                return {
+                    "output": [{"text": data.get("output", "")}]
+                }
+
+            # TOOL CALL
+            if data.get("type") == "tool":
+                tool_name = data.get("tool")
+                args = data.get("args", {})
+
+                if tool_name not in self.tools:
+                    tool_result = f"Tool not found: {tool_name}"
+                else:
+                    tool_result = self.tools[tool_name](**args)
+
+                if self.verbose:
+                    print(f"[TOOL] {tool_name} -> {tool_result}")
+
+                # feed result back into model
+                messages.append({
+                    "role": "assistant",
+                    "content": content
+                })
+
+                messages.append({
+                    "role": "user",
+                    "content": f"tool_result: {tool_result}"
+                })

agent_executer/llm.py

@@ -0,0 +1,29 @@
+import requests
+from .config import Config
+print("OLLAMA HOST =", Config.OLLAMA_HOST)
+
+class ChatOllama:
+    def __init__(self, model: str):
+        self.model = model
+
+    def __call__(self, messages):
+        payload = {
+            "model": self.model,
+            "messages": messages,
+            "stream": False
+        }
+
+        if Config.DEBUG:
+            print("\n[OLLAMA REQUEST]\n", payload)
+
+        res = requests.post(
+            f"{Config.OLLAMA_HOST}/api/chat",
+            json=payload
+        )
+
+        data = res.json()
+
+        if Config.DEBUG:
+            print("\n[OLLAMA RESPONSE]\n", data)
+
+        return data

agent_executer/parser.py

@@ -0,0 +1,12 @@
+class PydanticOutputParser:
+    def __init__(self, pydantic_object):
+        self.model = pydantic_object
+
+    def get_format_instructions(self):
+        return f"""
+Return ONLY valid JSON matching this schema:
+{self.model.model_json_schema()}
+"""
+
+    def parse(self, text: str):
+        return self.model.model_validate_json(text)

agent_executer/prompt.py

@@ -0,0 +1,25 @@
+class ChatPromptTemplate:
+    def __init__(self, messages):
+        self.messages = messages
+
+    @staticmethod
+    def from_messages(messages):
+        return ChatPromptTemplate(messages)
+
+    def partial(self, **kwargs):
+        updated = []
+
+        for role, content in self.messages:
+
+            # ONLY format real strings that contain format variables safely
+            if isinstance(content, str):
+                try:
+                    content = content.format(**kwargs)
+                except KeyError:
+                    # ignore missing keys like chat_history / agent_scratchpad
+                    pass
+
+            updated.append((role, content))
+
+        self.messages = updated
+        return self

agent_executer/tools.py

@@ -0,0 +1,9 @@
+from dataclasses import dataclass
+from typing import Callable, Any
+
+
+@dataclass
+class Tool:
+    name: str
+    func: Callable[..., Any]
+    description: str

agent_executor-0.1.0.dist-info/METADATA

@@ -0,0 +1,9 @@
+Metadata-Version: 2.4
+Name: agent-executor
+Version: 0.1.0
+Summary: Minimal Ollama agent executor runtime
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: pydantic>=2.13.4
+Requires-Dist: python-dotenv>=1.2.2
+Requires-Dist: requests>=2.34.2

agent_executor-0.1.0.dist-info/RECORD

@@ -0,0 +1,12 @@
+agent_executer/__init__.py,sha256=Ms5rWOMwtpt1K2K5yYO4MteYHGetopASO6eKD-y2diA,379
+agent_executer/agent.py,sha256=rbs8S0AbswmAQaPuA5tqg0cxgop9NmnIF5OzijbHaBU,580
+agent_executer/config.py,sha256=Nbl0_EaPLhe-9wdWXTJJw2-4ebwAtElNDzccVEOUK0k,287
+agent_executer/executor.py,sha256=pNBjFhKVwkzpVW8Xatm9BWyX8ll-qBjWThGjYub0oEs,1748
+agent_executer/llm.py,sha256=4wXHaayBQD9N-uiDHW_1Kz2uxrrdvQtvIDLD314MxPY,666
+agent_executer/parser.py,sha256=6_SFeYRJx6TuJW-VkkLWa7tP3Ado7N6aS35jubChCxw,343
+agent_executer/prompt.py,sha256=5_8rz82I8-WVJ_g35WPbw5kmwH-xoHLIVarnJChBJRw,741
+agent_executer/tools.py,sha256=5QMeSiOJi4LShQ5vKkX8yUgabN55SRJO6soknCVNljo,163
+agent_executor-0.1.0.dist-info/METADATA,sha256=NQSVPYilNtsjMadXf07G9eNg6O0fKjNjpher910Dl2U,278
+agent_executor-0.1.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+agent_executor-0.1.0.dist-info/top_level.txt,sha256=YPWwC1wrGP2lUVUmi5iQhpx2yotJeA1MEqgICoRlYwY,15
+agent_executor-0.1.0.dist-info/RECORD,,

agent_executor-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

agent_executor-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+agent_executer