prompt-caller 0.0.1__tar.gz

prompt_caller-0.0.1/PKG-INFO

@@ -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/README.md

@@ -0,0 +1,9 @@
+# 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/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-0.0.1/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/prompt_caller.egg-info/PKG-INFO

@@ -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/prompt_caller.egg-info/SOURCES.txt

@@ -0,0 +1,10 @@
+README.md
+setup.py
+prompt_caller/__init__.py
+prompt_caller/__main__.py
+prompt_caller/prompt_caller.py
+prompt_caller.egg-info/PKG-INFO
+prompt_caller.egg-info/SOURCES.txt
+prompt_caller.egg-info/dependency_links.txt
+prompt_caller.egg-info/requires.txt
+prompt_caller.egg-info/top_level.txt

prompt_caller-0.0.1/prompt_caller.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

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

@@ -0,0 +1 @@
+toml

prompt_caller-0.0.1/prompt_caller.egg-info/top_level.txt

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

prompt_caller-0.0.1/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

prompt_caller-0.0.1/setup.py

@@ -0,0 +1,55 @@
+import setuptools
+import shutil
+import os
+
+from setuptools.command.sdist import sdist
+from wheel.bdist_wheel import bdist_wheel
+
+with open("README.md", "r") as fh:
+    long_description = fh.read()
+
+
+def clean_folders():
+    print("Removing temporary folders.")
+
+    TMP_FOLDERS = ["neps_academy_ai.egg-info", "build"]
+    for folder in TMP_FOLDERS:
+        if os.path.exists(folder):
+            shutil.rmtree(folder)
+
+
+class SdistCommand(sdist):
+    """Custom build command."""
+
+    def run(self):
+        sdist.run(self)
+        clean_folders()
+
+
+class BdistWheelCommand(bdist_wheel):
+    """Custom build command."""
+
+    def run(self):
+        bdist_wheel.run(self)
+        clean_folders()
+
+
+setuptools.setup(
+    name="prompt_caller",
+    version="0.0.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",
+    long_description=long_description,
+    long_description_content_type="text/markdown",
+    include_package_data=True,
+    # url="https://github.com/username/neps_academy_ai",
+    packages=setuptools.find_packages(),
+    classifiers=[
+        "Programming Language :: Python :: 3",
+        "License :: OSI Approved :: MIT License",
+        "Operating System :: OS Independent",
+    ],
+    install_requires=["toml"],
+    cmdclass={"sdist": SdistCommand, "bdist_wheel": BdistWheelCommand},
+)