Skip to content

Commit 3a1f463

Browse files
KavinduZoysaNipunaRanasinghe
authored andcommitted
Resolve merge conflicts
1 parent 4022ab9 commit 3a1f463

File tree

2 files changed

+24
-7
lines changed

2 files changed

+24
-7
lines changed

flow-model-generator/modules/flow-model-generator-core/src/main/java/io/ballerina/flowmodelgenerator/core/AgentsGenerator.java

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -372,8 +372,8 @@ public JsonElement genTool(JsonElement node, String toolName, String connectionN
372372
Property.CONNECTION_KEY, Property.CHECK_ERROR_KEY));
373373
keys.removeAll(ignoredKeys);
374374
List<String> paramList = new ArrayList<>();
375-
for (String key : keys) {
376-
Property property = properties.get(key);
375+
for (String k : keys) {
376+
Property property = properties.get(k);
377377
if (property == null) {
378378
continue;
379379
}
@@ -385,6 +385,10 @@ public JsonElement genTool(JsonElement node, String toolName, String connectionN
385385
continue;
386386
}
387387
}
388+
String key = k;
389+
if (k.startsWith("$")) {
390+
key = "'" + k.substring(1);
391+
}
388392
if (hasDescription) {
389393
sourceBuilder.token().parameterDoc(key, property.metadata().description());
390394
}
@@ -465,8 +469,8 @@ public JsonElement genTool(JsonElement node, String toolName, String connectionN
465469
keys.removeAll(ignoredKeys);
466470
List<String> paramList = new ArrayList<>();
467471
Set<String> pathParams = new HashSet<>();
468-
for (String key : keys) {
469-
Property property = properties.get(key);
472+
for (String k : keys) {
473+
Property property = properties.get(k);
470474
if (property == null) {
471475
continue;
472476
}
@@ -481,6 +485,10 @@ public JsonElement genTool(JsonElement node, String toolName, String connectionN
481485
continue;
482486
}
483487
}
488+
String key = k;
489+
if (k.startsWith("$")) {
490+
key = "'" + k.substring(1);
491+
}
484492
if (hasDescription) {
485493
sourceBuilder.token().parameterDoc(key, property.metadata().description());
486494
}

flow-model-generator/modules/flow-model-generator-core/src/main/java/io/ballerina/flowmodelgenerator/core/model/SourceBuilder.java

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -446,15 +446,15 @@ public SourceBuilder functionParameters(FlowNode nodeTemplate, Set<String> ignor
446446
if (firstParamAdded) {
447447
tokenBuilder.keyword(SyntaxKind.COMMA_TOKEN);
448448
}
449-
tokenBuilder.expression(prop);
449+
tokenBuilder.param(prop);
450450
} else if (kind.equals(ParameterData.Kind.INCLUDED_RECORD.name())) {
451451
if (isPropValueEmpty(prop)) {
452452
continue;
453453
}
454454
if (firstParamAdded) {
455455
tokenBuilder.keyword(SyntaxKind.COMMA_TOKEN);
456456
}
457-
tokenBuilder.expression(prop);
457+
tokenBuilder.param(prop);
458458
} else if (kind.equals(ParameterData.Kind.DEFAULTABLE.name())) {
459459
if (isPropValueEmpty(prop)) {
460460
missedDefaultValue = true;
@@ -470,7 +470,7 @@ public SourceBuilder functionParameters(FlowNode nodeTemplate, Set<String> ignor
470470
tokenBuilder.name(prop.codedata().originalName()).whiteSpace()
471471
.keyword(SyntaxKind.EQUAL_TOKEN).expression(prop);
472472
} else {
473-
tokenBuilder.expression(prop);
473+
tokenBuilder.param(prop);
474474
}
475475
} else if (kind.equals(ParameterData.Kind.INCLUDED_FIELD.name())) {
476476
if (isPropValueEmpty(prop)) {
@@ -679,6 +679,15 @@ public TokenBuilder expression(Property property) {
679679
return this;
680680
}
681681

682+
public TokenBuilder param(Property property) {
683+
String source = property.toSourceCode();
684+
if (source.startsWith("$")) {
685+
source = "'" + source.substring(1);
686+
}
687+
sb.append(source);
688+
return this;
689+
}
690+
682691
public TokenBuilder expression(String exprAsStr) {
683692
sb.append(exprAsStr);
684693
return this;

0 commit comments

Comments
 (0)