terse-cli 0.1.16 → 0.1.17

package/dist/providers/python/PythonProvider.js

@@ -0,0 +1,369 @@
+import { execFileSync, spawn } from "node:child_process";
+import os from "node:os";
+import path from "node:path";
+import chalk from "chalk";
+import { ensureUvAvailable, execUv, loadDotenv, withTempScript } from "../shared/shellUtils.js";
+import { openSessionStream, promptForToolApproval, submitApprovalDecision } from "../shared/sessionStream.js";
+import { preparePythonTemplateContext } from "./preparePythonCodegenData.js";
+import { renderPythonGeneratedCode } from "./pythonTemplateEngine.js";
+const JOB_REGISTRY_MARKER = "__TERSE_JOB_REGISTRY__=";
+const JOB_SKIPPED_MARKER = "__TERSE_SKIPPED__";
+const PYTHON_SDK_DEPENDENCY = "terse-sdk~=0.1.8";
+export class PythonProvider {
+    language = "python";
+    displayName = "Python";
+    projectMarkers = {
+        requiredFiles: ["pyproject.toml", "src/main.py"],
+        description: "Python Terse project",
+    };
+    entryFile = "src/main.py";
+    generatedCodePath = "src/terse_generated.py";
+    deployExclusions = {
+        dirs: new Set([".venv", "__pycache__", ".pytest_cache", ".mypy_cache", ".ruff_cache", ".git"]),
+        files: new Set([".env", ".DS_Store"]),
+    };
+    scaffoldFiles() {
+        return [
+            { template: "python/init/pyproject.toml.hbs", output: "pyproject.toml" },
+            { template: "python/init/src/main.py.hbs", output: "src/main.py" },
+            { template: "python/init/.python-version.hbs", output: ".python-version" },
+            { template: "python/init/env.example.hbs", output: ".env.example" },
+            { template: "python/init/gitignore.hbs", output: ".gitignore" },
+            { template: "python/init/README.md.hbs", output: "README.md" },
+        ];
+    }
+    buildInitTemplateContext(projectName) {
+        return {
+            projectName,
+            sdkDependency: PYTHON_SDK_DEPENDENCY,
+        };
+    }
+    getPostInitSteps(_packageManager) {
+        return [
+            "uv run ty check  Type-check the project",
+            "terse test       Run a sample event locally",
+        ];
+    }
+    detectPackageManager() {
+        return "uv";
+    }
+    async installDependencies(targetDir) {
+        const env = {
+            ...loadDotenv(targetDir),
+            UV_CACHE_DIR: path.join(os.tmpdir(), "terse-uv-cache"),
+        };
+        await execUv(["sync"], { cwd: targetDir, env });
+    }
+    renderGeneratedCode(input) {
+        const code = renderPythonGeneratedCode(preparePythonTemplateContext(input));
+        validatePythonSyntax(code);
+        return code;
+    }
+    async loadJobRegistry() {
+        const cwd = process.cwd();
+        const env = loadDotenv(cwd);
+        try {
+            const script = buildLoadRegistryScript();
+            return await withTempScript(script, ".py", async (scriptPath) => {
+                const { stdout } = await execUv(["run", "python", scriptPath], { cwd, env });
+                const payload = extractRegistryPayload(stdout);
+                const parsed = JSON.parse(payload);
+                const registry = new Map();
+                for (const [name, data] of Object.entries(parsed)) {
+                    registry.set(name, createJobParametersFromPython(data));
+                }
+                if (registry.size === 0) {
+                    console.error(chalk.red(`No jobs found. Make sure your ${this.entryFile} registers at least one job with @app.job(...).`));
+                    process.exit(1);
+                }
+                return registry;
+            });
+        }
+        catch (error) {
+            console.error(chalk.red(`Error importing ${this.entryFile}:\n`));
+            printPythonCommandError(error);
+            process.exit(1);
+        }
+    }
+    async executeJob(job, event, opts) {
+        const cwd = process.cwd();
+        const env = {
+            ...loadDotenv(cwd),
+            TERSE_JOB_NAME: job.name,
+            TERSE_EVENT_JSON: JSON.stringify(event),
+        };
+        const isVerbose = opts?.verbose ?? false;
+        const isSandbox = !!process.env.TERSE_RUN_ID;
+        const apiKey = env.TERSE_API_KEY ?? null;
+        const approvalState = { paused: false };
+        let closeSession;
+        try {
+            if (apiKey && (isVerbose || !isSandbox)) {
+                const session = await openSessionStream(apiKey, {
+                    verbose: isVerbose,
+                    isPaused: () => approvalState.paused,
+                    onEvent: !isSandbox
+                        ? createApprovalHandler(apiKey, approvalState)
+                        : undefined,
+                });
+                closeSession = session.close;
+                env.TERSE_SESSION_ID = session.sessionId;
+            }
+            const script = buildExecuteJobScript();
+            await ensureUvAvailable(cwd);
+            await withTempScript(script, ".py", async (scriptPath) => {
+                await runStreamingPython(cwd, env, scriptPath, job.name);
+            });
+        }
+        catch (error) {
+            console.error(chalk.red(`\n  Job "${job.name}" failed.\n`));
+            printPythonCommandError(error);
+            process.exit(1);
+        }
+        finally {
+            closeSession?.();
+        }
+    }
+}
+export const pythonProvider = new PythonProvider();
+function createJobParametersFromPython(data) {
+    return {
+        name: data.name,
+        triggers: data.triggers.map(reconstructPythonConfig),
+        skills: data.skills.map(reconstructPythonConfig),
+        toolApprovals: data.toolApprovals ?? [],
+        webhookURL: data.webhookURL,
+        onTrigger: async () => {
+            throw new Error("Python job execution must go through provider.executeJob()");
+        },
+        filter: data.hasFilter
+            ? async () => {
+                throw new Error("Python filter execution must go through provider.executeJob()");
+            }
+            : undefined,
+    };
+}
+function reconstructPythonConfig(config) {
+    return {
+        ...config,
+        isComplete: () => true,
+        formatForAgent: () => JSON.stringify(config),
+    };
+}
+function buildLoadRegistryScript() {
+    return `
+import importlib.util
+import json
+import os
+import sys
+import uuid
+
+from terse_sdk import clear_job_registry, get_job_registry
+
+PROJECT_ROOT = os.getcwd()
+SRC_DIR = os.path.join(PROJECT_ROOT, "src")
+sys.path[:0] = [PROJECT_ROOT, SRC_DIR]
+
+clear_job_registry()
+module_name = f"__terse_main_{uuid.uuid4().hex}"
+spec = importlib.util.spec_from_file_location(module_name, os.path.join(SRC_DIR, "main.py"))
+if spec is None or spec.loader is None:
+    raise RuntimeError("Could not create an import spec for src/main.py.")
+
+module = importlib.util.module_from_spec(spec)
+sys.modules[module_name] = module
+spec.loader.exec_module(module)
+
+registry = get_job_registry()
+result = {}
+for name, job in registry.items():
+    def serialize_config(config):
+        payload = dict(config.config)
+        payload["integrationId"] = config.integration_id
+        payload["integrationType"] = config.integration_type
+        payload["configType"] = config.config_type
+        return payload
+
+    result[name] = {
+        "name": name,
+        "triggers": [serialize_config(trigger) for trigger in job.triggers],
+        "skills": [serialize_config(skill) for skill in job.skills],
+        "toolApprovals": list(job.tool_approvals or []),
+        "webhookURL": job.webhook_url,
+        "hasFilter": job.filter is not None,
+    }
+
+print(${JSON.stringify(JOB_REGISTRY_MARKER)} + json.dumps(result))
+`.trim();
+}
+function buildExecuteJobScript() {
+    return `
+import importlib.util
+import json
+import os
+import sys
+import uuid
+
+from terse_sdk import TerseAgent, clear_job_registry, deserialize_input_event, execute_registered_job, get_job_registry
+
+PROJECT_ROOT = os.getcwd()
+SRC_DIR = os.path.join(PROJECT_ROOT, "src")
+sys.path[:0] = [PROJECT_ROOT, SRC_DIR]
+
+clear_job_registry()
+module_name = f"__terse_main_{uuid.uuid4().hex}"
+spec = importlib.util.spec_from_file_location(module_name, os.path.join(SRC_DIR, "main.py"))
+if spec is None or spec.loader is None:
+    raise RuntimeError("Could not create an import spec for src/main.py.")
+
+module = importlib.util.module_from_spec(spec)
+sys.modules[module_name] = module
+spec.loader.exec_module(module)
+
+registry = get_job_registry()
+job_name = os.environ["TERSE_JOB_NAME"]
+event = deserialize_input_event(json.loads(os.environ["TERSE_EVENT_JSON"]))
+session_id = os.environ.get("TERSE_SESSION_ID")
+job = registry[job_name]
+agent = TerseAgent(job.skills, session_id=session_id, tool_approvals=job.tool_approvals)
+skipped = execute_registered_job(job, event, agent=agent)
+if skipped:
+    print(${JSON.stringify(JOB_SKIPPED_MARKER)})
+`.trim();
+}
+function extractRegistryPayload(stdout) {
+    const line = stdout
+        .split(/\r?\n/)
+        .map(entry => entry.trim())
+        .filter(Boolean)
+        .reverse()
+        .find(entry => entry.startsWith(JOB_REGISTRY_MARKER));
+    if (!line) {
+        throw new Error("Python job loader did not return registry JSON.");
+    }
+    return line.slice(JOB_REGISTRY_MARKER.length);
+}
+function createApprovalHandler(apiKey, state) {
+    return async (event) => {
+        if (event.type === "run_started") {
+            state.runId = event.runId;
+            return;
+        }
+        if (event.type !== "tool_approval_requested")
+            return;
+        if (!state.runId) {
+            throw new Error("Received tool approval request before run_started.");
+        }
+        state.paused = true;
+        try {
+            const info = event.toolApprovalRequested;
+            const approved = await promptForToolApproval(info.toolName, info.arguments);
+            await submitApprovalDecision(apiKey, {
+                runId: state.runId,
+                stepId: info.stepId,
+                approved,
+            });
+        }
+        finally {
+            state.paused = false;
+        }
+    };
+}
+async function runStreamingPython(cwd, env, scriptPath, jobName) {
+    await new Promise((resolve, reject) => {
+        const child = spawn("uv", ["run", "python", scriptPath], {
+            cwd,
+            env,
+            stdio: ["ignore", "pipe", "pipe"],
+        });
+        let skipped = false;
+        let stdoutBuffer = "";
+        child.stdout.on("data", (chunk) => {
+            stdoutBuffer += chunk.toString();
+            stdoutBuffer = flushStdoutBuffer(stdoutBuffer, line => {
+                if (line === JOB_SKIPPED_MARKER) {
+                    skipped = true;
+                    return;
+                }
+                process.stdout.write(`${line}\n`);
+            });
+        });
+        child.stdout.on("end", () => {
+            if (!stdoutBuffer)
+                return;
+            if (stdoutBuffer === JOB_SKIPPED_MARKER) {
+                skipped = true;
+                return;
+            }
+            process.stdout.write(stdoutBuffer);
+        });
+        child.stderr.on("data", (chunk) => {
+            process.stderr.write(chunk);
+        });
+        child.on("error", reject);
+        child.on("close", code => {
+            if (code !== 0) {
+                reject(new Error(`uv run python exited with code ${code ?? "unknown"}`));
+                return;
+            }
+            if (skipped) {
+                console.log(chalk.dim(`\n  Job "${jobName}" skipped (filter returned false).\n`));
+            }
+            else {
+                console.log(chalk.green(`\n  Job "${jobName}" completed successfully.\n`));
+            }
+            resolve();
+        });
+    });
+}
+function flushStdoutBuffer(buffer, onLine) {
+    let newlineIndex = buffer.indexOf("\n");
+    while (newlineIndex !== -1) {
+        const line = buffer.slice(0, newlineIndex).replace(/\r$/, "");
+        onLine(line);
+        buffer = buffer.slice(newlineIndex + 1);
+        newlineIndex = buffer.indexOf("\n");
+    }
+    return buffer;
+}
+function validatePythonSyntax(code) {
+    for (const executable of ["python3", "python"]) {
+        try {
+            execFileSync(executable, ["-c", "import sys; compile(sys.stdin.read(), 'terse_generated.py', 'exec')"], {
+                input: code,
+                stdio: ["pipe", "ignore", "pipe"],
+            });
+            return;
+        }
+        catch (error) {
+            const code = error.code;
+            if (code === "ENOENT")
+                continue;
+            const stderr = Buffer.isBuffer(error.stderr)
+                ? (error.stderr.toString().trim())
+                : "";
+            if (stderr) {
+                console.warn(chalk.yellow(`Warning: generated Python did not pass syntax validation.\n${stderr}`));
+            }
+            else {
+                console.warn(chalk.yellow("Warning: generated Python did not pass syntax validation."));
+            }
+            return;
+        }
+    }
+}
+function printPythonCommandError(error) {
+    if (error instanceof Error) {
+        console.error(error.message);
+        const stderr = error.stderr;
+        if (typeof stderr === "string" && stderr.trim()) {
+            console.error(stderr.trim());
+        }
+        else if (Buffer.isBuffer(stderr) && stderr.toString().trim()) {
+            console.error(stderr.toString().trim());
+        }
+        return;
+    }
+    console.error(String(error));
+}
+//# sourceMappingURL=PythonProvider.js.map

package/dist/providers/python/PythonProvider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"PythonProvider.js","sourceRoot":"","sources":["../../../src/providers/python/PythonProvider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAA;AACxD,OAAO,EAAE,MAAM,SAAS,CAAA;AACxB,OAAO,IAAI,MAAM,WAAW,CAAA;AAC5B,OAAO,KAAK,MAAM,OAAO,CAAA;AAKzB,OAAO,EAAE,iBAAiB,EAAE,MAAM,EAAE,UAAU,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAA;AAC/F,OAAO,EAAE,iBAAiB,EAAE,qBAAqB,EAAE,sBAAsB,EAAE,MAAM,4BAA4B,CAAA;AAC7G,OAAO,EAAE,4BAA4B,EAAE,MAAM,+BAA+B,CAAA;AAC5E,OAAO,EAAE,yBAAyB,EAAE,MAAM,2BAA2B,CAAA;AAErE,MAAM,mBAAmB,GAAG,yBAAyB,CAAA;AACrD,MAAM,kBAAkB,GAAG,mBAAmB,CAAA;AAC9C,MAAM,qBAAqB,GAAG,kBAAkB,CAAA;AAuBhD,MAAM,OAAO,cAAc;IACd,QAAQ,GAAG,QAAiB,CAAA;IAC5B,WAAW,GAAG,QAAQ,CAAA;IACtB,cAAc,GAAG;QACtB,aAAa,EAAE,CAAC,gBAAgB,EAAE,aAAa,CAAC;QAChD,WAAW,EAAE,sBAAsB;KACtC,CAAA;IACQ,SAAS,GAAG,aAAa,CAAA;IACzB,iBAAiB,GAAG,wBAAwB,CAAA;IAC5C,gBAAgB,GAAG;QACxB,IAAI,EAAE,IAAI,GAAG,CAAC,CAAC,OAAO,EAAE,aAAa,EAAE,eAAe,EAAE,aAAa,EAAE,aAAa,EAAE,MAAM,CAAC,CAAC;QAC9F,KAAK,EAAE,IAAI,GAAG,CAAC,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;KACxC,CAAA;IAED,aAAa;QACT,OAAO;YACH,EAAE,QAAQ,EAAE,gCAAgC,EAAE,MAAM,EAAE,gBAAgB,EAAE;YACxE,EAAE,QAAQ,EAAE,6BAA6B,EAAE,MAAM,EAAE,aAAa,EAAE;YAClE,EAAE,QAAQ,EAAE,iCAAiC,EAAE,MAAM,EAAE,iBAAiB,EAAE;YAC1E,EAAE,QAAQ,EAAE,6BAA6B,EAAE,MAAM,EAAE,cAAc,EAAE;YACnE,EAAE,QAAQ,EAAE,2BAA2B,EAAE,MAAM,EAAE,YAAY,EAAE;YAC/D,EAAE,QAAQ,EAAE,2BAA2B,EAAE,MAAM,EAAE,WAAW,EAAE;SACjE,CAAA;IACL,CAAC;IAED,wBAAwB,CAAC,WAAmB;QACxC,OAAO;YACH,WAAW;YACX,aAAa,EAAE,qBAAqB;SACvC,CAAA;IACL,CAAC;IAED,gBAAgB,CAAC,eAAuB;QACpC,OAAO;YACH,yCAAyC;YACzC,6CAA6C;SAChD,CAAA;IACL,CAAC;IAED,oBAAoB;QAChB,OAAO,IAAI,CAAA;IACf,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,SAAiB;QACvC,MAAM,GAAG,GAAG;YACR,GAAG,UAAU,CAAC,SAAS,CAAC;YACxB,YAAY,EAAE,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,gBAAgB,CAAC;SACzD,CAAA;QACD,MAAM,MAAM,CAAC,CAAC,MAAM,CAAC,EAAE,EAAE,GAAG,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,CAAA;IACnD,CAAC;IAED,mBAAmB,CAAC,KAAmB;QACnC,MAAM,IAAI,GAAG,yBAAyB,CAAC,4BAA4B,CAAC,KAAK,CAAC,CAAC,CAAA;QAC3E,oBAAoB,CAAC,IAAI,CAAC,CAAA;QAC1B,OAAO,IAAI,CAAA;IACf,CAAC;IAED,KAAK,CAAC,eAAe;QACjB,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,EAAE,CAAA;QACzB,MAAM,GAAG,GAAG,UAAU,CAAC,GAAG,CAAC,CAAA;QAE3B,IAAI,CAAC;YACD,MAAM,MAAM,GAAG,uBAAuB,EAAE,CAAA;YACxC,OAAO,MAAM,cAAc,CAAC,MAAM,EAAE,KAAK,EAAE,KAAK,EAAC,UAAU,EAAC,EAAE;gBAC1D,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,CAAC,KAAK,EAAE,QAAQ,EAAE,UAAU,CAAC,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAA;gBAC5E,MAAM,OAAO,GAAG,sBAAsB,CAAC,MAAM,CAAC,CAAA;gBAC9C,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAkC,CAAA;gBACnE,MAAM,QAAQ,GAAG,IAAI,GAAG,EAA+B,CAAA;gBAEvD,KAAK,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;oBAChD,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,6BAA6B,CAAC,IAAI,CAAC,CAAC,CAAA;gBAC3D,CAAC;gBAED,IAAI,QAAQ,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;oBACtB,OAAO,CAAC,KAAK,CACT,KAAK,CAAC,GAAG,CACL,iCAAiC,IAAI,CAAC,SAAS,iDAAiD,CACnG,CACJ,CAAA;oBACD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;gBACnB,CAAC;gBAED,OAAO,QAAQ,CAAA;YACnB,CAAC,CAAC,CAAA;QACN,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,mBAAmB,IAAI,CAAC,SAAS,KAAK,CAAC,CAAC,CAAA;YAChE,uBAAuB,CAAC,KAAK,CAAC,CAAA;YAC9B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QACnB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,UAAU,CACZ,GAAwB,EACxB,KAAsB,EACtB,IAA4B;QAE5B,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,EAAE,CAAA;QACzB,MAAM,GAAG,GAAsB;YAC3B,GAAG,UAAU,CAAC,GAAG,CAAC;YAClB,cAAc,EAAE,GAAG,CAAC,IAAI;YACxB,gBAAgB,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;SAC1C,CAAA;QACD,MAAM,SAAS,GAAG,IAAI,EAAE,OAAO,IAAI,KAAK,CAAA;QACxC,MAAM,SAAS,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,YAAY,CAAA;QAC5C,MAAM,MAAM,GAAG,GAAG,CAAC,aAAa,IAAI,IAAI,CAAA;QACxC,MAAM,aAAa,GAAkB,EAAE,MAAM,EAAE,KAAK,EAAE,CAAA;QAEtD,IAAI,YAAsC,CAAA;QAC1C,IAAI,CAAC;YACD,IAAI,MAAM,IAAI,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;gBACtC,MAAM,OAAO,GAAG,MAAM,iBAAiB,CAAC,MAAM,EAAE;oBAC5C,OAAO,EAAE,SAAS;oBAClB,QAAQ,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,MAAM;oBACpC,OAAO,EAAE,CAAC,SAAS;wBACf,CAAC,CAAC,qBAAqB,CAAC,MAAM,EAAE,aAAa,CAAC;wBAC9C,CAAC,CAAC,SAAS;iBAClB,CAAC,CAAA;gBACF,YAAY,GAAG,OAAO,CAAC,KAAK,CAAA;gBAC5B,GAAG,CAAC,gBAAgB,GAAG,OAAO,CAAC,SAAS,CAAA;YAC5C,CAAC;YAED,MAAM,MAAM,GAAG,qBAAqB,EAAE,CAAA;YACtC,MAAM,iBAAiB,CAAC,GAAG,CAAC,CAAA;YAC5B,MAAM,cAAc,CAAC,MAAM,EAAE,KAAK,EAAE,KAAK,EAAC,UAAU,EAAC,EAAE;gBACnD,MAAM,kBAAkB,CAAC,GAAG,EAAE,GAAG,EAAE,UAAU,EAAE,GAAG,CAAC,IAAI,CAAC,CAAA;YAC5D,CAAC,CAAC,CAAA;QACN,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,YAAY,GAAG,CAAC,IAAI,aAAa,CAAC,CAAC,CAAA;YAC3D,uBAAuB,CAAC,KAAK,CAAC,CAAA;YAC9B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QACnB,CAAC;gBAAS,CAAC;YACP,YAAY,EAAE,EAAE,CAAA;QACpB,CAAC;IACL,CAAC;CACJ;AAED,MAAM,CAAC,MAAM,cAAc,GAAG,IAAI,cAAc,EAAE,CAAA;AAElD,SAAS,6BAA6B,CAAC,IAAmB;IACtD,OAAO;QACH,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,uBAAuB,CAAC;QACpD,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,uBAAuB,CAAC;QAChD,aAAa,EAAE,IAAI,CAAC,aAAa,IAAI,EAAE;QACvC,UAAU,EAAE,IAAI,CAAC,UAAU;QAC3B,SAAS,EAAE,KAAK,IAAI,EAAE;YAClB,MAAM,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAA;QACjF,CAAC;QACD,MAAM,EAAE,IAAI,CAAC,SAAS;YAClB,CAAC,CAAC,KAAK,IAAI,EAAE;gBACT,MAAM,IAAI,KAAK,CAAC,+DAA+D,CAAC,CAAA;YACpF,CAAC;YACD,CAAC,CAAC,SAAS;KACK,CAAA;AAC5B,CAAC;AAED,SAAS,uBAAuB,CAAC,MAA8B;IAC3D,OAAO;QACH,GAAG,MAAM;QACT,UAAU,EAAE,GAAG,EAAE,CAAC,IAAI;QACtB,cAAc,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;KAC/C,CAAA;AACL,CAAC;AAED,SAAS,uBAAuB;IAC5B,OAAO;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;QA0CH,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC;CAC1C,CAAC,IAAI,EAAE,CAAA;AACR,CAAC;AAED,SAAS,qBAAqB;IAC1B,OAAO;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;YA+BC,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC;CAC7C,CAAC,IAAI,EAAE,CAAA;AACR,CAAC;AAED,SAAS,sBAAsB,CAAC,MAAc;IAC1C,MAAM,IAAI,GAAG,MAAM;SACd,KAAK,CAAC,OAAO,CAAC;SACd,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;SAC1B,MAAM,CAAC,OAAO,CAAC;SACf,OAAO,EAAE;SACT,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC,CAAA;IAEzD,IAAI,CAAC,IAAI,EAAE,CAAC;QACR,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAA;IACtE,CAAC;IAED,OAAO,IAAI,CAAC,KAAK,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAA;AACjD,CAAC;AAED,SAAS,qBAAqB,CAAC,MAAc,EAAE,KAAoB;IAC/D,OAAO,KAAK,EAAE,KAA2E,EAAE,EAAE;QACzF,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;YAC/B,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAA;YACzB,OAAM;QACV,CAAC;QAED,IAAI,KAAK,CAAC,IAAI,KAAK,yBAAyB;YAAE,OAAM;QACpD,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,oDAAoD,CAAC,CAAA;QACzE,CAAC;QAED,KAAK,CAAC,MAAM,GAAG,IAAI,CAAA;QACnB,IAAI,CAAC;YACD,MAAM,IAAI,GAAG,KAAK,CAAC,qBAAqB,CAAA;YACxC,MAAM,QAAQ,GAAG,MAAM,qBAAqB,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,CAAA;YAC3E,MAAM,sBAAsB,CAAC,MAAM,EAAE;gBACjC,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,QAAQ;aACX,CAAC,CAAA;QACN,CAAC;gBAAS,CAAC;YACP,KAAK,CAAC,MAAM,GAAG,KAAK,CAAA;QACxB,CAAC;IACL,CAAC,CAAA;AACL,CAAC;AAED,KAAK,UAAU,kBAAkB,CAC7B,GAAW,EACX,GAAsB,EACtB,UAAkB,EAClB,OAAe;IAEf,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACxC,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,KAAK,EAAE,QAAQ,EAAE,UAAU,CAAC,EAAE;YACrD,GAAG;YACH,GAAG;YACH,KAAK,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC;SACpC,CAAC,CAAA;QAEF,IAAI,OAAO,GAAG,KAAK,CAAA;QACnB,IAAI,YAAY,GAAG,EAAE,CAAA;QAErB,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;YACtC,YAAY,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAA;YAChC,YAAY,GAAG,iBAAiB,CAAC,YAAY,EAAE,IAAI,CAAC,EAAE;gBAClD,IAAI,IAAI,KAAK,kBAAkB,EAAE,CAAC;oBAC9B,OAAO,GAAG,IAAI,CAAA;oBACd,OAAM;gBACV,CAAC;gBACD,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,IAAI,IAAI,CAAC,CAAA;YACrC,CAAC,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;QAEF,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;YACxB,IAAI,CAAC,YAAY;gBAAE,OAAM;YACzB,IAAI,YAAY,KAAK,kBAAkB,EAAE,CAAC;gBACtC,OAAO,GAAG,IAAI,CAAA;gBACd,OAAM;YACV,CAAC;YACD,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAA;QACtC,CAAC,CAAC,CAAA;QAEF,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;YACtC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;QAC/B,CAAC,CAAC,CAAA;QAEF,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAA;QACzB,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,IAAI,CAAC,EAAE;YACrB,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;gBACb,MAAM,CAAC,IAAI,KAAK,CAAC,kCAAkC,IAAI,IAAI,SAAS,EAAE,CAAC,CAAC,CAAA;gBACxE,OAAM;YACV,CAAC;YAED,IAAI,OAAO,EAAE,CAAC;gBACV,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,YAAY,OAAO,sCAAsC,CAAC,CAAC,CAAA;YACrF,CAAC;iBAAM,CAAC;gBACJ,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,OAAO,6BAA6B,CAAC,CAAC,CAAA;YAC9E,CAAC;YACD,OAAO,EAAE,CAAA;QACb,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;AACN,CAAC;AAED,SAAS,iBAAiB,CAAC,MAAc,EAAE,MAA8B;IACrE,IAAI,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;IACvC,OAAO,YAAY,KAAK,CAAC,CAAC,EAAE,CAAC;QACzB,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAA;QAC7D,MAAM,CAAC,IAAI,CAAC,CAAA;QACZ,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,YAAY,GAAG,CAAC,CAAC,CAAA;QACvC,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;IACvC,CAAC;IACD,OAAO,MAAM,CAAA;AACjB,CAAC;AAED,SAAS,oBAAoB,CAAC,IAAY;IACtC,KAAK,MAAM,UAAU,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,EAAE,CAAC;QAC7C,IAAI,CAAC;YACD,YAAY,CACR,UAAU,EACV,CAAC,IAAI,EAAE,qEAAqE,CAAC,EAC7E;gBACI,KAAK,EAAE,IAAI;gBACX,KAAK,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,MAAM,CAAC;aACpC,CACJ,CAAA;YACD,OAAM;QACV,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,MAAM,IAAI,GAAI,KAA+B,CAAC,IAAI,CAAA;YAClD,IAAI,IAAI,KAAK,QAAQ;gBAAE,SAAQ;YAE/B,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAE,KAA8B,CAAC,MAAM,CAAC;gBAClE,CAAC,CAAC,CAAE,KAA4B,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,IAAI,EAAE,CAAC;gBAC1D,CAAC,CAAC,EAAE,CAAA;YACR,IAAI,MAAM,EAAE,CAAC;gBACT,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,8DAA8D,MAAM,EAAE,CAAC,CAAC,CAAA;YACtG,CAAC;iBAAM,CAAC;gBACJ,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,2DAA2D,CAAC,CAAC,CAAA;YAC3F,CAAC;YACD,OAAM;QACV,CAAC;IACL,CAAC;AACL,CAAC;AAED,SAAS,uBAAuB,CAAC,KAAc;IAC3C,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QACzB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAA;QAC5B,MAAM,MAAM,GAAI,KAA8B,CAAC,MAAM,CAAA;QACrD,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;YAC9C,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAA;QAChC,CAAC;aAAM,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC;YAC7D,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,IAAI,EAAE,CAAC,CAAA;QAC3C,CAAC;QACD,OAAM;IACV,CAAC;IAED,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAA;AAChC,CAAC"}

package/dist/providers/python/preparePythonCodegenData.d.ts

@@ -0,0 +1,73 @@
+import type { CodegenInput } from "../codegenTypes.js";
+export interface PythonToolCtx {
+    name: string;
+    nameRepr: string;
+    methodName: string;
+    outputModel: string;
+    approvable: boolean;
+}
+export interface PythonAttioAttrCtx {
+    apiSlug: string;
+    recordType: string;
+    inputType: string;
+    filterType: string;
+}
+export interface PythonAttioObjectCtx {
+    apiSlug: string;
+    apiSlugRepr: string;
+    singularNoun: string;
+    singularNounRepr: string;
+    pascal: string;
+    staticName: string;
+    attributes: PythonAttioAttrCtx[];
+    hasAttributes: boolean;
+    attrSlugsLiteral: string;
+    attrSlugsFrozensetRepr: string;
+    multiValueSlugs: string[];
+    hasMultiValueSlugs: boolean;
+    multiValueFrozensetRepr: string;
+    attributeTupleRepr: string;
+    recordValuesTypeName: string;
+    inputValuesTypeName: string;
+    attributeSlugTypeName: string;
+    filterTypeName: string;
+    staticDeclaration: string;
+}
+export interface PythonAttioCtx {
+    instanceIdRepr: string;
+    hasAttrs: boolean;
+    objects: PythonAttioObjectCtx[];
+    tools: PythonToolCtx[];
+    approvableTools: PythonToolCtx[];
+    skillToolType: string;
+    skillObjectType: string;
+}
+export interface PythonSnowflakeCtx {
+    instanceIdRepr: string;
+    tools: PythonToolCtx[];
+    approvableTools: PythonToolCtx[];
+    skillToolType: string;
+}
+export interface PythonTemplateContext {
+    attio: PythonAttioCtx | null;
+    snowflake: PythonSnowflakeCtx | null;
+    hasAttio: boolean;
+    hasAttioAttrs: boolean;
+    hasAttioTools: boolean;
+    hasSnowflake: boolean;
+    hasSnowflakeTools: boolean;
+    hasAnyToolNameTypes: boolean;
+    hasAnyDeterministicTools: boolean;
+    sdkImports: string[];
+    typingImportsLine?: string;
+    pydanticImportsLine?: string;
+    needsJsonImport: boolean;
+    attioToolNamesLiteral?: string;
+    snowflakeToolNamesLiteral?: string;
+    allToolNamesLiteral?: string;
+    exportedNamesRepr: string;
+}
+export declare function preparePythonTemplateContext(input: CodegenInput): PythonTemplateContext;
+export declare function pyRepr(value: string): string;
+export declare function pyListRepr(values: string[]): string;
+//# sourceMappingURL=preparePythonCodegenData.d.ts.map

package/dist/providers/python/preparePythonCodegenData.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"preparePythonCodegenData.d.ts","sourceRoot":"","sources":["../../../src/providers/python/preparePythonCodegenData.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAGR,YAAY,EAGf,MAAM,oBAAoB,CAAA;AA6B3B,MAAM,WAAW,aAAa;IAC1B,IAAI,EAAE,MAAM,CAAA;IACZ,QAAQ,EAAE,MAAM,CAAA;IAChB,UAAU,EAAE,MAAM,CAAA;IAClB,WAAW,EAAE,MAAM,CAAA;IACnB,UAAU,EAAE,OAAO,CAAA;CACtB;AAED,MAAM,WAAW,kBAAkB;IAC/B,OAAO,EAAE,MAAM,CAAA;IACf,UAAU,EAAE,MAAM,CAAA;IAClB,SAAS,EAAE,MAAM,CAAA;IACjB,UAAU,EAAE,MAAM,CAAA;CACrB;AAED,MAAM,WAAW,oBAAoB;IACjC,OAAO,EAAE,MAAM,CAAA;IACf,WAAW,EAAE,MAAM,CAAA;IACnB,YAAY,EAAE,MAAM,CAAA;IACpB,gBAAgB,EAAE,MAAM,CAAA;IACxB,MAAM,EAAE,MAAM,CAAA;IACd,UAAU,EAAE,MAAM,CAAA;IAClB,UAAU,EAAE,kBAAkB,EAAE,CAAA;IAChC,aAAa,EAAE,OAAO,CAAA;IACtB,gBAAgB,EAAE,MAAM,CAAA;IACxB,sBAAsB,EAAE,MAAM,CAAA;IAC9B,eAAe,EAAE,MAAM,EAAE,CAAA;IACzB,kBAAkB,EAAE,OAAO,CAAA;IAC3B,uBAAuB,EAAE,MAAM,CAAA;IAC/B,kBAAkB,EAAE,MAAM,CAAA;IAC1B,oBAAoB,EAAE,MAAM,CAAA;IAC5B,mBAAmB,EAAE,MAAM,CAAA;IAC3B,qBAAqB,EAAE,MAAM,CAAA;IAC7B,cAAc,EAAE,MAAM,CAAA;IACtB,iBAAiB,EAAE,MAAM,CAAA;CAC5B;AAED,MAAM,WAAW,cAAc;IAC3B,cAAc,EAAE,MAAM,CAAA;IACtB,QAAQ,EAAE,OAAO,CAAA;IACjB,OAAO,EAAE,oBAAoB,EAAE,CAAA;IAC/B,KAAK,EAAE,aAAa,EAAE,CAAA;IACtB,eAAe,EAAE,aAAa,EAAE,CAAA;IAChC,aAAa,EAAE,MAAM,CAAA;IACrB,eAAe,EAAE,MAAM,CAAA;CAC1B;AAED,MAAM,WAAW,kBAAkB;IAC/B,cAAc,EAAE,MAAM,CAAA;IACtB,KAAK,EAAE,aAAa,EAAE,CAAA;IACtB,eAAe,EAAE,aAAa,EAAE,CAAA;IAChC,aAAa,EAAE,MAAM,CAAA;CACxB;AAED,MAAM,WAAW,qBAAqB;IAClC,KAAK,EAAE,cAAc,GAAG,IAAI,CAAA;IAC5B,SAAS,EAAE,kBAAkB,GAAG,IAAI,CAAA;IACpC,QAAQ,EAAE,OAAO,CAAA;IACjB,aAAa,EAAE,OAAO,CAAA;IACtB,aAAa,EAAE,OAAO,CAAA;IACtB,YAAY,EAAE,OAAO,CAAA;IACrB,iBAAiB,EAAE,OAAO,CAAA;IAC1B,mBAAmB,EAAE,OAAO,CAAA;IAC5B,wBAAwB,EAAE,OAAO,CAAA;IACjC,UAAU,EAAE,MAAM,EAAE,CAAA;IACpB,iBAAiB,CAAC,EAAE,MAAM,CAAA;IAC1B,mBAAmB,CAAC,EAAE,MAAM,CAAA;IAC5B,eAAe,EAAE,OAAO,CAAA;IACxB,qBAAqB,CAAC,EAAE,MAAM,CAAA;IAC9B,yBAAyB,CAAC,EAAE,MAAM,CAAA;IAClC,mBAAmB,CAAC,EAAE,MAAM,CAAA;IAC5B,iBAAiB,EAAE,MAAM,CAAA;CAC5B;AAED,wBAAgB,4BAA4B,CAAC,KAAK,EAAE,YAAY,GAAG,qBAAqB,CA6DvF;AAED,wBAAgB,MAAM,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CA2B5C;AAED,wBAAgB,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,CAEnD"}