prompt-caller 0.0.5__py3-none-any.whl → 0.1.1__py3-none-any.whl

prompt_caller/prompt_caller.py

@@ -5,7 +5,8 @@ import requests
 import yaml
 from dotenv import load_dotenv
 from jinja2 import Template
-from langchain_core.messages import HumanMessage, SystemMessage
+from langchain_core.tools import tool
+from langchain_core.messages import HumanMessage, SystemMessage, ToolMessage
 from langchain_openai import ChatOpenAI
 from PIL import Image
 from pydantic import BaseModel, Field, create_model
@@ -18,6 +19,9 @@ load_dotenv()
 
 class PromptCaller:
 
+    def __init__(self, promptPath="prompts"):
+        self.promptPath = promptPath
+
     def _loadPrompt(self, file_path):
         with open(file_path, "r", encoding="utf-8") as file:
             content = file.read()
@@ -61,7 +65,7 @@ class PromptCaller:
             context = {}
 
         configuration, template = self._loadPrompt(
-            os.path.join("prompts", f"{promptName}.prompt")
+            os.path.join(self.promptPath, f"{promptName}.prompt")
         )
 
         template = self._renderTemplate(template, context)
@@ -124,3 +128,79 @@ class PromptCaller:
         response = chat.invoke(messages)
 
         return response
+
+    def agent(self, promptName, context=None, tools=None, allowed_steps=3):
+
+        configuration, messages = self.loadPrompt(promptName, context)
+
+        output = None
+
+        if "output" in configuration:
+            output = configuration.get("output")
+            configuration.pop("output")
+
+            for message in messages:
+                if isinstance(message, SystemMessage):
+                    message.content += "\nOnly use the tool DynamicModel when providing an output call."
+                    break
+
+        chat = ChatOpenAI(**configuration)
+
+        # Register the tools
+        if tools is None:
+            tools = []
+
+        # Transform functions in tools
+        tools = [tool(t) for t in tools]
+
+        tools_dict = {t.name.lower(): t for t in tools}
+
+        if output:
+            dynamicModel = self.createPydanticModel(output)
+
+        tools.extend([dynamicModel])
+        tools_dict["dynamicmodel"] = dynamicModel
+
+        chat = chat.bind_tools(tools)
+
+        try:
+            # First LLM invocation
+            response = chat.invoke(messages)
+            messages.append(response)
+
+            steps = 0
+            while response.tool_calls and steps < allowed_steps:
+                for tool_call in response.tool_calls:
+                    tool_name = tool_call["name"].lower()
+
+                    # If it's the final formatting tool, validate and return
+                    if tool_name == "dynamicmodel":
+                        return dynamicModel.model_validate(tool_call["args"])
+
+                    selected_tool = tools_dict.get(tool_name)
+                    if not selected_tool:
+                        raise ValueError(f"Unknown tool: {tool_name}")
+
+                    # Invoke the selected tool with provided arguments
+                    tool_response = selected_tool.invoke(tool_call)
+                    messages.append(tool_response)
+
+                # If the latest message is a ToolMessage, re-invoke the LLM
+                if isinstance(messages[-1], ToolMessage):
+                    response = chat.invoke(messages)
+                    messages.append(response)
+                else:
+                    break
+
+                steps += 1
+
+            # Final LLM call if the last message is still a ToolMessage
+            if isinstance(messages[-1], ToolMessage):
+                response = chat.invoke(messages)
+                messages.append(response)
+
+            return response
+
+        except Exception as e:
+            # Replace with appropriate logging in production
+            raise RuntimeError("Error during agent process") from e

{prompt_caller-0.0.5.dist-info → prompt_caller-0.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prompt_caller
-Version: 0.0.5
+Version: 0.1.1
 Summary: This package is responsible for calling prompts in a specific format. It uses LangChain and OpenAI API
 Home-page: https://github.com/ThiNepo/prompt-caller
 Author: Thiago Nepomuceno
@@ -87,6 +87,36 @@ In this example:
 - The `expression` value `3+8/9` is injected into the user message.
 - The model will respond with both the result of the expression and an explanation, as specified in the `output` section of the prompt.
 
+3. **Using the agent feature:**  
+
+The `agent` method allows you to enhance the prompt's functionality by integrating external tools. Here’s an example where we evaluate a mathematical expression using Python’s `eval` in a safe execution environment:
+
+```python
+from prompt_caller import PromptCaller
+
+ai = PromptCaller()
+
+def evaluate_expression(expression: str):
+      """
+      Evaluate a math expression using eval.
+      """
+      safe_globals = {"__builtins__": None}
+      return eval(expression, safe_globals, {})
+
+response = ai.agent(
+      "sample-agent", {"expression": "3+8/9"}, tools=[evaluate_expression]
+)
+
+print(response)
+```
+
+In this example:
+
+- The `agent` method is used to process the prompt while integrating external tools.
+- The `evaluate_expression` function evaluates the mathematical expression securely.
+- The response includes the processed result based on the prompt and tool execution.
+
+
 ## How It Works
 
 1. **\_loadPrompt:** Loads the prompt file, splits the YAML header from the body, and parses them.

prompt_caller-0.1.1.dist-info/RECORD

@@ -0,0 +1,8 @@
+prompt_caller/__init__.py,sha256=4EGdeAJ_Ig7A-b-e17-nYbiXjckT7uL3to5lchMsoW4,41
+prompt_caller/__main__.py,sha256=dJ0dYtVmnhZuoV79R6YiAIta1ZkUKb-TEX4VEuYbgk0,139
+prompt_caller/prompt_caller.py,sha256=fy-pLXmYD2j5fnAxgBvxCNBrkQvDPGX0nWyqnaWeqSo,6737
+prompt_caller-0.1.1.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+prompt_caller-0.1.1.dist-info/METADATA,sha256=j1xmg_Y_NAhh7CmlCgAcfSMuHChZ9C4DPcszLADg7dk,4909
+prompt_caller-0.1.1.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
+prompt_caller-0.1.1.dist-info/top_level.txt,sha256=iihiDRq-0VrKB8IKjxf7Lrtv-fLMq4tvgM4fH3x0I94,14
+prompt_caller-0.1.1.dist-info/RECORD,,

prompt_caller-0.0.5.dist-info/RECORD

@@ -1,8 +0,0 @@
-prompt_caller/__init__.py,sha256=4EGdeAJ_Ig7A-b-e17-nYbiXjckT7uL3to5lchMsoW4,41
-prompt_caller/__main__.py,sha256=dJ0dYtVmnhZuoV79R6YiAIta1ZkUKb-TEX4VEuYbgk0,139
-prompt_caller/prompt_caller.py,sha256=k5xiY_COGxvuRVOgaVuBzO9vbrUMFrzjG4EXarMj_jk,3899
-prompt_caller-0.0.5.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-prompt_caller-0.0.5.dist-info/METADATA,sha256=Xk8vrEi0fyyQtcXlPT4QhDtc4TPW6cps0cWHW2tYJI8,3947
-prompt_caller-0.0.5.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
-prompt_caller-0.0.5.dist-info/top_level.txt,sha256=iihiDRq-0VrKB8IKjxf7Lrtv-fLMq4tvgM4fH3x0I94,14
-prompt_caller-0.0.5.dist-info/RECORD,,