diff --git a/shared/ai/ai_test.go b/shared/ai/ai_test.go
index 17499e9..7576588 100644
--- a/shared/ai/ai_test.go
+++ b/shared/ai/ai_test.go
@@ -13,7 +13,7 @@ import (
 // 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") == "" {
-		if testutils.IsGithubAction() {
+		if testutils.IsGithubAction() && testutils.GetCurrentGitBranch(t) == testutils.DefaultGitBranchName {
 			t.Fatal("OPENAI_API_KEY is not set, cannot run TestLiveOpenAiApi")
 		} else {
 			t.Skip("Skipping test since OPENAI_API_KEY is not set")
diff --git a/shared/testutils/testutils.go b/shared/testutils/testutils.go
index adad836..bc186d5 100644
--- a/shared/testutils/testutils.go
+++ b/shared/testutils/testutils.go
@@ -425,3 +425,17 @@ func GetOsVersion(t *testing.T) string {
 	version := unix.ByteSliceToString(uts.Release[:])
 	return strings.Split(version, ".")[0]
 }
+
+const DefaultGitBranchName = "master"
+
+func GetCurrentGitBranch(t *testing.T) string {
+	cmd := exec.Command("git", "symbolic-ref", "--short", "HEAD")
+	var out bytes.Buffer
+	cmd.Stdout = &out
+
+	if err := cmd.Run(); err != nil {
+		t.Fatalf("failed to get current git branch: %v", err)
+	}
+
+	return strings.TrimSpace(out.String())
+}