diff --git a/examples/getting_started/function_calling.py b/examples/getting_started/function_calling.py index da8b87b..f5ca9d5 100644 --- a/examples/getting_started/function_calling.py +++ b/examples/getting_started/function_calling.py @@ -230,9 +230,20 @@ def handle_task(self, task): # Call the function if function_name in self.functions: result = self.functions[function_name](**params) - response = f"Function result: {result}" + task.artifacts = [{ + "parts": [{ + "type": "function_response", + "name": function_name, + "content": { + "result": result + } + }] + }] else: response = f"Unknown function: {function_name}" + task.artifacts = [{ + "parts": [{"type": "text", "text": response}] + }] else: # Extract text from message content = message_data.get("content", {}) @@ -266,12 +277,12 @@ def handle_task(self, task): task.artifacts = [{ "parts": [{ "type": "function_call", - "function": "calculate", - "args": { - "operation": operation, - "a": a, - "b": b - } + "name": "calculate", + "parameters": [ + {"name": "operation", "value": operation}, + {"name": "a", "value": a}, + {"name": "b", "value": b} + ] }] }] task.status = TaskStatus(state=TaskState.COMPLETED) @@ -306,12 +317,12 @@ def handle_task(self, task): task.artifacts = [{ "parts": [{ "type": "function_call", - "function": "convert", - "args": { - "value": value, - "from_unit": from_unit, - "to_unit": to_unit - } + "name": "convert", + "parameters": [ + {"name": "value", "value": value}, + {"name": "from_unit", "value": from_unit}, + {"name": "to_unit", "value": to_unit} + ] }] }] task.status = TaskStatus(state=TaskState.COMPLETED) @@ -326,8 +337,8 @@ def handle_task(self, task): task.artifacts = [{ "parts": [{ "type": "function_call", - "function": "get_time", - "args": {} + "name": "get_time", + "parameters": [] }] }] task.status = TaskStatus(state=TaskState.COMPLETED) @@ -443,4 +454,4 @@ def handle_task(self, task): sys.exit(main()) except KeyboardInterrupt: print("\n✅ Program interrupted by user") - sys.exit(0) \ No newline at end of file + sys.exit(0)