check for openapi better

Signed-off-by: Grant Linville <[email protected]>

Author: g-linville

pkg/loader/loader.go

@@ -20,6 +20,7 @@ import (
 	"github.com/gptscript-ai/gptscript/pkg/engine"
 	"github.com/gptscript-ai/gptscript/pkg/parser"
 	"github.com/gptscript-ai/gptscript/pkg/types"
+	"gopkg.in/yaml.v3"
 )
 
 type source struct {
@@ -128,9 +129,7 @@ func readTool(ctx context.Context, prg *types.Program, base *source, targetToolN
 	}
 
 	var tools []types.Tool
-	if strings.HasSuffix(base.Name, ".json") ||
-		strings.HasSuffix(base.Name, ".yaml") ||
-		strings.HasSuffix(base.Name, ".yml") {
+	if isOpenAPI(data) {
 		if t, err := openapi3.NewLoader().LoadFromData(data); err == nil {
 			tools, err = getOpenAPITools(t)
 			if err != nil {
@@ -317,3 +316,16 @@ func SplitToolRef(targetToolName string) (toolName, subTool string) {
 	}
 	return
 }
+
+func isOpenAPI(data []byte) bool {
+	var version struct {
+		OpenAPI string `json:"openapi" yaml:"openapi"`
+	}
+
+	if err := json.Unmarshal(data, &version); err != nil {
+		if err := yaml.Unmarshal(data, &version); err != nil {
+			return false
+		}
+	}
+	return strings.HasPrefix(version.OpenAPI, "3.")
+}