Add basic sanity test to confirm that our interactions with the OpenAI API are correct

.github/workflows/go-test.yml

@@ -66,6 +66,7 @@ jobs:
     - name: Go test
       env:
         DD_API_KEY: ${{ secrets.DD_API_KEY }}
+        OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
       run: |
 
           # Once https://github.com/gotestyourself/gotestsum/pull/377 is merged, update this to pull from main rather than my fork

shared/ai/ai_test.go

@@ -0,0 +1,25 @@
+package ai
+
+import (
+	"os"
+	"strings"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+// A basic sanity test that our integration with the OpenAI API is correct and is returning reasonable results (at least for a very basic query)
+func TestLiveOpenAiApi(t *testing.T) {
+	if os.Getenv("OPENAI_API_KEY") == "" {
+		t.Skip("Skipping test since OPENAI_API_KEY is not set")
+	}
+	results, _, err := GetAiSuggestionsViaOpenAiApi("list files in the current directory", 3)
+	require.NoError(t, err)
+	resultsContainsLs := false
+	for _, result := range results {
+		if strings.Contains(result, "ls") {
+			resultsContainsLs = true
+		}
+	}
+	require.Truef(t, resultsContainsLs, "expected results=%#v to contain ls", results)
+}