add input field to embedding (#16)

newrelic · Aug 2, 2023 · 0100b4c · 0100b4c
1 parent eaaa640
commit 0100b4c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/nr_openai_observability/build_events.py b/src/nr_openai_observability/build_events.py
@@ -110,6 +110,7 @@ def build_embedding_event(response, request, response_headers, response_time):
 
     embedding = {
         "id": embedding_id,
+        "input": request.get("input")[:4095],
         "api_key_last_four_digits": f"sk-{response.api_key[-4:]}",
         "timestamp": datetime.now(),
         "response_time": int(response_time * 1000),