Gemini command line example

Author: mark-watson

gemini_commandline/Main.hs

@@ -0,0 +1,97 @@
+import Control.Monad.IO.Class (liftIO)
+import System.Environment (getArgs, getEnv)
+import qualified Data.Aeson as Aeson
+import Data.Aeson (FromJSON, ToJSON)
+import GHC.Generics
+import Network.HTTP.Client.TLS (tlsManagerSettings)
+import Network.HTTP.Client (newManager, httpLbs, parseRequest, Request(..), RequestBody(..), responseBody, responseStatus)
+import Network.HTTP.Types.Status (statusCode)
+import qualified Data.Text as T
+import Data.Text.Encoding (encodeUtf8)
+import qualified Data.Vector as V
+
+data GeminiRequest = GeminiRequest
+  { prompt :: String
+  } deriving (Show, Generic, ToJSON)
+
+data GeminiResponse = GeminiResponse
+  { candidates :: [Candidate]  -- Changed from choices to candidates
+  } deriving (Show, Generic, FromJSON)
+
+data Candidate = Candidate
+  { content :: Content
+  } deriving (Show, Generic, FromJSON)
+
+data Content = Content
+  { parts :: [Part]
+  } deriving (Show, Generic, FromJSON)
+
+data Part = Part
+  { text :: String
+  } deriving (Show, Generic, FromJSON, ToJSON)
+
+data PromptFeedback = PromptFeedback
+  { blockReason :: Maybe String
+  , safetyRatings :: Maybe [SafetyRating]
+  } deriving (Show, Generic, FromJSON, ToJSON)
+
+data SafetyRating = SafetyRating
+  { category :: String
+  , probability :: String
+  } deriving (Show, Generic, FromJSON, ToJSON)
+
+main :: IO ()
+main = do
+  args <- getArgs
+  case args of
+    [] -> putStrLn "Error: Please provide a prompt as a command-line argument."
+    (arg:_) -> do  --  Extract the argument directly
+      apiKey <- getEnv "GOOGLE_API_KEY"
+
+      manager <- newManager tlsManagerSettings
+
+      initialRequest <- parseRequest "https://generativelanguage.googleapis.com/v1/models/gemini-1.5-pro:generateContent"
+
+      let geminiRequestBody = Aeson.object [
+            ("contents", Aeson.Array $ V.singleton $ Aeson.object [
+                ("parts", Aeson.Array $ V.singleton $ Aeson.object [
+                    ("text", Aeson.String $ T.pack  arg)
+                    ])
+            ]),
+            ("generationConfig", Aeson.object [
+                ("temperature", Aeson.Number 0.1),
+                ("maxOutputTokens", Aeson.Number 800)
+                ])
+            ]
+            
+      let request = initialRequest
+            { requestHeaders =
+                [ ("Content-Type", "application/json")
+                , ("x-goog-api-key", encodeUtf8 $ T.pack apiKey)
+                ]
+            , method = "POST"
+            , requestBody = RequestBodyLBS $ Aeson.encode geminiRequestBody
+            }
+
+      response <- httpLbs request manager
+      
+      let responseStatus' = responseStatus response
+
+      if statusCode responseStatus' == 200
+        then do
+        let maybeGeminiResponse = Aeson.decode (responseBody response) :: Maybe GeminiResponse
+        case maybeGeminiResponse of
+          Just geminiResponse -> do
+            case candidates geminiResponse of
+              (candidate:_) -> do
+                case parts (content candidate) of
+                  (part:_) -> do  -- Changed text_ to _ since it's unused
+                    liftIO $ putStrLn $ "Response:\n\n" ++ text part
+                  [] -> do
+                    liftIO $ putStrLn "Error: No parts in content"
+              [] -> do
+                liftIO $ putStrLn "Error: No candidates in response"
+          Nothing -> do
+             liftIO $ putStrLn "Error: Failed to parse response"
+        else do
+        putStrLn $ "Error: " ++ show responseStatus'

gemini_commandline/README.md

@@ -0,0 +1,30 @@
+# Command Line Utility To Use the Google Gemini APIs
+
+This example is similar to the example in ../webchat but here we build a command line application, not a web application to use the Google Gemini LLM APIs.
+
+Example:
+
+```
+$ gemini "what is the square of pi?"
+Response:
+
+The square of pi (π) is π multiplied by itself: π².  Since π is approximately 3.14159, π² is approximately 9.8696.
+```
+
+The executable file **gemini** is on my path because I copied the executable file to my personal bin directory:
+
+```
+$ cabal build
+$ find . -name gemini
+  ... output not shown
+$ cp ./dist-newstyle/build/aarch64-osx/ghc-9.4.8/gemini-0.1.0.0/x/gemini/build/gemini/gemini ~/bin
+```
+
+If you don’t want to permanently install this example on your laptop, then just run it with cabal:
+
+```
+$ cabal run gemini "what is 11 + 23?"
+Response:
+
+11 + 23 = 34
+```

gemini_commandline/gemini.cabal

@@ -0,0 +1,45 @@
+cabal-version:      2.4
+name:              gemini
+version:           0.1.0.0
+-- synopsis:
+-- description:
+license:            MIT
+license-file:       LICENSE
+author:             Bard
+maintainer:         Bard
+-- copyright:
+-- category:
+build-type:          Simple
+extra-source-files:  CHANGELOG.md
+
+executable gemini
+  main-is:             Main.hs
+  -- other-modules:
+  -- other-extensions:
+  build-depends:       base >= 4.7 && < 5
+                     , aeson
+                     , bytestring
+                     , text
+                     , http-client
+                     , http-client-tls
+                     , http-types
+                     , directory
+                     , vector
+  if os(darwin)
+      ghc-options:
+      ld-options:
+
+  -- Language extensions used in the code
+  default-extensions:
+    OverloadedStrings
+    DeriveGeneric
+    LambdaCase
+    DeriveAnyClass
+
+  program-default-options
+  hsc2hs: --with-hsc2hs=/opt/homebrew/bin/hsc2hs
+  
+  -- Compiler flags
+  ghc-options: -Wall -O2
+
+                     

webchat/main.hs

@@ -114,6 +114,7 @@ main = do
               )
             ]
 
+
       let request2 = initialRequest
             { requestHeaders =
                 [ ("Content-Type", "application/json")
@@ -135,7 +136,7 @@ main = do
 
       let maybeGeminiResponse = Aeson.decode (responseBody response2) :: Maybe GeminiResponse
 
-      liftIO $ putStrLn $ "Parsed response: " ++ show maybeGeminiResponse  -- Debug print
+      liftIO $ putStrLn $ "Parsed response: " ++ show maybeGeminiResponse
       liftIO $ hFlush stdout
 
       case maybeGeminiResponse of