Fix return type.

google · Jul 24, 2024 · 9f41f61 · 9f41f61
1 parent 42efe6d
commit 9f41f61
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llm_toolkit/models.py b/llm_toolkit/models.py
@@ -281,7 +281,7 @@ def prompt_type(self) -> type[prompts.Prompt]:
     """Returns the expected prompt type."""
     return prompts.ClaudePrompt
 
-  def get_model(self) -> Any:
+  def get_model(self) -> str:
     return self._vertex_ai_model
 
   # ============================== Generation ============================== #