chore: Formatting

sdk-go/inferable_test.go

@@ -199,6 +199,7 @@ func TestGetSchema(t *testing.T) {
         "TestFunc": {
             "name": "TestFunc",
             "input": {
+                "$schema": "https://json-schema.org/draft/2020-12/schema",
                 "type": "object",
                 "properties": {
                     "a": {"type": "integer"},
@@ -210,6 +211,7 @@ func TestGetSchema(t *testing.T) {
         "TestFunc2": {
             "name": "TestFunc2",
             "input": {
+                "$schema": "https://json-schema.org/draft/2020-12/schema",
                 "type": "object",
                 "properties": {
                     "c": {

sdk-go/main_test.go

@@ -62,12 +62,9 @@ func TestInferableFunctions(t *testing.T) {
 		t.Fatalf("Error registering reverse function: %v", err)
 	}
 
-	jsonDef, err := inferableInstance.toJSONDefinition()
 	if err != nil {
 		t.Fatalf("Error generating JSON definition: %v", err)
 	}
-	t.Logf("JSON Definition:\n%s\n", string(jsonDef))
-
 	t.Run("Echo Function", func(t *testing.T) {
 		testInput := EchoInput{Input: "Hello, Inferable!"}
 		result, err := inferableInstance.callFunc("string_operations", "echo", testInput)

sdk-go/service.go

@@ -106,18 +106,18 @@ func (s *service) RegisterFunc(fn Function) (*FunctionReference, error) {
 	}
 	argType := fnType.In(0)
 
-  // Set the argument type to the referenced type
-  if argType.Kind() == reflect.Ptr {
-    argType = argType.Elem()
-  }
-
-  if argType.Kind() != reflect.Struct {
-    return nil, fmt.Errorf("function '%s' first argument must be a struct or a pointer to a struct", fn.Name)
-  }
-
-  // Get the schema for the input struct
-  reflector := jsonschema.Reflector{DoNotReference: true}
-  schema := reflector.Reflect(reflect.New(argType).Interface())
+	// Set the argument type to the referenced type
+	if argType.Kind() == reflect.Ptr {
+		argType = argType.Elem()
+	}
+
+	if argType.Kind() != reflect.Struct {
+		return nil, fmt.Errorf("function '%s' first argument must be a struct or a pointer to a struct", fn.Name)
+	}
+
+	// Get the schema for the input struct
+	reflector := jsonschema.Reflector{DoNotReference: true, Anonymous: true}
+	schema := reflector.Reflect(reflect.New(argType).Interface())
 
 	if schema == nil {
 		return nil, fmt.Errorf("failed to get schema for function '%s'", fn.Name)