prompt-caller 0.1.0__py3-none-any.whl → 0.1.2__py3-none-any.whl

prompt_caller/prompt_caller.py

@@ -19,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()
@@ -62,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)
@@ -126,14 +129,16 @@ class PromptCaller:
 
         return response
 
-    def agent(self, promptName, context=None, tools=None, allowed_steps=3):
+    def agent(
+        self, promptName, context=None, tools=None, output=None, allowed_steps=10
+    ):
 
         configuration, messages = self.loadPrompt(promptName, context)
 
-        output = None
+        dynamicOutput = None
 
-        if "output" in configuration:
-            output = configuration.get("output")
+        if output is None and "output" in configuration:
+            dynamicOutput = configuration.get("output")
             configuration.pop("output")
 
             for message in messages:
@@ -153,10 +158,13 @@ class PromptCaller:
         tools_dict = {t.name.lower(): t for t in tools}
 
         if output:
+            tools.extend([output])
+            tools_dict[output.__name__.lower()] = output
+        elif dynamicOutput:
             dynamicModel = self.createPydanticModel(output)
 
-        tools.extend([dynamicModel])
-        tools_dict["dynamicmodel"] = dynamicModel
+            tools.extend([dynamicModel])
+            tools_dict["dynamicmodel"] = dynamicModel
 
         chat = chat.bind_tools(tools)
 
@@ -171,9 +179,12 @@ class PromptCaller:
                     tool_name = tool_call["name"].lower()
 
                     # If it's the final formatting tool, validate and return
-                    if tool_name == "dynamicmodel":
+                    if dynamicOutput and tool_name == "dynamicmodel":
                         return dynamicModel.model_validate(tool_call["args"])
 
+                    if output and tool_name == output.__name__.lower():
+                        return output.model_validate(tool_call["args"])
+
                     selected_tool = tools_dict.get(tool_name)
                     if not selected_tool:
                         raise ValueError(f"Unknown tool: {tool_name}")
@@ -199,5 +210,6 @@ class PromptCaller:
             return response
 
         except Exception as e:
+            print(e)
             # Replace with appropriate logging in production
             raise RuntimeError("Error during agent process") from e

{prompt_caller-0.1.0.dist-info → prompt_caller-0.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prompt_caller
-Version: 0.1.0
+Version: 0.1.2
 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
@@ -89,33 +89,33 @@ In this example:
 
 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:
+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
+```python
+from prompt_caller import PromptCaller
 
-   ai = PromptCaller()
+ai = PromptCaller()
 
-   def evaluate_expression(expression: str):
-       """
-       Evaluate a math expression using eval.
-       """
-       safe_globals = {"__builtins__": None}
-       return eval(expression, safe_globals, {})
+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]
-   )
+response = ai.agent(
+      "sample-agent", {"expression": "3+8/9"}, tools=[evaluate_expression]
+)
 
-   print(response)
-   ```
+print(response)
+```
+
+In this example:
 
-   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.
 
-   - 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
 

prompt_caller-0.1.2.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=zAJq_5v-ku_O9ACAw7C7JU1RmVwlIvunbd-31B0XX6E,7119
+prompt_caller-0.1.2.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+prompt_caller-0.1.2.dist-info/METADATA,sha256=IjSAGTvmJbi7X6_5f3OxvH6QA7l9H6Opx4d9wtGGRak,4909
+prompt_caller-0.1.2.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
+prompt_caller-0.1.2.dist-info/top_level.txt,sha256=iihiDRq-0VrKB8IKjxf7Lrtv-fLMq4tvgM4fH3x0I94,14
+prompt_caller-0.1.2.dist-info/RECORD,,

prompt_caller-0.1.0.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=FlQEmNJWrxrdLMaoTxsCfPPcbr3DXWN5Oq1pp45yVQM,6644
-prompt_caller-0.1.0.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-prompt_caller-0.1.0.dist-info/METADATA,sha256=0ciKS5ENrpqRA6EjrulhMu_R-7iwWwmMEbiYpjJbymk,4957
-prompt_caller-0.1.0.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
-prompt_caller-0.1.0.dist-info/top_level.txt,sha256=iihiDRq-0VrKB8IKjxf7Lrtv-fLMq4tvgM4fH3x0I94,14
-prompt_caller-0.1.0.dist-info/RECORD,,