promptlayer 0.5.6__tar.gz → 1.0.1__tar.gz

{promptlayer-0.5.6 → promptlayer-1.0.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: promptlayer
-Version: 0.5.6
+Version: 1.0.1
 Summary: PromptLayer is a platform for prompt engineering and tracks your LLM requests.
 License: Apache-2.0
 Author: Magniv
@@ -12,6 +12,9 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Requires-Dist: anthropic (>=0.25.8,<0.26.0)
+Requires-Dist: anyio (>=4.3.0,<5.0.0)
+Requires-Dist: openai (>=1.26.0,<2.0.0)
 Requires-Dist: requests (>=2.31.0,<3.0.0)
 Description-Content-Type: text/markdown
 
@@ -56,8 +59,9 @@ Once you have that all set up, [install PromptLayer using](https://pypi.org/proj
 In the Python file where you use OpenAI APIs, add the following. This allows us to keep track of your requests without needing any other code changes.
 
 ```python
-import promptlayer
-promptlayer.api_key = "<YOUR PromptLayer API KEY pl_xxxxxx>"
+from promptlayer import PromptLayer
+
+promptlayer = PromptLayer(api_key="<YOUR PromptLayer API KEY pl_xxxxxx>")
 openai = promptlayer.openai
 ```
 

{promptlayer-0.5.6 → promptlayer-1.0.1}/README.md

@@ -39,8 +39,9 @@ Once you have that all set up, [install PromptLayer using](https://pypi.org/proj
 In the Python file where you use OpenAI APIs, add the following. This allows us to keep track of your requests without needing any other code changes.
 
 ```python
-import promptlayer
-promptlayer.api_key = "<YOUR PromptLayer API KEY pl_xxxxxx>"
+from promptlayer import PromptLayer
+
+promptlayer = PromptLayer(api_key="<YOUR PromptLayer API KEY pl_xxxxxx>")
 openai = promptlayer.openai
 ```
 

promptlayer-1.0.1/promptlayer/__init__.py

@@ -0,0 +1,48 @@
+import os
+from typing import Literal, Union
+
+from promptlayer.groups import GroupManager
+from promptlayer.promptlayer import PromptLayerBase
+from promptlayer.templates import TemplateManager
+from promptlayer.track import TrackManager
+
+
+class PromptLayer:
+    def __init__(self, api_key: str = None):
+        if api_key is None:
+            api_key = os.environ.get("PROMPTLAYER_API_KEY")
+        if api_key is None:
+            raise ValueError(
+                "PromptLayer API key not provided. Please set the PROMPTLAYER_API_KEY environment variable or pass the api_key parameter."
+            )
+        self.api_key = api_key
+        self.templates = TemplateManager(api_key)
+        self.group = GroupManager(api_key)
+        self.track = TrackManager(api_key)
+
+    def __getattr__(
+        self,
+        name: Union[Literal["openai"], Literal["anthropic"], Literal["prompts"]],
+    ):
+        if name == "openai":
+            import openai as openai_module
+
+            openai = PromptLayerBase(
+                openai_module, function_name="openai", api_key=self.api_key
+            )
+            return openai
+        elif name == "anthropic":
+            import anthropic as anthropic_module
+
+            anthropic = PromptLayerBase(
+                anthropic_module,
+                function_name="anthropic",
+                provider_type="anthropic",
+                api_key=self.api_key,
+            )
+            return anthropic
+        else:
+            raise AttributeError(f"module {__name__} has no attribute {name}")
+
+
+__all__ = ["PromptLayer"]

promptlayer-1.0.1/promptlayer/groups/__init__.py

@@ -0,0 +1,12 @@
+from promptlayer.groups.groups import create
+
+
+class GroupManager:
+    def __init__(self, api_key: str):
+        self.api_key = api_key
+
+    def create(self):
+        return create(self.api_key)
+
+
+__all__ = ["GroupManager"]

{promptlayer-0.5.6 → promptlayer-1.0.1}/promptlayer/groups/groups.py

@@ -1,6 +1,6 @@
 from promptlayer.utils import promptlayer_create_group
 
 
-def create():
+def create(api_key: str = None):
     """Create a new group."""
-    return promptlayer_create_group()
+    return promptlayer_create_group(api_key)

{promptlayer-0.5.6 → promptlayer-1.0.1}/promptlayer/promptlayer.py

@@ -2,26 +2,27 @@ import datetime
 import inspect
 import re
 
-from promptlayer.utils import async_wrapper, get_api_key, promptlayer_api_handler
+from promptlayer.utils import async_wrapper, promptlayer_api_handler
 
 
 class PromptLayerBase(object):
-    __slots__ = ["_obj", "__weakref__", "_function_name", "_provider_type"]
+    __slots__ = ["_obj", "__weakref__", "_function_name", "_provider_type", "_api_key"]
 
-    def __init__(self, obj, function_name="", provider_type="openai"):
+    def __init__(self, obj, function_name="", provider_type="openai", api_key=None):
         object.__setattr__(self, "_obj", obj)
         object.__setattr__(self, "_function_name", function_name)
         object.__setattr__(self, "_provider_type", provider_type)
+        object.__setattr__(self, "_api_key", api_key)
 
     def __getattr__(self, name):
         attr = getattr(object.__getattribute__(self, "_obj"), name)
         if (
             name != "count_tokens"  # fix for anthropic count_tokens
             and not re.match(
-                "<class 'anthropic\..*Error'>", str(attr)
+                r"<class 'anthropic\..*Error'>", str(attr)
             )  # fix for anthropic errors
             and not re.match(
-                "<class 'openai\..*Error'>", str(attr)
+                r"<class 'openai\..*Error'>", str(attr)
             )  # fix for openai errors
             and (
                 inspect.isclass(attr)
@@ -34,13 +35,14 @@ class PromptLayerBase(object):
                 or str(type(attr)) == "<class 'anthropic.resources.messages.Messages'>"
                 or str(type(attr))
                 == "<class 'anthropic.resources.messages.AsyncMessages'>"
-                or re.match("<class 'openai\.resources.*'>", str(type(attr)))
+                or re.match(r"<class 'openai\.resources.*'>", str(type(attr)))
             )
         ):
             return PromptLayerBase(
                 attr,
                 function_name=f'{object.__getattribute__(self, "_function_name")}.{name}',
                 provider_type=object.__getattribute__(self, "_provider_type"),
+                api_key=object.__getattribute__(self, "_api_key"),
             )
         return attr
 
@@ -62,6 +64,7 @@ class PromptLayerBase(object):
                 function_object(*args, **kwargs),
                 function_name=object.__getattribute__(self, "_function_name"),
                 provider_type=object.__getattribute__(self, "_provider_type"),
+                api_key=object.__getattribute__(self, "_api_key"),
             )
         function_response = function_object(*args, **kwargs)
         if inspect.iscoroutinefunction(function_object) or inspect.iscoroutine(
@@ -74,6 +77,7 @@ class PromptLayerBase(object):
                 object.__getattribute__(self, "_function_name"),
                 object.__getattribute__(self, "_provider_type"),
                 tags,
+                api_key=object.__getattribute__(self, "_api_key"),
                 *args,
                 **kwargs,
             )
@@ -87,6 +91,6 @@ class PromptLayerBase(object):
             function_response,
             request_start_time,
             request_end_time,
-            get_api_key(),
+            object.__getattribute__(self, "_api_key"),
             return_pl_id=return_pl_id,
         )

promptlayer-1.0.1/promptlayer/templates.py

@@ -0,0 +1,22 @@
+from typing import Union
+
+from promptlayer.types.prompt_template import GetPromptTemplate, PublishPromptTemplate
+from promptlayer.utils import (
+    get_all_prompt_templates,
+    get_prompt_template,
+    publish_prompt_template,
+)
+
+
+class TemplateManager:
+    def __init__(self, api_key: str):
+        self.api_key = api_key
+
+    def get(self, prompt_name: str, params: Union[GetPromptTemplate, None] = None):
+        return get_prompt_template(prompt_name, params, self.api_key)
+
+    def publish(self, body: PublishPromptTemplate):
+        return publish_prompt_template(body, self.api_key)
+
+    def all(self, page: int = 1, per_page: int = 30):
+        return get_all_prompt_templates(page, per_page, self.api_key)

promptlayer-1.0.1/promptlayer/track/__init__.py

@@ -0,0 +1,33 @@
+from promptlayer.track.track import group
+from promptlayer.track.track import metadata as metadata_
+from promptlayer.track.track import prompt
+from promptlayer.track.track import score as score_
+
+
+class TrackManager:
+    def __init__(self, api_key: str):
+        self.api_key = api_key
+
+    def group(self, request_id, group_id):
+        return group(request_id, group_id, self.api_key)
+
+    def metadata(self, request_id, metadata):
+        return metadata_(request_id, metadata, self.api_key)
+
+    def prompt(
+        self, request_id, prompt_name, prompt_input_variables, version=None, label=None
+    ):
+        return prompt(
+            request_id,
+            prompt_name,
+            prompt_input_variables,
+            version,
+            label,
+            self.api_key,
+        )
+
+    def score(self, request_id, score, score_name=None):
+        return score_(request_id, score, score_name, self.api_key)
+
+
+__all__ = ["TrackManager"]

{promptlayer-0.5.6 → promptlayer-1.0.1}/promptlayer/track/track.py

@@ -1,5 +1,4 @@
 from promptlayer.utils import (
-    get_api_key,
     promptlayer_track_group,
     promptlayer_track_metadata,
     promptlayer_track_prompt,
@@ -7,15 +6,22 @@ from promptlayer.utils import (
 )
 
 
-def prompt(request_id, prompt_name, prompt_input_variables, version=None, label=None):
+def prompt(
+    request_id,
+    prompt_name,
+    prompt_input_variables,
+    version=None,
+    label=None,
+    api_key: str = 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(), version, label
+        request_id, prompt_name, prompt_input_variables, api_key, version, label
     )
 
 
-def metadata(request_id, metadata):
+def metadata(request_id, metadata, api_key: str = None):
     if not isinstance(metadata, dict):
         raise Exception("Please provide a dictionary of metadata.")
     for key, value in metadata.items():
@@ -23,18 +29,18 @@ def metadata(request_id, metadata):
             raise Exception(
                 "Please provide a dictionary of metadata with key value pair of strings."
             )
-    return promptlayer_track_metadata(request_id, metadata, get_api_key())
+    return promptlayer_track_metadata(request_id, metadata, api_key)
 
 
-def score(request_id, score, score_name=None):
+def score(request_id, score, score_name=None, api_key: str = None):
     if not isinstance(score, int):
         raise Exception("Please provide a int score.")
     if not isinstance(score_name, str) and score_name is not None:
         raise Exception("Please provide a string as score name.")
     if score < 0 or score > 100:
         raise Exception("Please provide a score between 0 and 100.")
-    return promptlayer_track_score(request_id, score, score_name, get_api_key())
+    return promptlayer_track_score(request_id, score, score_name, api_key)
 
 
-def group(request_id, group_id):
-    return promptlayer_track_group(request_id, group_id)
+def group(request_id, group_id, api_key: str = None):
+    return promptlayer_track_group(request_id, group_id, api_key)

{promptlayer-0.5.6 → promptlayer-1.0.1}/promptlayer/utils.py

@@ -12,7 +12,6 @@ from typing import List, Union
 
 import requests
 
-import promptlayer
 from promptlayer.types.prompt_template import (
     GetPromptTemplate,
     GetPromptTemplateResponse,
@@ -26,16 +25,6 @@ URL_API_PROMPTLAYER = os.environ.setdefault(
 )
 
 
-def get_api_key():
-    # raise an error if the api key is not set
-    if promptlayer.api_key is None:
-        raise Exception(
-            "Please set your PROMPTLAYER_API_KEY environment variable or set API KEY in code using 'promptlayer.api_key = <your_api_key>' "
-        )
-    else:
-        return promptlayer.api_key
-
-
 def promptlayer_api_handler(
     function_name,
     provider_type,
@@ -66,6 +55,7 @@ def promptlayer_api_handler(
                 "request_end_time": request_end_time,
                 "return_pl_id": return_pl_id,
             },
+            api_key,
         )
     else:
         request_id = promptlayer_api_request(
@@ -108,7 +98,7 @@ async def promptlayer_api_handler_async(
         response,
         request_start_time,
         request_end_time,
-        get_api_key(),
+        api_key,
         return_pl_id=return_pl_id,
     )
 
@@ -345,10 +335,11 @@ def promptlayer_track_score(request_id, score, score_name, api_key):
 
 
 class GeneratorProxy:
-    def __init__(self, generator, api_request_arguments):
+    def __init__(self, generator, api_request_arguments, api_key):
         self.generator = generator
         self.results = []
         self.api_request_arugments = api_request_arguments
+        self.api_key = api_key
 
     def __iter__(self):
         return self
@@ -362,6 +353,7 @@ class GeneratorProxy:
             return GeneratorProxy(
                 await self.generator._AsyncMessageStreamManager__api_request,
                 api_request_arguments,
+                self.api_key,
             )
 
     async def __aexit__(self, exc_type, exc_val, exc_tb):
@@ -378,7 +370,7 @@ class GeneratorProxy:
     def __getattr__(self, name):
         if name == "text_stream":  # anthropic async stream
             return GeneratorProxy(
-                self.generator.text_stream, self.api_request_arugments
+                self.generator.text_stream, self.api_request_arugments, self.api_key
             )
         return getattr(self.generator, name)
 
@@ -408,7 +400,7 @@ class GeneratorProxy:
                 self.cleaned_result(),
                 self.api_request_arugments["request_start_time"],
                 self.api_request_arugments["request_end_time"],
-                get_api_key(),
+                self.api_key,
                 return_pl_id=self.api_request_arugments["return_pl_id"],
             )
             if self.api_request_arugments["return_pl_id"]:
@@ -517,6 +509,7 @@ async def async_wrapper(
     function_name,
     provider_type,
     tags,
+    api_key: str = None,
     *args,
     **kwargs,
 ):
@@ -531,17 +524,17 @@ async def async_wrapper(
         response,
         request_start_time,
         request_end_time,
-        get_api_key(),
+        api_key,
         return_pl_id=return_pl_id,
     )
 
 
-def promptlayer_create_group():
+def promptlayer_create_group(api_key: str = None):
     try:
         request_response = requests.post(
             f"{URL_API_PROMPTLAYER}/create-group",
             json={
-                "api_key": get_api_key(),
+                "api_key": api_key,
             },
         )
         if request_response.status_code != 200:
@@ -558,12 +551,12 @@ def promptlayer_create_group():
     return request_response.json()["id"]
 
 
-def promptlayer_track_group(request_id, group_id):
+def promptlayer_track_group(request_id, group_id, api_key: str = None):
     try:
         request_response = requests.post(
             f"{URL_API_PROMPTLAYER}/track-group",
             json={
-                "api_key": get_api_key(),
+                "api_key": api_key,
                 "request_id": request_id,
                 "group_id": group_id,
             },
@@ -579,18 +572,19 @@ def promptlayer_track_group(request_id, group_id):
         raise Exception(
             f"PromptLayer had the following error while tracking your group: {e}"
         )
+    return True
 
 
 def get_prompt_template(
-    prompt_name: str, params: Union[GetPromptTemplate, None] = None
+    prompt_name: str, params: Union[GetPromptTemplate, None] = None, api_key: str = None
 ) -> GetPromptTemplateResponse:
     try:
-        json_body = {"api_key": get_api_key()}
+        json_body = {"api_key": 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()},
+            headers={"X-API-KEY": api_key},
             json=json_body,
         )
         if response.status_code != 200:
@@ -606,11 +600,12 @@ def get_prompt_template(
 
 def publish_prompt_template(
     body: PublishPromptTemplate,
+    api_key: str = None,
 ) -> PublishPromptTemplateResponse:
     try:
         response = requests.post(
             f"{URL_API_PROMPTLAYER}/rest/prompt-templates",
-            headers={"X-API-KEY": get_api_key()},
+            headers={"X-API-KEY": api_key},
             json={
                 "prompt_template": {**body},
                 "prompt_version": {**body},
@@ -629,12 +624,12 @@ def publish_prompt_template(
 
 
 def get_all_prompt_templates(
-    page: int = 1, per_page: int = 30
+    page: int = 1, per_page: int = 30, api_key: str = None
 ) -> List[ListPromptTemplateResponse]:
     try:
         response = requests.get(
             f"{URL_API_PROMPTLAYER}/prompt-templates",
-            headers={"X-API-KEY": get_api_key()},
+            headers={"X-API-KEY": api_key},
             params={"page": page, "per_page": per_page},
         )
         if response.status_code != 200:

{promptlayer-0.5.6 → promptlayer-1.0.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "promptlayer"
-version = "0.5.6"
+version = "1.0.1"
 description = "PromptLayer is a platform for prompt engineering and tracks your LLM requests."
 authors = ["Magniv <hello@magniv.io>"]
 license = "Apache-2.0"
@@ -9,11 +9,15 @@ readme = "README.md"
 [tool.poetry.dependencies]
 python = ">=3.8.1,<4.0"
 requests = "^2.31.0"
+openai = "^1.26.0"
+anyio = "^4.3.0"
+anthropic = "^0.25.8"
 
 [tool.poetry.group.dev.dependencies]
-openai = "^0.27.8"
 langchain = "^0.0.260"
 behave = "^1.2.6"
+pytest = "^8.2.0"
+pytest-asyncio = "^0.23.6"
 
 [build-system]
 requires = ["poetry-core"]

promptlayer-0.5.6/promptlayer/__init__.py

@@ -1,44 +0,0 @@
-import os
-from typing import Literal, Union
-
-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"]],
-):
-    if name == "openai":
-        import openai as openai_module
-
-        openai = PromptLayerBase(openai_module, function_name="openai")
-        return openai
-    elif name == "anthropic":
-        import anthropic as anthropic_module
-
-        anthropic = PromptLayerBase(
-            anthropic_module,
-            function_name="anthropic",
-            provider_type="anthropic",
-        )
-        return anthropic
-    elif name == "prompts":
-        import promptlayer.prompts as prompts
-
-        return prompts
-    elif name == "group":
-        import promptlayer.groups as group
-
-        return group
-    elif name == "track":
-        import promptlayer.track as track
-
-        return track
-    else:
-        raise AttributeError(f"module {__name__} has no attribute {name}")
-
-
-__all__ = ["api_key", "openai", "anthropic", "templates"]

promptlayer-0.5.6/promptlayer/groups/__init__.py

@@ -1,3 +0,0 @@
-from promptlayer.groups.groups import create
-
-__all__ = ["create"]

promptlayer-0.5.6/promptlayer/prompts/__init__.py

@@ -1,5 +0,0 @@
-from promptlayer.prompts.prompts import all
-from promptlayer.prompts.prompts import get_prompt as get
-from promptlayer.prompts.prompts import publish_prompt as publish
-
-__all__ = ["get", "all", "publish"]

promptlayer-0.5.6/promptlayer/prompts/chat.py

@@ -1,61 +0,0 @@
-import json
-
-from langchain import prompts
-
-CHAT_PROMPTLAYER_LANGCHAIN = "chat_promptlayer_langchain"
-ROLE_SYSTEM = "system"
-ROLE_ASSISTANT = "assistant"
-ROLE_USER = "user"
-
-
-def to_dict(prompt_template: prompts.ChatPromptTemplate):
-    prompt_dict = json.loads(prompt_template.json())
-    prompt_dict["_type"] = CHAT_PROMPTLAYER_LANGCHAIN
-    for index, message in enumerate(prompt_template.messages):
-        if isinstance(message, prompts.SystemMessagePromptTemplate):
-            prompt_dict["messages"][index]["role"] = ROLE_SYSTEM
-        elif isinstance(message, prompts.AIMessagePromptTemplate):
-            prompt_dict["messages"][index]["role"] = ROLE_ASSISTANT
-        elif isinstance(message, prompts.HumanMessagePromptTemplate):
-            prompt_dict["messages"][index]["role"] = ROLE_USER
-
-    return prompt_dict
-
-
-def to_prompt(prompt_dict: dict):
-    try:
-        messages = []
-        for message in prompt_dict.get("messages", []):
-            role = message.get("role")
-            prompt = message.get("prompt", {})
-            if not prompt:
-                continue
-            prompt_template = prompts.PromptTemplate(
-                **{k: v for k, v in prompt.items() if k != "_type"}
-            )
-            message_fields = {
-                k: v for k, v in message.items() if k != "role" and k != "prompt"
-            }
-            if role == ROLE_SYSTEM:
-                message = prompts.SystemMessagePromptTemplate(
-                    prompt=prompt_template, **message_fields
-                )
-            elif role == ROLE_ASSISTANT:
-                message = prompts.AIMessagePromptTemplate(
-                    prompt=prompt_template, **message_fields
-                )
-            elif role == ROLE_USER:
-                message = prompts.HumanMessagePromptTemplate(
-                    prompt=prompt_template, **message_fields
-                )
-            messages.append(message)
-        prompt_template = prompts.ChatPromptTemplate(
-            messages=messages,
-            input_variables=prompt_dict.get("input_variables", []),
-            output_parser=prompt_dict.get("output_parser", None),
-            partial_variables=prompt_dict.get("partial_variables", {}),
-        )
-        return prompt_template
-    except Exception as e:
-        print("Unknown error occurred. ", e)
-        return None

promptlayer-0.5.6/promptlayer/prompts/prompts.py

@@ -1,91 +0,0 @@
-from promptlayer.prompts.chat import CHAT_PROMPTLAYER_LANGCHAIN, to_dict, to_prompt
-from promptlayer.resources.prompt import Prompt
-from promptlayer.utils import (
-    get_api_key,
-    promptlayer_get_prompt,
-    promptlayer_publish_prompt,
-)
-
-
-def get_prompt(
-    prompt_name,
-    langchain=False,
-    version: int = None,
-    label: str = None,
-    include_metadata: bool = False,
-):
-    """
-    Get a prompt template from PromptLayer.
-    prompt_name: the prompt name
-    langchain: Enable this for langchain compatible prompt
-    version: The version of the prompt to get. If not specified, the latest version will be returned.
-    label: The specific label of a prompt you want to get. Setting this will supercede version
-    include_metadata: Whether or not to include the metadata of the prompt in the response.
-    """
-    api_key = get_api_key()
-    prompt = promptlayer_get_prompt(prompt_name, api_key, version, label)
-    if langchain:
-        try:
-            from langchain.prompts.loading import load_prompt_from_config
-        except ImportError:
-            raise Exception(
-                "Please install langchain to use langchain compatible prompts."
-            )
-        if "_type" not in prompt["prompt_template"]:
-            prompt["prompt_template"]["_type"] = "prompt"
-        if prompt["prompt_template"]["_type"] == CHAT_PROMPTLAYER_LANGCHAIN:
-            prompt_template = to_prompt(prompt["prompt_template"])
-        else:
-            prompt_template = load_prompt_from_config(prompt["prompt_template"])
-    else:
-        prompt_template = prompt["prompt_template"]
-    if include_metadata:
-        return prompt_template, prompt["metadata"]
-    return prompt_template
-
-
-def publish_prompt(
-    prompt_name, tags=[], commit_message=None, prompt_template=None, metadata=None
-):
-    try:
-        from langchain.prompts import ChatPromptTemplate, PromptTemplate
-    except ImportError:
-        raise Exception("Please install langchain to use langchain compatible prompts.")
-
-    api_key = get_api_key()
-    if commit_message is not None and len(commit_message) > 72:
-        raise Exception("Commit message must be less than 72 characters.")
-    if isinstance(prompt_template, ChatPromptTemplate):
-        prompt_template = to_dict(prompt_template)
-    elif isinstance(prompt_template, PromptTemplate):
-        prompt_template = prompt_template.dict()
-    elif not isinstance(prompt_template, dict):
-        raise Exception(
-            "Please provide either a JSON prompt template or a langchain prompt template."
-        )
-    promptlayer_publish_prompt(
-        prompt_name, prompt_template, commit_message, tags, api_key, metadata
-    )
-
-
-def all(page: int = 1, per_page: int = 30):
-    """
-    List all prompts on PromptLayer.
-
-    Parameters:
-    ----------
-    page: int
-        The page of prompts to get.
-    per_page: int
-        The number of prompts to get per page.
-
-    Returns:
-    -------
-    list of prompts
-    """
-    response = Prompt.list({"page": page, "per_page": per_page})
-    if not response.get("success", True):
-        raise Exception(
-            f"Failed to get prompts from PromptLayer. {response.get('message')}"
-        )
-    return response["items"]

promptlayer-0.5.6/promptlayer/resources/__init__.py

@@ -1,3 +0,0 @@
-from promptlayer.resources import prompt
-
-__all__ = ["prompt"]

promptlayer-0.5.6/promptlayer/resources/base.py

@@ -1,23 +0,0 @@
-import os
-
-import requests
-
-from promptlayer.utils import get_api_key
-
-
-class Base:
-    API_KEY = get_api_key()
-    BASE_URL = (
-        os.environ.get("URL_API_PROMPTLAYER", "https://api.promptlayer.com") + "/rest"
-    )
-
-    @classmethod
-    def list(cls, params={}):
-        """
-        List all resources
-        """
-        return requests.get(
-            cls.BASE_URL,
-            headers={"X-API-KEY": cls.API_KEY},
-            params=params,
-        ).json()

promptlayer-0.5.6/promptlayer/resources/prompt.py

@@ -1,12 +0,0 @@
-from .base import Base
-
-
-class Prompt(Base):
-    BASE_URL = Base.BASE_URL + "/prompts"
-
-    @classmethod
-    def list(cls, params={}):
-        """
-        List all prompts
-        """
-        return super().list(params)

promptlayer-0.5.6/promptlayer/templates.py

@@ -1,20 +0,0 @@
-from typing import Union
-
-from promptlayer.types.prompt_template import GetPromptTemplate, PublishPromptTemplate
-from promptlayer.utils import (
-    get_all_prompt_templates,
-    get_prompt_template,
-    publish_prompt_template,
-)
-
-
-def get(prompt_name: str, params: Union[GetPromptTemplate, None] = None):
-    return get_prompt_template(prompt_name, params)
-
-
-def publish(body: PublishPromptTemplate):
-    return publish_prompt_template(body)
-
-
-def all(page: int = 1, per_page: int = 30):
-    return get_all_prompt_templates(page, per_page)

promptlayer-0.5.6/promptlayer/track/__init__.py

@@ -1,3 +0,0 @@
-from promptlayer.track.track import group, metadata, prompt, score
-
-__all__ = ["group", "metadata", "prompt", "score"]