prompt message added to model

PromptSail · Jan 22, 2024 · f1c15e4 · f1c15e4
1 parent 5c4bb52
commit f1c15e4
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 0 deletions.
diff --git a/backend/src/transactions/models.py b/backend/src/transactions/models.py
@@ -22,6 +22,7 @@ class Transaction(BaseModel):
     library: str
     status_code: int
     message: str | None
+    prompt: str
     error_message: str | None
     request_time: datetime
     response_time: datetime = Field(

diff --git a/backend/src/transactions/schemas.py b/backend/src/transactions/schemas.py
@@ -16,6 +16,7 @@ class GetTransactionSchema(BaseModel):
     library: str
     status_code: int
     message: str | None
+    prompt: str
     error_message: str | None
     request_time: datetime
     response_time: datetime
@@ -35,6 +36,7 @@ class GetTransactionWithProjectSlugSchema(BaseModel):
     library: str
     status_code: int
     message: str | None
+    prompt: str
     error_message: str | None
     request_time: datetime
     response_time: datetime

diff --git a/backend/src/transactions/use_cases.py b/backend/src/transactions/use_cases.py
@@ -97,6 +97,7 @@ def store_transaction(
         ),
         tags=tags,
         model=params["model"],
+        prompt=params["prompt"],
         type=params["type"],
         os=params["os"],
         token_usage=params["token_usage"],

diff --git a/backend/src/utils.py b/backend/src/utils.py
@@ -91,6 +91,7 @@ def req_resp_to_transaction_parser(request, response, response_content) -> dict:
 
     if "openai.azure.com" in url and "embeddings" in url:
         transaction_params["type"] = "embedding"
+        transaction_params["prompt"] = request_content["input"]
         if response.__dict__["status_code"] > 200:
             transaction_params["error_message"] = response_content["error"]["message"]
             transaction_params["message"] = None
@@ -111,6 +112,7 @@ def req_resp_to_transaction_parser(request, response, response_content) -> dict:
 
     if "openai.azure.com" in url and "completions" in url:
         transaction_params["type"] = "chat"
+        transaction_params["prompt"] = request_content["messages"][0]["content"]
         if response.__dict__["status_code"] > 200:
             transaction_params["error_message"] = response_content["message"]
             transaction_params["message"] = None
@@ -126,6 +128,7 @@ def req_resp_to_transaction_parser(request, response, response_content) -> dict:
 
     if "api.openai.com" in url and "completions" in url:
         transaction_params["type"] = "chat"
+        transaction_params["prompt"] = request_content["messages"][0]["content"]
         if response.__dict__["status_code"] > 200:
             transaction_params["error_message"] = response_content["error"]["message"]
             transaction_params["message"] = None