promptlayer 0.1.77__tar.gz → 0.1.79__tar.gz

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

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

{promptlayer-0.1.77 → promptlayer-0.1.79}/promptlayer/promptlayer.py

@@ -1,6 +1,10 @@
 import datetime
 import inspect
-from promptlayer.utils import get_api_key, promptlayer_api_handler, run_in_thread_async
+from promptlayer.utils import (
+    get_api_key,
+    promptlayer_api_handler,
+    promptlayer_api_handler_async,
+)
 
 
 class PromptLayerBase(object):
@@ -36,9 +40,7 @@ class PromptLayerBase(object):
             async def async_wrapper(*args, **kwargs):
                 response = await function_object(*args, **kwargs)
                 request_end_time = datetime.datetime.now().timestamp()
-                return await run_in_thread_async(
-                    None,
-                    promptlayer_api_handler,
+                return await promptlayer_api_handler_async(
                     object.__getattribute__(self, "_function_name"),
                     object.__getattribute__(self, "_provider_type"),
                     args,

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

@@ -1,11 +1,18 @@
 import asyncio
 import contextvars
-from copy import deepcopy
 import functools
-import promptlayer
-import requests
+import os
 import sys
 import types
+from copy import deepcopy
+
+import requests
+
+import promptlayer
+
+URL_API_PROMPTLAYER = os.environ.setdefault(
+    "URL_API_PROMPTLAYER", "https://api.promptlayer.com"
+)
 
 
 def get_api_key():
@@ -64,6 +71,34 @@ def promptlayer_api_handler(
         return response
 
 
+async def promptlayer_api_handler_async(
+    function_name,
+    provider_type,
+    args,
+    kwargs,
+    tags,
+    response,
+    request_start_time,
+    request_end_time,
+    api_key,
+    return_pl_id=False,
+):
+    return await run_in_thread_async(
+        None,
+        promptlayer_api_handler,
+        function_name,
+        provider_type,
+        args,
+        kwargs,
+        tags,
+        response,
+        request_start_time,
+        request_end_time,
+        get_api_key(),
+        return_pl_id=return_pl_id,
+    )
+
+
 def promptlayer_api_request(
     function_name,
     provider_type,
@@ -80,7 +115,7 @@ def promptlayer_api_request(
         response = response.to_dict_recursive()
     try:
         request_response = requests.post(
-            "https://api.promptlayer.com/track-request",
+            f"{URL_API_PROMPTLAYER}/track-request",
             json={
                 "function_name": function_name,
                 "provider_type": provider_type,
@@ -113,10 +148,38 @@ def promptlayer_api_request(
         return request_response.json().get("request_id")
 
 
+def promptlayer_api_request_async(
+    function_name,
+    provider_type,
+    args,
+    kwargs,
+    tags,
+    response,
+    request_start_time,
+    request_end_time,
+    api_key,
+    return_pl_id=False,
+):
+    return run_in_thread_async(
+        None,
+        promptlayer_api_request,
+        function_name,
+        provider_type,
+        args,
+        kwargs,
+        tags,
+        response,
+        request_start_time,
+        request_end_time,
+        api_key,
+        return_pl_id=return_pl_id,
+    )
+
+
 def promptlayer_get_prompt(prompt_name, api_key):
     try:
         request_response = requests.post(
-            "https://api.promptlayer.com/library-get-prompt-template",
+            f"{URL_API_PROMPTLAYER}/library-get-prompt-template",
             json={"prompt_name": prompt_name, "api_key": api_key,},
         )
         if request_response.status_code != 200:
@@ -138,7 +201,7 @@ def promptlayer_get_prompt(prompt_name, api_key):
 def promptlayer_publish_prompt(prompt_name, prompt_template, tags, api_key):
     try:
         request_response = requests.post(
-            "https://api.promptlayer.com/library-publish-prompt-template",
+            f"{URL_API_PROMPTLAYER}/library-publish-prompt-template",
             json={
                 "prompt_name": prompt_name,
                 "prompt_template": prompt_template,
@@ -165,7 +228,7 @@ def promptlayer_publish_prompt(prompt_name, prompt_template, tags, api_key):
 def promptlayer_track_prompt(request_id, prompt_name, input_variables, api_key):
     try:
         request_response = requests.post(
-            "https://api.promptlayer.com/library-track-prompt",
+            f"{URL_API_PROMPTLAYER}/library-track-prompt",
             json={
                 "request_id": request_id,
                 "prompt_name": prompt_name,
@@ -198,7 +261,7 @@ def promptlayer_track_prompt(request_id, prompt_name, input_variables, api_key):
 def promptlayer_track_metadata(request_id, metadata, api_key):
     try:
         request_response = requests.post(
-            "https://api.promptlayer.com/library-track-metadata",
+            f"{URL_API_PROMPTLAYER}/library-track-metadata",
             json={"request_id": request_id, "metadata": metadata, "api_key": api_key,},
         )
         if request_response.status_code != 200:
@@ -226,7 +289,7 @@ def promptlayer_track_metadata(request_id, metadata, api_key):
 def promptlayer_track_score(request_id, score, api_key):
     try:
         request_response = requests.post(
-            "https://api.promptlayer.com/library-track-score",
+            f"{URL_API_PROMPTLAYER}/library-track-score",
             json={"request_id": request_id, "score": score, "api_key": api_key,},
         )
         if request_response.status_code != 200:
@@ -303,7 +366,9 @@ class OpenAIGeneratorProxy:
             final_result = deepcopy(self.results[-1])
             final_result.choices[0].text = response
             return final_result
-        elif hasattr(self.results[0].choices[0], "delta"):  # this is completion with delta
+        elif hasattr(
+            self.results[0].choices[0], "delta"
+        ):  # this is completion with delta
             response = {"message": {"role": "", "content": ""}}
             for result in self.results:
                 if hasattr(result.choices[0].delta, "role"):

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

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

{promptlayer-0.1.77 → promptlayer-0.1.79}/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.77",
+    version="0.1.79",
     py_modules=["promptlayer"],
     packages=find_packages(),
     install_requires=["requests", "openai", "langchain"],