From c5cd7a553fc8f29f530484110b126f23507cc4aa Mon Sep 17 00:00:00 2001
From: itsmvd <mvd@google.com>
Date: Mon, 23 Dec 2024 10:18:04 +0000
Subject: [PATCH] Rephrase pydantic->JSON

---
 timesketch/lib/llms/aistudio.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/timesketch/lib/llms/aistudio.py b/timesketch/lib/llms/aistudio.py
index 9e60d87666..0f00ea9be9 100644
--- 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).