Rephrase pydantic->JSON

google · Dec 23, 2024 · c5cd7a5 · c5cd7a5
1 parent 3f66d89
commit c5cd7a5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/timesketch/lib/llms/aistudio.py b/timesketch/lib/llms/aistudio.py
@@ -56,7 +56,7 @@ def generate(self, prompt: str, response_schema: Optional[dict] = None) -> str:
 
         Args:
             prompt: The prompt to use for the generation.
-            response_schema: An optional Pydantic model to define the expected response format (JSON).
+            response_schema: An optional JSON schema to define the expected response format.
 
         Returns:
             The generated text as a string (or parsed data if response_schema is provided).