fixes

Author: boazdavid

src/toolguard/buildtime/gen_py/mellea_simple.py

@@ -16,7 +16,7 @@ def __init__(self, llm: I_TG_LLM):
         self.llm = llm
         self.formatter = TemplateFormatter(model_id="")
 
-    async def generate_from_context(
+    async def _generate_from_context(
         self,
         action: Component | CBlock,
         ctx: Context,

src/toolguard/buildtime/utils/datamodel_codegen.py

@@ -1,4 +1,5 @@
 import asyncio
+import sys
 from pathlib import Path
 
 
@@ -16,7 +17,9 @@ async def run(oas_file: Path) -> str:
     """
     # see https://github.com/koxudaxi/datamodel-code-generator
     process = await asyncio.create_subprocess_exec(
-        "datamodel-codegen",
+        sys.executable,  # = Python executable
+        "-m",
+        "datamodel_code_generator",
         "--use-field-description",
         "--use-schema-description",
         "--output-model-type",

src/toolguard/extra/langchain_to_oas.py

@@ -27,9 +27,10 @@ def langchain_tools_to_openapi(
     for tool in tools:
         # Get JSON schema from the args model
         if tool.get_input_schema():
-            components["schemas"][tool.name + "Args"] = (
-                tool.get_input_schema().model_json_schema()
-            )
+            in_schema = tool.get_input_schema().model_json_schema()
+            in_schema = substitute_refs(in_schema)
+            in_schema.pop("$defs", None)
+            components["schemas"][tool.name + "Args"] = in_schema
 
             request_body = {
                 "description": tool.description,