From 502d1442a1be69fb32c82ffdd35fc46278aed891 Mon Sep 17 00:00:00 2001
From: Eugene Chernyavsky <anti.eugene@gmail.com>
Date: Fri, 2 May 2025 20:50:20 +0300
Subject: [PATCH] google-genai and google.generativeai requirement from import
 (#5699)

---
 backend/parsers/windmill-parser-py-imports/src/mapping.rs | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/backend/parsers/windmill-parser-py-imports/src/mapping.rs b/backend/parsers/windmill-parser-py-imports/src/mapping.rs
index 4c7d9a7212..1bd0688951 100644
--- a/backend/parsers/windmill-parser-py-imports/src/mapping.rs
+++ b/backend/parsers/windmill-parser-py-imports/src/mapping.rs
@@ -142,6 +142,8 @@ pub static FULL_IMPORTS_MAP: PyMap = phf_map! {
     "google.cloud.dns" => "google-cloud-dns",
     "google.cloud.runtimeconfig" => "google-cloud-runtimeconfig",
     "google.cloud.iot" => "google-cloud-iot",
+    "google.generativeai" => "google-generativeai",
+    "google.genai" => "google-genai",
     // Azure
     "azure.mgmt.hybridkubernetes" => "azure-mgmt-hybridkubernetes",
     "azure.mgmt.sql" => "azure-mgmt-sql",