Add error message when cannot convert to pydantic

chainlite/llm_generate.py

@@ -234,7 +234,14 @@ async def postprocess_generations(input_: AsyncIterator[str]) -> AsyncIterator[s
 
 @chain
 def string_to_pydantic_object(llm_output: str, pydantic_class: BaseModel):
-    return pydantic_class.model_validate(json.loads(llm_output))
+    try:
+        return pydantic_class.model_validate(json.loads(llm_output))
+    except Exception as e:
+        logger.exception(
+            f"Error decoding JSON: {e}. This might be resolved by increasing `max_tokens`"
+        )
+        logger.error(f"LLM output: {llm_output}")
+        return None
 
 
 def is_list(obj):

setup.py

@@ -2,7 +2,7 @@
 
 setup(
     name="chainlite",
-    version="0.2.1",
+    version="0.2.2",
     author="Sina Semnani",
     author_email="[email protected]",
     description="A Python package that uses LangChain and LiteLLM to call large language model APIs easily",