prompt-caller 0.1.1__tar.gz → 0.1.2__tar.gz

{prompt_caller-0.1.1 → prompt_caller-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prompt_caller
-Version: 0.1.1
+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

{prompt_caller-0.1.1 → prompt_caller-0.1.2}/prompt_caller/prompt_caller.py

@@ -129,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:
@@ -156,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)
 
@@ -174,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}")
@@ -202,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.1 → prompt_caller-0.1.2}/prompt_caller.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prompt_caller
-Version: 0.1.1
+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

{prompt_caller-0.1.1 → prompt_caller-0.1.2}/setup.py

@@ -35,7 +35,7 @@ class BdistWheelCommand(bdist_wheel):
 
 setuptools.setup(
     name="prompt_caller",
-    version="0.1.1",
+    version="0.1.2",
     author="Thiago Nepomuceno",
     author_email="thiago@neps.academy",
     description="This package is responsible for calling prompts in a specific format. It uses LangChain and OpenAI API",