updates

Author: torrmal

flat_ai/__about__.py

@@ -1,6 +1,6 @@
 __title__ = 'flat-ai'
 __package_name__ = 'flat_ai'
-__version__ = '0.3.2'
+__version__ = '0.3.3'
 __description__ = 'F.L.A.T. (Frameworkless LLM Agent Thing) for building AI Agents'
 __email__ = 'hello@mindsdb.com'
 __author__ = 'Yours truly Jorge Torres and an LLM'

flat_ai/flat_ai.py

@@ -159,11 +159,10 @@ def _execute():
                 }
                 schema_name = schema_class.__args__[0].__name__+"Array"
             # Handle Pydantic models
-            elif isinstance(schema_class, type) and issubclass(schema_class, BaseModel):
+            else:
                 schema = schema_class.model_json_schema()
                 schema_name = schema_class.__name__
-            else:
-                raise ValueError(f"Unsupported schema type: {schema_class}")
+            
 
             messages = self._build_messages(
                 {