Expose Check utility functions in TypeScript

engine/language_client_typescript/checked.d.ts

@@ -2,13 +2,14 @@ export interface Checked<T, K extends BaseChecks> {
     value: T;
     checks: K;
 }
-interface Check {
+export interface Check {
     name: string;
     expr: string;
-    result: "succeeded" | "failed";
+    status: "succeeded" | "failed";
 }
-interface BaseChecks {
+export interface BaseChecks {
     [key: string]: Check;
 }
-export {};
+export declare function all_succeeded<K extends BaseChecks>(checks: K): boolean;
+export declare function get_checks<K extends BaseChecks>(checks: K): Check[];
 //# sourceMappingURL=checked.d.ts.map

engine/language_client_typescript/checked.js

@@ -1,5 +1,11 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.get_checks = exports.all_succeeded = void 0;
 function all_succeeded(checks) {
-    return Object.values(checks).every(value => value.result == "succeeded");
+    return Object.values(checks).every(value => value.status == "succeeded");
 }
+exports.all_succeeded = all_succeeded;
+function get_checks(checks) {
+    return Object.values(checks);
+}
+exports.get_checks = get_checks;

engine/language_client_typescript/typescript_src/checked.ts

@@ -3,16 +3,20 @@ export interface Checked<T,K extends BaseChecks> {
     checks: K,
 }
 
-interface Check {
+export interface Check {
     name: string,
     expr: string
     status: "succeeded" | "failed"
 }
 
-interface BaseChecks {
+export interface BaseChecks {
     [key: string]: Check
 }
 
-function all_succeeded<K extends BaseChecks>(checks: K): boolean {
+export function all_succeeded<K extends BaseChecks>(checks: K): boolean {
     return Object.values(checks).every(value => value.status == "succeeded");
 }
+
+export function get_checks<K extends BaseChecks>(checks: K): Check[] {
+    return Object.values(checks)
+}

integ-tests/typescript/tests/integ-tests.test.ts

@@ -18,6 +18,7 @@ import { RecursivePartialNull } from '../baml_client/async_client'
 import { b as b_sync } from '../baml_client/sync_client'
 import { config } from 'dotenv'
 import { BamlLogEvent, BamlRuntime } from '@boundaryml/baml/native'
+import { all_succeeded, get_checks } from '@boundaryml/baml/checked'
 import { AsyncLocalStorage } from 'async_hooks'
 import { DO_NOT_USE_DIRECTLY_UNLESS_YOU_KNOW_WHAT_YOURE_DOING_RUNTIME, resetBamlEnvVars } from '../baml_client/globals'
 config()
@@ -648,6 +649,8 @@ describe('Integ tests', () => {
     expect(res.primary.value.checks.valid_phone_number.status).toEqual("succeeded");
     expect(res.secondary?.value.value).toEqual("[email protected]");
     expect(res.secondary?.value.checks.valid_email.status).toEqual("succeeded");
+    expect(all_succeeded(res.primary.value.checks));
+    expect(get_checks(res.primary.value.checks).length).toEqual(1);
   })
 })