Output audit log fix (#148)

openshieldai · Sep 12, 2024 · 68a73b6 · 68a73b6
2 parents 7fdd981 + 9747e8f
commit 68a73b6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lib/openai/handlers.go b/lib/openai/handlers.go
@@ -590,7 +590,7 @@ func performResponseAuditLogging(r *http.Request, resp openai.ChatCompletionResp
 		return
 	}
 
-	auditLog := lib.AuditLogs(string(responseJSON), "openai_chat_completion", apiKeyId, "input", productID, r)
+	auditLog := lib.AuditLogs(string(responseJSON), "openai_chat_completion", apiKeyId, "output", productID, r)
 
 	if auditLog == nil {
 		log.Printf("Failed to create audit log")

diff --git a/rules/rule-service/rule_service/main.py b/rules/rule-service/rule_service/main.py
@@ -105,7 +105,7 @@ async def execute_plugin(rule: Rule):
     logger.debug(f"Rule engine result: match={match}")
 
     response = {"match": match, "inspection": plugin_result}
-    logger.debug(f"API response: {response}")
+    logger.debug(f"Plugin Name: {plugin_name} API response: {response}")
 
     return response