Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion engine/baml-runtime/src/cli/generate.rs
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,11 @@ impl GenerateArgs {
}

fn generate_clients(&self, defaults: super::RuntimeCliDefaults) -> Result<()> {
let runtime = BamlRuntime::from_directory(&self.from, std::env::vars().collect())
// Set BAML_GENERATE to prevent starting the tracing publisher
let mut env_vars: std::collections::HashMap<String, String> = std::env::vars().collect();
env_vars.insert("BAML_GENERATE".to_string(), "1".to_string());

let runtime = BamlRuntime::from_directory(&self.from, env_vars)
.context("Failed to build BAML runtime")?;
let src_files = baml_src_files(&self.from)
.context("Failed while searching for .baml files in baml_src/")?;
Expand Down
4 changes: 4 additions & 0 deletions engine/baml-runtime/src/tracingv2/publisher/publisher.rs
Original file line number Diff line number Diff line change
Expand Up @@ -183,6 +183,10 @@ pub fn start_publisher(
lookup: Arc<AstSignatureWrapper>,
#[cfg(not(target_arch = "wasm32"))] rt: Arc<tokio::runtime::Runtime>,
) {
if lookup.env_var("BAML_GENERATE").is_some() {
log::debug!("Skipping publisher because BAML_GENERATE is set");
return;
}
if lookup.env_var("BOUNDARY_API_KEY").is_none() {
log::debug!("Skipping publisher because BOUNDARY_API_KEY is not set");
return;
Expand Down
3 changes: 2 additions & 1 deletion integ-tests/typescript/package.json
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,8 @@
"integ-tests:dotenv": "tsc && pnpm test -- --silent false --testTimeout 30000",
"generate": "baml-cli generate --from ../baml_src",
"memory-test": "BAML_LOG=info infisical run --env=test -- pnpm test -- --silent false --testTimeout 60000 -t 'memory'",
"typecheck": "tsc --noEmit"
"typecheck": "tsc --noEmit",
"build:debug": "cd ../../engine/language_client_typescript && pnpm build:debug"
},
"keywords": [],
"author": "",
Expand Down
2 changes: 1 addition & 1 deletion package.json
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@
"format:fix": "biome check --write",
"docker-build": "bash build-docker.sh",
"test": "turbo run test",
"integ-tests": "pnpm run generate && turbo run integ-tests"
"integ-tests:ts": "turbo run build:debug --filter=@boundaryml/baml && cd integ-tests/typescript && pnpm generate && pnpm integ-tests"
},
"devDependencies": {
"@infisical/cli": "0.41.89",
Expand Down
21 changes: 21 additions & 0 deletions turbo.json
Original file line number Diff line number Diff line change
Expand Up @@ -191,6 +191,27 @@
],
"cache": false
},
"@boundaryml/baml#build:debug": {
"outputs": ["dist/**", "build/**", "artifacts/**"],
"cache": false
},
"integ-tests:ts": {
"dependsOn": ["build:debug"],
"outputs": [
"coverage/**",
".coverage/**",
"junit.xml",
"test-results/**"
],
"cache": false,
"env": [
"OPENAI_API_KEY",
"ANTHROPIC_API_KEY",
"GOOGLE_API_KEY",
"AWS_*",
"INTEG_TESTS_*"
]
},
"integ-tests": {
"dependsOn": ["build"],
"outputs": [
Expand Down
Loading