mirror of
https://github.com/hanzoai/go-openai.git
synced 2026-10-30 14:51:51 +00:00
zeekayandHanzo Dev
88762272c3
The fork kept the upstream module path (module github.com/sashabaranov/ go-openai), so every consumer could only reach it through a "replace github.com/sashabaranov/go-openai => github.com/hanzoai/go-openai" directive — and Go replace directives do NOT propagate to consumers, so each downstream repo had to copy the same replace or fail to compile against fork-only fields (e.g. ReasoningContent). That is the opposite of one-and- only-one-way. Declare the fork's OWN module path so it can be required directly and inherited transitively. Package name stays 'openai' (import identifier unchanged); only the module path + internal self-imports (/internal, /jsonschema) are renamed. Upstream issue-URL comments are left intact as attribution. Builds + vets clean. Co-authored-by: Hanzo Dev <[email protected]>
212 lines
4.3 KiB
Go
212 lines
4.3 KiB
Go
package jsonschema_test
|
|
|
|
import (
|
|
"encoding/json"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/hanzoai/go-openai/jsonschema"
|
|
)
|
|
|
|
func TestDefinition_MarshalJSON(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
def jsonschema.Definition
|
|
want string
|
|
}{
|
|
{
|
|
name: "Test with empty Definition",
|
|
def: jsonschema.Definition{},
|
|
want: `{"properties":{}}`,
|
|
},
|
|
{
|
|
name: "Test with Definition properties set",
|
|
def: jsonschema.Definition{
|
|
Type: jsonschema.String,
|
|
Description: "A string type",
|
|
Properties: map[string]jsonschema.Definition{
|
|
"name": {
|
|
Type: jsonschema.String,
|
|
},
|
|
},
|
|
},
|
|
want: `{
|
|
"type":"string",
|
|
"description":"A string type",
|
|
"properties":{
|
|
"name":{
|
|
"type":"string",
|
|
"properties":{}
|
|
}
|
|
}
|
|
}`,
|
|
},
|
|
{
|
|
name: "Test with nested Definition properties",
|
|
def: jsonschema.Definition{
|
|
Type: jsonschema.Object,
|
|
Properties: map[string]jsonschema.Definition{
|
|
"user": {
|
|
Type: jsonschema.Object,
|
|
Properties: map[string]jsonschema.Definition{
|
|
"name": {
|
|
Type: jsonschema.String,
|
|
},
|
|
"age": {
|
|
Type: jsonschema.Integer,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
want: `{
|
|
"type":"object",
|
|
"properties":{
|
|
"user":{
|
|
"type":"object",
|
|
"properties":{
|
|
"name":{
|
|
"type":"string",
|
|
"properties":{}
|
|
},
|
|
"age":{
|
|
"type":"integer",
|
|
"properties":{}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}`,
|
|
},
|
|
{
|
|
name: "Test with complex nested Definition",
|
|
def: jsonschema.Definition{
|
|
Type: jsonschema.Object,
|
|
Properties: map[string]jsonschema.Definition{
|
|
"user": {
|
|
Type: jsonschema.Object,
|
|
Properties: map[string]jsonschema.Definition{
|
|
"name": {
|
|
Type: jsonschema.String,
|
|
},
|
|
"age": {
|
|
Type: jsonschema.Integer,
|
|
},
|
|
"address": {
|
|
Type: jsonschema.Object,
|
|
Properties: map[string]jsonschema.Definition{
|
|
"city": {
|
|
Type: jsonschema.String,
|
|
},
|
|
"country": {
|
|
Type: jsonschema.String,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
want: `{
|
|
"type":"object",
|
|
"properties":{
|
|
"user":{
|
|
"type":"object",
|
|
"properties":{
|
|
"name":{
|
|
"type":"string",
|
|
"properties":{}
|
|
},
|
|
"age":{
|
|
"type":"integer",
|
|
"properties":{}
|
|
},
|
|
"address":{
|
|
"type":"object",
|
|
"properties":{
|
|
"city":{
|
|
"type":"string",
|
|
"properties":{}
|
|
},
|
|
"country":{
|
|
"type":"string",
|
|
"properties":{}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}`,
|
|
},
|
|
{
|
|
name: "Test with Array type Definition",
|
|
def: jsonschema.Definition{
|
|
Type: jsonschema.Array,
|
|
Items: &jsonschema.Definition{
|
|
Type: jsonschema.String,
|
|
},
|
|
Properties: map[string]jsonschema.Definition{
|
|
"name": {
|
|
Type: jsonschema.String,
|
|
},
|
|
},
|
|
},
|
|
want: `{
|
|
"type":"array",
|
|
"items":{
|
|
"type":"string",
|
|
"properties":{
|
|
|
|
}
|
|
},
|
|
"properties":{
|
|
"name":{
|
|
"type":"string",
|
|
"properties":{}
|
|
}
|
|
}
|
|
}`,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
wantBytes := []byte(tt.want)
|
|
var want map[string]interface{}
|
|
err := json.Unmarshal(wantBytes, &want)
|
|
if err != nil {
|
|
t.Errorf("Failed to Unmarshal JSON: error = %v", err)
|
|
return
|
|
}
|
|
|
|
got := structToMap(t, tt.def)
|
|
gotPtr := structToMap(t, &tt.def)
|
|
|
|
if !reflect.DeepEqual(got, want) {
|
|
t.Errorf("MarshalJSON() got = %v, want %v", got, want)
|
|
}
|
|
if !reflect.DeepEqual(gotPtr, want) {
|
|
t.Errorf("MarshalJSON() gotPtr = %v, want %v", gotPtr, want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func structToMap(t *testing.T, v any) map[string]any {
|
|
t.Helper()
|
|
gotBytes, err := json.Marshal(v)
|
|
if err != nil {
|
|
t.Errorf("Failed to Marshal JSON: error = %v", err)
|
|
return nil
|
|
}
|
|
|
|
var got map[string]interface{}
|
|
err = json.Unmarshal(gotBytes, &got)
|
|
if err != nil {
|
|
t.Errorf("Failed to Unmarshal JSON: error = %v", err)
|
|
return nil
|
|
}
|
|
return got
|
|
}
|