prompt-caller 0.0.3__tar.gz → 0.0.5__tar.gz

{prompt_caller-0.0.3 → prompt_caller-0.0.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prompt_caller
-Version: 0.0.3
+Version: 0.0.5
 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,9 @@ License-File: LICENSE
 Requires-Dist: pyyaml>=6.0.2
 Requires-Dist: python-dotenv>=1.0.1
 Requires-Dist: Jinja2>=3.1.4
-Requires-Dist: langchain-openai>=0.1.23
+Requires-Dist: langchain-openai>=0.3.5
+Requires-Dist: openai>=1.63.0
+Requires-Dist: pillow>=11.0.0
 
 # PromptCaller
 

{prompt_caller-0.0.3 → prompt_caller-0.0.5}/prompt_caller/prompt_caller.py

@@ -1,13 +1,18 @@
 import os
 import re
 
+import requests
 import yaml
 from dotenv import load_dotenv
 from jinja2 import Template
 from langchain_core.messages import HumanMessage, SystemMessage
 from langchain_openai import ChatOpenAI
+from PIL import Image
 from pydantic import BaseModel, Field, create_model
 
+from io import BytesIO
+import base64
+
 load_dotenv()
 
 
@@ -32,7 +37,7 @@ class PromptCaller:
 
     def _parseJSXBody(self, body):
         elements = []
-        tag_pattern = r"<(system|user|assistant)>(.*?)</\1>"
+        tag_pattern = r"<(system|user|assistant|image)>(.*?)</\1>"
 
         matches = re.findall(tag_pattern, body, re.DOTALL)
 
@@ -41,6 +46,15 @@ class PromptCaller:
 
         return elements
 
+    def getImageBase64(self, url: str) -> str:
+        response = requests.get(url)
+        response.raise_for_status()
+        img = Image.open(BytesIO(response.content))
+        buffered = BytesIO()
+        img.save(buffered, format="PNG")
+        img_base64 = base64.b64encode(buffered.getvalue()).decode("utf-8")
+        return f"data:image/png;base64,{img_base64}"
+
     def loadPrompt(self, promptName, context=None):
         # initialize context
         if context is None:
@@ -63,6 +77,23 @@ class PromptCaller:
             if message.get("role") == "user":
                 messages.append(HumanMessage(content=message.get("content")))
 
+            if message.get("role") == "image":
+                base64_image = message.get("content")
+
+                if base64_image.startswith("http"):
+                    base64_image = self.getImageBase64(base64_image)
+
+                messages.append(
+                    HumanMessage(
+                        content=[
+                            {
+                                "type": "image_url",
+                                "image_url": {"url": base64_image},
+                            }
+                        ]
+                    )
+                )
+
         return configuration, messages
 
     def createPydanticModel(self, dynamic_dict):

{prompt_caller-0.0.3 → prompt_caller-0.0.5}/prompt_caller.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prompt_caller
-Version: 0.0.3
+Version: 0.0.5
 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,9 @@ License-File: LICENSE
 Requires-Dist: pyyaml>=6.0.2
 Requires-Dist: python-dotenv>=1.0.1
 Requires-Dist: Jinja2>=3.1.4
-Requires-Dist: langchain-openai>=0.1.23
+Requires-Dist: langchain-openai>=0.3.5
+Requires-Dist: openai>=1.63.0
+Requires-Dist: pillow>=11.0.0
 
 # PromptCaller
 

prompt_caller-0.0.5/prompt_caller.egg-info/requires.txt

@@ -0,0 +1,6 @@
+pyyaml>=6.0.2
+python-dotenv>=1.0.1
+Jinja2>=3.1.4
+langchain-openai>=0.3.5
+openai>=1.63.0
+pillow>=11.0.0

{prompt_caller-0.0.3 → prompt_caller-0.0.5}/setup.py

@@ -35,7 +35,7 @@ class BdistWheelCommand(bdist_wheel):
 
 setuptools.setup(
     name="prompt_caller",
-    version="0.0.3",
+    version="0.0.5",
     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,9 @@ setuptools.setup(
         "pyyaml>=6.0.2",
         "python-dotenv>=1.0.1",
         "Jinja2>=3.1.4",
-        "langchain-openai>=0.1.23",
+        "langchain-openai>=0.3.5",
+        "openai>=1.63.0",
+        "pillow>=11.0.0",
     ],
     cmdclass={"sdist": SdistCommand, "bdist_wheel": BdistWheelCommand},
 )

prompt_caller-0.0.3/prompt_caller.egg-info/requires.txt

@@ -1,4 +0,0 @@
-pyyaml>=6.0.2
-python-dotenv>=1.0.1
-Jinja2>=3.1.4
-langchain-openai>=0.1.23