fix: for function call, role should be model.

PiperOrigin-RevId: 613639813

Author: yyyu-google

README.md

@@ -259,7 +259,7 @@ async function functionCallingGenerateContentStream() {
     contents: [
       {role: 'user', parts: [{text: 'What is the weather in Boston?'}]},
       {role: 'model', parts: [{functionCall: {name: 'get_current_weather', args: {'location': 'Boston'}}}]},
-      {role: 'function', parts: functionResponseParts}
+      {role: 'model', parts: functionResponseParts}
     ],
     tools: functionDeclarations,
   };

src/models/chat_session.ts

@@ -407,7 +407,7 @@ function assignRoleToPartsAndValidateSendMessageRequest(
   parts: Array<Part>
 ): Content[] {
   const userContent: Content = {role: constants.USER_ROLE, parts: []};
-  const functionContent: Content = {role: constants.FUNCTION_ROLE, parts: []};
+  const functionContent: Content = {role: constants.MODEL_ROLE, parts: []};
   let hasUserContent = false;
   let hasFunctionContent = false;
   for (const part of parts) {

src/util/constants.ts

@@ -18,7 +18,6 @@ export const GENERATE_CONTENT_METHOD = 'generateContent';
 export const STREAMING_GENERATE_CONTENT_METHOD = 'streamGenerateContent';
 export const USER_ROLE = 'user';
 export const MODEL_ROLE = 'model';
-export const FUNCTION_ROLE = 'function';
 const USER_AGENT_PRODUCT = 'model-builder';
 const CLIENT_LIBRARY_VERSION = '0.5.0'; // x-release-please-version
 const CLIENT_LIBRARY_LANGUAGE = `grpc-node/${CLIENT_LIBRARY_VERSION}`;

system_test/end_to_end_sample_test.ts

@@ -367,7 +367,7 @@ describe('generateContentStream', () => {
       contents: [
         {role: 'user', parts: [{text: 'What is the weather in Boston?'}]},
         {role: 'model', parts: FUNCTION_CALL},
-        {role: 'function', parts: FUNCTION_RESPONSE_PART},
+        {role: 'model', parts: FUNCTION_RESPONSE_PART},
       ],
       tools: TOOLS_WITH_FUNCTION_DECLARATION,
     };
@@ -387,7 +387,7 @@ describe('generateContentStream', () => {
       contents: [
         {role: 'user', parts: [{text: 'What is the weather in Boston?'}]},
         {role: 'model', parts: FUNCTION_CALL},
-        {role: 'function', parts: FUNCTION_RESPONSE_PART},
+        {role: 'model', parts: FUNCTION_RESPONSE_PART},
       ],
       tools: TOOLS_WITH_FUNCTION_DECLARATION,
     };
@@ -534,7 +534,7 @@ describe('generateContent', () => {
       contents: [
         {role: 'user', parts: [{text: 'What is the weather in Boston?'}]},
         {role: 'model', parts: FUNCTION_CALL},
-        {role: 'function', parts: FUNCTION_RESPONSE_PART},
+        {role: 'model', parts: FUNCTION_RESPONSE_PART},
       ],
       tools: TOOLS_WITH_FUNCTION_DECLARATION,
     };
@@ -552,7 +552,7 @@ describe('generateContent', () => {
       contents: [
         {role: 'user', parts: [{text: 'What is the weather in Boston?'}]},
         {role: 'model', parts: FUNCTION_CALL},
-        {role: 'function', parts: FUNCTION_RESPONSE_PART},
+        {role: 'model', parts: FUNCTION_RESPONSE_PART},
       ],
       tools: TOOLS_WITH_FUNCTION_DECLARATION,
     };