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

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.dist-info → prompt_caller-0.0.5.dist-info}/METADATA

@@ -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.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=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,,

prompt_caller-0.0.3.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=KRQYxvAOPMFIqhoLgJWQYr1ahGEWefNWtIjw1W_v8BM,2842
-prompt_caller-0.0.3.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-prompt_caller-0.0.3.dist-info/METADATA,sha256=LDXDLiUlMe1U3LQOhKNgM3xrqiMfSsVQglAEXK0tA4g,3886
-prompt_caller-0.0.3.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
-prompt_caller-0.0.3.dist-info/top_level.txt,sha256=iihiDRq-0VrKB8IKjxf7Lrtv-fLMq4tvgM4fH3x0I94,14
-prompt_caller-0.0.3.dist-info/RECORD,,