fix: google-genai version

Author: EYH0602

.github/workflows/mypy.yml

@@ -17,6 +17,8 @@ jobs:
       - name: Set up Python
         run: uv sync
 
+      - name: install mypy
+        run: uv pip install mypy
+
       - name: Type Check Source Code
-        run: uv run mypy src
-      
+        run: mypy src

.github/workflows/pylint.yml

@@ -16,6 +16,9 @@ jobs:
 
       - name: Set up Python
         run: uv sync
+      
+      - name: Install Pylint
+        run: uv pip install pylint
 
       - name: Lint Source Code
-        run: uv run pylint src
+        run: pylint src

pyproject.toml

@@ -5,22 +5,21 @@ description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.12"
 dependencies = [
-    "anthropic>=0.49.0",
+    "anthropic==0.49.0",
     "dacite>=1.8.1",
     "deprecated>=1.2.18",
     "dotenv>=0.9.9",
     "fire==0.5.0",
     "funcy==2.0",
     "funcy-chain==0.2.0",
-    "google-genai>=1.11.0",
+    "google-genai==1.31.0",
     "hypothesis>=6.98.6",
     "markdown-to-json==2.1.2",
     "matplotlib>=3.8.3",
     "numpy>=1.26.4",
-    "ollama>=0.5.3",
-    "openai>=1.99.9",
+    "ollama==0.5.3",
+    "openai==1.99.9",
     "pathos>=0.3.3",
-    "pylint>=3.3.6",
     "pytest>=8.0.0",
     "python-dotenv==1.0.1",
     "requests==2.32.3",

src/tfbench/lm/_google.py

@@ -56,7 +56,7 @@ def _gen(self, prompt: str) -> LMAnswer:
 
         response = self.client.models.generate_content(
             model=self.model_name,
-            contents=[prompt],
+            contents=prompt,
             config=GenerateContentConfig(
                 system_instruction=[self.instruction],
             ),
@@ -90,7 +90,7 @@ def _gen(self, prompt: str) -> LMAnswer:
 
         response = self.client.models.generate_content(
             model=self.model_name,
-            contents=[prompt],
+            contents=prompt,
             config=GenerateContentConfig(
                 system_instruction=[self.instruction],
                 thinking_config=ThinkingConfig(