fix: tests

langgenius · Dec 24, 2024 · 24c9665 · 24c9665
1 parent 1a694a9
commit 24c9665
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/internal/types/entities/plugin_entities/tool_declaration.go b/internal/types/entities/plugin_entities/tool_declaration.go
@@ -110,7 +110,7 @@ type ToolDeclaration struct {
 
 func isJSONSchema(fl validator.FieldLevel) bool {
 	// get schema from interface
-	schemaMap, ok := fl.Field().Interface().(map[string]any)
+	schemaMap, ok := fl.Field().Interface().(ToolOutputSchema)
 	if !ok {
 		return false
 	}

diff --git a/internal/types/entities/plugin_entities/tool_declaration_test.go b/internal/types/entities/plugin_entities/tool_declaration_test.go
@@ -1178,7 +1178,7 @@ func TestParameterScope_Validate(t *testing.T) {
 
 func TestInvalidJSONSchemaToolProvider_Validate(t *testing.T) {
 	type Test struct {
-		Text map[string]any `json:"text" validate:"json_schema"`
+		Text ToolOutputSchema `json:"text" validate:"json_schema"`
 	}
 
 	data := parser.MarshalJsonBytes(Test{