prompt-caller 0.0.1__py3-none-any.whl

prompt_caller/__main__.py

@@ -0,0 +1,7 @@
+from prompt_caller import PromptCaller
+
+ai = PromptCaller()
+
+response = ai.call("sample", {"expression": "3+8/9"})
+
+print(response)

prompt_caller/prompt_caller.py

@@ -0,0 +1,95 @@
+import os
+import re
+
+import yaml
+from dotenv import load_dotenv
+from jinja2 import Template
+from langchain_core.messages import HumanMessage, SystemMessage
+from langchain_openai import ChatOpenAI
+from pydantic import BaseModel, Field, create_model
+
+load_dotenv()
+
+
+class PromptCaller:
+
+    def _loadPrompt(self, file_path):
+        with open(file_path, "r") as file:
+            content = file.read()
+
+        # Split YAML header and the body
+        header, body = content.split("---", 2)[1:]
+
+        # Parse the YAML header
+        model_config = yaml.safe_load(header.strip())
+
+        # Step 2: Parse the JSX body and return it
+        return model_config, body.strip()
+
+    def _renderTemplate(self, body, context):
+        template = Template(body)
+        return template.render(context)
+
+    def _parseJSXBody(self, body):
+        elements = []
+        tag_pattern = r"<(system|user|assistant)>(.*?)</\1>"
+
+        matches = re.findall(tag_pattern, body, re.DOTALL)
+
+        for tag, content in matches:
+            elements.append({"role": tag, "content": content.strip()})
+
+        return elements
+
+    def loadPrompt(self, promptName, context=None):
+        # initialize context
+        if context is None:
+            context = {}
+
+        configuration, template = self._loadPrompt(
+            os.path.join("prompts", f"{promptName}.prompt")
+        )
+
+        template = self._renderTemplate(template, context)
+
+        parsedMessages = self._parseJSXBody(template)
+
+        messages = []
+
+        for message in parsedMessages:
+            if message.get("role") == "system":
+                messages.append(SystemMessage(content=message.get("content")))
+
+            if message.get("role") == "user":
+                messages.append(HumanMessage(content=message.get("content")))
+
+        return configuration, messages
+
+    def createPydanticModel(self, dynamic_dict):
+        # Create a dynamic Pydantic model from the dictionary
+        fields = {
+            key: (str, Field(description=f"Description for {key}"))
+            for key in dynamic_dict.keys()
+        }
+        # Dynamically create the Pydantic model with the fields
+        return create_model("DynamicModel", **fields)
+
+    def call(self, promptName, context=None):
+
+        configuration, messages = self.loadPrompt(promptName, context)
+
+        output = None
+
+        if "output" in configuration:
+            output = configuration.get("output")
+            configuration.pop("output")
+
+        chat = ChatOpenAI(**configuration)
+
+        if output:
+            dynamicModel = self.createPydanticModel(output)
+            chat = chat.with_structured_output(dynamicModel)
+
+        response = chat.invoke(messages)
+
+        return response

prompt_caller-0.0.1.dist-info/METADATA

@@ -0,0 +1,21 @@
+Metadata-Version: 2.1
+Name: prompt_caller
+Version: 0.0.1
+Summary: This package is responsible for calling prompts in a specific format. It uses LangChain and OpenAI API
+Author: Thiago Nepomuceno
+Author-email: thiago@neps.academy
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Description-Content-Type: text/markdown
+Requires-Dist: toml
+
+# Neps Academy AI
+
+This package is responsible for all Neps Academy features related to AI
+
+# Build and Upload
+
+python setup.py sdist bdist_wheel
+
+twine upload dist/\*

prompt_caller-0.0.1.dist-info/RECORD

@@ -0,0 +1,7 @@
+prompt_caller/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+prompt_caller/__main__.py,sha256=dJ0dYtVmnhZuoV79R6YiAIta1ZkUKb-TEX4VEuYbgk0,139
+prompt_caller/prompt_caller.py,sha256=VmmMLiOWsTC-545ejXcfOP0neA3k4aDA2SLCYNGD2nI,2824
+prompt_caller-0.0.1.dist-info/METADATA,sha256=B8L0iifaUnM3idkMsLV_AE455LVTE51WMFwKKnuhdCI,625
+prompt_caller-0.0.1.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
+prompt_caller-0.0.1.dist-info/top_level.txt,sha256=iihiDRq-0VrKB8IKjxf7Lrtv-fLMq4tvgM4fH3x0I94,14
+prompt_caller-0.0.1.dist-info/RECORD,,

prompt_caller-0.0.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: bdist_wheel (0.44.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

prompt_caller-0.0.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+prompt_caller