promptlayer 0.1.79__tar.gz → 0.1.80__tar.gz

{promptlayer-0.1.79 → promptlayer-0.1.80}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: promptlayer
-Version: 0.1.79
+Version: 0.1.80
 Summary: PromptLayer is a package to keep track of your GPT models training
 Home-page: https://www.promptlayer.com
 Author: Magniv

{promptlayer-0.1.79 → promptlayer-0.1.80}/promptlayer/prompts/prompts.py

@@ -1,22 +1,32 @@
-from promptlayer.utils import get_api_key, promptlayer_get_prompt, promptlayer_publish_prompt
-from langchain.prompts.loading import load_prompt_from_config
 from langchain import PromptTemplate
+from langchain.prompts.loading import load_prompt_from_config
+
+from promptlayer.utils import (get_api_key, promptlayer_get_prompt,
+                               promptlayer_publish_prompt)
 
-def get_prompt(prompt_name, langchain=False):
+
+def get_prompt(prompt_name, langchain=False, version=None):
+    """
+    Get a prompt template from PromptLayer.
+    version: The version of the prompt to get. If not specified, the latest version will be returned.
+    """
     api_key = get_api_key()
-    prompt = promptlayer_get_prompt(prompt_name, api_key)
+    prompt = promptlayer_get_prompt(prompt_name, api_key, version)
     if langchain:
         if "_type" not in prompt["prompt_template"]:
             prompt["prompt_template"]["_type"] = "prompt"
         return load_prompt_from_config(prompt["prompt_template"])
     else:
         return prompt["prompt_template"]
-    
+
+
 def publish_prompt(prompt_name, tags=[], prompt_template=None):
     api_key = get_api_key()
     if type(prompt_template) == dict:
         promptlayer_publish_prompt(prompt_name, prompt_template, tags, api_key)
     elif isinstance(prompt_template, PromptTemplate):
-        promptlayer_publish_prompt(prompt_name, prompt_template.dict(), tags, api_key)
+        promptlayer_publish_prompt(
+            prompt_name, prompt_template.dict(), tags, api_key)
     else:
-        raise Exception("Please provide either a JSON prompt template or a langchain prompt template.")
+        raise Exception(
+            "Please provide either a JSON prompt template or a langchain prompt template.")

{promptlayer-0.1.79 → promptlayer-0.1.80}/promptlayer/track/track.py

@@ -1,9 +1,9 @@
 from promptlayer.utils import get_api_key, promptlayer_track_prompt, promptlayer_track_metadata, promptlayer_track_score
 
-def prompt(request_id, prompt_name, prompt_input_variables):
+def prompt(request_id, prompt_name, prompt_input_variables, version=None):
     if not isinstance(prompt_input_variables, dict):
         raise Exception("Please provide a dictionary of input variables.")
-    return promptlayer_track_prompt(request_id, prompt_name, prompt_input_variables, get_api_key())
+    return promptlayer_track_prompt(request_id, prompt_name, prompt_input_variables, get_api_key(), version)
 
 
 def metadata(request_id, metadata):

{promptlayer-0.1.79 → promptlayer-0.1.80}/promptlayer/utils.py

@@ -176,11 +176,16 @@ def promptlayer_api_request_async(
     )
 
 
-def promptlayer_get_prompt(prompt_name, api_key):
+def promptlayer_get_prompt(prompt_name, api_key, version=None):
+    """
+    Get a prompt from the PromptLayer library
+    version: version of the prompt to get, None for latest
+    """
     try:
         request_response = requests.post(
             f"{URL_API_PROMPTLAYER}/library-get-prompt-template",
-            json={"prompt_name": prompt_name, "api_key": api_key,},
+            json={"prompt_name": prompt_name,
+                  "api_key": api_key, 'version': version},
         )
         if request_response.status_code != 200:
             if hasattr(request_response, "json"):
@@ -225,7 +230,7 @@ def promptlayer_publish_prompt(prompt_name, prompt_template, tags, api_key):
     return True
 
 
-def promptlayer_track_prompt(request_id, prompt_name, input_variables, api_key):
+def promptlayer_track_prompt(request_id, prompt_name, input_variables, api_key, version):
     try:
         request_response = requests.post(
             f"{URL_API_PROMPTLAYER}/library-track-prompt",
@@ -234,6 +239,7 @@ def promptlayer_track_prompt(request_id, prompt_name, input_variables, api_key):
                 "prompt_name": prompt_name,
                 "prompt_input_variables": input_variables,
                 "api_key": api_key,
+                "version": version,
             },
         )
         if request_response.status_code != 200:
@@ -262,7 +268,8 @@ def promptlayer_track_metadata(request_id, metadata, api_key):
     try:
         request_response = requests.post(
             f"{URL_API_PROMPTLAYER}/library-track-metadata",
-            json={"request_id": request_id, "metadata": metadata, "api_key": api_key,},
+            json={"request_id": request_id,
+                  "metadata": metadata, "api_key": api_key, },
         )
         if request_response.status_code != 200:
             if hasattr(request_response, "json"):
@@ -290,7 +297,8 @@ def promptlayer_track_score(request_id, score, api_key):
     try:
         request_response = requests.post(
             f"{URL_API_PROMPTLAYER}/library-track-score",
-            json={"request_id": request_id, "score": score, "api_key": api_key,},
+            json={"request_id": request_id,
+                  "score": score, "api_key": api_key, },
         )
         if request_response.status_code != 200:
             if hasattr(request_response, "json"):

{promptlayer-0.1.79 → promptlayer-0.1.80}/promptlayer.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: promptlayer
-Version: 0.1.79
+Version: 0.1.80
 Summary: PromptLayer is a package to keep track of your GPT models training
 Home-page: https://www.promptlayer.com
 Author: Magniv

{promptlayer-0.1.79 → promptlayer-0.1.80}/setup.py

@@ -17,7 +17,7 @@ setup(
     author_email="hello@magniv.io",
     url="https://www.promptlayer.com",
     project_urls={"Documentation": "https://magniv.notion.site/Prompt-Layer-Docs-db0e6f50cacf4564a6d09824ba17a629",},
-    version="0.1.79",
+    version="0.1.80",
     py_modules=["promptlayer"],
     packages=find_packages(),
     install_requires=["requests", "openai", "langchain"],