improved OpenAI API example

Author: mark-watson

OpenAiApiClient/GenText.hs

@@ -1,51 +1,49 @@
--- Example from https://github.com/agrafix/openai-hs/tree/main/openai-hs
--- code by Alexander Thiemann
-
 {-# LANGUAGE OverloadedStrings #-}
 import OpenAI.Client
 
 import Network.HTTP.Client
-
 import Network.HTTP.Client.TLS
 import System.Environment (getEnv)
 import qualified Data.Text as T
 import Data.Maybe (fromMaybe)
 
-request :: ChatCompletionRequest
-request = ChatCompletionRequest 
-         { chcrModel = ModelId "gpt-4o"
-         , chcrMessages = 
-            [ChatMessage { chmContent = Just "Write a hello world program in Haskell"
-                         , chmRole = "user"
-                         , chmFunctionCall = Nothing
-                         , chmName = Nothing
-                         }
-            ]
-         , chcrFunctions = Nothing
-         , chcrTemperature = Nothing
-         , chcrTopP = Nothing
-         , chcrN = Nothing
-         , chcrStream = Nothing
-         , chcrStop = Nothing
-         , chcrMaxTokens = Nothing
-         , chcrPresencePenalty = Nothing
-         , chcrFrequencyPenalty = Nothing
-         , chcrLogitBias = Nothing
-         , chcrUser = Nothing
-         }
+completionRequestToString :: String -> IO String
+completionRequestToString prompt = do
+    manager <- newManager tlsManagerSettings
+    apiKey <- T.pack <$> getEnv "OPENAI_KEY"
+    let client = makeOpenAIClient apiKey manager 4
+    let request = ChatCompletionRequest
+                 { chcrModel = ModelId "gpt-4o"
+                 , chcrMessages =
+                    [ ChatMessage
+                        { chmContent = Just (T.pack prompt)
+                        , chmRole = "user"
+                        , chmFunctionCall = Nothing
+                        , chmName = Nothing
+                        }
+                    ]
+                 , chcrFunctions = Nothing
+                 , chcrTemperature = Nothing
+                 , chcrTopP = Nothing
+                 , chcrN = Nothing
+                 , chcrStream = Nothing
+                 , chcrStop = Nothing
+                 , chcrMaxTokens = Nothing
+                 , chcrPresencePenalty = Nothing
+                 , chcrFrequencyPenalty = Nothing
+                 , chcrLogitBias = Nothing
+                 , chcrUser = Nothing
+                 }
+    result <- completeChat client request
+    case result of
+        Left failure -> return (show failure)
+        Right success ->
+            case chrChoices success of
+                (ChatChoice {chchMessage = ChatMessage {chmContent = content}} : _) ->
+                    return $ fromMaybe "No content" $ T.unpack <$> content
+                _ -> return "No choices returned"
 
 main :: IO ()
 main = do
-  manager <- newManager tlsManagerSettings
-  apiKey <- T.pack <$> (getEnv "OPENAI_KEY")
-  -- create a openai client that automatically retries up to 4 times on network
-  -- errors
-  let client = makeOpenAIClient apiKey manager 4
-  result <- completeChat client request        
-  case result of
-    Left failure -> print failure
-    Right success ->  -- print $ chrChoices success
-      case chrChoices success of
-        (ChatChoice {chchMessage = ChatMessage {chmContent = content}}:_) ->
-          putStrLn $ fromMaybe "No content" $ T.unpack <$> content
-        _ -> putStrLn "No choices returned"
+    response <- completionRequestToString "Write a hello world program in Haskell"
+    putStrLn response

OpenAiApiClient/OpenAiApiClient.cabal

@@ -17,9 +17,3 @@ executable GenText
   default-language:    Haskell2010
   build-depends:       base >= 4.7 && < 5, mtl >= 2.2.2, text, http-client >= 0.7.13.1, openai-hs, http-client-tls
 
-executable Completion
-  hs-source-dirs:      .
-  main-is:             GenText.hs
-  default-language:    Haskell2010
-  build-depends:       base >= 4.7 && < 5, mtl >= 2.2.2, text, http-client >= 0.7.13.1, openai-hs, http-client-tls, http-conduit, aeson
-

OpenAiApiClient/README.md

@@ -78,4 +78,12 @@ Alternatively, you can run the Haskell code without compiling by using an interp
    ```
 
 This should also produce the output "Hello, World!" in the terminal.
-```
+```
+
+
+## Run using Replit.com, Nix, Cabal
+
+    cabal build
+    cabal run
+
+Note: I had to manually install:  cabal install cpphs