From 16403cec663f2ba7507c8c2177a05f3b073324b4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jochen=20Kirst=C3=A4tter?=
 <7329802+jochenkirstaetter@users.noreply.github.com>
Date: Thu, 21 Mar 2024 16:24:49 +0400
Subject: [PATCH] add GoogleAI type #3

---
 src/Mscc.GenerativeAI/GoogleAI.cs | 50 +++++++++++++++++++++++++++++++
 1 file changed, 50 insertions(+)
 create mode 100644 src/Mscc.GenerativeAI/GoogleAI.cs

diff --git a/src/Mscc.GenerativeAI/GoogleAI.cs b/src/Mscc.GenerativeAI/GoogleAI.cs
new file mode 100644
index 0000000..c03c978
--- /dev/null
+++ b/src/Mscc.GenerativeAI/GoogleAI.cs
@@ -0,0 +1,50 @@
+#if NET472_OR_GREATER || NETSTANDARD2_0
+using System;
+using System.Collections.Generic;
+#endif
+
+namespace Mscc.GenerativeAI
+{
+    /// <summary>
+    /// Entry point to access Gemini API running in Google AI.
+    /// </summary>
+    public sealed class GoogleAI
+    {
+        private readonly string? _apiKey;
+
+        /// <summary>
+        /// Default constructor attempts to read environment variables and
+        /// sets default values, if available
+        /// </summary>
+        private GoogleAI()
+        {
+            _apiKey = Environment.GetEnvironmentVariable("GOOGLE_API_KEY");
+        }
+        
+        /// <summary>
+        /// Constructor to initialize access to Google AI Gemini API.
+        /// </summary>
+        /// <param name="apiKey">Identifier of the Google Cloud project</param>
+        public GoogleAI(string apiKey) : this()
+        {
+            _apiKey ??= apiKey;
+        }
+
+        /// <summary>
+        /// Create a generative model on Google AI to use.
+        /// </summary>
+        /// <param name="model">Model to use (default: "gemini-1.0-pro")</param>
+        /// <param name="generationConfig">Optional. Configuration options for model generation and outputs.</param>
+        /// <param name="safetySettings">Optional. A list of unique SafetySetting instances for blocking unsafe content.</param>
+        /// <exception cref="ArgumentNullException"></exception>
+        /// <returns></returns>
+        public GenerativeModel GenerativeModel(string model = Model.Gemini10Pro,
+            GenerationConfig? generationConfig = null,
+            List<SafetySetting>? safetySettings = null)
+        {
+            if (_apiKey is null) throw new ArgumentNullException(nameof(_apiKey));
+            
+            return new GenerativeModel(_apiKey, model, generationConfig, safetySettings);
+        }
+    }
+}