Skip to content

Commit fc479ca

Browse files
update function template
1 parent 10adca3 commit fc479ca

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

src/datacustomcode/templates/function/payload/entrypoint.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -60,14 +60,14 @@ def make_einstein_prediction(runtime: Runtime) -> None:
6060

6161
prediction_response = runtime.einstein_predictions.predict(prediction_request)
6262
print(
63-
f"Einstein prediction results - success: {prediction_response.is_success} \
64-
response data: {prediction_response.data}"
63+
f"Einstein prediction results - success: [{prediction_response.is_success}] \
64+
response data: {prediction_response.data}"
6565
)
6666

6767

6868
def generate_text(runtime: Runtime):
6969
builder = GenerateTextRequestBuilder()
70-
llm_request = builder.set_prompt("Hello").set_model("modelName").build()
70+
llm_request = builder.set_prompt("Generate 2 dog names").set_model("sfdc_ai__DefaultGPT52").build()
7171
llm_response = runtime.llm_gateway.generate_text(llm_request)
7272

7373
if llm_response.is_success:

0 commit comments

Comments
 (0)