feat(cli): remove tag from llm-sfn init template (#1024)

Author: woorui

cli/serverless/golang/serverless.go

@@ -60,6 +60,7 @@ func (s *GolangServerless) Init(opts *serverless.Options) error {
 		WithWantedTarget: opt.WithWantedTarget,
 		WithDescription:  opt.WithDescription,
 		WithInputSchema:  opt.WithInputSchema,
+		WithDataTags:     opt.WithDataTags,
 	}
 
 	isWasi := opts.WASI
@@ -251,6 +252,7 @@ type AppOpts struct {
 	WithWantedTarget bool
 	WithDescription  bool
 	WithInputSchema  bool
+	WithDataTags     bool
 }
 
 // ParseSrc parse app option from source code to run serverless
@@ -274,6 +276,8 @@ func ParseSrc(appFile string) (*AppOpts, error) {
 				opts.WithInputSchema = true
 			case "WantedTarget":
 				opts.WithWantedTarget = true
+			case "DataTags":
+				opts.WithDataTags = true
 			}
 		}
 	}

cli/serverless/golang/template.go

@@ -28,6 +28,8 @@ type Context struct {
 	WithDescription bool
 	// WithInputSchema determines whether to work with input schema
 	WithInputSchema bool
+	// WithDataTags determines whether to work with data tags
+	WithDataTags bool
 }
 
 // RenderTmpl renders the template with the given context

cli/serverless/golang/templates/main.tmpl

@@ -4,6 +4,12 @@ func InputSchema() any {
 }
 {{end}}
 
+{{if not .WithDataTags}}
+func DataTags() []uint32 {
+	return []uint32{}
+}
+{{end}}
+
 // Serverless main function
 func main() {
 	if err := rootCmd.Execute(); err != nil {

cli/serverless/nodejs/templates/wrapper_ts.tmpl

@@ -7,26 +7,30 @@ import {
 	_FunctionCall,
 	_createConnection,
 } from '@yomo/sfn'
-import { description, tag, handler } from '{{ .FileName }}'
+import * as app from '{{ .FileName }}'
 
+const { description, handler, tag } = app as {
+	description: string
+	handler: (args: unknown) => Promise<unknown>
+	tag?: number
+}
 const WORK_DIR = '{{ .WorkDir }}'
 const FUNCTION_NAME = '{{ .FunctionName }}'
 const SFN_FILE_PATH = '{{ .FilePath }}'
 const SOCK_PATH = join(WORK_DIR, 'sfn.sock');
 const REDUCE_TAG = 0xe001;
 
 function run() {
-	if (
-		!description ||
-		!handler ||
-		tag === undefined ||
-		tag === null
-	) {
-		throw Error('description, tags, handler signature must be exported!')
+	if (!description || !handler) {
+		throw Error('description, handler signature must be exported!')
+	}
+	let tags: number[] = []
+	if (tag) {
+		tags = [tag]
 	}
 	const tools = _genTools(FUNCTION_NAME, description, SFN_FILE_PATH)
 	const header = JSON.stringify({
-		tags: [tag],
+		tags: tags,
 		function_definition: JSON.stringify(tools, null, 2)
 	})
 	const { conn } = _createConnection(SOCK_PATH, {

cli/template/go/init_llm.tmpl

@@ -44,19 +44,9 @@ type LLMArguments struct {
 	Longitude float64 `json:"longitude" jsonschema:"description=The longitude of the city, in decimal format, range should be in (-180, 180)"`
 }
 
-// DataTags specifies the data tags to which this serverless function
-// subscribes, essential for data reception. Upon receiving data with these
-// tags, the Handler function is triggered.
-func DataTags() []uint32 {
-	return []uint32{0x30}
-}
-
 // Handler orchestrates the core processing logic of this function.
 // - ctx.ReadLLMArguments() parses LLM Function Calling Arguments (skip if none).
 // - ctx.WriteLLMResult() sends the retrieval result back to LLM.
-// - ctx.Tag() identifies the tag of the incoming data.
-// - ctx.Data() accesses the raw data.
-// - ctx.Write() forwards processed data downstream.
 func Handler(ctx serverless.Context) {
 	var p LLMArguments
 	// deserilize the arguments from llm tool_call response

cli/template/go/init_llm_test.tmpl

@@ -16,7 +16,7 @@ func TestHandler(t *testing.T) {
 	}{
 		{
 			name: "get weather",
-			ctx:  mock.NewArgumentsContext(`{"city":"New York","latitude":40.7128,"longitude":-74.0060}`, 0x33),
+			ctx:  mock.NewArgumentsContext(`{"city":"New York","latitude":40.7128,"longitude":-74.0060}`),
 			want: "The current weather in New York (40.712800,-74.006000) is sunny",
 		},
 		// TODO: add more test cases.

cli/template/node/init_llm.tmpl

@@ -1,7 +1,5 @@
 export const description = 'Get the current weather for `city_name`'
 
-export const tag = 0x33
-
 // For jsonschema in TypeScript, see: https://github.com/YousefED/typescript-json-schema
 export type Argument = {
   /**
@@ -15,6 +13,7 @@ async function getWeather(city_name: string) {
   console.log(`get weather for ${city_name} with temperature ${tempraure}°C`)
   return { city_name: city_name, temperature: tempraure }
 }
+
 export async function handler(args: Argument) {
   const result = await getWeather(args.city_name)
   return result

example/10-ai/llm-sfn-currency-converter/app.go

@@ -54,10 +54,6 @@ func Handler(ctx serverless.Context) {
 	}
 }
 
-func DataTags() []uint32 {
-	return []uint32{0x10}
-}
-
 type Rates struct {
 	Rates map[string]float64 `json:"rates"`
 }

example/10-ai/llm-sfn-get-ip-and-latency/app.go

@@ -65,7 +65,3 @@ func Handler(ctx serverless.Context) {
 
 	ctx.WriteLLMResult(val)
 }
-
-func DataTags() []uint32 {
-	return []uint32{0x10}
-}

example/10-ai/llm-sfn-get-weather/app.go

@@ -10,8 +10,6 @@ import (
 	"github.com/yomorun/yomo/serverless"
 )
 
-var tag uint32 = 0x11
-
 type Parameter struct {
 	CityName string `json:"city_name" jsonschema:"description=The name of a city to be queried"`
 }
@@ -35,7 +33,7 @@ func Handler(ctx serverless.Context) {
 		slog.Error("[sfn] ReadLLMArguments error", "err", err)
 		return
 	}
-	slog.Info("[sfn] << receive", "tag", tag, "msg", msg)
+	slog.Info("[sfn] << receive", "msg", msg)
 	data := fmt.Sprintf("[%s] temperature: %d°C", msg.CityName, rand.Intn(40))
 	time.Sleep(time.Millisecond * 300)
 	// helper ai function
@@ -50,7 +48,3 @@ func Handler(ctx serverless.Context) {
 		slog.Info("[sfn] >> write", "tag", ai.ReducerTag, "fnCall", fnCall)
 	}
 }
-
-func DataTags() []uint32 {
-	return []uint32{tag}
-}