Add LLM SDKs: AzureAI, Mistral, and StabilityAI

Privado-Inc · Feb 2, 2024 · e5edfdd · e5edfdd
1 parent a9a2caa
commit e5edfdd
Show file tree

Hide file tree

Showing 3 changed files with 39 additions and 0 deletions.
diff --git a/rules/sinks/third_parties/sdk/azureai/python.yml 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
@@ -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
@@ -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: