kader 0.1.5__py3-none-any.whl

kader/prompts/__init__.py

@@ -0,0 +1,9 @@
+from .agent_prompts import BasicAssistancePrompt, PlanningAgentPrompt, ReActAgentPrompt
+from .base import PromptBase
+
+__all__ = [
+    "PromptBase",
+    "BasicAssistancePrompt",
+    "ReActAgentPrompt",
+    "PlanningAgentPrompt",
+]

kader/prompts/agent_prompts.py

@@ -0,0 +1,27 @@
+from datetime import datetime
+from typing import Any
+
+from .base import PromptBase
+
+
+class BasicAssistancePrompt(PromptBase):
+    """Basic assistance prompt with date context."""
+
+    def __init__(self, **kwargs: Any) -> None:
+        template = "You are a helpful AI assistant. Today is {{ date }}."
+        kwargs.setdefault("date", datetime.now().strftime("%Y-%m-%d"))
+        super().__init__(template=template, **kwargs)
+
+
+class ReActAgentPrompt(PromptBase):
+    """Prompt for ReAct Agent."""
+
+    def __init__(self, **kwargs: Any) -> None:
+        super().__init__(template_path="react_agent.j2", **kwargs)
+
+
+class PlanningAgentPrompt(PromptBase):
+    """Prompt for Planning Agent."""
+
+    def __init__(self, **kwargs: Any) -> None:
+        super().__init__(template_path="planning_agent.j2", **kwargs)

kader/prompts/base.py

@@ -0,0 +1,81 @@
+import os
+import re
+from pathlib import Path
+
+from jinja2 import Environment, FileSystemLoader
+
+
+class PromptBase:
+    """
+    Base class for handling Jinja2 templates for prompts.
+
+    Attributes
+    ----------
+    template : str, optional
+        A string containing the Jinja2 template.
+    template_path : str, optional
+        The path to the Jinja2 template file.
+    """
+
+    template: str | None = None
+    template_path: str | None = None
+
+    def __init__(self, **kwargs) -> None:
+        """Initialize the PromptBase instance.
+
+        Parameters
+        ----------
+        **kwargs
+            Keyword arguments that will be available as variables in the template.
+            Special keys 'template' and 'template_path' are used to initialize the prompt.
+        """
+        self.template = kwargs.pop("template", self.template)
+        self.template_path = kwargs.pop("template_path", self.template_path)
+        self.vars = kwargs
+
+        if self.template:
+            env = Environment()
+            self.prompt = env.from_string(self.template)
+        elif self.template_path:
+            current_dir = Path(__file__).parent
+            path = os.path.join(current_dir, "templates")
+            loader = FileSystemLoader(path)
+            env = Environment(loader=loader)
+            self.prompt = env.get_template(self.template_path)
+        self._resolved_prompt = None
+
+    def render_template(self) -> str:
+        """Render the Jinja2 template with the provided variables.
+
+        Returns
+        -------
+        str
+            The rendered prompt as a string.
+        """
+        render = self.prompt.render(**self.vars)
+        render = re.sub(r"\n{3,}", "\n\n", render)
+        return render
+
+    def resolve_prompt(self) -> str:
+        """Resolve the prompt by rendering it if it hasn't been already.
+
+        This method ensures the template is rendered only once and the result is cached.
+
+        Returns
+        -------
+        str
+            The resolved prompt.
+        """
+        if self._resolved_prompt is None:
+            self._resolved_prompt = self.render_template()
+        return self._resolved_prompt
+
+    def __str__(self) -> str:
+        """Return the resolved prompt when the object is converted to a string.
+
+        Returns
+        -------
+        str
+            The resolved prompt.
+        """
+        return self.resolve_prompt()

kader/prompts/templates/planning_agent.j2

@@ -0,0 +1,26 @@
+You are a planning agent. You have access to the following tools:
+
+{% for tool in tools %}
+{{ tool.name }}: {{ tool.description }}
+{% endfor %}
+
+Your goal is to complete the user's request by creating a plan and executing it step-by-step.
+
+Instructions:
+1. Break down the user request into logical steps.
+2. For each step, determine if a tool is needed.
+3. Execute necessary tools.
+4. CRITICAL: Always create a TODO Using TODO Tool for the logical.
+5. CRITICAL: After completing a step (e.g., creating a file, running a test), you MUST immediately use the TODO Tool to update the status of that item to 'completed'. Do not proceed to the next item without updating the TODO list.
+6. Planning Agent must use TODO Tool to ensure that the plan is completed when execution occurs.
+7. If you have enough information, provide the final answer.
+
+Example (programming task):
+  Task:
+    - Implement a ML Inference Endpoint.
+  Plan:
+    - Analyze the requirements and design the system architecture.
+    - Implement the endpoint using a web framework.
+    - Test the endpoint to ensure it works as expected.
+    - Deploy the endpoint to a production environment.
+this requires going back and forth with the user and todo tools, and also using other tools to complete the task. You must update the todo list after each step.

kader/prompts/templates/react_agent.j2

@@ -0,0 +1,18 @@
+Answer the following questions as best you can. You have access to the following tools:
+
+{% for tool in tools %}
+{{ tool.name }}: {{ tool.description }}
+{% endfor %}
+
+Use the following format:
+
+Question: the input question you must answer
+Thought: you should always think about what to do
+Action: the action to take, should be one of [{{ tool_names }}]
+Action Input: the input to the action
+Observation: the result of the action
+... (this Thought/Action/Action Input/Observation can repeat N times)
+Thought: I now know the final answer
+Final Answer: the final answer to the original input question
+
+Begin!

kader/providers/__init__.py

@@ -0,0 +1,9 @@
+from .base import Message
+from .mock import MockLLM
+from .ollama import OllamaProvider
+
+__all__ = [
+    "Message",
+    "OllamaProvider",
+    "MockLLM",
+]