using json_schema

Author: torrmal

flat_ai/__about__.py

@@ -1,6 +1,6 @@
 __title__ = 'flat-ai'
 __package_name__ = 'flat_ai'
-__version__ = '0.3.0'
+__version__ = '0.3.1'
 __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

@@ -202,25 +202,19 @@ def _execute():
                 **kwargs,
             )
 
-            if isinstance(self.client, MyOpenAI):
-                response = self.client.chat.completions.create(
-                    model=self.model,
-                    response_format={
-                        "type": "json_schema",
-                        "json_schema": {
-                            "name": schema_class.__name__,
-                            "schema": schema,
-                        },
+            
+            response = self.client.chat.completions.create(
+                model=self.model,
+                response_format={
+                    "type": "json_schema",
+                    "json_schema": {
+                        "name": schema_class.__name__,
+                        "schema": schema,
                     },
-                    messages=messages,
-                )
-            else:
-                response = self.client.chat.completions.create(
-                    model=self.model,
-                    response_format={"type": "json_object", "schema": schema},
-                    messages=messages,
-                )
-
+                },
+                messages=messages,
+            )
+            
             result = json.loads(response.choices[0].message.content)
 
             # Handle list of Pydantic models