Add templates module and update version (#98)

* Add templates module and update version

* Add import statement to __init__.py and remove blank lines from templates.py

* Change API request method from GET to POST in get_prompt_template function

* Refactor get_prompt_template function signature

* Refactor GetPromptTemplate class to remove optional types

* Refactor get() function signature in templates.py

Update get_prompt_template() function signature in utils.py

promptlayer/__init__.py

@@ -3,11 +3,13 @@
 
 from promptlayer.promptlayer import PromptLayerBase
 
+from . import templates
+
 api_key = os.environ.get("PROMPTLAYER_API_KEY")
 
 
 def __getattr__(
-    name: Union[Literal["openai"], Literal["anthropic"], Literal["prompts"]]
+    name: Union[Literal["openai"], Literal["anthropic"], Literal["prompts"]],
 ):
     if name == "openai":
         import openai as openai_module
@@ -39,4 +41,4 @@ def __getattr__(
         raise AttributeError(f"module {__name__} has no attribute {name}")
 
 
-__all__ = ["api_key", "openai", "anthropic"]
+__all__ = ["api_key", "openai", "anthropic", "templates"]

promptlayer/templates.py

@@ -0,0 +1,8 @@
+from typing import Union
+
+from promptlayer.types.prompt_template import GetPromptTemplate
+from promptlayer.utils import get_prompt_template
+
+
+def get(prompt_name: str, params: Union[GetPromptTemplate, None] = None):
+    return get_prompt_template(prompt_name, params)

promptlayer/types/__init__.py

@@ -0,0 +1,3 @@
+from . import prompt_template
+
+__all__ = ["prompt_template"]

promptlayer/types/prompt_template.py

@@ -0,0 +1,8 @@
+from typing import Dict, TypedDict
+
+
+class GetPromptTemplate(TypedDict, total=False):
+    version: int
+    label: str
+    provider: str
+    input_variables: Dict[str, str]

promptlayer/utils.py

@@ -7,10 +7,12 @@
 import sys
 import types
 from copy import deepcopy
+from typing import Union
 
 import requests
 
 import promptlayer
+from promptlayer.types.prompt_template import GetPromptTemplate
 
 URL_API_PROMPTLAYER = os.environ.setdefault(
     "URL_API_PROMPTLAYER", "https://api.promptlayer.com"
@@ -514,3 +516,26 @@ def promptlayer_track_group(request_id, group_id):
         raise Exception(
             f"PromptLayer had the following error while tracking your group: {e}"
         )
+
+
+def get_prompt_template(
+    prompt_name: str, params: Union[GetPromptTemplate, None] = None
+):
+    try:
+        json_body = {"api_key": get_api_key()}
+        if params:
+            json_body = {**json_body, **params}
+        response = requests.post(
+            f"{URL_API_PROMPTLAYER}/prompt-templates/{prompt_name}",
+            headers={"X-API-KEY": get_api_key()},
+            json=json_body,
+        )
+        if response.status_code != 200:
+            raise Exception(
+                f"PromptLayer had the following error while getting your prompt template: {response.text}"
+            )
+        return response.json()
+    except requests.exceptions.RequestException as e:
+        raise Exception(
+            f"PromptLayer had the following error while getting your prompt template: {e}"
+        )

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "promptlayer"
-version = "0.4.0"
+version = "0.4.1"
 description = "PromptLayer is a platform for prompt engineering and tracks your LLM requests."
 authors = ["Magniv <[email protected]>"]
 license = "Apache-2.0"