feat: change google api request

Author: terryyz

bigcodebench/gen/util/google_request.py

@@ -6,14 +6,13 @@
 
 def make_request(
     client: genai.GenerativeModel,
-    messages: List,
+    message: str,
     temperature: float,
     n: int,
     max_new_tokens: int = 2048,
 ) -> genai.types.GenerateContentResponse:
-    messages = [{"role": m["role"], "parts": [m["content"]]} for m in messages]
     response = client.generate_content(
-        messages,
+        [{"role": "user", "parts": [message]}],
         generation_config=genai.types.GenerationConfig(
             candidate_count=n,
             max_output_tokens=max_new_tokens,

bigcodebench/provider/google.py

@@ -35,7 +35,7 @@ def codegen(
             )
             ret = make_auto_request(
                 self.client,
-                message,
+                message=message,
                 n=num_samples,
                 temperature=self.temperature,
                 max_new_tokens=self.max_new_tokens,