chore: work with schemav2 but still write files for debugging

Author: jbarreau

src/agent_toolkit/forestadmin/agent_toolkit/agent.py

@@ -200,13 +200,13 @@ async def _start(self):
         ForestLogger.log("debug", "Starting agent")
 
         if self.options["skip_schema_update"] is False:
-            try:
-                api_map = await SchemaEmitter.get_serialized_schema(
-                    self.options, await self.customizer.get_datasource(), self.meta
-                )
-                await ForestHttpApi.send_schema(self.options, api_map)
-            except Exception:
-                ForestLogger.log("exception", "Error generating/sending forest schema V1")
+            # try:
+            #     api_map = await SchemaEmitter.get_serialized_schema(
+            #         self.options, await self.customizer.get_datasource(), self.meta
+            #     )
+            #     await ForestHttpApi.send_schema(self.options, api_map)
+            # except Exception:
+            #     ForestLogger.log("exception", "Error generating/sending forest schema V1")
 
             try:
                 api_map = await SchemaEmitterV2.get_serialized_schema(

src/agent_toolkit/forestadmin/agent_toolkit/utils/forest_schema/emitter.py

@@ -78,7 +78,10 @@ async def get_serialized_schema(
                 )
                 raise
 
-        return cls.serialize(collections_schema, meta)
+        serialized = cls.serialize(collections_schema, meta)
+        with open(f'{options["schema_path"].split(".json")[0]}_json_api.json', "w") as fout:
+            json.dump(serialized, fout, indent=4)
+        return serialized
 
     @staticmethod
     async def generate(

src/agent_toolkit/forestadmin/agent_toolkit/utils/http.py

@@ -170,6 +170,11 @@ async def send_schema_v2(cls, options: HttpOptions, schema: ForestSchemaV2) -> b
             {"forest-secret-key": options["env_secret"], "content-type": "application/json"},
             options["verify_ssl"],
         )
-        # TODO: this dump is only to compare schema during poc
-        with open(options["schema_path"], "w") as fout:
-            json.dump(ret, fout, indent=4)
+        pass
+        # TODO: remove file writing just after this dump is only to compare schema during poc
+        if ret:
+            with open(options["schema_path"], "w") as fout:
+                json.dump(ret["schemaV1"], fout, indent=4)
+
+            with open(f'{options["schema_path"].split(".json")[0]}_json_api.json', "w") as fout:
+                json.dump(ret["schemaV1JSONAPI"], fout, indent=4)