prompt-caller 0.2.0__tar.gz → 0.2.1__tar.gz

{prompt_caller-0.2.0 → prompt_caller-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prompt_caller
-Version: 0.2.0
+Version: 0.2.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
@@ -13,7 +13,7 @@ License-File: LICENSE
 Requires-Dist: pyyaml>=6.0.2
 Requires-Dist: python-dotenv>=1.2.1
 Requires-Dist: Jinja2>=3.1.4
-Requires-Dist: langchain-core>=1.2.7
+Requires-Dist: langchain>=1.2.7
 Requires-Dist: langchain-openai>=1.1.7
 Requires-Dist: langchain-google-genai>=4.2.0
 Requires-Dist: openai>=2.16.0

{prompt_caller-0.2.0 → prompt_caller-0.2.1}/prompt_caller/prompt_caller.py

@@ -161,72 +161,40 @@ class PromptCaller:
 
         return response
 
-    def _create_pdf_middleware(self):
-        """Middleware to handle tool responses that contain pdf content."""
+    def _create_media_middleware(self):
+        """Middleware to handle tool responses that contain media content (images, PDFs)."""
 
         @wrap_tool_call
-        def handle_pdf_response(request, handler):
-            # Execute the actual tool
+        def handle_media_response(request, handler):
             result = handler(request)
 
-            # Check if result content is pdf data
             if hasattr(result, "content"):
                 content = result.content
-                # Try to parse if it's a string representation of a list
-                if isinstance(content, str) and content.startswith("["):
-                    try:
-                        content = ast.literal_eval(content)
-                    except (ValueError, SyntaxError):
-                        pass
-
-                if (
-                    isinstance(content, list)
-                    and content
-                    and isinstance(content[0], dict)
-                    and "input_file" in content[0]
-                    and "pdf" in content[0]["file_data"]
-                ):
-                    # Use Command to add both tool result and image to messages
-                    return Command(
-                        update={"messages": [result, HumanMessage(content=content)]}
-                    )
-
-            return result  # Return normal result
-
-        return handle_pdf_response
-
-    def _create_image_middleware(self):
-        """Middleware to handle tool responses that contain image content."""
 
-        @wrap_tool_call
-        def handle_image_response(request, handler):
-            # Execute the actual tool
-            result = handler(request)
-
-            # Check if result content is image data (list with image_url dict)
-            if hasattr(result, "content"):
-                content = result.content
-                # Try to parse if it's a string representation of a list
                 if isinstance(content, str) and content.startswith("["):
                     try:
                         content = ast.literal_eval(content)
                     except (ValueError, SyntaxError):
                         pass
 
+                # Check if content is media (image or PDF)
                 if (
                     isinstance(content, list)
                     and content
                     and isinstance(content[0], dict)
-                    and "image_url" in content[0]
                 ):
-                    # Use Command to add both tool result and image to messages
-                    return Command(
-                        update={"messages": [result, HumanMessage(content=content)]}
+                    is_media = (
+                        "image_url" in content[0]  # Image
+                        or ("input_file" == content[0].get("type", "").strip())  # PDF
                     )
+                    if is_media:
+                        return Command(
+                            update={"messages": [result, HumanMessage(content=content)]}
+                        )
 
-            return result  # Return normal result
+            return result
 
-        return handle_image_response
+        return handle_media_response
 
     def agent(
         self, promptName, context=None, tools=None, output=None, allowed_steps=10
@@ -267,10 +235,7 @@ class PromptCaller:
             tools=tools,
             system_prompt=system_prompt,
             response_format=response_format,
-            middleware=[
-                self._create_image_middleware(),
-                self._create_pdf_middleware(),
-            ],
+            middleware=[self._create_media_middleware()],
         )
 
         result = agent_graph.invoke(

{prompt_caller-0.2.0 → prompt_caller-0.2.1}/prompt_caller.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prompt_caller
-Version: 0.2.0
+Version: 0.2.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
@@ -13,7 +13,7 @@ License-File: LICENSE
 Requires-Dist: pyyaml>=6.0.2
 Requires-Dist: python-dotenv>=1.2.1
 Requires-Dist: Jinja2>=3.1.4
-Requires-Dist: langchain-core>=1.2.7
+Requires-Dist: langchain>=1.2.7
 Requires-Dist: langchain-openai>=1.1.7
 Requires-Dist: langchain-google-genai>=4.2.0
 Requires-Dist: openai>=2.16.0

{prompt_caller-0.2.0 → prompt_caller-0.2.1}/prompt_caller.egg-info/requires.txt

@@ -1,7 +1,7 @@
 pyyaml>=6.0.2
 python-dotenv>=1.2.1
 Jinja2>=3.1.4
-langchain-core>=1.2.7
+langchain>=1.2.7
 langchain-openai>=1.1.7
 langchain-google-genai>=4.2.0
 openai>=2.16.0

{prompt_caller-0.2.0 → prompt_caller-0.2.1}/setup.py

@@ -35,7 +35,7 @@ class BdistWheelCommand(bdist_wheel):
 
 setuptools.setup(
     name="prompt_caller",
-    version="0.2.0",
+    version="0.2.1",
     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",
@@ -53,7 +53,7 @@ setuptools.setup(
         "pyyaml>=6.0.2",
         "python-dotenv>=1.2.1",
         "Jinja2>=3.1.4",
-        "langchain-core>=1.2.7",
+        "langchain>=1.2.7",
         "langchain-openai>=1.1.7",
         "langchain-google-genai>=4.2.0",
         "openai>=2.16.0",