🔖 kwargs dict type example

shroominic · Feb 12, 2024 · 195a77d · 195a77d
1 parent bdab3d8
commit 195a77d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/examples/dict_type.py b/examples/dict_type.py
@@ -5,7 +5,12 @@
 settings.llm = "llamacpp/openchat-3.5-0106:Q3_K_M"
 
 
-def get_kwargs(user_query: str) -> dict:
+class Output(BaseModel):
+    plot: str
+    kwargs: dict
+
+
+def get_kwargs(user_query: str) -> Output:
     """
     Translate the user query to a dictionary of kwargs
     """