add prompt layer integration

This commit is contained in:
ishaan-jaff 2023-08-26 11:28:19 -07:00
parent 03d023e372
commit 7d99e09a4a

View file

@ -0,0 +1,43 @@
#### What this does ####
# On success, logs events to Helicone
import dotenv, os
import requests
import requests
dotenv.load_dotenv() # Loading env variables using dotenv
import traceback
class PromptLayer:
# Class variables or attributes
def __init__(self):
# Instance variables
self.key = os.getenv("PROMPTLAYER_API_KEY")
def log_event(self, model, response_obj, start_time, end_time, print_verbose):
# Method definition
try:
print_verbose(
f"Prompt Layer Logging - Enters logging function for model {model}"
)
request_response = requests.post(
"https://api.promptlayer.com/rest/track-request",
json={
"function_name": "openai.Completion.create",
"kwargs": {"engine": "text-ada-001", "prompt": "My name is"},
"tags": ["hello", "world"],
"request_response": response_obj,
"request_start_time": start_time,
"request_end_time": end_time,
"prompt_id": "<PROMPT ID>",
"prompt_input_variables": "<Dictionary of variables for prompt>",
"prompt_version":1,
"api_key": self.key
},
)
print_verbose(f"Prompt Layer Logging - final response object: {request_response}")
except:
# traceback.print_exc()
print_verbose(f"Prompt Layer Error - {traceback.format_exc()}")
pass