pylint

openlit · Feb 11, 2024 · 0620e6d · 0620e6d
1 parent a7a02da
commit 0620e6d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/src/dokumetry/cohere.py b/src/dokumetry/cohere.py
@@ -169,15 +169,15 @@ def stream_generator():
                 accumulated_content = ""
                 for event in original_chat(*args, **kwargs):
                     if event.event_type == "stream-start":
-                        responseId = event.generation_id
+                        response_id = event.generation_id
                     if event.event_type == "text-generation":
                         accumulated_content += event.text
                     yield event
                 end_time = time.time()
                 duration = end_time - start_time
                 prompt = kwargs.get('message')
                 data = {
-                    "llmReqId": responseId,
+                    "llmReqId": response_id,
                     "environment": environment,
                     "applicationName": application_name,
                     "sourceLanguage": "python",
@@ -242,9 +242,9 @@ def summarize_generate(*args, **kwargs):
         prompt = kwargs.get('text')
 
         data = {
+                "applicationName": application_name,
                 "llmReqId": response.id,
                 "environment": environment,
-                "applicationName": application_name,
                 "sourceLanguage": "python",
                 "endpoint": "cohere.summarize",
                 "skipResp": skip_resp,

diff --git a/src/dokumetry/openai.py b/src/dokumetry/openai.py
@@ -53,7 +53,7 @@ def stream_generator():
                         if content:
                             accumulated_content += content
                     yield chunk
-                    responseId = chunk.id
+                    response_id = chunk.id
                 end_time = time.time()
                 duration = end_time - start_time
                 message_prompt = kwargs.get('messages', "No prompt provided")
@@ -75,8 +75,8 @@ def stream_generator():
 
                 prompt = "\n".join(formatted_messages)
                 data = {
-                    "llmReqId": responseId,
                     "environment": environment,
+                    "llmReqId": response_id,
                     "applicationName": application_name,
                     "sourceLanguage": "python",
                     "endpoint": "openai.chat.completions",
@@ -175,12 +175,12 @@ def stream_generator():
                         if content:
                             accumulated_content += content
                     yield chunk
-                    responseId = chunk.id
+                    response_id = chunk.id
                 end_time = time.time()
                 duration = end_time - start_time
                 prompt = kwargs.get('prompt', "No prompt provided")
                 data = {
-                    "llmReqId": responseId,
+                    "llmReqId": response_id,
                     "environment": environment,
                     "applicationName": application_name,
                     "sourceLanguage": "python",