From e5edfddbdf25a4618b3d4620827be52f1983c23f Mon Sep 17 00:00:00 2001
From: Pawan Kartik <pawan.kartik@privado.ai>
Date: Fri, 2 Feb 2024 09:23:22 +0000
Subject: [PATCH] Add LLM SDKs: AzureAI, Mistral, and StabilityAI

---
 rules/sinks/third_parties/sdk/azureai/python.yml    | 13 +++++++++++++
 .../sinks/third_parties/sdk/mistral/javascript.yml  | 13 +++++++++++++
 .../sinks/third_parties/sdk/stabilityai/python.yml  | 13 +++++++++++++
 3 files changed, 39 insertions(+)
 create mode 100644 rules/sinks/third_parties/sdk/azureai/python.yml
 create mode 100644 rules/sinks/third_parties/sdk/mistral/javascript.yml
 create mode 100644 rules/sinks/third_parties/sdk/stabilityai/python.yml

diff --git a/rules/sinks/third_parties/sdk/azureai/python.yml b/rules/sinks/third_parties/sdk/azureai/python.yml
new file mode 100644
index 00000000..82aba4f6
--- /dev/null
+++ b/rules/sinks/third_parties/sdk/azureai/python.yml
@@ -0,0 +1,13 @@
+
+#  Sink rule for ThirdParty SDK
+#  The id follows a format : "ThirdParties.SDK.<THIRD_PARTY_ORGANISATION>.<SUB_ORGANISATION_IF_APPLICABLE>"
+
+sinks:
+
+  - id: ThirdParties.SDK.AzureAI.AzureOpenAI
+    name: AzureAI
+    domains:
+      - "azure.com"
+    patterns:
+      - "(?i)(azure.AzureOpenAI).*"
+    tags:
diff --git a/rules/sinks/third_parties/sdk/mistral/javascript.yml b/rules/sinks/third_parties/sdk/mistral/javascript.yml
new file mode 100644
index 00000000..033e33b5
--- /dev/null
+++ b/rules/sinks/third_parties/sdk/mistral/javascript.yml
@@ -0,0 +1,13 @@
+
+#  Sink rule for ThirdParty SDK
+#  The id follows a format : "ThirdParties.SDK.<THIRD_PARTY_ORGANISATION>.<SUB_ORGANISATION_IF_APPLICABLE>"
+
+sinks:
+
+  - id: ThirdParties.SDK.Mistral
+    name: Mistral
+    domains:
+      - "sdk.vercel.ai"
+    patterns:
+      - "@mistralai-ai\\/mistralai"
+    tags:
diff --git a/rules/sinks/third_parties/sdk/stabilityai/python.yml b/rules/sinks/third_parties/sdk/stabilityai/python.yml
new file mode 100644
index 00000000..07edcb5e
--- /dev/null
+++ b/rules/sinks/third_parties/sdk/stabilityai/python.yml
@@ -0,0 +1,13 @@
+
+#  Sink rule for ThirdParty SDK
+#  The id follows a format : "ThirdParties.SDK.<THIRD_PARTY_ORGANISATION>.<SUB_ORGANISATION_IF_APPLICABLE>"
+
+sinks:
+
+  - id: ThirdParties.SDK.StabilityAI
+    name: StabilityAI
+    domains:
+      - "https://platform.stability.ai/"
+    patterns:
+      - "(?i)(stability_sdk.interfaces.gooseai).*"
+    tags: