Skip to content

Commit 97a87b7

Browse files
authored
Merge pull request #140 from schemacrawler/mcp-json
Use library to create inputSchema
2 parents 0bdc4a0 + c5dac9a commit 97a87b7

File tree

4 files changed

+9
-129
lines changed

4 files changed

+9
-129
lines changed

schemacrawler-ai-core/pom.xml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,11 @@
6565
<groupId>tools.jackson.core</groupId>
6666
<artifactId>jackson-databind</artifactId>
6767
</dependency>
68+
<dependency>
69+
<groupId>us.fatehi</groupId>
70+
<artifactId>mcp-json-schema</artifactId>
71+
<version>1.0.1</version>
72+
</dependency>
6873

6974
<!-- Test dependencies -->
7075
<dependency>

schemacrawler-ai-core/src/main/java/schemacrawler/tools/ai/tools/JsonSchemaGenerator.java

Lines changed: 0 additions & 125 deletions
This file was deleted.

schemacrawler-ai-core/src/test/java/schemacrawler/tools/ai/tools/test/JsonSchemaGeneratorTest.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@
2323
import schemacrawler.tools.ai.functions.ListFunctionParameters;
2424
import schemacrawler.tools.ai.functions.TableSampleFunctionParameters;
2525
import schemacrawler.tools.ai.tools.FunctionParameters;
26-
import schemacrawler.tools.ai.tools.JsonSchemaGenerator;
2726
import tools.jackson.databind.JsonNode;
27+
import us.fatehi.mcp_json_schema.McpJsonSchemaUtility;
2828
import us.fatehi.test.utility.TestWriter;
2929

3030
public class JsonSchemaGeneratorTest {
@@ -45,7 +45,7 @@ public void functionParameters() {
4545
final TestWriter testout = new TestWriter();
4646
try (final TestWriter out = testout) {
4747
for (final Class<?> parametersClass : functionParameters) {
48-
final JsonNode schemaNode = JsonSchemaGenerator.generateSchema(parametersClass);
48+
final JsonNode schemaNode = McpJsonSchemaUtility.generateJsonSchema(parametersClass);
4949
// final JsonNode schemaNode =
5050
// ToolUtility.extractParametersSchemaNode(parametersClass);
5151

schemacrawler-ai-mcpserver/src/main/java/schemacrawler/tools/ai/mcpserver/server/ToolHelper.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,8 +33,8 @@
3333
import schemacrawler.tools.ai.tools.FunctionDefinition;
3434
import schemacrawler.tools.ai.tools.FunctionParameters;
3535
import schemacrawler.tools.ai.tools.FunctionReturn;
36-
import schemacrawler.tools.ai.tools.JsonSchemaGenerator;
3736
import tools.jackson.databind.JsonNode;
37+
import us.fatehi.mcp_json_schema.McpJsonSchemaUtility;
3838

3939
@Component
4040
public class ToolHelper {
@@ -103,7 +103,7 @@ private <P extends FunctionParameters> Tool toTool(
103103
final String toolName = functionDefinition.getName();
104104

105105
final Class<P> parametersClass = functionDefinition.getParametersClass();
106-
final JsonNode parametersSchemaNode = JsonSchemaGenerator.generateSchema(parametersClass);
106+
final JsonNode parametersSchemaNode = McpJsonSchemaUtility.generateJsonSchema(parametersClass);
107107

108108
final McpSchema.Tool tool =
109109
McpSchema.Tool.builder()

0 commit comments

Comments
 (0)